带记忆的 Agent

代码

cookbook/models/vllm/memory.py

from agno.agent import Agent
from agno.memory.v2.db.postgres import PostgresMemoryDb
from agno.memory.v2.memory import Memory
from agno.models.vllm import vLLM
from agno.storage.postgres import PostgresStorage

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

agent = Agent(
    model=vLLM(id="microsoft/Phi-3-mini-128k-instruct"),
    memory=Memory(
        db=PostgresMemoryDb(table_name="agent_memory", db_url=DB_URL),
    ),
    enable_user_memories=True,
    enable_session_summaries=True,
    storage=PostgresStorage(
        table_name="personalized_agent_sessions",
        db_url=DB_URL,
    ),
)

# 分享个人信息；Agent 应该能够记住它们。
agent.print_response("My name is John Billings.", stream=True)
print("Current memories →")
pprint(agent.memory.memories)
print("Current summary →")
pprint(agent.memory.summaries)

agent.print_response("I live in NYC.", stream=True)
print("Memories →")
pprint(agent.memory.memories)
print("Summary →")
pprint(agent.memory.summaries)

agent.print_response("I'm going to a concert tomorrow.", stream=True)
print("Memories →")
pprint(agent.memory.memories)
print("Summary →")
pprint(agent.memory.summaries)

# 让 Agent 回忆信息
agent.print_response(
    "What have we been talking about, do you know my name?", stream=True
)

请确保 Postgres 数据库正在运行。

使用方法

1. 创建虚拟环境

   打开 Terminal 并创建一个 python 虚拟环境。

   Mac

   python3 -m venv .venv
   source .venv/bin/activate
   
2. 启动 Postgres 数据库

   ./cookbook/scripts/run_pgvector.sh
   
3. 安装库

   pip install -U agno openai vllm sqlalchemy psycopg[binary] pgvector
   
4. 启动 vLLM 服务器

   vllm serve microsoft/Phi-3-mini-128k-instruct \
       --dtype float32 \
       --enable-auto-tool-choice \
       --tool-call-parser pythonic
   
5. 运行 Agent

   python cookbook/models/vllm/memory.py