[BUG] [HELP] Postgres Storage is not working

[Anan7Codes]

I have followed the docs and the cookbook. But the storage conversations are not being stored in the db.

I have run the following Docker command,

docker run -d \
  -e POSTGRES_DB=ai \
  -e POSTGRES_USER=ai \
  -e POSTGRES_PASSWORD=ai \
  -e PGDATA=/var/lib/postgresql/data/pgdata \
  -v pgvolume:/var/lib/postgresql/data \
  -p 5532:5432 \
  --name pgvector \
  agnohq/pgvector:16

Docs are outdated I guess but there is already a PR raised for it. It is "agnohq" and not "agno".

The following contains the necessary functions from my code. I have two different databases. One for querying(SQL agent) and one for storage.

# Initialize PostgresTools with connection details from environment variables
postgres_tools = PostgresTools(
    host=os.getenv('DB_HOST'),
    port=int(os.getenv('DB_PORT', '5432')),
    db_name=os.getenv('DB_NAME'),
    user=os.getenv('DB_USER'),
    password=os.getenv('DB_PASSWORD'),
    run_queries=True,
)

# Initialize PostgresStorage with separate connection details
storage_db_url = f"postgresql+psycopg://{os.getenv('STORAGE_DB_USER')}:{os.getenv('STORAGE_DB_PASSWORD')}@{os.getenv('STORAGE_DB_HOST')}:{os.getenv('STORAGE_DB_PORT')}/{os.getenv('STORAGE_DB_NAME')}"
storage = PostgresStorage(
    table_name="agent_sessions",
    db_url=storage_db_url,
    auto_upgrade_schema=True
)

def create_user_agent(user_id):
    agent = Agent(
        model=Claude(
            # id="apac.anthropic.claude-3-5-sonnet-20240620-v1:0",
            id="anthropic.claude-3-sonnet-20240229-v1:0",
            temperature=0.1,
            max_tokens=4096
        ),
        debug_mode=True,
        tools=[postgres_tools],
        storage=storage,
        add_history_to_messages=True,
        show_tool_calls=True,
        markdown=True,
    )

My code works as it should but I dont see anything being saved on the DB. Wanted to understand this first before I dived into how I could handle different sessions(or pass session to an agent or storage?)

[monali7-d]

Hey @Anan7Codes

Thank you for reaching out and for using Agno! We sincerely apologize for missing your query yesterday. We're prioritizing resolving this bug and will get back to you asap.

Thank you for your patience!

[kausmeows]

Hey @Anan7Codes, can you first confirm me this-
for example if let's say you're using tableplus to see the table updates, are you on public or ai section. See the image below for better understanding.

And yes, thanks for pointing out, the model storage docs are not update, we'll get to it asap.
The storage import is now like this-

from agno.storage.postgres import PostgresStorage

instead of

from agno.storage.agent.postgres import PostgresAgentStorage

Here's the code i used to test if things are getting saved-

"""Run `pip install duckduckgo-search sqlalchemy openai` to install dependencies."""

from agno.agent import Agent
from agno.models.anthropic import Claude
from agno.storage.postgres import PostgresStorage
from agno.tools.duckduckgo import DuckDuckGoTools

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

agent = Agent(
    model=Claude(id="claude-3-5-sonnet-20241022"),
    storage=PostgresStorage(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?")

Let me know if this solves it?

[kausmeows]

@Anan7Codes here's a loom walkthrough with somewhat of the code you gave above. I altered it a bit to test! Saves in db.

https://www.loom.com/share/6d309d4c37024b1e9108a2066f6af78d?sid=b572fe91-0796-4095-a628-9504850e6af0

[Anan7Codes]

Thank you very much for the reply and absolutely no problem with delay. Amazing stuff going on here.

Also this was working as expected. I was trying to query on the cli itself and that was showing empty. Ran it on pgadmin and it actually has been working. My bad.

Thank you and keep kiling it out there 🔥

[kausmeows]

Thanks a lot for the kind words 🧡