MemoryCard

import { tool } from "ai"
import { MemoryCardSchema } from "@/components/taw/memory-card"

export const getMemories = tool({
  description: "Show what you remember about the user",
  parameters: z.object({ scope: z.string().optional() }),
  outputSchema: MemoryCardSchema,
  execute: async ({ scope }) => {
    const memories = await memoryStore.getAll(userId, scope)
    return {
      id: "user-memory",
      title: "What I Remember",
      description: `From our last ${memories.count} conversations`,
      memories: memories.items.map(m => ({
        id: m.id,
        content: m.content,
        category: m.category,
        learnedFrom: m.source,
        confidence: m.confidence,
      })),
      confidence: memories.overallConfidence,
      source: { label: "Memory Store" },
    }
  },
})

import { MemoryCard } from "@/components/taw/memory-card"

function ToolOutput({ part }) {
  const [receipt, setReceipt] = useState()

  const handleAction = (id, payload) => {
    if (payload.receipt) setReceipt(payload.receipt)
    // Send verdicts to your memory store
    await updateMemories(payload.verdicts)
  }

  return (
    <MemoryCard
      part={part}
      onAction={handleAction}
      receipt={receipt}
    />
  )
}