@ainetwork/adk-provider-memory-mongodb 0.2.4 → 0.3.2

package/dist/models/agent.model.d.ts

@@ -0,0 +1,23 @@
+import mongoose, { Schema, Document } from 'mongoose';
+
+declare const AgentObjectSchema: Schema<any, mongoose.Model<any, any, any, any, any, any>, {}, {}, {}, {}, mongoose.DefaultSchemaOptions, {
+    agent_prompt?: string | null | undefined;
+}, Document<unknown, {}, mongoose.FlatRecord<{
+    agent_prompt?: string | null | undefined;
+}>, {}> & mongoose.FlatRecord<{
+    agent_prompt?: string | null | undefined;
+}> & {
+    _id: mongoose.Types.ObjectId;
+} & {
+    __v: number;
+}>;
+interface AgentDocument extends Document {
+    agent_prompt: string;
+}
+declare const AgentModel: mongoose.Model<AgentDocument, {}, {}, {}, Document<unknown, {}, AgentDocument, {}> & AgentDocument & Required<{
+    _id: unknown;
+}> & {
+    __v: number;
+}, any>;
+
+export { type AgentDocument, AgentModel, AgentObjectSchema };

package/dist/models/agent.model.js

@@ -0,0 +1,9 @@
+import {
+  AgentModel,
+  AgentObjectSchema
+} from "../chunk-OWGGE5EW.js";
+export {
+  AgentModel,
+  AgentObjectSchema
+};
+//# sourceMappingURL=agent.model.js.map

package/dist/models/agent.model.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/implements/agent.memory.ts

@@ -0,0 +1,46 @@
+import { IAgentMemory } from "@ainetwork/adk/modules";
+import { AgentModel } from "../models/agent.model";
+
+export type ExecuteWithRetryFn = <T>(
+  operation: () => Promise<T>,
+  operationName?: string
+) => Promise<T>;
+
+export type GetOperationTimeoutFn = () => number;
+
+type AgentMetadata = {
+  agent_prompt: string;
+}
+
+export class MongoDBAgent implements IAgentMemory {
+  private executeWithRetry: ExecuteWithRetryFn;
+  private getOperationTimeout: GetOperationTimeoutFn;
+
+  constructor(
+    executeWithRetry: ExecuteWithRetryFn,
+    getOperationTimeout: GetOperationTimeoutFn
+  ) {
+    this.executeWithRetry = executeWithRetry;
+    this.getOperationTimeout = getOperationTimeout;
+  }
+
+  public async getAgentPrompt(): Promise<string> {
+    return this.executeWithRetry(async () => {
+      const timeout = this.getOperationTimeout();
+      const metadata = await AgentModel.findOne({
+        id: "agent_metadata"
+      }).maxTimeMS(timeout)
+        .lean<AgentMetadata>();
+      return metadata?.agent_prompt || "";
+    }, "getAgentPrompt()");
+  };
+  
+  public async updateAgentPrompt(prompt: string): Promise<void> {
+    return this.executeWithRetry(async () => {
+      const timeout = this.getOperationTimeout();
+      await AgentModel.updateOne({
+        id: "agent_metadata",
+      }, { "agent_prompt": prompt }).maxTimeMS(timeout);
+    }, "updateAgentPrompt()");
+  };
+}

package/implements/base.memory.ts

@@ -1,6 +1,9 @@
-import { IMemory } from "node_modules/@ainetwork/adk/dist/esm/modules/memory/base.memory";
+import { IAgentMemory, IIntentMemory, IMemory, IThreadMemory } from "node_modules/@ainetwork/adk/dist/esm/modules/memory/base.memory";
 import mongoose from "mongoose";
 import { loggers } from "@ainetwork/adk/utils/logger";
