Mastra Agent Memory Integration

npm install @mastra/core @sovseal/sdk @ai-sdk/openai zod

import { createTool } from "@mastra/core/tools";
import { AgentStateClient, EngineUnavailableError } from "@sovseal/sdk";
import { z } from "zod";

// store/recall need no endpoint/apiKey — they talk to the local native host.
const client = new AgentStateClient();

export const recallMemoryTool = createTool({
  id: "sovseal-recall",
  description: "Search the local memory store for user preferences, previous decisions, and system history.",
  inputSchema: z.object({
    query: z.string().describe("The search query to match against memories."),
  }),
  outputSchema: z.object({
    context: z.string().describe("The recalled memory text, newline-separated."),
  }),
  execute: async ({ context }) => {
    try {
      const hits = await client.recall(context.query, { topK: 5 });
      return { context: hits.map((h) => h.text).join("\n") || "No memories found." };
    } catch (err) {
      if (err instanceof EngineUnavailableError) {
        return { context: `Recall unavailable: ${err.reason}` };
      }
      throw err;
    }
  },
});

export const storeMemoryTool = createTool({
  id: "sovseal-store",
  description: "Store a new preference, convention, or fact in persistent memory.",
  inputSchema: z.object({
    fact: z.string().describe("The declarative fact to save."),
  }),
  execute: async ({ context }) => {
    try {
      const result = await client.store(context.fact);
      return { success: true, reinforced: result.reinforced };
    } catch (err) {
      if (err instanceof EngineUnavailableError) {
        return { success: false, reason: err.reason };
      }
      throw err;
    }
  },
});

import { Agent } from "@mastra/core/agent";
import { openai } from "@ai-sdk/openai";
import { recallMemoryTool, storeMemoryTool } from "./tools";

export const memoryAgent = new Agent({
  name: "SovereignAssistant",
  instructions: `
    You are a personalized AI helper.
    Always call the 'sovseal-recall' tool first on starting a chat to search for user context.
    Use the 'sovseal-store' tool whenever the user volunteers useful personal preferences or workflow guidelines.
  `,
  model: openai("gpt-4o"),
  tools: { recallMemoryTool, storeMemoryTool },
});