@ainetwork/adk-provider-memory-inmemory 0.1.1 → 0.1.3

package/dist/index.cjs

@@ -20,31 +20,106 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // index.ts
 var index_exports = {};
 __export(index_exports, {
-  InMemoryMemory: () => InMemoryMemory
+  InMemoryIntent: () => InMemoryIntent,
+  InMemorySession: () => InMemorySession
 });
 module.exports = __toCommonJS(index_exports);
+
+// implements/session.memory.ts
 var import_node_crypto = require("crypto");
-var import_modules = require("@ainetwork/adk/modules");
-var InMemoryMemory = class extends import_modules.BaseMemory {
-  sessionHistory;
-  constructor() {
-    super();
-    this.sessionHistory = /* @__PURE__ */ new Map();
+var InMemorySession = class {
+  sessions = /* @__PURE__ */ new Map();
+  userSessionIndex = /* @__PURE__ */ new Map();
+  async connect() {
+  }
+  async disconnect() {
+  }
+  isConnected() {
+    return true;
+  }
+  generateKey(userId, sessionId) {
+    return `${userId}:${sessionId}`;
   }
-  async getSessionHistory(sessionId) {
-    return this.sessionHistory.get(sessionId) || { chats: {} };
+  async getSession(userId, sessionId) {
+    const key = this.generateKey(userId, sessionId);
+    const res = this.sessions.get(key);
+    if (res) {
+      const sessionObject = {
+        chats: Object.fromEntries(res.chats)
+      };
+      return sessionObject;
+    }
+    return void 0;
   }
-  async updateSessionHistory(sessionId, chat) {
+  async createSession(userId, sessionId) {
+    const key = this.generateKey(userId, sessionId);
+    if (!this.userSessionIndex.has(userId)) {
+      this.userSessionIndex.set(userId, /* @__PURE__ */ new Set());
+    }
+    if (!this.sessions.has(key)) {
+      this.sessions.set(key, { chats: /* @__PURE__ */ new Map() });
+      const metadata = {
+        sessionId,
+        createdAt: Date.now(),
+        updatedAt: Date.now()
+      };
+      this.userSessionIndex.get(userId)?.add(metadata);
+    }
+  }
+  async addChatToSession(userId, sessionId, chat) {
+    const key = this.generateKey(userId, sessionId);
     const newChatId = (0, import_node_crypto.randomUUID)();
-    const history = await this.getSessionHistory(sessionId);
-    history.chats[newChatId] = chat;
-    this.sessionHistory.set(sessionId, history);
+    if (!this.sessions.has(key)) {
+      await this.createSession(userId, sessionId);
+    }
+    const sessions = this.sessions.get(key);
+    this.sessions.get(key)?.chats.set(newChatId, chat);
+  }
+  async deleteSession(userId, sessionId) {
+    const key = this.generateKey(userId, sessionId);
+    this.sessions.delete(key);
+    this.userSessionIndex.delete(sessionId);
+  }
+  async listSessions(userId) {
+    const sessions = this.userSessionIndex.get(userId);
+    if (sessions) {
+      return Array.from(sessions);
+    }
+    return [];
+  }
+};
+
+// implements/intent.memory.ts
+var import_node_crypto2 = require("crypto");
+var InMemoryIntent = class {
+  intents = /* @__PURE__ */ new Map();
+  async connect() {
+  }
+  async disconnect() {
+  }
+  isConnected() {
+    return true;
+  }
+  async getIntent(intentId) {
+    return this.intents.get(intentId);
+  }
+  async saveIntent(intent) {
+    const newId = (0, import_node_crypto2.randomUUID)();
+    this.intents.set(newId, intent);
+  }
+  async updateIntent(intentId, intent) {
+    this.intents.set(intentId, intent);
+  }
+  async deleteIntent(intentId) {
+    this.intents.delete(intentId);
   }
-  async storeQueryAndIntent(query, intent, sessionId) {
+  async listIntents() {
+    return Array.from(this.intents.values());
   }
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  InMemoryMemory
+  InMemoryIntent,
+  InMemorySession
 });
 //# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../index.ts"],"sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport type { ChatObject, SessionObject } from \"@ainetwork/adk/types/memory\";\nimport { BaseMemory } from \"@ainetwork/adk/modules\";\n\nexport class InMemoryMemory extends BaseMemory {\n\tpublic sessionHistory: Map<string, SessionObject>;\n\n\tconstructor() {\n\t\tsuper();\n\t\tthis.sessionHistory = new Map<string, SessionObject>();\n\t}\n\n\tpublic async getSessionHistory(sessionId: string): Promise<SessionObject> {\n\t\treturn this.sessionHistory.get(sessionId) || { chats: {} };\n\t}\n\n\tpublic async updateSessionHistory(\n\t\tsessionId: string,\n\t\tchat: ChatObject,\n\t): Promise<void> {\n\t\tconst newChatId = randomUUID();\n\t\tconst history = await this.getSessionHistory(sessionId);\n\t\thistory.chats[newChatId] = chat;\n\t\tthis.sessionHistory.set(sessionId, history);\n\t}\n\n\tpublic async storeQueryAndIntent(\n\t\tquery: string,\n\t\tintent: string,\n\t\tsessionId: string,\n\t) {}\n}"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,yBAA2B;AAE3B,qBAA2B;AAEpB,IAAM,iBAAN,cAA6B,0BAAW;AAAA,EACvC;AAAA,EAEP,cAAc;AACb,UAAM;AACN,SAAK,iBAAiB,oBAAI,IAA2B;AAAA,EACtD;AAAA,EAEA,MAAa,kBAAkB,WAA2C;AACzE,WAAO,KAAK,eAAe,IAAI,SAAS,KAAK,EAAE,OAAO,CAAC,EAAE;AAAA,EAC1D;AAAA,EAEA,MAAa,qBACZ,WACA,MACgB;AAChB,UAAM,gBAAY,+BAAW;AAC7B,UAAM,UAAU,MAAM,KAAK,kBAAkB,SAAS;AACtD,YAAQ,MAAM,SAAS,IAAI;AAC3B,SAAK,eAAe,IAAI,WAAW,OAAO;AAAA,EAC3C;AAAA,EAEA,MAAa,oBACZ,OACA,QACA,WACC;AAAA,EAAC;AACJ;","names":[]}
+{"version":3,"sources":["../index.ts","../implements/session.memory.ts","../implements/intent.memory.ts"],"sourcesContent":["export { InMemorySession } from \"./implements/session.memory\";\nexport { InMemoryIntent } from \"./implements/intent.memory\";","import { randomUUID } from \"node:crypto\";\nimport type { ChatObject, SessionObject, SessionMetadata } from \"@ainetwork/adk/types/memory\";\nimport { ISessionMemory } from \"@ainetwork/adk/modules\";\n\ntype InMemorySessionObject = {\n  chats: Map<string, ChatObject>\n}\n\ntype InMemorySessionMetadata = {\n  sessionId: string;\n  title?: string;\n  updatedAt: number;\n  createdAt: number;\n}\n\nexport class InMemorySession implements ISessionMemory {\n\tpublic sessions: Map<string, InMemorySessionObject> = new Map();\n  public userSessionIndex: Map<string, Set<InMemorySessionMetadata>> = new Map();\n\n  public async connect(): Promise<void> {}\n  public async disconnect(): Promise<void> {}\n  public isConnected(): boolean {\n    return true;\n  }\n\n  private generateKey(userId: string, sessionId: string) {\n    return `${userId}:${sessionId}`;\n  }\n\n  public async getSession(userId: string, sessionId: string): Promise<SessionObject | undefined> {\n    const key = this.generateKey(userId, sessionId);\n    const res = this.sessions.get(key);\n    if (res) {\n      const sessionObject: SessionObject = {\n        chats: Object.fromEntries(res.chats)\n      };\n      return sessionObject;\n    }\n    return undefined;\n  };\n\n\tpublic async createSession(userId: string, sessionId: string): Promise<void> {\n    const key = this.generateKey(userId, sessionId);\n    if (!this.userSessionIndex.has(userId)) {\n      this.userSessionIndex.set(userId, new Set());\n    }\n    if (!this.sessions.has(key)) {\n      this.sessions.set(key, { chats: new Map() });\n      const metadata: InMemorySessionMetadata = {\n        sessionId, createdAt: Date.now(), updatedAt: Date.now(),\n      }\n      this.userSessionIndex.get(userId)?.add(metadata);\n    }\n  };\n\n\tpublic async addChatToSession(userId: string, sessionId: string, chat: ChatObject): Promise<void> {\n    const key = this.generateKey(userId, sessionId);\n    const newChatId = randomUUID();\n    if (!this.sessions.has(key)) {\n      await this.createSession(userId, sessionId);\n    }\n    const sessions = this.sessions.get(key);\n    this.sessions.get(key)?.chats.set(newChatId, chat);\n  };\n\n\tpublic async deleteSession(userId: string, sessionId: string): Promise<void> {\n    const key = this.generateKey(userId, sessionId);\n    this.sessions.delete(key);\n    this.userSessionIndex.delete(sessionId);\n  };\n\n\tpublic async listSessions(userId: string): Promise<SessionMetadata[]> {\n    const sessions = this.userSessionIndex.get(userId);\n    if (sessions) {\n      return Array.from(sessions);\n    }\n    return [];\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\tpublic intents: Map<string, Intent> = new Map();\n\n  public async connect(): Promise<void> {}\n  public async disconnect(): Promise<void> {}\n  public isConnected(): boolean {\n    return true;\n  }\n\n  public async getIntent(intentId: string): Promise<Intent | undefined> {\n    return this.intents.get(intentId);\n  };\n\n\tpublic async saveIntent(intent: Intent): Promise<void> {\n    const newId = randomUUID();\n    this.intents.set(newId, intent);\n  };\n\n\tpublic async updateIntent(intentId: string, intent: Intent): Promise<void> {\n    this.intents.set(intentId, intent);\n  };\n\n\tpublic async deleteIntent(intentId: string): Promise<void> {\n    this.intents.delete(intentId);\n  };\n\n\tpublic async listIntents(): Promise<Intent[]> {\n    return Array.from(this.intents.values());\n  };\n}"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,yBAA2B;AAepB,IAAM,kBAAN,MAAgD;AAAA,EAC/C,WAA+C,oBAAI,IAAI;AAAA,EACtD,mBAA8D,oBAAI,IAAI;AAAA,EAE7E,MAAa,UAAyB;AAAA,EAAC;AAAA,EACvC,MAAa,aAA4B;AAAA,EAAC;AAAA,EACnC,cAAuB;AAC5B,WAAO;AAAA,EACT;AAAA,EAEQ,YAAY,QAAgB,WAAmB;AACrD,WAAO,GAAG,MAAM,IAAI,SAAS;AAAA,EAC/B;AAAA,EAEA,MAAa,WAAW,QAAgB,WAAuD;AAC7F,UAAM,MAAM,KAAK,YAAY,QAAQ,SAAS;AAC9C,UAAM,MAAM,KAAK,SAAS,IAAI,GAAG;AACjC,QAAI,KAAK;AACP,YAAM,gBAA+B;AAAA,QACnC,OAAO,OAAO,YAAY,IAAI,KAAK;AAAA,MACrC;AACA,aAAO;AAAA,IACT;AACA,WAAO;AAAA,EACT;AAAA,EAED,MAAa,cAAc,QAAgB,WAAkC;AAC1E,UAAM,MAAM,KAAK,YAAY,QAAQ,SAAS;AAC9C,QAAI,CAAC,KAAK,iBAAiB,IAAI,MAAM,GAAG;AACtC,WAAK,iBAAiB,IAAI,QAAQ,oBAAI,IAAI,CAAC;AAAA,IAC7C;AACA,QAAI,CAAC,KAAK,SAAS,IAAI,GAAG,GAAG;AAC3B,WAAK,SAAS,IAAI,KAAK,EAAE,OAAO,oBAAI,IAAI,EAAE,CAAC;AAC3C,YAAM,WAAoC;AAAA,QACxC;AAAA,QAAW,WAAW,KAAK,IAAI;AAAA,QAAG,WAAW,KAAK,IAAI;AAAA,MACxD;AACA,WAAK,iBAAiB,IAAI,MAAM,GAAG,IAAI,QAAQ;AAAA,IACjD;AAAA,EACF;AAAA,EAED,MAAa,iBAAiB,QAAgB,WAAmB,MAAiC;AAC/F,UAAM,MAAM,KAAK,YAAY,QAAQ,SAAS;AAC9C,UAAM,gBAAY,+BAAW;AAC7B,QAAI,CAAC,KAAK,SAAS,IAAI,GAAG,GAAG;AAC3B,YAAM,KAAK,cAAc,QAAQ,SAAS;AAAA,IAC5C;AACA,UAAM,WAAW,KAAK,SAAS,IAAI,GAAG;AACtC,SAAK,SAAS,IAAI,GAAG,GAAG,MAAM,IAAI,WAAW,IAAI;AAAA,EACnD;AAAA,EAED,MAAa,cAAc,QAAgB,WAAkC;AAC1E,UAAM,MAAM,KAAK,YAAY,QAAQ,SAAS;AAC9C,SAAK,SAAS,OAAO,GAAG;AACxB,SAAK,iBAAiB,OAAO,SAAS;AAAA,EACxC;AAAA,EAED,MAAa,aAAa,QAA4C;AACnE,UAAM,WAAW,KAAK,iBAAiB,IAAI,MAAM;AACjD,QAAI,UAAU;AACZ,aAAO,MAAM,KAAK,QAAQ;AAAA,IAC5B;AACA,WAAO,CAAC;AAAA,EACV;AACF;;;AC9EA,IAAAA,sBAA2B;AAIpB,IAAM,iBAAN,MAA8C;AAAA,EAC7C,UAA+B,oBAAI,IAAI;AAAA,EAE7C,MAAa,UAAyB;AAAA,EAAC;AAAA,EACvC,MAAa,aAA4B;AAAA,EAAC;AAAA,EACnC,cAAuB;AAC5B,WAAO;AAAA,EACT;AAAA,EAEA,MAAa,UAAU,UAA+C;AACpE,WAAO,KAAK,QAAQ,IAAI,QAAQ;AAAA,EAClC;AAAA,EAED,MAAa,WAAW,QAA+B;AACpD,UAAM,YAAQ,gCAAW;AACzB,SAAK,QAAQ,IAAI,OAAO,MAAM;AAAA,EAChC;AAAA,EAED,MAAa,aAAa,UAAkB,QAA+B;AACxE,SAAK,QAAQ,IAAI,UAAU,MAAM;AAAA,EACnC;AAAA,EAED,MAAa,aAAa,UAAiC;AACxD,SAAK,QAAQ,OAAO,QAAQ;AAAA,EAC9B;AAAA,EAED,MAAa,cAAiC;AAC3C,WAAO,MAAM,KAAK,KAAK,QAAQ,OAAO,CAAC;AAAA,EACzC;AACF;","names":["import_node_crypto"]}

package/dist/index.d.cts

@@ -1,12 +1,39 @@
-import { SessionObject, ChatObject } from '@ainetwork/adk/types/memory';
-import { BaseMemory } from '@ainetwork/adk/modules';
+import { ChatObject, SessionObject, SessionMetadata, Intent } from '@ainetwork/adk/types/memory';
+import { ISessionMemory, IIntentMemory } from '@ainetwork/adk/modules';
 
-declare class InMemoryMemory extends BaseMemory {
-    sessionHistory: Map<string, SessionObject>;
-    constructor();
-    getSessionHistory(sessionId: string): Promise<SessionObject>;
-    updateSessionHistory(sessionId: string, chat: ChatObject): Promise<void>;
-    storeQueryAndIntent(query: string, intent: string, sessionId: string): Promise<void>;
+type InMemorySessionObject = {
+    chats: Map<string, ChatObject>;
+};
+type InMemorySessionMetadata = {
+    sessionId: string;
+    title?: string;
+    updatedAt: number;
+    createdAt: number;
+};
+declare class InMemorySession implements ISessionMemory {
+    sessions: Map<string, InMemorySessionObject>;
+    userSessionIndex: Map<string, Set<InMemorySessionMetadata>>;
+    connect(): Promise<void>;
+    disconnect(): Promise<void>;
+    isConnected(): boolean;
+    private generateKey;
+    getSession(userId: string, sessionId: string): Promise<SessionObject | undefined>;
+    createSession(userId: string, sessionId: string): Promise<void>;
+    addChatToSession(userId: string, sessionId: string, chat: ChatObject): Promise<void>;
+    deleteSession(userId: string, sessionId: string): Promise<void>;
+    listSessions(userId: string): Promise<SessionMetadata[]>;
 }
 
-export { InMemoryMemory };
+declare class InMemoryIntent implements IIntentMemory {
+    intents: Map<string, Intent>;
+    connect(): Promise<void>;
+    disconnect(): Promise<void>;
+    isConnected(): boolean;
+    getIntent(intentId: string): Promise<Intent | undefined>;
+    saveIntent(intent: Intent): Promise<void>;
+    updateIntent(intentId: string, intent: Intent): Promise<void>;
+    deleteIntent(intentId: string): Promise<void>;
+    listIntents(): Promise<Intent[]>;
+}
+
+export { InMemoryIntent, InMemorySession };

package/dist/index.d.ts

@@ -1,12 +1,39 @@
-import { SessionObject, ChatObject } from '@ainetwork/adk/types/memory';
-import { BaseMemory } from '@ainetwork/adk/modules';
+import { ChatObject, SessionObject, SessionMetadata, Intent } from '@ainetwork/adk/types/memory';
+import { ISessionMemory, IIntentMemory } from '@ainetwork/adk/modules';
 
-declare class InMemoryMemory extends BaseMemory {
-    sessionHistory: Map<string, SessionObject>;
-    constructor();
-    getSessionHistory(sessionId: string): Promise<SessionObject>;
-    updateSessionHistory(sessionId: string, chat: ChatObject): Promise<void>;
-    storeQueryAndIntent(query: string, intent: string, sessionId: string): Promise<void>;
+type InMemorySessionObject = {
+    chats: Map<string, ChatObject>;
+};
+type InMemorySessionMetadata = {
+    sessionId: string;
+    title?: string;
+    updatedAt: number;
+    createdAt: number;
+};
+declare class InMemorySession implements ISessionMemory {
+    sessions: Map<string, InMemorySessionObject>;
+    userSessionIndex: Map<string, Set<InMemorySessionMetadata>>;
+    connect(): Promise<void>;
+    disconnect(): Promise<void>;
+    isConnected(): boolean;
+    private generateKey;
+    getSession(userId: string, sessionId: string): Promise<SessionObject | undefined>;
+    createSession(userId: string, sessionId: string): Promise<void>;
+    addChatToSession(userId: string, sessionId: string, chat: ChatObject): Promise<void>;
+    deleteSession(userId: string, sessionId: string): Promise<void>;
+    listSessions(userId: string): Promise<SessionMetadata[]>;
 }
 
-export { InMemoryMemory };
+declare class InMemoryIntent implements IIntentMemory {
+    intents: Map<string, Intent>;
+    connect(): Promise<void>;
+    disconnect(): Promise<void>;
+    isConnected(): boolean;
+    getIntent(intentId: string): Promise<Intent | undefined>;
+    saveIntent(intent: Intent): Promise<void>;
+    updateIntent(intentId: string, intent: Intent): Promise<void>;
+    deleteIntent(intentId: string): Promise<void>;
+    listIntents(): Promise<Intent[]>;
+}
+
+export { InMemoryIntent, InMemorySession };

package/dist/index.js

@@ -1,25 +1,97 @@
-// index.ts
+// implements/session.memory.ts
 import { randomUUID } from "crypto";
-import { BaseMemory } from "@ainetwork/adk/modules";
-var InMemoryMemory = class extends BaseMemory {
-  sessionHistory;
-  constructor() {
-    super();
-    this.sessionHistory = /* @__PURE__ */ new Map();
+var InMemorySession = class {
+  sessions = /* @__PURE__ */ new Map();
+  userSessionIndex = /* @__PURE__ */ new Map();
+  async connect() {
   }
-  async getSessionHistory(sessionId) {
-    return this.sessionHistory.get(sessionId) || { chats: {} };
+  async disconnect() {
   }
-  async updateSessionHistory(sessionId, chat) {
+  isConnected() {
+    return true;
+  }
+  generateKey(userId, sessionId) {
+    return `${userId}:${sessionId}`;
+  }
+  async getSession(userId, sessionId) {
+    const key = this.generateKey(userId, sessionId);
+    const res = this.sessions.get(key);
+    if (res) {
+      const sessionObject = {
+        chats: Object.fromEntries(res.chats)
+      };
+      return sessionObject;
+    }
+    return void 0;
+  }
+  async createSession(userId, sessionId) {
+    const key = this.generateKey(userId, sessionId);
+    if (!this.userSessionIndex.has(userId)) {
+      this.userSessionIndex.set(userId, /* @__PURE__ */ new Set());
+    }
+    if (!this.sessions.has(key)) {
+      this.sessions.set(key, { chats: /* @__PURE__ */ new Map() });
+      const metadata = {
+        sessionId,
+        createdAt: Date.now(),
+        updatedAt: Date.now()
+      };
+      this.userSessionIndex.get(userId)?.add(metadata);
+    }
+  }
+  async addChatToSession(userId, sessionId, chat) {
+    const key = this.generateKey(userId, sessionId);
     const newChatId = randomUUID();
-    const history = await this.getSessionHistory(sessionId);
-    history.chats[newChatId] = chat;
-    this.sessionHistory.set(sessionId, history);
+    if (!this.sessions.has(key)) {
+      await this.createSession(userId, sessionId);
+    }
+    const sessions = this.sessions.get(key);
+    this.sessions.get(key)?.chats.set(newChatId, chat);
+  }
+  async deleteSession(userId, sessionId) {
+    const key = this.generateKey(userId, sessionId);
+    this.sessions.delete(key);
+    this.userSessionIndex.delete(sessionId);
+  }
+  async listSessions(userId) {
+    const sessions = this.userSessionIndex.get(userId);
+    if (sessions) {
+      return Array.from(sessions);
+    }
+    return [];
+  }
+};
+
+// implements/intent.memory.ts
+import { randomUUID as randomUUID2 } from "crypto";
+var InMemoryIntent = class {
+  intents = /* @__PURE__ */ new Map();
+  async connect() {
+  }
+  async disconnect() {
+  }
+  isConnected() {
+    return true;
+  }
+  async getIntent(intentId) {
+    return this.intents.get(intentId);
+  }
+  async saveIntent(intent) {
+    const newId = randomUUID2();
+    this.intents.set(newId, intent);
+  }
+  async updateIntent(intentId, intent) {
+    this.intents.set(intentId, intent);
+  }
+  async deleteIntent(intentId) {
+    this.intents.delete(intentId);
   }
-  async storeQueryAndIntent(query, intent, sessionId) {
+  async listIntents() {
+    return Array.from(this.intents.values());
   }
 };
 export {
-  InMemoryMemory
+  InMemoryIntent,
+  InMemorySession
 };
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../index.ts"],"sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport type { ChatObject, SessionObject } from \"@ainetwork/adk/types/memory\";\nimport { BaseMemory } from \"@ainetwork/adk/modules\";\n\nexport class InMemoryMemory extends BaseMemory {\n\tpublic sessionHistory: Map<string, SessionObject>;\n\n\tconstructor() {\n\t\tsuper();\n\t\tthis.sessionHistory = new Map<string, SessionObject>();\n\t}\n\n\tpublic async getSessionHistory(sessionId: string): Promise<SessionObject> {\n\t\treturn this.sessionHistory.get(sessionId) || { chats: {} };\n\t}\n\n\tpublic async updateSessionHistory(\n\t\tsessionId: string,\n\t\tchat: ChatObject,\n\t): Promise<void> {\n\t\tconst newChatId = randomUUID();\n\t\tconst history = await this.getSessionHistory(sessionId);\n\t\thistory.chats[newChatId] = chat;\n\t\tthis.sessionHistory.set(sessionId, history);\n\t}\n\n\tpublic async storeQueryAndIntent(\n\t\tquery: string,\n\t\tintent: string,\n\t\tsessionId: string,\n\t) {}\n}"],"mappings":";AAAA,SAAS,kBAAkB;AAE3B,SAAS,kBAAkB;AAEpB,IAAM,iBAAN,cAA6B,WAAW;AAAA,EACvC;AAAA,EAEP,cAAc;AACb,UAAM;AACN,SAAK,iBAAiB,oBAAI,IAA2B;AAAA,EACtD;AAAA,EAEA,MAAa,kBAAkB,WAA2C;AACzE,WAAO,KAAK,eAAe,IAAI,SAAS,KAAK,EAAE,OAAO,CAAC,EAAE;AAAA,EAC1D;AAAA,EAEA,MAAa,qBACZ,WACA,MACgB;AAChB,UAAM,YAAY,WAAW;AAC7B,UAAM,UAAU,MAAM,KAAK,kBAAkB,SAAS;AACtD,YAAQ,MAAM,SAAS,IAAI;AAC3B,SAAK,eAAe,IAAI,WAAW,OAAO;AAAA,EAC3C;AAAA,EAEA,MAAa,oBACZ,OACA,QACA,WACC;AAAA,EAAC;AACJ;","names":[]}
+{"version":3,"sources":["../implements/session.memory.ts","../implements/intent.memory.ts"],"sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport type { ChatObject, SessionObject, SessionMetadata } from \"@ainetwork/adk/types/memory\";\nimport { ISessionMemory } from \"@ainetwork/adk/modules\";\n\ntype InMemorySessionObject = {\n  chats: Map<string, ChatObject>\n}\n\ntype InMemorySessionMetadata = {\n  sessionId: string;\n  title?: string;\n  updatedAt: number;\n  createdAt: number;\n}\n\nexport class InMemorySession implements ISessionMemory {\n\tpublic sessions: Map<string, InMemorySessionObject> = new Map();\n  public userSessionIndex: Map<string, Set<InMemorySessionMetadata>> = new Map();\n\n  public async connect(): Promise<void> {}\n  public async disconnect(): Promise<void> {}\n  public isConnected(): boolean {\n    return true;\n  }\n\n  private generateKey(userId: string, sessionId: string) {\n    return `${userId}:${sessionId}`;\n  }\n\n  public async getSession(userId: string, sessionId: string): Promise<SessionObject | undefined> {\n    const key = this.generateKey(userId, sessionId);\n    const res = this.sessions.get(key);\n    if (res) {\n      const sessionObject: SessionObject = {\n        chats: Object.fromEntries(res.chats)\n      };\n      return sessionObject;\n    }\n    return undefined;\n  };\n\n\tpublic async createSession(userId: string, sessionId: string): Promise<void> {\n    const key = this.generateKey(userId, sessionId);\n    if (!this.userSessionIndex.has(userId)) {\n      this.userSessionIndex.set(userId, new Set());\n    }\n    if (!this.sessions.has(key)) {\n      this.sessions.set(key, { chats: new Map() });\n      const metadata: InMemorySessionMetadata = {\n        sessionId, createdAt: Date.now(), updatedAt: Date.now(),\n      }\n      this.userSessionIndex.get(userId)?.add(metadata);\n    }\n  };\n\n\tpublic async addChatToSession(userId: string, sessionId: string, chat: ChatObject): Promise<void> {\n    const key = this.generateKey(userId, sessionId);\n    const newChatId = randomUUID();\n    if (!this.sessions.has(key)) {\n      await this.createSession(userId, sessionId);\n    }\n    const sessions = this.sessions.get(key);\n    this.sessions.get(key)?.chats.set(newChatId, chat);\n  };\n\n\tpublic async deleteSession(userId: string, sessionId: string): Promise<void> {\n    const key = this.generateKey(userId, sessionId);\n    this.sessions.delete(key);\n    this.userSessionIndex.delete(sessionId);\n  };\n\n\tpublic async listSessions(userId: string): Promise<SessionMetadata[]> {\n    const sessions = this.userSessionIndex.get(userId);\n    if (sessions) {\n      return Array.from(sessions);\n    }\n    return [];\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\tpublic intents: Map<string, Intent> = new Map();\n\n  public async connect(): Promise<void> {}\n  public async disconnect(): Promise<void> {}\n  public isConnected(): boolean {\n    return true;\n  }\n\n  public async getIntent(intentId: string): Promise<Intent | undefined> {\n    return this.intents.get(intentId);\n  };\n\n\tpublic async saveIntent(intent: Intent): Promise<void> {\n    const newId = randomUUID();\n    this.intents.set(newId, intent);\n  };\n\n\tpublic async updateIntent(intentId: string, intent: Intent): Promise<void> {\n    this.intents.set(intentId, intent);\n  };\n\n\tpublic async deleteIntent(intentId: string): Promise<void> {\n    this.intents.delete(intentId);\n  };\n\n\tpublic async listIntents(): Promise<Intent[]> {\n    return Array.from(this.intents.values());\n  };\n}"],"mappings":";AAAA,SAAS,kBAAkB;AAepB,IAAM,kBAAN,MAAgD;AAAA,EAC/C,WAA+C,oBAAI,IAAI;AAAA,EACtD,mBAA8D,oBAAI,IAAI;AAAA,EAE7E,MAAa,UAAyB;AAAA,EAAC;AAAA,EACvC,MAAa,aAA4B;AAAA,EAAC;AAAA,EACnC,cAAuB;AAC5B,WAAO;AAAA,EACT;AAAA,EAEQ,YAAY,QAAgB,WAAmB;AACrD,WAAO,GAAG,MAAM,IAAI,SAAS;AAAA,EAC/B;AAAA,EAEA,MAAa,WAAW,QAAgB,WAAuD;AAC7F,UAAM,MAAM,KAAK,YAAY,QAAQ,SAAS;AAC9C,UAAM,MAAM,KAAK,SAAS,IAAI,GAAG;AACjC,QAAI,KAAK;AACP,YAAM,gBAA+B;AAAA,QACnC,OAAO,OAAO,YAAY,IAAI,KAAK;AAAA,MACrC;AACA,aAAO;AAAA,IACT;AACA,WAAO;AAAA,EACT;AAAA,EAED,MAAa,cAAc,QAAgB,WAAkC;AAC1E,UAAM,MAAM,KAAK,YAAY,QAAQ,SAAS;AAC9C,QAAI,CAAC,KAAK,iBAAiB,IAAI,MAAM,GAAG;AACtC,WAAK,iBAAiB,IAAI,QAAQ,oBAAI,IAAI,CAAC;AAAA,IAC7C;AACA,QAAI,CAAC,KAAK,SAAS,IAAI,GAAG,GAAG;AAC3B,WAAK,SAAS,IAAI,KAAK,EAAE,OAAO,oBAAI,IAAI,EAAE,CAAC;AAC3C,YAAM,WAAoC;AAAA,QACxC;AAAA,QAAW,WAAW,KAAK,IAAI;AAAA,QAAG,WAAW,KAAK,IAAI;AAAA,MACxD;AACA,WAAK,iBAAiB,IAAI,MAAM,GAAG,IAAI,QAAQ;AAAA,IACjD;AAAA,EACF;AAAA,EAED,MAAa,iBAAiB,QAAgB,WAAmB,MAAiC;AAC/F,UAAM,MAAM,KAAK,YAAY,QAAQ,SAAS;AAC9C,UAAM,YAAY,WAAW;AAC7B,QAAI,CAAC,KAAK,SAAS,IAAI,GAAG,GAAG;AAC3B,YAAM,KAAK,cAAc,QAAQ,SAAS;AAAA,IAC5C;AACA,UAAM,WAAW,KAAK,SAAS,IAAI,GAAG;AACtC,SAAK,SAAS,IAAI,GAAG,GAAG,MAAM,IAAI,WAAW,IAAI;AAAA,EACnD;AAAA,EAED,MAAa,cAAc,QAAgB,WAAkC;AAC1E,UAAM,MAAM,KAAK,YAAY,QAAQ,SAAS;AAC9C,SAAK,SAAS,OAAO,GAAG;AACxB,SAAK,iBAAiB,OAAO,SAAS;AAAA,EACxC;AAAA,EAED,MAAa,aAAa,QAA4C;AACnE,UAAM,WAAW,KAAK,iBAAiB,IAAI,MAAM;AACjD,QAAI,UAAU;AACZ,aAAO,MAAM,KAAK,QAAQ;AAAA,IAC5B;AACA,WAAO,CAAC;AAAA,EACV;AACF;;;AC9EA,SAAS,cAAAA,mBAAkB;AAIpB,IAAM,iBAAN,MAA8C;AAAA,EAC7C,UAA+B,oBAAI,IAAI;AAAA,EAE7C,MAAa,UAAyB;AAAA,EAAC;AAAA,EACvC,MAAa,aAA4B;AAAA,EAAC;AAAA,EACnC,cAAuB;AAC5B,WAAO;AAAA,EACT;AAAA,EAEA,MAAa,UAAU,UAA+C;AACpE,WAAO,KAAK,QAAQ,IAAI,QAAQ;AAAA,EAClC;AAAA,EAED,MAAa,WAAW,QAA+B;AACpD,UAAM,QAAQA,YAAW;AACzB,SAAK,QAAQ,IAAI,OAAO,MAAM;AAAA,EAChC;AAAA,EAED,MAAa,aAAa,UAAkB,QAA+B;AACxE,SAAK,QAAQ,IAAI,UAAU,MAAM;AAAA,EACnC;AAAA,EAED,MAAa,aAAa,UAAiC;AACxD,SAAK,QAAQ,OAAO,QAAQ;AAAA,EAC9B;AAAA,EAED,MAAa,cAAiC;AAC3C,WAAO,MAAM,KAAK,KAAK,QAAQ,OAAO,CAAC;AAAA,EACzC;AACF;","names":["randomUUID"]}

package/implements/intent.memory.ts

@@ -0,0 +1,34 @@
+import { randomUUID } from "node:crypto";
+import type { Intent } from "@ainetwork/adk/types/memory";
+import { IIntentMemory } from "@ainetwork/adk/modules";
+
+export class InMemoryIntent implements IIntentMemory {
+	public intents: Map<string, Intent> = new Map();
+
+  public async connect(): Promise<void> {}
+  public async disconnect(): Promise<void> {}
+  public isConnected(): boolean {
+    return true;
+  }
+
+  public async getIntent(intentId: string): Promise<Intent | undefined> {
+    return this.intents.get(intentId);
+  };
+
+	public async saveIntent(intent: Intent): Promise<void> {
+    const newId = randomUUID();
+    this.intents.set(newId, intent);
+  };
+
+	public async updateIntent(intentId: string, intent: Intent): Promise<void> {
+    this.intents.set(intentId, intent);
+  };
+
+	public async deleteIntent(intentId: string): Promise<void> {
+    this.intents.delete(intentId);
+  };
+
+	public async listIntents(): Promise<Intent[]> {
+    return Array.from(this.intents.values());
+  };
+}

package/implements/session.memory.ts

@@ -0,0 +1,79 @@
+import { randomUUID } from "node:crypto";
+import type { ChatObject, SessionObject, SessionMetadata } from "@ainetwork/adk/types/memory";
+import { ISessionMemory } from "@ainetwork/adk/modules";
+
+type InMemorySessionObject = {
+  chats: Map<string, ChatObject>
+}
+
+type InMemorySessionMetadata = {
+  sessionId: string;
+  title?: string;
+  updatedAt: number;
+  createdAt: number;
+}
+
+export class InMemorySession implements ISessionMemory {
+	public sessions: Map<string, InMemorySessionObject> = new Map();
+  public userSessionIndex: Map<string, Set<InMemorySessionMetadata>> = new Map();
+
+  public async connect(): Promise<void> {}
+  public async disconnect(): Promise<void> {}
+  public isConnected(): boolean {
+    return true;
+  }
+
+  private generateKey(userId: string, sessionId: string) {
+    return `${userId}:${sessionId}`;
+  }
+
+  public async getSession(userId: string, sessionId: string): Promise<SessionObject | undefined> {
+    const key = this.generateKey(userId, sessionId);
+    const res = this.sessions.get(key);
+    if (res) {
+      const sessionObject: SessionObject = {
+        chats: Object.fromEntries(res.chats)
+      };
+      return sessionObject;
+    }
+    return undefined;
+  };
+
+	public async createSession(userId: string, sessionId: string): Promise<void> {
+    const key = this.generateKey(userId, sessionId);
+    if (!this.userSessionIndex.has(userId)) {
+      this.userSessionIndex.set(userId, new Set());
+    }
+    if (!this.sessions.has(key)) {
+      this.sessions.set(key, { chats: new Map() });
+      const metadata: InMemorySessionMetadata = {
+        sessionId, createdAt: Date.now(), updatedAt: Date.now(),
+      }
+      this.userSessionIndex.get(userId)?.add(metadata);
+    }
+  };
+
+	public async addChatToSession(userId: string, sessionId: string, chat: ChatObject): Promise<void> {
+    const key = this.generateKey(userId, sessionId);
+    const newChatId = randomUUID();
+    if (!this.sessions.has(key)) {
+      await this.createSession(userId, sessionId);
+    }
+    const sessions = this.sessions.get(key);
+    this.sessions.get(key)?.chats.set(newChatId, chat);
+  };
+
+	public async deleteSession(userId: string, sessionId: string): Promise<void> {
+    const key = this.generateKey(userId, sessionId);
+    this.sessions.delete(key);
+    this.userSessionIndex.delete(sessionId);
+  };
+
+	public async listSessions(userId: string): Promise<SessionMetadata[]> {
+    const sessions = this.userSessionIndex.get(userId);
+    if (sessions) {
+      return Array.from(sessions);
+    }
+    return [];
+  };
+}

package/index.ts

@@ -1,32 +1,2 @@
-import { randomUUID } from "node:crypto";
-import type { ChatObject, SessionObject } from "@ainetwork/adk/types/memory";
-import { BaseMemory } from "@ainetwork/adk/modules";
-
-export class InMemoryMemory extends BaseMemory {
-	public sessionHistory: Map<string, SessionObject>;
-
-	constructor() {
-		super();
-		this.sessionHistory = new Map<string, SessionObject>();
-	}
-
-	public async getSessionHistory(sessionId: string): Promise<SessionObject> {
-		return this.sessionHistory.get(sessionId) || { chats: {} };
-	}
-
-	public async updateSessionHistory(
-		sessionId: string,
-		chat: ChatObject,
-	): Promise<void> {
-		const newChatId = randomUUID();
-		const history = await this.getSessionHistory(sessionId);
-		history.chats[newChatId] = chat;
-		this.sessionHistory.set(sessionId, history);
-	}
-
-	public async storeQueryAndIntent(
-		query: string,
-		intent: string,
-		sessionId: string,
-	) {}
-}
+export { InMemorySession } from "./implements/session.memory";
+export { InMemoryIntent } from "./implements/intent.memory";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ainetwork/adk-provider-memory-inmemory",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "author": "AI Network (https://ainetwork.ai)",
   "type": "module",
   "engines": {
@@ -21,7 +21,7 @@
     "clean": "rm -rf dist"
   },
   "dependencies": {
-    "@ainetwork/adk": "^0.1.2"
+    "@ainetwork/adk": "^0.1.7"
   },
   "devDependencies": {
     "typescript": "^5.0.0"
@@ -30,5 +30,5 @@
   "publishConfig": {
     "access": "public"
   },
-  "gitHead": "cb6205738fd84533ce56bbeeab00ea9dfe2aaf3a"
+  "gitHead": "179c9e22bac074444e9a1ddf656eb52d6fe8d34a"
 }