Mastra Working Memory + Zod 스키마로 에이전트 사용자 상태를 구조적으로 추적하기

import { Memory } from "@mastra/memory";
import { z } from "zod";
 
const memory = new Memory({
  options: {
    workingMemory: {
      enabled: true,
      schema: z.object({
        name: z.string().optional(),
        location: z.string().optional(),
        preferences: z.object({
          communicationStyle: z.enum(["formal", "casual"]).optional(),
          projectGoal: z.string().optional(),
        }).optional(),
        activeDeadlines: z.array(z.object({
          label: z.string(),
          date: z.string(),
        })).optional(),
        lastTopicDiscussed: z.string().optional(),
      }),
    },
  },
});

# Working Memory (Current State)
{
  "name": null,
  "location": null,
  "preferences": null,
  "activeDeadlines": [],
  "lastTopicDiscussed": null
}
 
대화에서 사용자 정보를 파악하면 위 JSON 구조에 맞춰 업데이트하세요.
변경이 필요한 필드만 포함해도 됩니다 — 나머지는 자동으로 보존됩니다.

const userProfileSchema = z.object({
  name: z.string().optional(),
  location: z.string().optional(),
});
 
type UserProfile = z.infer<typeof userProfileSchema>;
// → { name?: string | undefined; location?: string | undefined }
// 이 타입으로 메모리 조회 결과를 안전하게 다룰 수 있습니다

import { Agent } from "@mastra/core";
import { Memory } from "@mastra/memory";
import { z } from "zod";
 
const userProfileSchema = z.object({
  profile: z.object({
    name: z.string().optional(),
    timezone: z.string().optional(),
    preferredLanguage: z.string().optional(),
  }),
  currentSession: z.object({
    topic: z.string().optional(),
    openQuestions: z.array(z.string()).default([]),
  }),
  goals: z.array(z.object({
    description: z.string(),
    deadline: z.string().optional(),
    status: z.enum(["active", "completed", "paused"]),
  })).default([]),
});
 
type UserProfile = z.infer<typeof userProfileSchema>;
 
const agent = new Agent({
  name: "PersonalAssistant",
  instructions: `당신은 사용자의 개인 어시스턴트입니다.
 
대화에서 아래 정보를 파악하면 즉시 working memory를 업데이트하세요:
- 사용자 이름, 시간대, 선호 언어 → profile 필드
- 현재 대화 주제나 미해결 질문 → currentSession 필드
- 사용자가 언급한 목표나 마감 → goals 배열에 추가
 
goals에 새 항목을 추가할 때는 기존 항목을 포함한 전체 배열을 반환하세요.
working memory에 이미 있는 정보는 사용자에게 다시 묻지 말고 자연스럽게 활용하세요.`,
  memory: new Memory({
    options: {
      workingMemory: {
        enabled: true,
        schema: userProfileSchema,
        scope: "resource", // 동일 사용자의 모든 스레드에서 상태 공유
      },
    },
  }),
});
 
const response = await agent.generate("안녕하세요, 저는 김민수입니다.", {
  resourceId: "user-kim-minsu",
  threadId: crypto.randomUUID(),
});

const taskSchema = z.object({
  tasks: z.array(z.object({
    id: z.string(),
    title: z.string(),
    priority: z.enum(["high", "medium", "low"]),
    done: z.boolean().default(false),
    createdAt: z.string(),
  })).default([]),
  userContext: z.object({
    focusArea: z.string().optional(),
    workingHours: z.string().optional(),
    preferredNotification: z.enum(["email", "slack", "none"]).optional(),
  }).optional(),
});
 
const taskAgent = new Agent({
  name: "TaskManager",
  instructions: `태스크를 추가하거나 상태를 변경할 때는 해당 필드만 업데이트하세요.
tasks 배열에 새 항목을 추가할 때는 기존 항목을 포함한 전체 배열을 반환하세요.
userContext는 명시적으로 변경 요청이 없는 한 업데이트하지 마세요.`,
  memory: new Memory({
    options: {
      workingMemory: {
        enabled: true,
        schema: taskSchema,
      },
    },
  }),
});

const result = await taskAgent.generate(
  "내일 오후 3시까지 API 문서 작성해야 해. 우선순위 높음으로 추가해줘.",
  {
    resourceId: "user-dev",
    threadId: crypto.randomUUID(),
  }
);
 
// 에이전트가 내부적으로 업데이트하는 구조 (예시)
// {
//   tasks: [
//     ...기존태스크,
//     {
//       id: crypto.randomUUID(),
//       title: "API 문서 작성",
//       priority: "high",
//       done: false,
//       createdAt: "2026-05-21"
//     }
//   ]
//   // userContext는 건드리지 않음 → 기존 값 자동 보존
// }

const sharedMemory = new Memory({
  options: {
    workingMemory: {
      enabled: true,
      schema: z.object({
        customerProfile: z.object({
          name: z.string().optional(),
          purchaseHistory: z.array(z.string()).default([]),
          preferences: z.array(z.string()).default([]),
        }),
        currentIssue: z.object({
          description: z.string().optional(),
          status: z.enum(["open", "resolved", "escalated"]).optional(),
        }).optional(),
      }),
    },
  },
});
 
const supportAgent = new Agent({ name: "SupportBot", memory: sharedMemory });
const recommendAgent = new Agent({ name: "RecommendBot", memory: sharedMemory });
 
// 동일 resourceId → 스토리지에서 동일한 working memory 레코드를 읽고 씀
await supportAgent.generate("제품에 문제가 있어요.", {
  resourceId: "user-123",
  threadId: crypto.randomUUID(),
});
 
await recommendAgent.generate("다른 제품 추천해주세요.", {
  resourceId: "user-123", // SupportBot이 기록한 customerProfile을 그대로 참조
  threadId: crypto.randomUUID(),
});