Memory That Scales

Nodes:
  Alice (person)
  Q3 Review (event, date: 2024-09-15)
  Pivot to B2B (decision, date: 2024-09-16)
  Project Phoenix (project, started: 2024-09-20)

Edges:
  Alice → attended → Q3 Review
  Q3 Review → resulted in → Pivot to B2B
  Pivot to B2B → spawned → Project Phoenix
  Alice → leads → Project Phoenix

import { MemoryClient } from 'mem0ai';

const mem0 = new MemoryClient({
  apiKey: process.env.MEM0_API_KEY,
});

// Store a memory
export async function addMemory(
  content: string,
  userId: string,
  metadata?: Record<string, any>
) {
  return await mem0.add(
    [{ role: 'user', content }],
    { user_id: userId, metadata }
  );
}

// Search memories
export async function searchMemory(query: string, userId: string) {
  return await mem0.search(query, {
    user_id: userId,
    limit: 10,
  });
}

async function agentSearch(query: string, agentId: string) {
  // Try graph first for complex queries
  if (isComplexQuery(query)) {
    try {
      const graphResults = await searchMem0(query, agentId);
      if (graphResults.length > 0) return graphResults;
    } catch (e) {
      console.warn('Graph search failed, falling back');
    }
  }
  // Fall back to vector/markdown search
  return await vectorSearch(query, agentId);
}

function isComplexQuery(query: string): boolean {
  const indicators = [
    'before', 'after', 'because', 'led to',
    'resulted in', 'changed', 'decided',
    'when did', 'why did', 'how did',
  ];
  return indicators.some(i =>
    query.toLowerCase().includes(i)
  );
}