Skip to main content

Base kernel for pydantic-ai kernels

Project description

Pydantic AI Base Kernel

This is wrapper around pydantic-ai agent, that allows to requests it through jupyter kernel.

It is meant to be subclassed to create new kernel-based agent, for adding tools or any special application.

Getting started

Within a python venv,

pip install pydantic-ai-kernel

In order to specify information about the agent, you have to set up a config file, and place it in : ~/.jupyter/jupyter_pydantic_ai_kernel_config.yaml. See Configuration File to get details.

Then, any jupyter frontend should be able to treat with this agent, for example :

Notebook (you might need to restart the IDE) : select 'pydantic_ai' on top right of the notebook

CLI : Install jupyter-console (pip install jupyter-console); and run jupyter console --kernel pydantic_ai

Silik Signal Messaging : Access the kernel through Signal Message Application, see here

Configuration file

The configuration standard is a little bit cumbersome but is made to match the description of agents in pydantic-ai. We describe hereafter this standard.

An Pydantic AI agent is made of :

  • a model

  • information about the agent : tools, system prompt, agent name

The model itself, is composed of several parts :

Here is an example for a local ollama instance :

agent_name: pydantic_ai
system_prompt: You are a specialist in cooking, and you are always ready to help people creating new cooking recipees.
model:
  model_name: qwen3:1.7b
  model_type: openai
  model_provider:
    name: ollama
    params:
      base_url: http://localhost:11434/v1

If you just need to use an external open-ai provider, put :

model:
  model_name: openai:gpt-4o

and specify API key in environment variable.

Scheme can be found here.

Creating your own agents

In order to create custom agents, you just need to create a new kernel, and subclass PydanticAIBaseKernel from this library. See an example here : [https://github.com/mariusgarenaux/rudi-kernel]

You can then create tools, or any mechanism you want. We provide here juste the communication protocol between agent and user, through well known and proven jupyter kernels.

The configuration file for any subclass of PydanticAIBaseKernel will be fetched from : ~/.jupyter/jupyter_<kernel_name>_config.yaml; and must follows the same scheme as the one of pydantic_ai_kernel.

Dealing with multi-agents

Multi-agents means here several agents that have access to the same context. To do so, you can for example use silik-kernel; an other kernel that allows several kernels to be started and managed through a single one.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

pydantic_ai_kernel-1.2.2.tar.gz (12.5 kB view details)

Uploaded Source

File details

Details for the file pydantic_ai_kernel-1.2.2.tar.gz.

File metadata

  • Download URL: pydantic_ai_kernel-1.2.2.tar.gz
  • Upload date:
  • Size: 12.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.12

File hashes

Hashes for pydantic_ai_kernel-1.2.2.tar.gz
Algorithm Hash digest
SHA256 8a398052298c8abeb5cb8b58c6d14d2f988628050b6b84da1ab8d5f50c583ebe
MD5 3cdca7c2b5b20449484e147ae947db36
BLAKE2b-256 82e121a707bed4e95a162c6fcf5c788d309b10630f93b45acf678919ff79baf1

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page