import { streamText } from 'ai';
import { openai } from '@ai-sdk/openai';
import { memoryTools, getMemoryContext } from '@usesatori/tools';

// Create memory tools scoped to a user
const tools = memoryTools({
  apiKey: process.env.SATORI_API_KEY!,
  baseUrl: 'https://api.satori.dev',
  userId: 'user-123',
});

// Pre-fetch relevant memories for context
const memoryContext = await getMemoryContext(
  {
    apiKey: process.env.SATORI_API_KEY!,
    baseUrl: 'https://api.satori.dev',
    userId: 'user-123',
  },
  userMessage,
  { limit: 5 }
);

// Use with AI SDK
const result = await streamText({
  model: openai('gpt-4o'),
  system: `You are a helpful assistant with memory.
    
What you know about this user:
${memoryContext}

Use add_memory to save important information the user shares.`,
  messages,
  tools,
});

// LLM will call this automatically when appropriate
tools.add_memory({ memory: "User prefers dark mode" })

tools.delete_memory({ memoryId: "memory-uuid" })

import { MemoryClient } from '@usesatori/tools';

const client = new MemoryClient({
  apiKey: process.env.SATORI_API_KEY!,
  baseUrl: 'https://api.satori.dev',
  userId: 'user-123',
});

// Store a memory
await client.addMemory('User loves TypeScript');

// Search memories
const memories = await client.searchMemories('programming languages');

// Get all memories
const allMemories = await client.getAllMemories();

// Delete a memory
await client.deleteMemory('memory-id');