+import { MongoDBAgent } from "./agent.memory";
+import { MongoDBIntent } from "./intent.memory";
+import { MongoDBThread } from "./thread.memory";
 
 export interface MongoDBMemoryConfig {
   uri: string;
@@ -25,6 +28,10 @@ export class MongoDBMemory implements IMemory {
   private eventListenersSetup: boolean = false;
   private operationTimeoutMS: number;
 
+  private agentMemory: MongoDBAgent;
+  private intentMemory: MongoDBIntent;
+  private threadMemory: MongoDBThread;
+
   constructor(config: string | MongoDBMemoryConfig) {
     const cfg = typeof config === 'string' ? { uri: config } : config;
 
@@ -48,6 +55,33 @@ export class MongoDBMemory implements IMemory {
       this.connected = MongoDBMemory.instance.connected;
       this.operationTimeoutMS = MongoDBMemory.instance.operationTimeoutMS;
     }
+
+		this.agentMemory = new MongoDBAgent(
+			this.executeWithRetry.bind(this),
+			this.getOperationTimeout.bind(this)
+		);
+
+		this.threadMemory = new MongoDBThread(
+			this.executeWithRetry.bind(this),
+			this.getOperationTimeout.bind(this)
+		);
+
+		this.intentMemory = new MongoDBIntent(
+			this.executeWithRetry.bind(this),
+			this.getOperationTimeout.bind(this)
+		);
+  }
+
+  public getAgentMemory(): IAgentMemory {
+    return this.agentMemory;
+  }
+
+  public getThreadMemory(): IThreadMemory {
+    return this.threadMemory;
+  }
+
+  public getIntentMemory(): IIntentMemory {
+    return this.intentMemory;
   }
 
   private setupMongooseEventListeners(): void {

package/implements/intent.memory.ts

@@ -1,9 +1,26 @@
 import type { Intent } from "@ainetwork/adk/types/memory";
 import { IIntentMemory } from "@ainetwork/adk/modules";
-import { MongoDBMemory } from "./base.memory";
 import { IntentModel } from "../models/intent.model";
 
-export class MongoDBIntent extends MongoDBMemory implements IIntentMemory {
+export type ExecuteWithRetryFn = <T>(
+  operation: () => Promise<T>,
+  operationName?: string
+) => Promise<T>;
+
+export type GetOperationTimeoutFn = () => number;
+
+export class MongoDBIntent implements IIntentMemory {
+  private executeWithRetry: ExecuteWithRetryFn;
+  private getOperationTimeout: GetOperationTimeoutFn;
+
+  constructor(
+    executeWithRetry: ExecuteWithRetryFn,
+    getOperationTimeout: GetOperationTimeoutFn
+  ) {
+    this.executeWithRetry = executeWithRetry;
+    this.getOperationTimeout = getOperationTimeout;
+  }
+
   public async getIntent(intentId: string): Promise<Intent | undefined> {
     return this.executeWithRetry(async () => {
       const timeout = this.getOperationTimeout();

package/implements/thread.memory.ts

@@ -1,14 +1,27 @@
 import type { MessageObject, ThreadMetadata, ThreadObject, ThreadType } from "@ainetwork/adk/types/memory";
 import { MessageRole } from "@ainetwork/adk/types/memory";
 import { IThreadMemory } from "@ainetwork/adk/modules";
-import { MongoDBMemory } from "./base.memory";
 import { ThreadDocument, ThreadModel } from "../models/threads.model";
 import { MessageDocument, MessageModel } from "../models/messages.model";
 import { loggers } from "@ainetwork/adk/utils/logger";
 
-export class MongoDBThread extends MongoDBMemory implements IThreadMemory {
-  constructor(uri: string) {
-    super(uri);
+export type ExecuteWithRetryFn = <T>(
+  operation: () => Promise<T>,
+  operationName?: string
+) => Promise<T>;
+
+export type GetOperationTimeoutFn = () => number;
+
+export class MongoDBThread implements IThreadMemory {
+  private executeWithRetry: ExecuteWithRetryFn;
+  private getOperationTimeout: GetOperationTimeoutFn;
+
+  constructor(
+    executeWithRetry: ExecuteWithRetryFn,
+    getOperationTimeout: GetOperationTimeoutFn
+  ) {
+    this.executeWithRetry = executeWithRetry;
+    this.getOperationTimeout = getOperationTimeout;
   }
 
   public async getThread(

package/index.ts

@@ -1,4 +1,2 @@
-export { MongoDBThread } from "./implements/thread.memory";
-export { MongoDBIntent } from "./implements/intent.memory";
 export { MongoDBMemory } from "./implements/base.memory";
 export type { MongoDBMemoryConfig } from "./implements/base.memory";

package/models/agent.model.ts

@@ -0,0 +1,16 @@
+import { type Document, Schema } from "mongoose";
+import mongoose from "mongoose";
+
+export const AgentObjectSchema = new Schema(
+	{
+		agent_prompt: {
+			type: String,
+		},
+	},
+);
+
+export interface AgentDocument extends Document {
+	agent_prompt: string;
+}
+
+export const AgentModel = mongoose.model<AgentDocument>("Agent", AgentObjectSchema);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ainetwork/adk-provider-memory-mongodb",
-  "version": "0.2.4",
+  "version": "0.3.2",
   "author": "AI Network (https://ainetwork.ai)",
   "type": "module",
   "engines": {
@@ -28,7 +28,7 @@
     "clean": "rm -rf dist"
   },
   "dependencies": {
-    "@ainetwork/adk": "0.2.9",
+    "@ainetwork/adk": "^0.3.2",
     "mongoose": "^8.16.5"
   },
   "devDependencies": {
@@ -38,5 +38,5 @@
   "publishConfig": {
     "access": "public"
   },
-  "gitHead": "f1315510a23dd44492dc20737c4be71188d01e26"
+  "gitHead": "4bfa5afae29304e6cb5f106c7327f5f2092f6601"
 }