Memory for CrewAI

import os, requests

AULD = os.environ["AULD_API_BASE"]          # e.g. https://YOUR-AULD-HOST
HEADERS = {"Authorization": f"Bearer {os.environ['AULD_API_KEY']}"}

def auld_post(path: str, body: dict) -> dict:
    r = requests.post(f"{AULD}/v1/{path}", json=body, headers=HEADERS, timeout=30)
    r.raise_for_status()
    return r.json()

def remember(text: str, user_id: str, source_id: str, occurred_at: str) -> str:
    """Store a dated record for one user. Returns the job id."""
    return auld_post("records", {"sourceId": source_id, "text": text,
                                 "occurredAt": occurred_at, "author": user_id,
                                 "visibility": f"user:{user_id}"})["jobId"]

def recall(query: str, user_id: str, max_tokens: int = 2000) -> str:
    """Current memories with sources, formatted for a prompt."""
    res = auld_post("recall", {"query": query, "viewer": user_id, "maxTokens": max_tokens})
    return "\n".join(f"- {m['text']} ({m['sources'][0]['title']}, {m['sources'][0]['occurredAt'][:10]})"
                     for m in res["memories"]) or "No relevant memory."

from crewai import Agent, Task, Crew
from crewai.tools import tool
from memory_client import recall, remember

def tools_for(customer_id: str):
    @tool("Recall customer memory")
    def recall_customer(question: str) -> str:
        """Return current facts and history about this customer, with sources."""
        return recall(question, customer_id)
    return [recall_customer]

researcher = Agent(role="Account researcher", goal="Summarise the account's current state",
                   backstory="You check memory before drawing conclusions.",
                   tools=tools_for(job.customer_id))
task = Task(description="Prepare a renewal brief.", expected_output="A one-page brief", agent=researcher)
result = Crew(agents=[researcher], tasks=[task]).kickoff()
remember(str(result), job.customer_id, f"brief-{job.id}", job.started_at)