Code

cookbook/models/cohere/storage.py
from agno.agent import Agent
from agno.models.cohere import Cohere
from agno.storage.agent.postgres import PostgresAgentStorage
from agno.tools.duckduckgo import DuckDuckGoTools

db_url = "postgresql+psycopg://ai:ai@localhost:5532/ai"

agent = Agent(
    model=Cohere(id="command-r-08-2024"),
    storage=PostgresAgentStorage(table_name="agent_sessions", db_url=db_url),
    tools=[DuckDuckGoTools()],
    add_history_to_messages=True,
)
agent.print_response("How many people live in Canada?")
agent.print_response("What is their national anthem called?")

Usage

1

Create a virtual environment

Open the Terminal and create a python virtual environment.

2

Set your API key

export CO_API_KEY=xxx
3

Install libraries

pip install -U cohere sqlalchemy psycopg duckduckgo-search agno
4

Run Agent