Rorkで作るアプリに「記憶するAI」を実装する — ユーザー文脈を保持するパーソナライゼーション設計

import { MMKV } from 'react-native-mmkv';
 
const storage = new MMKV({ id: 'user-context' });
 
// ユーザーコンテキストの型定義
interface UserContext {
  preferences: {
    language: string;
    topics: string[];
    tone: 'casual' | 'formal';
  };
  recentConversations: Array<{
    summary: string;
    timestamp: number;
  }>;
  profileFacts: string[]; // 「東京在住」「30代」等、会話から抽出した事実
}
 
// 保存
function saveUserContext(ctx: UserContext) {
  storage.set('user_context', JSON.stringify(ctx));
}
 
// 取得
function getUserContext(): UserContext | null {
  const raw = storage.getString('user_context');
  return raw ? JSON.parse(raw) : null;
}

async function extractAndSaveUserFacts(
  userMessage: string,
  existingFacts: string[]
): Promise<string[]> {
  const response = await fetch('https://api.anthropic.com/v1/messages', {
    method: 'POST',
    headers: {
      'x-api-key': process.env.EXPO_PUBLIC_ANTHROPIC_KEY!,
      'anthropic-version': '2023-06-01',
      'content-type': 'application/json',
    },
    body: JSON.stringify({
      model: 'claude-haiku-4-5-20251001',
      max_tokens: 256,
      system: `ユーザーのメッセージから、将来の会話で役立つ個人情報を抽出してください。
既知の情報: ${existingFacts.join('、')}
新しい事実のみをJSON配列で返してください。新しい情報がなければ空配列を返してください。
例: ["東京在住", "旅行が趣味", "3歳の子供がいる"]`,
      messages: [{ role: 'user', content: userMessage }],
    }),
  });
 
  const data = await response.json();
  try {
    const newFacts = JSON.parse(data.content[0].text);
    return [...new Set([...existingFacts, ...newFacts])];
  } catch {
    return existingFacts;
  }
}

function buildSystemPrompt(ctx: UserContext | null): string {
  if (!ctx) {
    return 'あなたは親切なアシスタントです。';
  }
 
  const parts = ['あなたは親切なアシスタントです。'];
 
  if (ctx.profileFacts.length > 0) {
    parts.push(`\n【ユーザーについて知っていること】\n${ctx.profileFacts.join('\n')}`);
  }
 
  if (ctx.recentConversations.length > 0) {
    const recent = ctx.recentConversations
      .slice(-3) // 直近3件のみ
      .map(c => `- ${c.summary}`)
      .join('\n');
    parts.push(`\n【最近の会話の概要】\n${recent}`);
  }
 
  if (ctx.preferences.tone === 'casual') {
    parts.push('\nフレンドリーで砕けた口調で話してください。');
  }
 
  return parts.join('');
}
 
// 実際のAPI呼び出し
async function chat(userMessage: string): Promise<string> {
  const ctx = getUserContext();
  const systemPrompt = buildSystemPrompt(ctx);
 
  const response = await fetch('https://api.anthropic.com/v1/messages', {
    method: 'POST',
    headers: { /* ... */ },
    body: JSON.stringify({
      model: 'claude-haiku-4-5-20251001',
      max_tokens: 1024,
      system: systemPrompt,
      messages: [{ role: 'user', content: userMessage }],
    }),
  });
 
  // レスポンス後にコンテキストを更新
  if (ctx) {
    const newFacts = await extractAndSaveUserFacts(userMessage, ctx.profileFacts);
    saveUserContext({ ...ctx, profileFacts: newFacts });
  }
 
  const data = await response.json();
  return data.content[0].text;
}

// ユーザーが「リセット」した場合
function clearUserContext() {
  storage.delete('user_context');
}