Memory

pip install "mem0ai>=2" orgo openai

export ORGO_API_KEY=sk_live_...
export OPENAI_API_KEY=sk-...

import os

from mem0 import Memory
from openai import OpenAI
from orgo import Computer

memory = Memory()
# Pass a size: SDKs before 0.0.47 default to 2 GB, which the API rejects.
computer = Computer(ram=4, cpu=1)

client = OpenAI(
    base_url="https://www.orgo.ai/api/v1",
    api_key=os.environ["ORGO_API_KEY"],
)

user_id = "alice"

# Teach preferences
memory.add(
    messages=[{"role": "user", "content": "I search the web with DuckDuckGo"}],
    user_id=user_id,
)
memory.add(
    messages=[{"role": "user", "content": "I live in Chicago"}],
    user_id=user_id,
)

task = "Find tomorrow's weather forecast where I live"

# Recall memories before a task
results = memory.search(query=task, filters={"user_id": user_id}, top_k=5)
context = "\n".join(f"- {m['memory']}" for m in results.get("results", []))

# Use memories as context
response = client.chat.completions.create(
    model="orgo-agent",
    messages=[{
        "role": "user",
        "content": f"User preferences:\n{context}\n\nTask: {task}"
    }],
    extra_body={"computer_id": computer.computer_id},
)

reply = response.choices[0].message.content
print(reply)

# Store the interaction for next time
memory.add(
    messages=[
        {"role": "user", "content": task},
        {"role": "assistant", "content": reply},
    ],
    user_id=user_id,
)

computer.destroy()

from mem0 import Memory

memory = Memory()

# View memories (20 by default; pass top_k for more)
for m in memory.get_all(filters={"user_id": "alice"}).get("results", []):
    print(f"  {m['memory']}")

# Search
results = memory.search(query="weather", filters={"user_id": "alice"}, top_k=5)

# Clear
memory.delete_all(user_id="alice")