Edge-ready AI with persistent memory

// lib/stompy.ts - Shared context utilities
const STOMPY_URL = 'https://mcp.stompy.ai/sse';
const STOMPY_TOKEN = process.env.STOMPY_TOKEN!;

export async function getProjectContext(topic: string): Promise<string> {
  const response = await fetch(STOMPY_URL, {
    method: 'POST',
    headers: {
      'Authorization': `Bearer ${STOMPY_TOKEN}`,
      'Content-Type': 'application/json',
    },
    body: JSON.stringify({ tool: 'recall_context', topic }),
  });

  const data = await response.json();
  return data.content || '';
}

export async function searchContext(query: string, limit = 3): Promise<string[]> {
  const response = await fetch(STOMPY_URL, {
    method: 'POST',
    headers: {
      'Authorization': `Bearer ${STOMPY_TOKEN}`,
      'Content-Type': 'application/json',
    },
    body: JSON.stringify({ tool: 'context_search', query, limit }),
  });

  const data = await response.json();
  return data.contexts?.map((c: any) => c.content) || [];
}

// app/api/chat/route.ts
import { streamText } from 'ai';
import { anthropic } from '@ai-sdk/anthropic';
import { getProjectContext, searchContext } from '@/lib/stompy';

export async function POST(req: Request) {
  const { messages } = await req.json();
  const lastMessage = messages[messages.length - 1]?.content || '';

  // Fetch context in parallel for speed
  const [projectRules, relevantContext] = await Promise.all([
    getProjectContext('project_rules'),
    searchContext(lastMessage, 3),
  ]);

  // Build context-aware system prompt
  const systemPrompt = `You are a helpful assistant for this project.

PROJECT RULES (always follow these):
${projectRules}

RELEVANT CONTEXT (from previous work):
${relevantContext.join('\n\n')}

Respond helpfully while respecting our project's established patterns and decisions.`;

  const result = await streamText({
    model: anthropic('claude-sonnet-4-20250514'),
    system: systemPrompt,
    messages,
  });

  return result.toDataStreamResponse();
}

export const runtime = 'edge';  // Still runs at the edge!

// app/api/save-insight/route.ts
import { NextResponse } from 'next/server';

const STOMPY_URL = 'https://mcp.stompy.ai/sse';

export async function POST(req: Request) {
  const { topic, content, tags } = await req.json();

  const response = await fetch(STOMPY_URL, {
    method: 'POST',
    headers: {
      'Authorization': `Bearer ${process.env.STOMPY_TOKEN}`,
      'Content-Type': 'application/json',
    },
    body: JSON.stringify({
      tool: 'lock_context',
      topic,
      content,
      tags,
    }),
  });

  const result = await response.json();
  return NextResponse.json(result);
}

// Client-side: Call this when user confirms a decision in chat
// await fetch('/api/save-insight', {
//   method: 'POST',
//   body: JSON.stringify({
//     topic: 'authentication_approach',
//     content: 'Decided to use NextAuth.js with Google OAuth',
//     tags: 'auth,decisions,nextauth'
//   })
// });