Set Client

Code

1from kern.agent import Agent, RunOutput  # noqa
2from kern.models.ollama import Ollama
3from ollama import Client as OllamaClient
4
5agent = Agent(
6    model=Ollama(id="llama3.1:8b", client=OllamaClient()),
7    markdown=True,
8)
9
10# Print the response in the terminal
11agent.print_response("Share a 2 sentence horror story")

Usage

Set up your virtual environment

1uv venv --python 3.12
2source .venv/bin/activate

1uv venv --python 3.12
2.venv\Scripts\activate

Install Ollama

Follow the Ollama installation guide and run:

1ollama pull llama3.1:8b

Install dependencies

1uv pip install -U ollama kern-ai

Run Agent

1python cookbook/11_models/ollama/set_client.py