@ainetwork/adk-provider-memory-inmemory 0.3.3 → 0.3.5

package/dist/index.cjs

@@ -46,8 +46,9 @@ var InMemoryIntent = class {
     return Array.from(this.intents.values()).find((intent) => intent.name === intentName);
   }
   async saveIntent(intent) {
-    const newId = (0, import_node_crypto.randomUUID)();
-    this.intents.set(newId, intent);
+    const intentId = intent.id || (0, import_node_crypto.randomUUID)();
+    const intentToSave = { ...intent, id: intentId };
+    this.intents.set(intentId, intentToSave);
   }
   async updateIntent(intentId, intent) {
     this.intents.set(intentId, intent);
@@ -112,7 +113,15 @@ var InMemoryThread = class {
   async deleteThread(userId, threadId) {
     const key = this.generateKey(userId, threadId);
     this.threads.delete(key);
-    this.userThreadIndex.delete(threadId);
+    const userThreads = this.userThreadIndex.get(userId);
+    if (userThreads) {
+      const metadataToDelete = Array.from(userThreads).find(
+        (metadata) => metadata.threadId === threadId
+      );
+      if (metadataToDelete) {
+        userThreads.delete(metadataToDelete);
+      }
+    }
   }
   async listThreads(userId) {
     const threads = this.userThreadIndex.get(userId);
@@ -123,6 +132,66 @@ var InMemoryThread = class {
   }
 };
 
+// implements/workflow.memory.ts
+var InMemoryWorkflow = class {
+  workflows = /* @__PURE__ */ new Map();
+  userWorkflowIndex = /* @__PURE__ */ new Map();
+  async createWorkflow(workflow) {
+    this.workflows.set(workflow.workflowId, workflow);
+    if (workflow.userId) {
+      if (!this.userWorkflowIndex.has(workflow.userId)) {
+        this.userWorkflowIndex.set(workflow.userId, /* @__PURE__ */ new Set());
+      }
+      this.userWorkflowIndex.get(workflow.userId)?.add(workflow.workflowId);
+    }
+    return workflow;
+  }
+  async getWorkflow(workflowId) {
+    return this.workflows.get(workflowId);
+  }
+  async updateWorkflow(workflowId, updates) {
+    if (!updates.userId) {
+      throw new Error("userId is required for updateWorkflow");
+    }
+    const existing = this.workflows.get(workflowId);
+    if (!existing) {
+      throw new Error(`Workflow not found: ${workflowId}`);
+    }
+    if (existing.userId !== updates.userId) {
+      throw new Error("Unauthorized: userId does not match workflow owner");
+    }
+    const updated = { ...existing, ...updates, workflowId };
+    this.workflows.set(workflowId, updated);
+  }
+  async deleteWorkflow(workflowId, userId) {
+    const workflow = this.workflows.get(workflowId);
+    if (workflow && workflow.userId === userId) {
+      this.workflows.delete(workflowId);
+      this.userWorkflowIndex.get(userId)?.delete(workflowId);
+    }
+  }
+  async listWorkflows(userId) {
+    const workflows = [];
+    for (const workflow of this.workflows.values()) {
+      if (!workflow.userId) {
+        workflows.push(workflow);
+      }
+    }
+    if (userId) {
+      const userWorkflowIds = this.userWorkflowIndex.get(userId);
+      if (userWorkflowIds) {
+        for (const workflowId of userWorkflowIds) {
+          const workflow = this.workflows.get(workflowId);
+          if (workflow) {
+            workflows.push(workflow);
+          }
+        }
+      }
+    }
+    return workflows;
+  }
+};
+
 // implements/base.memory.ts
 var InMemoryMemory = class _InMemoryMemory {
   static instance;
@@ -130,6 +199,7 @@ var InMemoryMemory = class _InMemoryMemory {
   agentMemory;
   intentMemory;
   threadMemory;
+  workflowMemory;
   constructor() {
     if (!_InMemoryMemory.instance) {
       _InMemoryMemory.instance = this;
@@ -137,6 +207,7 @@ var InMemoryMemory = class _InMemoryMemory {
     this.agentMemory = new InMemoryAgent();
     this.threadMemory = new InMemoryThread();
     this.intentMemory = new InMemoryIntent();
+    this.workflowMemory = new InMemoryWorkflow();
   }
   async connect() {
     this.connected = true;
@@ -156,6 +227,9 @@ var InMemoryMemory = class _InMemoryMemory {
   getIntentMemory() {
     return this.intentMemory;
   }
+  getWorkflowMemory() {
+    return this.workflowMemory;
+  }
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {

package/dist/index.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../index.ts","../implements/agent.memory.ts","../implements/intent.memory.ts","../implements/thread.memory.ts","../implements/base.memory.ts"],"sourcesContent":["export { InMemoryMemory } from \"./implements/base.memory\";","import { IAgentMemory } from \"@ainetwork/adk/modules\";\n\nexport class InMemoryAgent implements IAgentMemory {\n  private prompt: string = \"\";\n\n  public async getAgentPrompt(): Promise<string> {\n    return this.prompt;\n  }\n\n  public async updateAgentPrompt(prompt: string): Promise<void> {\n    this.prompt = prompt;\n  }\n}\n","import { randomUUID } from \"node:crypto\";\nimport type { Intent } from \"@ainetwork/adk/types/memory\";\nimport { IIntentMemory } from \"@ainetwork/adk/modules\";\n\nexport class InMemoryIntent implements IIntentMemory {\n  public intents: Map<string, Intent> = new Map();\n\n  public async getIntent(intentId: string): Promise<Intent | undefined> {\n    return this.intents.get(intentId);\n  }\n\n  public async getIntentByName(intentName: string): Promise<Intent | undefined> {\n    return Array.from(this.intents.values()).find(intent => intent.name === intentName);\n  }\n\n  public async saveIntent(intent: Intent): Promise<void> {\n    const newId = randomUUID();\n    this.intents.set(newId, intent);\n  }\n\n  public async updateIntent(intentId: string, intent: Intent): Promise<void> {\n    this.intents.set(intentId, intent);\n  }\n\n  public async deleteIntent(intentId: string): Promise<void> {\n    this.intents.delete(intentId);\n  }\n\n  public async listIntents(): Promise<Intent[]> {\n    return Array.from(this.intents.values());\n  }\n}","import type { MessageObject, ThreadObject, ThreadMetadata, ThreadType } from \"@ainetwork/adk/types/memory\";\nimport { IThreadMemory } from \"@ainetwork/adk/modules\";\n\ntype InMemoryThreadObject = {\n  type: ThreadType;\n  title: string;\n  messages: Array<MessageObject>;\n}\n\ntype InMemoryThreadMetadata = {\n  type: ThreadType;\n  userId: string;\n  threadId: string;\n  title: string;\n  updatedAt: number;\n  createdAt: number;\n}\n\nexport class InMemoryThread implements IThreadMemory {\n  public threads: Map<string, InMemoryThreadObject> = new Map();\n  public userThreadIndex: Map<string, Set<InMemoryThreadMetadata>> = new Map();\n\n  private generateKey(userId: string, threadId: string) {\n    return `${userId}:${threadId}`;\n  }\n\n  public async getThread(\n    userId: string,\n    threadId: string\n  ): Promise<ThreadObject | undefined> {\n    const key = this.generateKey(userId, threadId);\n    const res = this.threads.get(key);\n    if (res) {\n      const threadObject: ThreadObject = {\n        threadId,\n        userId,\n        type: res.type,\n        title: res.title,\n        messages: res.messages,\n      };\n      return threadObject;\n    }\n    return undefined;\n  }\n\n  public async createThread(\n    type: ThreadType,\n    userId: string,\n    threadId: string,\n    title: string\n  ): Promise<ThreadObject> {\n    const now = Date.now();\n    const key = this.generateKey(userId, threadId);\n    if (!this.userThreadIndex.has(userId)) {\n      this.userThreadIndex.set(userId, new Set());\n    }\n    if (!this.threads.has(key)) {\n      this.threads.set(key, { type, title, messages: [] });\n      const metadata: InMemoryThreadMetadata = {\n        type, userId, threadId, title, createdAt: now, updatedAt: now,\n      }\n      this.userThreadIndex.get(userId)?.add(metadata);\n    }\n\n    return { type, title, threadId, userId, messages: [] };\n  }\n\n  public async addMessagesToThread(\n    userId: string,\n    threadId: string,\n    messages: MessageObject[]\n  ): Promise<void> {\n    const key = this.generateKey(userId, threadId);\n    const thread = this.threads.get(key);\n    for (const message of messages) {\n      thread?.messages.push(message);\n    }\n  }\n\n  public async deleteThread(userId: string, threadId: string): Promise<void> {\n    const key = this.generateKey(userId, threadId);\n    this.threads.delete(key);\n    this.userThreadIndex.delete(threadId);\n  }\n\n  public async listThreads(userId: string): Promise<ThreadMetadata[]> {\n    const threads = this.userThreadIndex.get(userId);\n    if (threads) {\n      return Array.from(threads);\n    }\n    return [];\n  }\n}","import { IAgentMemory, IIntentMemory, IMemory, IThreadMemory } from \"node_modules/@ainetwork/adk/dist/esm/modules/memory/base.memory\";\nimport { InMemoryAgent } from \"./agent.memory\";\nimport { InMemoryIntent } from \"./intent.memory\";\nimport { InMemoryThread } from \"./thread.memory\";\n\nexport class InMemoryMemory implements IMemory {\n  private static instance: InMemoryMemory;\n  private connected: boolean = false;\n\n  private agentMemory: InMemoryAgent;\n  private intentMemory: InMemoryIntent;\n  private threadMemory: InMemoryThread;\n\n  constructor() {\n    if (!InMemoryMemory.instance) {\n      InMemoryMemory.instance = this;\n    }\n\n    this.agentMemory = new InMemoryAgent();\n    this.threadMemory = new InMemoryThread();\n    this.intentMemory = new InMemoryIntent();\n  }\n\n  public async connect(): Promise<void> {\n    this.connected = true;\n  }\n\n  public async disconnect(): Promise<void> {\n    this.connected = false;\n  }\n\n  public isConnected(): boolean {\n    return this.connected;\n  }\n\n  public getAgentMemory(): IAgentMemory {\n    return this.agentMemory;\n  }\n\n  public getThreadMemory(): IThreadMemory {\n    return this.threadMemory;\n  }\n\n  public getIntentMemory(): IIntentMemory {\n    return this.intentMemory;\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACEO,IAAM,gBAAN,MAA4C;AAAA,EACzC,SAAiB;AAAA,EAEzB,MAAa,iBAAkC;AAC7C,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,MAAa,kBAAkB,QAA+B;AAC5D,SAAK,SAAS;AAAA,EAChB;AACF;;;ACZA,yBAA2B;AAIpB,IAAM,iBAAN,MAA8C;AAAA,EAC5C,UAA+B,oBAAI,IAAI;AAAA,EAE9C,MAAa,UAAU,UAA+C;AACpE,WAAO,KAAK,QAAQ,IAAI,QAAQ;AAAA,EAClC;AAAA,EAEA,MAAa,gBAAgB,YAAiD;AAC5E,WAAO,MAAM,KAAK,KAAK,QAAQ,OAAO,CAAC,EAAE,KAAK,YAAU,OAAO,SAAS,UAAU;AAAA,EACpF;AAAA,EAEA,MAAa,WAAW,QAA+B;AACrD,UAAM,YAAQ,+BAAW;AACzB,SAAK,QAAQ,IAAI,OAAO,MAAM;AAAA,EAChC;AAAA,EAEA,MAAa,aAAa,UAAkB,QAA+B;AACzE,SAAK,QAAQ,IAAI,UAAU,MAAM;AAAA,EACnC;AAAA,EAEA,MAAa,aAAa,UAAiC;AACzD,SAAK,QAAQ,OAAO,QAAQ;AAAA,EAC9B;AAAA,EAEA,MAAa,cAAiC;AAC5C,WAAO,MAAM,KAAK,KAAK,QAAQ,OAAO,CAAC;AAAA,EACzC;AACF;;;ACbO,IAAM,iBAAN,MAA8C;AAAA,EAC5C,UAA6C,oBAAI,IAAI;AAAA,EACrD,kBAA4D,oBAAI,IAAI;AAAA,EAEnE,YAAY,QAAgB,UAAkB;AACpD,WAAO,GAAG,MAAM,IAAI,QAAQ;AAAA,EAC9B;AAAA,EAEA,MAAa,UACX,QACA,UACmC;AACnC,UAAM,MAAM,KAAK,YAAY,QAAQ,QAAQ;AAC7C,UAAM,MAAM,KAAK,QAAQ,IAAI,GAAG;AAChC,QAAI,KAAK;AACP,YAAM,eAA6B;AAAA,QACjC;AAAA,QACA;AAAA,QACA,MAAM,IAAI;AAAA,QACV,OAAO,IAAI;AAAA,QACX,UAAU,IAAI;AAAA,MAChB;AACA,aAAO;AAAA,IACT;AACA,WAAO;AAAA,EACT;AAAA,EAEA,MAAa,aACX,MACA,QACA,UACA,OACuB;AACvB,UAAM,MAAM,KAAK,IAAI;AACrB,UAAM,MAAM,KAAK,YAAY,QAAQ,QAAQ;AAC7C,QAAI,CAAC,KAAK,gBAAgB,IAAI,MAAM,GAAG;AACrC,WAAK,gBAAgB,IAAI,QAAQ,oBAAI,IAAI,CAAC;AAAA,IAC5C;AACA,QAAI,CAAC,KAAK,QAAQ,IAAI,GAAG,GAAG;AAC1B,WAAK,QAAQ,IAAI,KAAK,EAAE,MAAM,OAAO,UAAU,CAAC,EAAE,CAAC;AACnD,YAAM,WAAmC;AAAA,QACvC;AAAA,QAAM;AAAA,QAAQ;AAAA,QAAU;AAAA,QAAO,WAAW;AAAA,QAAK,WAAW;AAAA,MAC5D;AACA,WAAK,gBAAgB,IAAI,MAAM,GAAG,IAAI,QAAQ;AAAA,IAChD;AAEA,WAAO,EAAE,MAAM,OAAO,UAAU,QAAQ,UAAU,CAAC,EAAE;AAAA,EACvD;AAAA,EAEA,MAAa,oBACX,QACA,UACA,UACe;AACf,UAAM,MAAM,KAAK,YAAY,QAAQ,QAAQ;AAC7C,UAAM,SAAS,KAAK,QAAQ,IAAI,GAAG;AACnC,eAAW,WAAW,UAAU;AAC9B,cAAQ,SAAS,KAAK,OAAO;AAAA,IAC/B;AAAA,EACF;AAAA,EAEA,MAAa,aAAa,QAAgB,UAAiC;AACzE,UAAM,MAAM,KAAK,YAAY,QAAQ,QAAQ;AAC7C,SAAK,QAAQ,OAAO,GAAG;AACvB,SAAK,gBAAgB,OAAO,QAAQ;AAAA,EACtC;AAAA,EAEA,MAAa,YAAY,QAA2C;AAClE,UAAM,UAAU,KAAK,gBAAgB,IAAI,MAAM;AAC/C,QAAI,SAAS;AACX,aAAO,MAAM,KAAK,OAAO;AAAA,IAC3B;AACA,WAAO,CAAC;AAAA,EACV;AACF;;;ACvFO,IAAM,iBAAN,MAAM,gBAAkC;AAAA,EAC7C,OAAe;AAAA,EACP,YAAqB;AAAA,EAErB;AAAA,EACA;AAAA,EACA;AAAA,EAER,cAAc;AACZ,QAAI,CAAC,gBAAe,UAAU;AAC5B,sBAAe,WAAW;AAAA,IAC5B;AAEA,SAAK,cAAc,IAAI,cAAc;AACrC,SAAK,eAAe,IAAI,eAAe;AACvC,SAAK,eAAe,IAAI,eAAe;AAAA,EACzC;AAAA,EAEA,MAAa,UAAyB;AACpC,SAAK,YAAY;AAAA,EACnB;AAAA,EAEA,MAAa,aAA4B;AACvC,SAAK,YAAY;AAAA,EACnB;AAAA,EAEO,cAAuB;AAC5B,WAAO,KAAK;AAAA,EACd;AAAA,EAEO,iBAA+B;AACpC,WAAO,KAAK;AAAA,EACd;AAAA,EAEO,kBAAiC;AACtC,WAAO,KAAK;AAAA,EACd;AAAA,EAEO,kBAAiC;AACtC,WAAO,KAAK;AAAA,EACd;AACF;","names":[]}
+{"version":3,"sources":["../index.ts","../implements/agent.memory.ts","../implements/intent.memory.ts","../implements/thread.memory.ts","../implements/workflow.memory.ts","../implements/base.memory.ts"],"sourcesContent":["export { InMemoryMemory } from \"./implements/base.memory\";","import { IAgentMemory } from \"@ainetwork/adk/modules\";\n\nexport class InMemoryAgent implements IAgentMemory {\n  private prompt: string = \"\";\n\n  public async getAgentPrompt(): Promise<string> {\n    return this.prompt;\n  }\n\n  public async updateAgentPrompt(prompt: string): Promise<void> {\n    this.prompt = prompt;\n  }\n}\n","import { randomUUID } from \"node:crypto\";\nimport type { Intent } from \"@ainetwork/adk/types/memory\";\nimport { IIntentMemory } from \"@ainetwork/adk/modules\";\n\nexport class InMemoryIntent implements IIntentMemory {\n  public intents: Map<string, Intent> = new Map();\n\n  public async getIntent(intentId: string): Promise<Intent | undefined> {\n    return this.intents.get(intentId);\n  }\n\n  public async getIntentByName(intentName: string): Promise<Intent | undefined> {\n    return Array.from(this.intents.values()).find(intent => intent.name === intentName);\n  }\n\n  public async saveIntent(intent: Intent): Promise<void> {\n    // Intent에 이미 id가 있으면 그것을 사용하고, 없으면 새로 생성\n    const intentId = intent.id || randomUUID();\n    const intentToSave = { ...intent, id: intentId };\n    this.intents.set(intentId, intentToSave);\n  }\n\n  public async updateIntent(intentId: string, intent: Intent): Promise<void> {\n    this.intents.set(intentId, intent);\n  }\n\n  public async deleteIntent(intentId: string): Promise<void> {\n    this.intents.delete(intentId);\n  }\n\n  public async listIntents(): Promise<Intent[]> {\n    return Array.from(this.intents.values());\n  }\n}","import type { MessageObject, ThreadObject, ThreadMetadata, ThreadType } from \"@ainetwork/adk/types/memory\";\nimport { IThreadMemory } from \"@ainetwork/adk/modules\";\n\ntype InMemoryThreadObject = {\n  type: ThreadType;\n  title: string;\n  messages: Array<MessageObject>;\n}\n\ntype InMemoryThreadMetadata = {\n  type: ThreadType;\n  userId: string;\n  threadId: string;\n  title: string;\n  updatedAt: number;\n  createdAt: number;\n}\n\nexport class InMemoryThread implements IThreadMemory {\n  public threads: Map<string, InMemoryThreadObject> = new Map();\n  public userThreadIndex: Map<string, Set<InMemoryThreadMetadata>> = new Map();\n\n  private generateKey(userId: string, threadId: string) {\n    return `${userId}:${threadId}`;\n  }\n\n  public async getThread(\n    userId: string,\n    threadId: string\n  ): Promise<ThreadObject | undefined> {\n    const key = this.generateKey(userId, threadId);\n    const res = this.threads.get(key);\n    if (res) {\n      const threadObject: ThreadObject = {\n        threadId,\n        userId,\n        type: res.type,\n        title: res.title,\n        messages: res.messages,\n      };\n      return threadObject;\n    }\n    return undefined;\n  }\n\n  public async createThread(\n    type: ThreadType,\n    userId: string,\n    threadId: string,\n    title: string\n  ): Promise<ThreadObject> {\n    const now = Date.now();\n    const key = this.generateKey(userId, threadId);\n    if (!this.userThreadIndex.has(userId)) {\n      this.userThreadIndex.set(userId, new Set());\n    }\n    if (!this.threads.has(key)) {\n      this.threads.set(key, { type, title, messages: [] });\n      const metadata: InMemoryThreadMetadata = {\n        type, userId, threadId, title, createdAt: now, updatedAt: now,\n      }\n      this.userThreadIndex.get(userId)?.add(metadata);\n    }\n\n    return { type, title, threadId, userId, messages: [] };\n  }\n\n  public async addMessagesToThread(\n    userId: string,\n    threadId: string,\n    messages: MessageObject[]\n  ): Promise<void> {\n    const key = this.generateKey(userId, threadId);\n    const thread = this.threads.get(key);\n    for (const message of messages) {\n      thread?.messages.push(message);\n    }\n  }\n\n  public async deleteThread(userId: string, threadId: string): Promise<void> {\n    const key = this.generateKey(userId, threadId);\n    this.threads.delete(key);\n\n    // userThreadIndex에서 해당 thread metadata 제거\n    const userThreads = this.userThreadIndex.get(userId);\n    if (userThreads) {\n      const metadataToDelete = Array.from(userThreads).find(\n        metadata => metadata.threadId === threadId\n      );\n      if (metadataToDelete) {\n        userThreads.delete(metadataToDelete);\n      }\n    }\n  }\n\n  public async listThreads(userId: string): Promise<ThreadMetadata[]> {\n    const threads = this.userThreadIndex.get(userId);\n    if (threads) {\n      return Array.from(threads);\n    }\n    return [];\n  }\n}","import { IWorkflowMemory } from \"@ainetwork/adk/modules\";\nimport type { Workflow } from \"@ainetwork/adk/types/memory\";\n\nexport class InMemoryWorkflow implements IWorkflowMemory {\n  private workflows: Map<string, Workflow> = new Map();\n  private userWorkflowIndex: Map<string, Set<string>> = new Map();\n\n  public async createWorkflow(workflow: Workflow): Promise<Workflow> {\n    this.workflows.set(workflow.workflowId, workflow);\n\n    if (workflow.userId) {\n      if (!this.userWorkflowIndex.has(workflow.userId)) {\n        this.userWorkflowIndex.set(workflow.userId, new Set());\n      }\n      this.userWorkflowIndex.get(workflow.userId)?.add(workflow.workflowId);\n    }\n\n    return workflow;\n  }\n\n  public async getWorkflow(workflowId: string): Promise<Workflow | undefined> {\n    return this.workflows.get(workflowId);\n  }\n\n  public async updateWorkflow(workflowId: string, updates: Partial<Workflow>): Promise<void> {\n    if (!updates.userId) {\n      throw new Error(\"userId is required for updateWorkflow\");\n    }\n\n    const existing = this.workflows.get(workflowId);\n    if (!existing) {\n      throw new Error(`Workflow not found: ${workflowId}`);\n    }\n\n    if (existing.userId !== updates.userId) {\n      throw new Error(\"Unauthorized: userId does not match workflow owner\");\n    }\n\n    const updated = { ...existing, ...updates, workflowId };\n    this.workflows.set(workflowId, updated);\n  }\n\n  public async deleteWorkflow(workflowId: string, userId: string): Promise<void> {\n    const workflow = this.workflows.get(workflowId);\n    if (workflow && workflow.userId === userId) {\n      this.workflows.delete(workflowId);\n      this.userWorkflowIndex.get(userId)?.delete(workflowId);\n    }\n  }\n\n  public async listWorkflows(userId?: string): Promise<Workflow[]> {\n    const workflows: Workflow[] = [];\n\n    // 템플릿 workflow (userId가 없는 것)은 항상 포함\n    for (const workflow of this.workflows.values()) {\n      if (!workflow.userId) {\n        workflows.push(workflow);\n      }\n    }\n\n    // userId가 있으면 해당 유저 소유 workflow도 포함\n    if (userId) {\n      const userWorkflowIds = this.userWorkflowIndex.get(userId);\n      if (userWorkflowIds) {\n        for (const workflowId of userWorkflowIds) {\n          const workflow = this.workflows.get(workflowId);\n          if (workflow) {\n            workflows.push(workflow);\n          }\n        }\n      }\n    }\n\n    return workflows;\n  }\n}\n","import { IAgentMemory, IIntentMemory, IMemory, IThreadMemory, IWorkflowMemory } from \"@ainetwork/adk/modules\";\nimport { InMemoryAgent } from \"./agent.memory\";\nimport { InMemoryIntent } from \"./intent.memory\";\nimport { InMemoryThread } from \"./thread.memory\";\nimport { InMemoryWorkflow } from \"./workflow.memory\";\n\nexport class InMemoryMemory implements IMemory {\n  private static instance: InMemoryMemory;\n  private connected: boolean = false;\n\n  private agentMemory: InMemoryAgent;\n  private intentMemory: InMemoryIntent;\n  private threadMemory: InMemoryThread;\n  private workflowMemory: InMemoryWorkflow;\n\n  constructor() {\n    if (!InMemoryMemory.instance) {\n      InMemoryMemory.instance = this;\n    }\n\n    this.agentMemory = new InMemoryAgent();\n    this.threadMemory = new InMemoryThread();\n    this.intentMemory = new InMemoryIntent();\n    this.workflowMemory = new InMemoryWorkflow();\n  }\n\n  public async connect(): Promise<void> {\n    this.connected = true;\n  }\n\n  public async disconnect(): Promise<void> {\n    this.connected = false;\n  }\n\n  public isConnected(): boolean {\n    return this.connected;\n  }\n\n  public getAgentMemory(): IAgentMemory {\n    return this.agentMemory;\n  }\n\n  public getThreadMemory(): IThreadMemory {\n    return this.threadMemory;\n  }\n\n  public getIntentMemory(): IIntentMemory {\n    return this.intentMemory;\n  }\n\n  public getWorkflowMemory(): IWorkflowMemory {\n    return this.workflowMemory;\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACEO,IAAM,gBAAN,MAA4C;AAAA,EACzC,SAAiB;AAAA,EAEzB,MAAa,iBAAkC;AAC7C,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,MAAa,kBAAkB,QAA+B;AAC5D,SAAK,SAAS;AAAA,EAChB;AACF;;;ACZA,yBAA2B;AAIpB,IAAM,iBAAN,MAA8C;AAAA,EAC5C,UAA+B,oBAAI,IAAI;AAAA,EAE9C,MAAa,UAAU,UAA+C;AACpE,WAAO,KAAK,QAAQ,IAAI,QAAQ;AAAA,EAClC;AAAA,EAEA,MAAa,gBAAgB,YAAiD;AAC5E,WAAO,MAAM,KAAK,KAAK,QAAQ,OAAO,CAAC,EAAE,KAAK,YAAU,OAAO,SAAS,UAAU;AAAA,EACpF;AAAA,EAEA,MAAa,WAAW,QAA+B;AAErD,UAAM,WAAW,OAAO,UAAM,+BAAW;AACzC,UAAM,eAAe,EAAE,GAAG,QAAQ,IAAI,SAAS;AAC/C,SAAK,QAAQ,IAAI,UAAU,YAAY;AAAA,EACzC;AAAA,EAEA,MAAa,aAAa,UAAkB,QAA+B;AACzE,SAAK,QAAQ,IAAI,UAAU,MAAM;AAAA,EACnC;AAAA,EAEA,MAAa,aAAa,UAAiC;AACzD,SAAK,QAAQ,OAAO,QAAQ;AAAA,EAC9B;AAAA,EAEA,MAAa,cAAiC;AAC5C,WAAO,MAAM,KAAK,KAAK,QAAQ,OAAO,CAAC;AAAA,EACzC;AACF;;;ACfO,IAAM,iBAAN,MAA8C;AAAA,EAC5C,UAA6C,oBAAI,IAAI;AAAA,EACrD,kBAA4D,oBAAI,IAAI;AAAA,EAEnE,YAAY,QAAgB,UAAkB;AACpD,WAAO,GAAG,MAAM,IAAI,QAAQ;AAAA,EAC9B;AAAA,EAEA,MAAa,UACX,QACA,UACmC;AACnC,UAAM,MAAM,KAAK,YAAY,QAAQ,QAAQ;AAC7C,UAAM,MAAM,KAAK,QAAQ,IAAI,GAAG;AAChC,QAAI,KAAK;AACP,YAAM,eAA6B;AAAA,QACjC;AAAA,QACA;AAAA,QACA,MAAM,IAAI;AAAA,QACV,OAAO,IAAI;AAAA,QACX,UAAU,IAAI;AAAA,MAChB;AACA,aAAO;AAAA,IACT;AACA,WAAO;AAAA,EACT;AAAA,EAEA,MAAa,aACX,MACA,QACA,UACA,OACuB;AACvB,UAAM,MAAM,KAAK,IAAI;AACrB,UAAM,MAAM,KAAK,YAAY,QAAQ,QAAQ;AAC7C,QAAI,CAAC,KAAK,gBAAgB,IAAI,MAAM,GAAG;AACrC,WAAK,gBAAgB,IAAI,QAAQ,oBAAI,IAAI,CAAC;AAAA,IAC5C;AACA,QAAI,CAAC,KAAK,QAAQ,IAAI,GAAG,GAAG;AAC1B,WAAK,QAAQ,IAAI,KAAK,EAAE,MAAM,OAAO,UAAU,CAAC,EAAE,CAAC;AACnD,YAAM,WAAmC;AAAA,QACvC;AAAA,QAAM;AAAA,QAAQ;AAAA,QAAU;AAAA,QAAO,WAAW;AAAA,QAAK,WAAW;AAAA,MAC5D;AACA,WAAK,gBAAgB,IAAI,MAAM,GAAG,IAAI,QAAQ;AAAA,IAChD;AAEA,WAAO,EAAE,MAAM,OAAO,UAAU,QAAQ,UAAU,CAAC,EAAE;AAAA,EACvD;AAAA,EAEA,MAAa,oBACX,QACA,UACA,UACe;AACf,UAAM,MAAM,KAAK,YAAY,QAAQ,QAAQ;AAC7C,UAAM,SAAS,KAAK,QAAQ,IAAI,GAAG;AACnC,eAAW,WAAW,UAAU;AAC9B,cAAQ,SAAS,KAAK,OAAO;AAAA,IAC/B;AAAA,EACF;AAAA,EAEA,MAAa,aAAa,QAAgB,UAAiC;AACzE,UAAM,MAAM,KAAK,YAAY,QAAQ,QAAQ;AAC7C,SAAK,QAAQ,OAAO,GAAG;AAGvB,UAAM,cAAc,KAAK,gBAAgB,IAAI,MAAM;AACnD,QAAI,aAAa;AACf,YAAM,mBAAmB,MAAM,KAAK,WAAW,EAAE;AAAA,QAC/C,cAAY,SAAS,aAAa;AAAA,MACpC;AACA,UAAI,kBAAkB;AACpB,oBAAY,OAAO,gBAAgB;AAAA,MACrC;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAa,YAAY,QAA2C;AAClE,UAAM,UAAU,KAAK,gBAAgB,IAAI,MAAM;AAC/C,QAAI,SAAS;AACX,aAAO,MAAM,KAAK,OAAO;AAAA,IAC3B;AACA,WAAO,CAAC;AAAA,EACV;AACF;;;ACnGO,IAAM,mBAAN,MAAkD;AAAA,EAC/C,YAAmC,oBAAI,IAAI;AAAA,EAC3C,oBAA8C,oBAAI,IAAI;AAAA,EAE9D,MAAa,eAAe,UAAuC;AACjE,SAAK,UAAU,IAAI,SAAS,YAAY,QAAQ;AAEhD,QAAI,SAAS,QAAQ;AACnB,UAAI,CAAC,KAAK,kBAAkB,IAAI,SAAS,MAAM,GAAG;AAChD,aAAK,kBAAkB,IAAI,SAAS,QAAQ,oBAAI,IAAI,CAAC;AAAA,MACvD;AACA,WAAK,kBAAkB,IAAI,SAAS,MAAM,GAAG,IAAI,SAAS,UAAU;AAAA,IACtE;AAEA,WAAO;AAAA,EACT;AAAA,EAEA,MAAa,YAAY,YAAmD;AAC1E,WAAO,KAAK,UAAU,IAAI,UAAU;AAAA,EACtC;AAAA,EAEA,MAAa,eAAe,YAAoB,SAA2C;AACzF,QAAI,CAAC,QAAQ,QAAQ;AACnB,YAAM,IAAI,MAAM,uCAAuC;AAAA,IACzD;AAEA,UAAM,WAAW,KAAK,UAAU,IAAI,UAAU;AAC9C,QAAI,CAAC,UAAU;AACb,YAAM,IAAI,MAAM,uBAAuB,UAAU,EAAE;AAAA,IACrD;AAEA,QAAI,SAAS,WAAW,QAAQ,QAAQ;AACtC,YAAM,IAAI,MAAM,oDAAoD;AAAA,IACtE;AAEA,UAAM,UAAU,EAAE,GAAG,UAAU,GAAG,SAAS,WAAW;AACtD,SAAK,UAAU,IAAI,YAAY,OAAO;AAAA,EACxC;AAAA,EAEA,MAAa,eAAe,YAAoB,QAA+B;AAC7E,UAAM,WAAW,KAAK,UAAU,IAAI,UAAU;AAC9C,QAAI,YAAY,SAAS,WAAW,QAAQ;AAC1C,WAAK,UAAU,OAAO,UAAU;AAChC,WAAK,kBAAkB,IAAI,MAAM,GAAG,OAAO,UAAU;AAAA,IACvD;AAAA,EACF;AAAA,EAEA,MAAa,cAAc,QAAsC;AAC/D,UAAM,YAAwB,CAAC;AAG/B,eAAW,YAAY,KAAK,UAAU,OAAO,GAAG;AAC9C,UAAI,CAAC,SAAS,QAAQ;AACpB,kBAAU,KAAK,QAAQ;AAAA,MACzB;AAAA,IACF;AAGA,QAAI,QAAQ;AACV,YAAM,kBAAkB,KAAK,kBAAkB,IAAI,MAAM;AACzD,UAAI,iBAAiB;AACnB,mBAAW,cAAc,iBAAiB;AACxC,gBAAM,WAAW,KAAK,UAAU,IAAI,UAAU;AAC9C,cAAI,UAAU;AACZ,sBAAU,KAAK,QAAQ;AAAA,UACzB;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AACF;;;ACrEO,IAAM,iBAAN,MAAM,gBAAkC;AAAA,EAC7C,OAAe;AAAA,EACP,YAAqB;AAAA,EAErB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,cAAc;AACZ,QAAI,CAAC,gBAAe,UAAU;AAC5B,sBAAe,WAAW;AAAA,IAC5B;AAEA,SAAK,cAAc,IAAI,cAAc;AACrC,SAAK,eAAe,IAAI,eAAe;AACvC,SAAK,eAAe,IAAI,eAAe;AACvC,SAAK,iBAAiB,IAAI,iBAAiB;AAAA,EAC7C;AAAA,EAEA,MAAa,UAAyB;AACpC,SAAK,YAAY;AAAA,EACnB;AAAA,EAEA,MAAa,aAA4B;AACvC,SAAK,YAAY;AAAA,EACnB;AAAA,EAEO,cAAuB;AAC5B,WAAO,KAAK;AAAA,EACd;AAAA,EAEO,iBAA+B;AACpC,WAAO,KAAK;AAAA,EACd;AAAA,EAEO,kBAAiC;AACtC,WAAO,KAAK;AAAA,EACd;AAAA,EAEO,kBAAiC;AACtC,WAAO,KAAK;AAAA,EACd;AAAA,EAEO,oBAAqC;AAC1C,WAAO,KAAK;AAAA,EACd;AACF;","names":[]}

package/dist/index.d.cts

@@ -1,4 +1,4 @@
-import { IMemory, IAgentMemory, IThreadMemory, IIntentMemory } from 'node_modules/@ainetwork/adk/dist/esm/modules/memory/base.memory';
+import { IMemory, IAgentMemory, IThreadMemory, IIntentMemory, IWorkflowMemory } from '@ainetwork/adk/modules';
 
 declare class InMemoryMemory implements IMemory {
     private static instance;
@@ -6,6 +6,7 @@ declare class InMemoryMemory implements IMemory {
     private agentMemory;
     private intentMemory;
     private threadMemory;
+    private workflowMemory;
     constructor();
     connect(): Promise<void>;
     disconnect(): Promise<void>;
@@ -13,6 +14,7 @@ declare class InMemoryMemory implements IMemory {
     getAgentMemory(): IAgentMemory;
     getThreadMemory(): IThreadMemory;
     getIntentMemory(): IIntentMemory;
+    getWorkflowMemory(): IWorkflowMemory;
 }
 
 export { InMemoryMemory };

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-import { IMemory, IAgentMemory, IThreadMemory, IIntentMemory } from 'node_modules/@ainetwork/adk/dist/esm/modules/memory/base.memory';
+import { IMemory, IAgentMemory, IThreadMemory, IIntentMemory, IWorkflowMemory } from '@ainetwork/adk/modules';
 
 declare class InMemoryMemory implements IMemory {
     private static instance;
@@ -6,6 +6,7 @@ declare class InMemoryMemory implements IMemory {
     private agentMemory;
     private intentMemory;
     private threadMemory;
+    private workflowMemory;
     constructor();
     connect(): Promise<void>;
     disconnect(): Promise<void>;
@@ -13,6 +14,7 @@ declare class InMemoryMemory implements IMemory {
     getAgentMemory(): IAgentMemory;
     getThreadMemory(): IThreadMemory;
     getIntentMemory(): IIntentMemory;
+    getWorkflowMemory(): IWorkflowMemory;
 }
 
 export { InMemoryMemory };

package/dist/index.js

@@ -20,8 +20,9 @@ var InMemoryIntent = class {
     return Array.from(this.intents.values()).find((intent) => intent.name === intentName);
   }
   async saveIntent(intent) {
-    const newId = randomUUID();
-    this.intents.set(newId, intent);
+    const intentId = intent.id || randomUUID();
+    const intentToSave = { ...intent, id: intentId };
+    this.intents.set(intentId, intentToSave);
   }
   async updateIntent(intentId, intent) {
     this.intents.set(intentId, intent);
@@ -86,7 +87,15 @@ var InMemoryThread = class {
   async deleteThread(userId, threadId) {
     const key = this.generateKey(userId, threadId);
     this.threads.delete(key);
-    this.userThreadIndex.delete(threadId);
+    const userThreads = this.userThreadIndex.get(userId);
+    if (userThreads) {
+      const metadataToDelete = Array.from(userThreads).find(
+        (metadata) => metadata.threadId === threadId
+      );
+      if (metadataToDelete) {
+        userThreads.delete(metadataToDelete);
+      }
+    }
   }
   async listThreads(userId) {
     const threads = this.userThreadIndex.get(userId);
@@ -97,6 +106,66 @@ var InMemoryThread = class {
   }
 };
 
+// implements/workflow.memory.ts
+var InMemoryWorkflow = class {
+  workflows = /* @__PURE__ */ new Map();
+  userWorkflowIndex = /* @__PURE__ */ new Map();
+  async createWorkflow(workflow) {
+    this.workflows.set(workflow.workflowId, workflow);
+    if (workflow.userId) {
+      if (!this.userWorkflowIndex.has(workflow.userId)) {
+        this.userWorkflowIndex.set(workflow.userId, /* @__PURE__ */ new Set());
+      }
+      this.userWorkflowIndex.get(workflow.userId)?.add(workflow.workflowId);
+    }
+    return workflow;
+  }
+  async getWorkflow(workflowId) {
+    return this.workflows.get(workflowId);
+  }
+  async updateWorkflow(workflowId, updates) {
+    if (!updates.userId) {
+      throw new Error("userId is required for updateWorkflow");
+    }
+    const existing = this.workflows.get(workflowId);
+    if (!existing) {
+      throw new Error(`Workflow not found: ${workflowId}`);
+    }
+    if (existing.userId !== updates.userId) {
+      throw new Error("Unauthorized: userId does not match workflow owner");
+    }
+    const updated = { ...existing, ...updates, workflowId };
+    this.workflows.set(workflowId, updated);
+  }
+  async deleteWorkflow(workflowId, userId) {
+    const workflow = this.workflows.get(workflowId);
+    if (workflow && workflow.userId === userId) {
+      this.workflows.delete(workflowId);
+      this.userWorkflowIndex.get(userId)?.delete(workflowId);
+    }
+  }
+  async listWorkflows(userId) {
+    const workflows = [];
+    for (const workflow of this.workflows.values()) {
+      if (!workflow.userId) {
+        workflows.push(workflow);
+      }
+    }
+    if (userId) {
+      const userWorkflowIds = this.userWorkflowIndex.get(userId);
+      if (userWorkflowIds) {
+        for (const workflowId of userWorkflowIds) {
+          const workflow = this.workflows.get(workflowId);
+          if (workflow) {
+            workflows.push(workflow);
+          }
+        }
+      }
+    }
+    return workflows;
+  }
+};
+
 // implements/base.memory.ts
 var InMemoryMemory = class _InMemoryMemory {
   static instance;
@@ -104,6 +173,7 @@ var InMemoryMemory = class _InMemoryMemory {
   agentMemory;
   intentMemory;
   threadMemory;
+  workflowMemory;
   constructor() {
     if (!_InMemoryMemory.instance) {
       _InMemoryMemory.instance = this;
@@ -111,6 +181,7 @@ var InMemoryMemory = class _InMemoryMemory {
     this.agentMemory = new InMemoryAgent();
     this.threadMemory = new InMemoryThread();
     this.intentMemory = new InMemoryIntent();
+    this.workflowMemory = new InMemoryWorkflow();
   }
   async connect() {
     this.connected = true;
@@ -130,6 +201,9 @@ var InMemoryMemory = class _InMemoryMemory {
   getIntentMemory() {
     return this.intentMemory;
   }
+  getWorkflowMemory() {
+    return this.workflowMemory;
+  }
 };
 export {
   InMemoryMemory

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../implements/agent.memory.ts","../implements/intent.memory.ts","../implements/thread.memory.ts","../implements/base.memory.ts"],"sourcesContent":["import { IAgentMemory } from \"@ainetwork/adk/modules\";\n\nexport class InMemoryAgent implements IAgentMemory {\n  private prompt: string = \"\";\n\n  public async getAgentPrompt(): Promise<string> {\n    return this.prompt;\n  }\n\n  public async updateAgentPrompt(prompt: string): Promise<void> {\n    this.prompt = prompt;\n  }\n}\n","import { randomUUID } from \"node:crypto\";\nimport type { Intent } from \"@ainetwork/adk/types/memory\";\nimport { IIntentMemory } from \"@ainetwork/adk/modules\";\n\nexport class InMemoryIntent implements IIntentMemory {\n  public intents: Map<string, Intent> = new Map();\n\n  public async getIntent(intentId: string): Promise<Intent | undefined> {\n    return this.intents.get(intentId);\n  }\n\n  public async getIntentByName(intentName: string): Promise<Intent | undefined> {\n    return Array.from(this.intents.values()).find(intent => intent.name === intentName);\n  }\n\n  public async saveIntent(intent: Intent): Promise<void> {\n    const newId = randomUUID();\n    this.intents.set(newId, intent);\n  }\n\n  public async updateIntent(intentId: string, intent: Intent): Promise<void> {\n    this.intents.set(intentId, intent);\n  }\n\n  public async deleteIntent(intentId: string): Promise<void> {\n    this.intents.delete(intentId);\n  }\n\n  public async listIntents(): Promise<Intent[]> {\n    return Array.from(this.intents.values());\n  }\n}","import type { MessageObject, ThreadObject, ThreadMetadata, ThreadType } from \"@ainetwork/adk/types/memory\";\nimport { IThreadMemory } from \"@ainetwork/adk/modules\";\n\ntype InMemoryThreadObject = {\n  type: ThreadType;\n  title: string;\n  messages: Array<MessageObject>;\n}\n\ntype InMemoryThreadMetadata = {\n  type: ThreadType;\n  userId: string;\n  threadId: string;\n  title: string;\n  updatedAt: number;\n  createdAt: number;\n}\n\nexport class InMemoryThread implements IThreadMemory {\n  public threads: Map<string, InMemoryThreadObject> = new Map();\n  public userThreadIndex: Map<string, Set<InMemoryThreadMetadata>> = new Map();\n\n  private generateKey(userId: string, threadId: string) {\n    return `${userId}:${threadId}`;\n  }\n\n  public async getThread(\n    userId: string,\n    threadId: string\n  ): Promise<ThreadObject | undefined> {\n    const key = this.generateKey(userId, threadId);\n    const res = this.threads.get(key);\n    if (res) {\n      const threadObject: ThreadObject = {\n        threadId,\n        userId,\n        type: res.type,\n        title: res.title,\n        messages: res.messages,\n      };\n      return threadObject;\n    }\n    return undefined;\n  }\n\n  public async createThread(\n    type: ThreadType,\n    userId: string,\n    threadId: string,\n    title: string\n  ): Promise<ThreadObject> {\n    const now = Date.now();\n    const key = this.generateKey(userId, threadId);\n    if (!this.userThreadIndex.has(userId)) {\n      this.userThreadIndex.set(userId, new Set());\n    }\n    if (!this.threads.has(key)) {\n      this.threads.set(key, { type, title, messages: [] });\n      const metadata: InMemoryThreadMetadata = {\n        type, userId, threadId, title, createdAt: now, updatedAt: now,\n      }\n      this.userThreadIndex.get(userId)?.add(metadata);\n    }\n\n    return { type, title, threadId, userId, messages: [] };\n  }\n\n  public async addMessagesToThread(\n    userId: string,\n    threadId: string,\n    messages: MessageObject[]\n  ): Promise<void> {\n    const key = this.generateKey(userId, threadId);\n    const thread = this.threads.get(key);\n    for (const message of messages) {\n      thread?.messages.push(message);\n    }\n  }\n\n  public async deleteThread(userId: string, threadId: string): Promise<void> {\n    const key = this.generateKey(userId, threadId);\n    this.threads.delete(key);\n    this.userThreadIndex.delete(threadId);\n  }\n\n  public async listThreads(userId: string): Promise<ThreadMetadata[]> {\n    const threads = this.userThreadIndex.get(userId);\n    if (threads) {\n      return Array.from(threads);\n    }\n    return [];\n  }\n}","import { IAgentMemory, IIntentMemory, IMemory, IThreadMemory } from \"node_modules/@ainetwork/adk/dist/esm/modules/memory/base.memory\";\nimport { InMemoryAgent } from \"./agent.memory\";\nimport { InMemoryIntent } from \"./intent.memory\";\nimport { InMemoryThread } from \"./thread.memory\";\n\nexport class InMemoryMemory implements IMemory {\n  private static instance: InMemoryMemory;\n  private connected: boolean = false;\n\n  private agentMemory: InMemoryAgent;\n  private intentMemory: InMemoryIntent;\n  private threadMemory: InMemoryThread;\n\n  constructor() {\n    if (!InMemoryMemory.instance) {\n      InMemoryMemory.instance = this;\n    }\n\n    this.agentMemory = new InMemoryAgent();\n    this.threadMemory = new InMemoryThread();\n    this.intentMemory = new InMemoryIntent();\n  }\n\n  public async connect(): Promise<void> {\n    this.connected = true;\n  }\n\n  public async disconnect(): Promise<void> {\n    this.connected = false;\n  }\n\n  public isConnected(): boolean {\n    return this.connected;\n  }\n\n  public getAgentMemory(): IAgentMemory {\n    return this.agentMemory;\n  }\n\n  public getThreadMemory(): IThreadMemory {\n    return this.threadMemory;\n  }\n\n  public getIntentMemory(): IIntentMemory {\n    return this.intentMemory;\n  }\n}\n"],"mappings":";AAEO,IAAM,gBAAN,MAA4C;AAAA,EACzC,SAAiB;AAAA,EAEzB,MAAa,iBAAkC;AAC7C,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,MAAa,kBAAkB,QAA+B;AAC5D,SAAK,SAAS;AAAA,EAChB;AACF;;;ACZA,SAAS,kBAAkB;AAIpB,IAAM,iBAAN,MAA8C;AAAA,EAC5C,UAA+B,oBAAI,IAAI;AAAA,EAE9C,MAAa,UAAU,UAA+C;AACpE,WAAO,KAAK,QAAQ,IAAI,QAAQ;AAAA,EAClC;AAAA,EAEA,MAAa,gBAAgB,YAAiD;AAC5E,WAAO,MAAM,KAAK,KAAK,QAAQ,OAAO,CAAC,EAAE,KAAK,YAAU,OAAO,SAAS,UAAU;AAAA,EACpF;AAAA,EAEA,MAAa,WAAW,QAA+B;AACrD,UAAM,QAAQ,WAAW;AACzB,SAAK,QAAQ,IAAI,OAAO,MAAM;AAAA,EAChC;AAAA,EAEA,MAAa,aAAa,UAAkB,QAA+B;AACzE,SAAK,QAAQ,IAAI,UAAU,MAAM;AAAA,EACnC;AAAA,EAEA,MAAa,aAAa,UAAiC;AACzD,SAAK,QAAQ,OAAO,QAAQ;AAAA,EAC9B;AAAA,EAEA,MAAa,cAAiC;AAC5C,WAAO,MAAM,KAAK,KAAK,QAAQ,OAAO,CAAC;AAAA,EACzC;AACF;;;ACbO,IAAM,iBAAN,MAA8C;AAAA,EAC5C,UAA6C,oBAAI,IAAI;AAAA,EACrD,kBAA4D,oBAAI,IAAI;AAAA,EAEnE,YAAY,QAAgB,UAAkB;AACpD,WAAO,GAAG,MAAM,IAAI,QAAQ;AAAA,EAC9B;AAAA,EAEA,MAAa,UACX,QACA,UACmC;AACnC,UAAM,MAAM,KAAK,YAAY,QAAQ,QAAQ;AAC7C,UAAM,MAAM,KAAK,QAAQ,IAAI,GAAG;AAChC,QAAI,KAAK;AACP,YAAM,eAA6B;AAAA,QACjC;AAAA,QACA;AAAA,QACA,MAAM,IAAI;AAAA,QACV,OAAO,IAAI;AAAA,QACX,UAAU,IAAI;AAAA,MAChB;AACA,aAAO;AAAA,IACT;AACA,WAAO;AAAA,EACT;AAAA,EAEA,MAAa,aACX,MACA,QACA,UACA,OACuB;AACvB,UAAM,MAAM,KAAK,IAAI;AACrB,UAAM,MAAM,KAAK,YAAY,QAAQ,QAAQ;AAC7C,QAAI,CAAC,KAAK,gBAAgB,IAAI,MAAM,GAAG;AACrC,WAAK,gBAAgB,IAAI,QAAQ,oBAAI,IAAI,CAAC;AAAA,IAC5C;AACA,QAAI,CAAC,KAAK,QAAQ,IAAI,GAAG,GAAG;AAC1B,WAAK,QAAQ,IAAI,KAAK,EAAE,MAAM,OAAO,UAAU,CAAC,EAAE,CAAC;AACnD,YAAM,WAAmC;AAAA,QACvC;AAAA,QAAM;AAAA,QAAQ;AAAA,QAAU;AAAA,QAAO,WAAW;AAAA,QAAK,WAAW;AAAA,MAC5D;AACA,WAAK,gBAAgB,IAAI,MAAM,GAAG,IAAI,QAAQ;AAAA,IAChD;AAEA,WAAO,EAAE,MAAM,OAAO,UAAU,QAAQ,UAAU,CAAC,EAAE;AAAA,EACvD;AAAA,EAEA,MAAa,oBACX,QACA,UACA,UACe;AACf,UAAM,MAAM,KAAK,YAAY,QAAQ,QAAQ;AAC7C,UAAM,SAAS,KAAK,QAAQ,IAAI,GAAG;AACnC,eAAW,WAAW,UAAU;AAC9B,cAAQ,SAAS,KAAK,OAAO;AAAA,IAC/B;AAAA,EACF;AAAA,EAEA,MAAa,aAAa,QAAgB,UAAiC;AACzE,UAAM,MAAM,KAAK,YAAY,QAAQ,QAAQ;AAC7C,SAAK,QAAQ,OAAO,GAAG;AACvB,SAAK,gBAAgB,OAAO,QAAQ;AAAA,EACtC;AAAA,EAEA,MAAa,YAAY,QAA2C;AAClE,UAAM,UAAU,KAAK,gBAAgB,IAAI,MAAM;AAC/C,QAAI,SAAS;AACX,aAAO,MAAM,KAAK,OAAO;AAAA,IAC3B;AACA,WAAO,CAAC;AAAA,EACV;AACF;;;ACvFO,IAAM,iBAAN,MAAM,gBAAkC;AAAA,EAC7C,OAAe;AAAA,EACP,YAAqB;AAAA,EAErB;AAAA,EACA;AAAA,EACA;AAAA,EAER,cAAc;AACZ,QAAI,CAAC,gBAAe,UAAU;AAC5B,sBAAe,WAAW;AAAA,IAC5B;AAEA,SAAK,cAAc,IAAI,cAAc;AACrC,SAAK,eAAe,IAAI,eAAe;AACvC,SAAK,eAAe,IAAI,eAAe;AAAA,EACzC;AAAA,EAEA,MAAa,UAAyB;AACpC,SAAK,YAAY;AAAA,EACnB;AAAA,EAEA,MAAa,aAA4B;AACvC,SAAK,YAAY;AAAA,EACnB;AAAA,EAEO,cAAuB;AAC5B,WAAO,KAAK;AAAA,EACd;AAAA,EAEO,iBAA+B;AACpC,WAAO,KAAK;AAAA,EACd;AAAA,EAEO,kBAAiC;AACtC,WAAO,KAAK;AAAA,EACd;AAAA,EAEO,kBAAiC;AACtC,WAAO,KAAK;AAAA,EACd;AACF;","names":[]}
+{"version":3,"sources":["../implements/agent.memory.ts","../implements/intent.memory.ts","../implements/thread.memory.ts","../implements/workflow.memory.ts","../implements/base.memory.ts"],"sourcesContent":["import { IAgentMemory } from \"@ainetwork/adk/modules\";\n\nexport class InMemoryAgent implements IAgentMemory {\n  private prompt: string = \"\";\n\n  public async getAgentPrompt(): Promise<string> {\n    return this.prompt;\n  }\n\n  public async updateAgentPrompt(prompt: string): Promise<void> {\n    this.prompt = prompt;\n  }\n}\n","import { randomUUID } from \"node:crypto\";\nimport type { Intent } from \"@ainetwork/adk/types/memory\";\nimport { IIntentMemory } from \"@ainetwork/adk/modules\";\n\nexport class InMemoryIntent implements IIntentMemory {\n  public intents: Map<string, Intent> = new Map();\n\n  public async getIntent(intentId: string): Promise<Intent | undefined> {\n    return this.intents.get(intentId);\n  }\n\n  public async getIntentByName(intentName: string): Promise<Intent | undefined> {\n    return Array.from(this.intents.values()).find(intent => intent.name === intentName);\n  }\n\n  public async saveIntent(intent: Intent): Promise<void> {\n    // Intent에 이미 id가 있으면 그것을 사용하고, 없으면 새로 생성\n    const intentId = intent.id || randomUUID();\n    const intentToSave = { ...intent, id: intentId };\n    this.intents.set(intentId, intentToSave);\n  }\n\n  public async updateIntent(intentId: string, intent: Intent): Promise<void> {\n    this.intents.set(intentId, intent);\n  }\n\n  public async deleteIntent(intentId: string): Promise<void> {\n    this.intents.delete(intentId);\n  }\n\n  public async listIntents(): Promise<Intent[]> {\n    return Array.from(this.intents.values());\n  }\n}","import type { MessageObject, ThreadObject, ThreadMetadata, ThreadType } from \"@ainetwork/adk/types/memory\";\nimport { IThreadMemory } from \"@ainetwork/adk/modules\";\n\ntype InMemoryThreadObject = {\n  type: ThreadType;\n  title: string;\n  messages: Array<MessageObject>;\n}\n\ntype InMemoryThreadMetadata = {\n  type: ThreadType;\n  userId: string;\n  threadId: string;\n  title: string;\n  updatedAt: number;\n  createdAt: number;\n}\n\nexport class InMemoryThread implements IThreadMemory {\n  public threads: Map<string, InMemoryThreadObject> = new Map();\n  public userThreadIndex: Map<string, Set<InMemoryThreadMetadata>> = new Map();\n\n  private generateKey(userId: string, threadId: string) {\n    return `${userId}:${threadId}`;\n  }\n\n  public async getThread(\n    userId: string,\n    threadId: string\n  ): Promise<ThreadObject | undefined> {\n    const key = this.generateKey(userId, threadId);\n    const res = this.threads.get(key);\n    if (res) {\n      const threadObject: ThreadObject = {\n        threadId,\n        userId,\n        type: res.type,\n        title: res.title,\n        messages: res.messages,\n      };\n      return threadObject;\n    }\n    return undefined;\n  }\n\n  public async createThread(\n    type: ThreadType,\n    userId: string,\n    threadId: string,\n    title: string\n  ): Promise<ThreadObject> {\n    const now = Date.now();\n    const key = this.generateKey(userId, threadId);\n    if (!this.userThreadIndex.has(userId)) {\n      this.userThreadIndex.set(userId, new Set());\n    }\n    if (!this.threads.has(key)) {\n      this.threads.set(key, { type, title, messages: [] });\n      const metadata: InMemoryThreadMetadata = {\n        type, userId, threadId, title, createdAt: now, updatedAt: now,\n      }\n      this.userThreadIndex.get(userId)?.add(metadata);\n    }\n\n    return { type, title, threadId, userId, messages: [] };\n  }\n\n  public async addMessagesToThread(\n    userId: string,\n    threadId: string,\n    messages: MessageObject[]\n  ): Promise<void> {\n    const key = this.generateKey(userId, threadId);\n    const thread = this.threads.get(key);\n    for (const message of messages) {\n      thread?.messages.push(message);\n    }\n  }\n\n  public async deleteThread(userId: string, threadId: string): Promise<void> {\n    const key = this.generateKey(userId, threadId);\n    this.threads.delete(key);\n\n    // userThreadIndex에서 해당 thread metadata 제거\n    const userThreads = this.userThreadIndex.get(userId);\n    if (userThreads) {\n      const metadataToDelete = Array.from(userThreads).find(\n        metadata => metadata.threadId === threadId\n      );\n      if (metadataToDelete) {\n        userThreads.delete(metadataToDelete);\n      }\n    }\n  }\n\n  public async listThreads(userId: string): Promise<ThreadMetadata[]> {\n    const threads = this.userThreadIndex.get(userId);\n    if (threads) {\n      return Array.from(threads);\n    }\n    return [];\n  }\n}","import { IWorkflowMemory } from \"@ainetwork/adk/modules\";\nimport type { Workflow } from \"@ainetwork/adk/types/memory\";\n\nexport class InMemoryWorkflow implements IWorkflowMemory {\n  private workflows: Map<string, Workflow> = new Map();\n  private userWorkflowIndex: Map<string, Set<string>> = new Map();\n\n  public async createWorkflow(workflow: Workflow): Promise<Workflow> {\n    this.workflows.set(workflow.workflowId, workflow);\n\n    if (workflow.userId) {\n      if (!this.userWorkflowIndex.has(workflow.userId)) {\n        this.userWorkflowIndex.set(workflow.userId, new Set());\n      }\n      this.userWorkflowIndex.get(workflow.userId)?.add(workflow.workflowId);\n    }\n\n    return workflow;\n  }\n\n  public async getWorkflow(workflowId: string): Promise<Workflow | undefined> {\n    return this.workflows.get(workflowId);\n  }\n\n  public async updateWorkflow(workflowId: string, updates: Partial<Workflow>): Promise<void> {\n    if (!updates.userId) {\n      throw new Error(\"userId is required for updateWorkflow\");\n    }\n\n    const existing = this.workflows.get(workflowId);\n    if (!existing) {\n      throw new Error(`Workflow not found: ${workflowId}`);\n    }\n\n    if (existing.userId !== updates.userId) {\n      throw new Error(\"Unauthorized: userId does not match workflow owner\");\n    }\n\n    const updated = { ...existing, ...updates, workflowId };\n    this.workflows.set(workflowId, updated);\n  }\n\n  public async deleteWorkflow(workflowId: string, userId: string): Promise<void> {\n    const workflow = this.workflows.get(workflowId);\n    if (workflow && workflow.userId === userId) {\n      this.workflows.delete(workflowId);\n      this.userWorkflowIndex.get(userId)?.delete(workflowId);\n    }\n  }\n\n  public async listWorkflows(userId?: string): Promise<Workflow[]> {\n    const workflows: Workflow[] = [];\n\n    // 템플릿 workflow (userId가 없는 것)은 항상 포함\n    for (const workflow of this.workflows.values()) {\n      if (!workflow.userId) {\n        workflows.push(workflow);\n      }\n    }\n\n    // userId가 있으면 해당 유저 소유 workflow도 포함\n    if (userId) {\n      const userWorkflowIds = this.userWorkflowIndex.get(userId);\n      if (userWorkflowIds) {\n        for (const workflowId of userWorkflowIds) {\n          const workflow = this.workflows.get(workflowId);\n          if (workflow) {\n            workflows.push(workflow);\n          }\n        }\n      }\n    }\n\n    return workflows;\n  }\n}\n","import { IAgentMemory, IIntentMemory, IMemory, IThreadMemory, IWorkflowMemory } from \"@ainetwork/adk/modules\";\nimport { InMemoryAgent } from \"./agent.memory\";\nimport { InMemoryIntent } from \"./intent.memory\";\nimport { InMemoryThread } from \"./thread.memory\";\nimport { InMemoryWorkflow } from \"./workflow.memory\";\n\nexport class InMemoryMemory implements IMemory {\n  private static instance: InMemoryMemory;\n  private connected: boolean = false;\n\n  private agentMemory: InMemoryAgent;\n  private intentMemory: InMemoryIntent;\n  private threadMemory: InMemoryThread;\n  private workflowMemory: InMemoryWorkflow;\n\n  constructor() {\n    if (!InMemoryMemory.instance) {\n      InMemoryMemory.instance = this;\n    }\n\n    this.agentMemory = new InMemoryAgent();\n    this.threadMemory = new InMemoryThread();\n    this.intentMemory = new InMemoryIntent();\n    this.workflowMemory = new InMemoryWorkflow();\n  }\n\n  public async connect(): Promise<void> {\n    this.connected = true;\n  }\n\n  public async disconnect(): Promise<void> {\n    this.connected = false;\n  }\n\n  public isConnected(): boolean {\n    return this.connected;\n  }\n\n  public getAgentMemory(): IAgentMemory {\n    return this.agentMemory;\n  }\n\n  public getThreadMemory(): IThreadMemory {\n    return this.threadMemory;\n  }\n\n  public getIntentMemory(): IIntentMemory {\n    return this.intentMemory;\n  }\n\n  public getWorkflowMemory(): IWorkflowMemory {\n    return this.workflowMemory;\n  }\n}\n"],"mappings":";AAEO,IAAM,gBAAN,MAA4C;AAAA,EACzC,SAAiB;AAAA,EAEzB,MAAa,iBAAkC;AAC7C,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,MAAa,kBAAkB,QAA+B;AAC5D,SAAK,SAAS;AAAA,EAChB;AACF;;;ACZA,SAAS,kBAAkB;AAIpB,IAAM,iBAAN,MAA8C;AAAA,EAC5C,UAA+B,oBAAI,IAAI;AAAA,EAE9C,MAAa,UAAU,UAA+C;AACpE,WAAO,KAAK,QAAQ,IAAI,QAAQ;AAAA,EAClC;AAAA,EAEA,MAAa,gBAAgB,YAAiD;AAC5E,WAAO,MAAM,KAAK,KAAK,QAAQ,OAAO,CAAC,EAAE,KAAK,YAAU,OAAO,SAAS,UAAU;AAAA,EACpF;AAAA,EAEA,MAAa,WAAW,QAA+B;AAErD,UAAM,WAAW,OAAO,MAAM,WAAW;AACzC,UAAM,eAAe,EAAE,GAAG,QAAQ,IAAI,SAAS;AAC/C,SAAK,QAAQ,IAAI,UAAU,YAAY;AAAA,EACzC;AAAA,EAEA,MAAa,aAAa,UAAkB,QAA+B;AACzE,SAAK,QAAQ,IAAI,UAAU,MAAM;AAAA,EACnC;AAAA,EAEA,MAAa,aAAa,UAAiC;AACzD,SAAK,QAAQ,OAAO,QAAQ;AAAA,EAC9B;AAAA,EAEA,MAAa,cAAiC;AAC5C,WAAO,MAAM,KAAK,KAAK,QAAQ,OAAO,CAAC;AAAA,EACzC;AACF;;;ACfO,IAAM,iBAAN,MAA8C;AAAA,EAC5C,UAA6C,oBAAI,IAAI;AAAA,EACrD,kBAA4D,oBAAI,IAAI;AAAA,EAEnE,YAAY,QAAgB,UAAkB;AACpD,WAAO,GAAG,MAAM,IAAI,QAAQ;AAAA,EAC9B;AAAA,EAEA,MAAa,UACX,QACA,UACmC;AACnC,UAAM,MAAM,KAAK,YAAY,QAAQ,QAAQ;AAC7C,UAAM,MAAM,KAAK,QAAQ,IAAI,GAAG;AAChC,QAAI,KAAK;AACP,YAAM,eAA6B;AAAA,QACjC;AAAA,QACA;AAAA,QACA,MAAM,IAAI;AAAA,QACV,OAAO,IAAI;AAAA,QACX,UAAU,IAAI;AAAA,MAChB;AACA,aAAO;AAAA,IACT;AACA,WAAO;AAAA,EACT;AAAA,EAEA,MAAa,aACX,MACA,QACA,UACA,OACuB;AACvB,UAAM,MAAM,KAAK,IAAI;AACrB,UAAM,MAAM,KAAK,YAAY,QAAQ,QAAQ;AAC7C,QAAI,CAAC,KAAK,gBAAgB,IAAI,MAAM,GAAG;AACrC,WAAK,gBAAgB,IAAI,QAAQ,oBAAI,IAAI,CAAC;AAAA,IAC5C;AACA,QAAI,CAAC,KAAK,QAAQ,IAAI,GAAG,GAAG;AAC1B,WAAK,QAAQ,IAAI,KAAK,EAAE,MAAM,OAAO,UAAU,CAAC,EAAE,CAAC;AACnD,YAAM,WAAmC;AAAA,QACvC;AAAA,QAAM;AAAA,QAAQ;AAAA,QAAU;AAAA,QAAO,WAAW;AAAA,QAAK,WAAW;AAAA,MAC5D;AACA,WAAK,gBAAgB,IAAI,MAAM,GAAG,IAAI,QAAQ;AAAA,IAChD;AAEA,WAAO,EAAE,MAAM,OAAO,UAAU,QAAQ,UAAU,CAAC,EAAE;AAAA,EACvD;AAAA,EAEA,MAAa,oBACX,QACA,UACA,UACe;AACf,UAAM,MAAM,KAAK,YAAY,QAAQ,QAAQ;AAC7C,UAAM,SAAS,KAAK,QAAQ,IAAI,GAAG;AACnC,eAAW,WAAW,UAAU;AAC9B,cAAQ,SAAS,KAAK,OAAO;AAAA,IAC/B;AAAA,EACF;AAAA,EAEA,MAAa,aAAa,QAAgB,UAAiC;AACzE,UAAM,MAAM,KAAK,YAAY,QAAQ,QAAQ;AAC7C,SAAK,QAAQ,OAAO,GAAG;AAGvB,UAAM,cAAc,KAAK,gBAAgB,IAAI,MAAM;AACnD,QAAI,aAAa;AACf,YAAM,mBAAmB,MAAM,KAAK,WAAW,EAAE;AAAA,QAC/C,cAAY,SAAS,aAAa;AAAA,MACpC;AACA,UAAI,kBAAkB;AACpB,oBAAY,OAAO,gBAAgB;AAAA,MACrC;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAa,YAAY,QAA2C;AAClE,UAAM,UAAU,KAAK,gBAAgB,IAAI,MAAM;AAC/C,QAAI,SAAS;AACX,aAAO,MAAM,KAAK,OAAO;AAAA,IAC3B;AACA,WAAO,CAAC;AAAA,EACV;AACF;;;ACnGO,IAAM,mBAAN,MAAkD;AAAA,EAC/C,YAAmC,oBAAI,IAAI;AAAA,EAC3C,oBAA8C,oBAAI,IAAI;AAAA,EAE9D,MAAa,eAAe,UAAuC;AACjE,SAAK,UAAU,IAAI,SAAS,YAAY,QAAQ;AAEhD,QAAI,SAAS,QAAQ;AACnB,UAAI,CAAC,KAAK,kBAAkB,IAAI,SAAS,MAAM,GAAG;AAChD,aAAK,kBAAkB,IAAI,SAAS,QAAQ,oBAAI,IAAI,CAAC;AAAA,MACvD;AACA,WAAK,kBAAkB,IAAI,SAAS,MAAM,GAAG,IAAI,SAAS,UAAU;AAAA,IACtE;AAEA,WAAO;AAAA,EACT;AAAA,EAEA,MAAa,YAAY,YAAmD;AAC1E,WAAO,KAAK,UAAU,IAAI,UAAU;AAAA,EACtC;AAAA,EAEA,MAAa,eAAe,YAAoB,SAA2C;AACzF,QAAI,CAAC,QAAQ,QAAQ;AACnB,YAAM,IAAI,MAAM,uCAAuC;AAAA,IACzD;AAEA,UAAM,WAAW,KAAK,UAAU,IAAI,UAAU;AAC9C,QAAI,CAAC,UAAU;AACb,YAAM,IAAI,MAAM,uBAAuB,UAAU,EAAE;AAAA,IACrD;AAEA,QAAI,SAAS,WAAW,QAAQ,QAAQ;AACtC,YAAM,IAAI,MAAM,oDAAoD;AAAA,IACtE;AAEA,UAAM,UAAU,EAAE,GAAG,UAAU,GAAG,SAAS,WAAW;AACtD,SAAK,UAAU,IAAI,YAAY,OAAO;AAAA,EACxC;AAAA,EAEA,MAAa,eAAe,YAAoB,QAA+B;AAC7E,UAAM,WAAW,KAAK,UAAU,IAAI,UAAU;AAC9C,QAAI,YAAY,SAAS,WAAW,QAAQ;AAC1C,WAAK,UAAU,OAAO,UAAU;AAChC,WAAK,kBAAkB,IAAI,MAAM,GAAG,OAAO,UAAU;AAAA,IACvD;AAAA,EACF;AAAA,EAEA,MAAa,cAAc,QAAsC;AAC/D,UAAM,YAAwB,CAAC;AAG/B,eAAW,YAAY,KAAK,UAAU,OAAO,GAAG;AAC9C,UAAI,CAAC,SAAS,QAAQ;AACpB,kBAAU,KAAK,QAAQ;AAAA,MACzB;AAAA,IACF;AAGA,QAAI,QAAQ;AACV,YAAM,kBAAkB,KAAK,kBAAkB,IAAI,MAAM;AACzD,UAAI,iBAAiB;AACnB,mBAAW,cAAc,iBAAiB;AACxC,gBAAM,WAAW,KAAK,UAAU,IAAI,UAAU;AAC9C,cAAI,UAAU;AACZ,sBAAU,KAAK,QAAQ;AAAA,UACzB;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AACF;;;ACrEO,IAAM,iBAAN,MAAM,gBAAkC;AAAA,EAC7C,OAAe;AAAA,EACP,YAAqB;AAAA,EAErB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,cAAc;AACZ,QAAI,CAAC,gBAAe,UAAU;AAC5B,sBAAe,WAAW;AAAA,IAC5B;AAEA,SAAK,cAAc,IAAI,cAAc;AACrC,SAAK,eAAe,IAAI,eAAe;AACvC,SAAK,eAAe,IAAI,eAAe;AACvC,SAAK,iBAAiB,IAAI,iBAAiB;AAAA,EAC7C;AAAA,EAEA,MAAa,UAAyB;AACpC,SAAK,YAAY;AAAA,EACnB;AAAA,EAEA,MAAa,aAA4B;AACvC,SAAK,YAAY;AAAA,EACnB;AAAA,EAEO,cAAuB;AAC5B,WAAO,KAAK;AAAA,EACd;AAAA,EAEO,iBAA+B;AACpC,WAAO,KAAK;AAAA,EACd;AAAA,EAEO,kBAAiC;AACtC,WAAO,KAAK;AAAA,EACd;AAAA,EAEO,kBAAiC;AACtC,WAAO,KAAK;AAAA,EACd;AAAA,EAEO,oBAAqC;AAC1C,WAAO,KAAK;AAAA,EACd;AACF;","names":[]}

package/implements/base.memory.ts

@@ -1,7 +1,8 @@
-import { IAgentMemory, IIntentMemory, IMemory, IThreadMemory } from "node_modules/@ainetwork/adk/dist/esm/modules/memory/base.memory";
+import { IAgentMemory, IIntentMemory, IMemory, IThreadMemory, IWorkflowMemory } from "@ainetwork/adk/modules";
 import { InMemoryAgent } from "./agent.memory";
 import { InMemoryIntent } from "./intent.memory";
 import { InMemoryThread } from "./thread.memory";
+import { InMemoryWorkflow } from "./workflow.memory";
 
 export class InMemoryMemory implements IMemory {
   private static instance: InMemoryMemory;
@@ -10,6 +11,7 @@ export class InMemoryMemory implements IMemory {
   private agentMemory: InMemoryAgent;
   private intentMemory: InMemoryIntent;
   private threadMemory: InMemoryThread;
+  private workflowMemory: InMemoryWorkflow;
 
   constructor() {
     if (!InMemoryMemory.instance) {
@@ -19,6 +21,7 @@ export class InMemoryMemory implements IMemory {
     this.agentMemory = new InMemoryAgent();
     this.threadMemory = new InMemoryThread();
     this.intentMemory = new InMemoryIntent();
+    this.workflowMemory = new InMemoryWorkflow();
   }
 
   public async connect(): Promise<void> {
@@ -44,4 +47,8 @@ export class InMemoryMemory implements IMemory {
   public getIntentMemory(): IIntentMemory {
     return this.intentMemory;
   }
+
+  public getWorkflowMemory(): IWorkflowMemory {
+    return this.workflowMemory;
+  }
 }

package/implements/intent.memory.ts

@@ -14,8 +14,10 @@ export class InMemoryIntent implements IIntentMemory {
   }
 
   public async saveIntent(intent: Intent): Promise<void> {
-    const newId = randomUUID();
-    this.intents.set(newId, intent);
+    // Intent에 이미 id가 있으면 그것을 사용하고, 없으면 새로 생성
+    const intentId = intent.id || randomUUID();
+    const intentToSave = { ...intent, id: intentId };
+    this.intents.set(intentId, intentToSave);
   }
 
   public async updateIntent(intentId: string, intent: Intent): Promise<void> {

package/implements/thread.memory.ts

@@ -80,7 +80,17 @@ export class InMemoryThread implements IThreadMemory {
   public async deleteThread(userId: string, threadId: string): Promise<void> {
     const key = this.generateKey(userId, threadId);
     this.threads.delete(key);
-    this.userThreadIndex.delete(threadId);
+
+    // userThreadIndex에서 해당 thread metadata 제거
+    const userThreads = this.userThreadIndex.get(userId);
+    if (userThreads) {
+      const metadataToDelete = Array.from(userThreads).find(
+        metadata => metadata.threadId === threadId
+      );
+      if (metadataToDelete) {
+        userThreads.delete(metadataToDelete);
+      }
+    }
   }
 
   public async listThreads(userId: string): Promise<ThreadMetadata[]> {

package/implements/workflow.memory.ts

@@ -0,0 +1,76 @@
+import { IWorkflowMemory } from "@ainetwork/adk/modules";
+import type { Workflow } from "@ainetwork/adk/types/memory";
+
+export class InMemoryWorkflow implements IWorkflowMemory {
+  private workflows: Map<string, Workflow> = new Map();
+  private userWorkflowIndex: Map<string, Set<string>> = new Map();
+
+  public async createWorkflow(workflow: Workflow): Promise<Workflow> {
+    this.workflows.set(workflow.workflowId, workflow);
+
+    if (workflow.userId) {
+      if (!this.userWorkflowIndex.has(workflow.userId)) {
+        this.userWorkflowIndex.set(workflow.userId, new Set());
+      }
+      this.userWorkflowIndex.get(workflow.userId)?.add(workflow.workflowId);
+    }
+
+    return workflow;
+  }
+
+  public async getWorkflow(workflowId: string): Promise<Workflow | undefined> {
+    return this.workflows.get(workflowId);
+  }
+
+  public async updateWorkflow(workflowId: string, updates: Partial<Workflow>): Promise<void> {
+    if (!updates.userId) {
+      throw new Error("userId is required for updateWorkflow");
+    }
+
+    const existing = this.workflows.get(workflowId);
+    if (!existing) {
+      throw new Error(`Workflow not found: ${workflowId}`);
+    }
+
+    if (existing.userId !== updates.userId) {
+      throw new Error("Unauthorized: userId does not match workflow owner");
+    }
+
+    const updated = { ...existing, ...updates, workflowId };
+    this.workflows.set(workflowId, updated);
+  }
+
+  public async deleteWorkflow(workflowId: string, userId: string): Promise<void> {
+    const workflow = this.workflows.get(workflowId);
+    if (workflow && workflow.userId === userId) {
+      this.workflows.delete(workflowId);
+      this.userWorkflowIndex.get(userId)?.delete(workflowId);
+    }
+  }
+
+  public async listWorkflows(userId?: string): Promise<Workflow[]> {
+    const workflows: Workflow[] = [];
+
+    // 템플릿 workflow (userId가 없는 것)은 항상 포함
+    for (const workflow of this.workflows.values()) {
+      if (!workflow.userId) {
+        workflows.push(workflow);
+      }
+    }
+
+    // userId가 있으면 해당 유저 소유 workflow도 포함
+    if (userId) {
+      const userWorkflowIds = this.userWorkflowIndex.get(userId);
+      if (userWorkflowIds) {
+        for (const workflowId of userWorkflowIds) {
+          const workflow = this.workflows.get(workflowId);
+          if (workflow) {
+            workflows.push(workflow);
+          }
+        }
+      }
+    }
+
+    return workflows;
+  }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ainetwork/adk-provider-memory-inmemory",
-  "version": "0.3.3",
+  "version": "0.3.5",
   "author": "AI Network (https://ainetwork.ai)",
   "type": "module",
   "engines": {
@@ -21,7 +21,7 @@
     "clean": "rm -rf dist"
   },
   "dependencies": {
-    "@ainetwork/adk": "^0.3.3"
+    "@ainetwork/adk": "^0.3.5"
   },
   "devDependencies": {
     "typescript": "^5.0.0"
@@ -30,5 +30,5 @@
   "publishConfig": {
     "access": "public"
   },
-  "gitHead": "e46af4210c3dc6ea073797d4ba465a24651c8ca2"
+  "gitHead": "0361ececa3c7404035d1b67c4830f52fa904d3d8"
 }