Streaming Agent

Code

1from typing import Iterator  # noqa
2from kern.agent import Agent, RunOutput  # noqa
3from kern.models.aws import AwsBedrock
4
5agent = Agent(
6    model=AwsBedrock(id="mistral.mistral-large-2402-v1:0"), markdown=True
7)
8
9# Get the response in a variable
10# run_response: Iterator[RunOutputEvent] = agent.run("Share a 2 sentence horror story", stream=True)
11# for chunk in run_response:
12#     print(chunk.content)
13
14# Print the response in the terminal
15agent.print_response("Share a 2 sentence horror story", stream=True)

Usage

Set up your virtual environment

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

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

Set your AWS Credentials

1export AWS_ACCESS_KEY_ID=***
2export AWS_SECRET_ACCESS_KEY=***
3export AWS_REGION=***

Install dependencies

1uv pip install -U boto3 kern-ai

Run Agent

1python basic_stream.py