@ainetwork/adk-provider-memory-mongodb 0.6.2 → 0.7.0

package/dist/chunk-JMRBIQNX.js

@@ -0,0 +1,96 @@
+// models/document.model.ts
+import {
+  DocumentFormat,
+  DocumentSource
+} from "@ainetwork/adk/types/document";
+import { Schema } from "mongoose";
+import mongoose from "mongoose";
+var DocumentObjectSchema = new Schema(
+  {
+    documentId: {
+      type: String,
+      required: true,
+      unique: true
+    },
+    userId: {
+      type: String,
+      required: true,
+      index: true
+    },
+    title: {
+      type: String,
+      required: true
+    },
+    format: {
+      type: String,
+      enum: Object.values(DocumentFormat),
+      required: true,
+      default: DocumentFormat.MARKDOWN
+    },
+    content: {
+      type: String,
+      required: true
+    },
+    blocks: {
+      type: Schema.Types.Mixed
+    },
+    slots: {
+      type: Schema.Types.Mixed
+    },
+    // Cached AI advice ({ content, generatedAt }). Free-form so the schema
+    // doesn't strip it on update (mongoose strict mode).
+    advice: {
+      type: Schema.Types.Mixed
+    },
+    // Faceted grouping (e.g. category/workplaceId/month). Stored as a free
+    // map; index specific keys (e.g. `labels.workplaceId`) if query volume
+    // warrants it.
+    labels: {
+      type: Schema.Types.Mixed
+    },
+    source: {
+      type: String,
+      enum: Object.values(DocumentSource),
+      required: true
+    },
+    workflowId: {
+      type: String,
+      index: true
+    },
+    threadId: {
+      type: String,
+      index: true
+    },
+    version: {
+      type: Number,
+      required: true,
+      default: 1
+    },
+    editedManually: {
+      type: Boolean
+    },
+    createdAt: {
+      type: String,
+      required: true
+    },
+    updatedAt: {
+      type: String,
+      required: true
+    }
+  },
+  // `createdAt`/`updatedAt` are caller-supplied ISO strings (see Document type),
+  // so mongoose's automatic Date timestamps are intentionally disabled.
+  {
+    timestamps: false
+  }
+);
+var DocumentModel = mongoose.model(
+  "Document",
+  DocumentObjectSchema
+);
+
+export {
+  DocumentObjectSchema,
+  DocumentModel
+};
+//# sourceMappingURL=chunk-JMRBIQNX.js.map

package/dist/chunk-JMRBIQNX.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../models/document.model.ts"],"sourcesContent":["import {\n\ttype DocumentAdvice,\n\tDocumentFormat,\n\ttype DocumentSlot,\n\tDocumentSource,\n} from \"@ainetwork/adk/types/document\";\nimport type { WorkflowRenderedBlock } from \"@ainetwork/adk/types/memory\";\nimport { type Document, Schema } from \"mongoose\";\nimport mongoose from \"mongoose\";\n\nexport const DocumentObjectSchema = new Schema(\n\t{\n\t\tdocumentId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tunique: true,\n\t\t},\n\t\tuserId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\ttitle: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t},\n\t\tformat: {\n\t\t\ttype: String,\n\t\t\tenum: Object.values(DocumentFormat),\n\t\t\trequired: true,\n\t\t\tdefault: DocumentFormat.MARKDOWN,\n\t\t},\n\t\tcontent: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t},\n\t\tblocks: {\n\t\t\ttype: Schema.Types.Mixed,\n\t\t},\n\t\tslots: {\n\t\t\ttype: Schema.Types.Mixed,\n\t\t},\n\t\t// Cached AI advice ({ content, generatedAt }). Free-form so the schema\n\t\t// doesn't strip it on update (mongoose strict mode).\n\t\tadvice: {\n\t\t\ttype: Schema.Types.Mixed,\n\t\t},\n\t\t// Faceted grouping (e.g. category/workplaceId/month). Stored as a free\n\t\t// map; index specific keys (e.g. `labels.workplaceId`) if query volume\n\t\t// warrants it.\n\t\tlabels: {\n\t\t\ttype: Schema.Types.Mixed,\n\t\t},\n\t\tsource: {\n\t\t\ttype: String,\n\t\t\tenum: Object.values(DocumentSource),\n\t\t\trequired: true,\n\t\t},\n\t\tworkflowId: {\n\t\t\ttype: String,\n\t\t\tindex: true,\n\t\t},\n\t\tthreadId: {\n\t\t\ttype: String,\n\t\t\tindex: true,\n\t\t},\n\t\tversion: {\n\t\t\ttype: Number,\n\t\t\trequired: true,\n\t\t\tdefault: 1,\n\t\t},\n\t\teditedManually: {\n\t\t\ttype: Boolean,\n\t\t},\n\t\tcreatedAt: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t},\n\t\tupdatedAt: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t},\n\t},\n\t// `createdAt`/`updatedAt` are caller-supplied ISO strings (see Document type),\n\t// so mongoose's automatic Date timestamps are intentionally disabled.\n\t{\n\t\ttimestamps: false,\n\t}\n);\n\nexport interface DocumentDocument extends Document {\n\tdocumentId: string;\n\tuserId: string;\n\ttitle: string;\n\tformat: DocumentFormat;\n\tcontent: string;\n\tblocks?: WorkflowRenderedBlock[];\n\tslots?: DocumentSlot[];\n\tadvice?: DocumentAdvice;\n\tlabels?: Record<string, string>;\n\tsource: DocumentSource;\n\tworkflowId?: string;\n\tthreadId?: string;\n\tversion: number;\n\teditedManually?: boolean;\n\tcreatedAt: string;\n\tupdatedAt: string;\n}\n\nexport const DocumentModel = mongoose.model<DocumentDocument>(\n\t\"Document\",\n\tDocumentObjectSchema\n);\n"],"mappings":";AAAA;AAAA,EAEC;AAAA,EAEA;AAAA,OACM;AAEP,SAAwB,cAAc;AACtC,OAAO,cAAc;AAEd,IAAM,uBAAuB,IAAI;AAAA,EACvC;AAAA,IACC,YAAY;AAAA,MACX,MAAM;AAAA,MACN,UAAU;AAAA,MACV,QAAQ;AAAA,IACT;AAAA,IACA,QAAQ;AAAA,MACP,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,OAAO;AAAA,MACN,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,QAAQ;AAAA,MACP,MAAM;AAAA,MACN,MAAM,OAAO,OAAO,cAAc;AAAA,MAClC,UAAU;AAAA,MACV,SAAS,eAAe;AAAA,IACzB;AAAA,IACA,SAAS;AAAA,MACR,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,QAAQ;AAAA,MACP,MAAM,OAAO,MAAM;AAAA,IACpB;AAAA,IACA,OAAO;AAAA,MACN,MAAM,OAAO,MAAM;AAAA,IACpB;AAAA;AAAA;AAAA,IAGA,QAAQ;AAAA,MACP,MAAM,OAAO,MAAM;AAAA,IACpB;AAAA;AAAA;AAAA;AAAA,IAIA,QAAQ;AAAA,MACP,MAAM,OAAO,MAAM;AAAA,IACpB;AAAA,IACA,QAAQ;AAAA,MACP,MAAM;AAAA,MACN,MAAM,OAAO,OAAO,cAAc;AAAA,MAClC,UAAU;AAAA,IACX;AAAA,IACA,YAAY;AAAA,MACX,MAAM;AAAA,MACN,OAAO;AAAA,IACR;AAAA,IACA,UAAU;AAAA,MACT,MAAM;AAAA,MACN,OAAO;AAAA,IACR;AAAA,IACA,SAAS;AAAA,MACR,MAAM;AAAA,MACN,UAAU;AAAA,MACV,SAAS;AAAA,IACV;AAAA,IACA,gBAAgB;AAAA,MACf,MAAM;AAAA,IACP;AAAA,IACA,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,EACD;AAAA;AAAA;AAAA,EAGA;AAAA,IACC,YAAY;AAAA,EACb;AACD;AAqBO,IAAM,gBAAgB,SAAS;AAAA,EACrC;AAAA,EACA;AACD;","names":[]}

package/dist/index.cjs

@@ -35,7 +35,7 @@ __export(index_exports, {
 module.exports = __toCommonJS(index_exports);
 
 // implements/base.memory.ts
-var import_mongoose12 = __toESM(require("mongoose"), 1);
+var import_mongoose14 = __toESM(require("mongoose"), 1);
 var import_logger2 = require("@ainetwork/adk/utils/logger");
 
 // models/agent.model.ts
@@ -145,9 +145,153 @@ var MongoDBAgent = class {
   }
 };
 
+// models/document.model.ts
+var import_document = require("@ainetwork/adk/types/document");
+var import_mongoose3 = require("mongoose");
+var import_mongoose4 = __toESM(require("mongoose"), 1);
+var DocumentObjectSchema = new import_mongoose3.Schema(
+  {
+    documentId: {
+      type: String,
+      required: true,
+      unique: true
+    },
+    userId: {
+      type: String,
+      required: true,
+      index: true
+    },
+    title: {
+      type: String,
+      required: true
+    },
+    format: {
+      type: String,
+      enum: Object.values(import_document.DocumentFormat),
+      required: true,
+      default: import_document.DocumentFormat.MARKDOWN
+    },
+    content: {
+      type: String,
+      required: true
+    },
+    blocks: {
+      type: import_mongoose3.Schema.Types.Mixed
+    },
+    slots: {
+      type: import_mongoose3.Schema.Types.Mixed
+    },
+    // Cached AI advice ({ content, generatedAt }). Free-form so the schema
+    // doesn't strip it on update (mongoose strict mode).
+    advice: {
+      type: import_mongoose3.Schema.Types.Mixed
+    },
+    // Faceted grouping (e.g. category/workplaceId/month). Stored as a free
+    // map; index specific keys (e.g. `labels.workplaceId`) if query volume
+    // warrants it.
+    labels: {
+      type: import_mongoose3.Schema.Types.Mixed
+    },
+    source: {
+      type: String,
+      enum: Object.values(import_document.DocumentSource),
+      required: true
+    },
+    workflowId: {
+      type: String,
+      index: true
+    },
+    threadId: {
+      type: String,
+      index: true
+    },
+    version: {
+      type: Number,
+      required: true,
+      default: 1
+    },
+    editedManually: {
+      type: Boolean
+    },
+    createdAt: {
+      type: String,
+      required: true
+    },
+    updatedAt: {
+      type: String,
+      required: true
+    }
+  },
+  // `createdAt`/`updatedAt` are caller-supplied ISO strings (see Document type),
+  // so mongoose's automatic Date timestamps are intentionally disabled.
+  {
+    timestamps: false
+  }
+);
+var DocumentModel = import_mongoose4.default.model(
+  "Document",
+  DocumentObjectSchema
+);
+
+// implements/document.memory.ts
+var MongoDBDocument = class {
+  executeWithRetry;
+  getOperationTimeout;
+  constructor(executeWithRetry, getOperationTimeout) {
+    this.executeWithRetry = executeWithRetry;
+    this.getOperationTimeout = getOperationTimeout;
+  }
+  async getDocument(documentId) {
+    return this.executeWithRetry(async () => {
+      const timeout = this.getOperationTimeout();
+      const document = await DocumentModel.findOne({ documentId }).maxTimeMS(timeout).lean();
+      return document || void 0;
+    }, "getDocument()");
+  }
+  async createDocument(document) {
+    return this.executeWithRetry(async () => {
+      const created = await DocumentModel.create(document);
+      return created.toObject();
+    }, "createDocument()");
+  }
+  async updateDocument(documentId, document) {
+    const { documentId: _documentId, ...mutableUpdates } = document;
+    return this.executeWithRetry(async () => {
+      const timeout = this.getOperationTimeout();
+      await DocumentModel.updateOne(
+        { documentId },
+        { $set: mutableUpdates }
+      ).maxTimeMS(timeout);
+    }, "updateDocument()");
+  }
+  async deleteDocument(documentId) {
+    return this.executeWithRetry(async () => {
+      const timeout = this.getOperationTimeout();
+      await DocumentModel.deleteOne({ documentId }).maxTimeMS(timeout);
+    }, "deleteDocument()");
+  }
+  async listDocuments(userId, filter) {
+    return this.executeWithRetry(async () => {
+      const timeout = this.getOperationTimeout();
+      const query = {};
+      if (userId) query.userId = userId;
+      if (filter?.workflowId) query.workflowId = filter.workflowId;
+      if (filter?.threadId) query.threadId = filter.threadId;
+      if (filter?.source) query.source = filter.source;
+      if (filter?.labels) {
+        for (const [key, value] of Object.entries(filter.labels)) {
+          query[`labels.${key}`] = value;
+        }
+      }
+      const documents = await DocumentModel.find(query).maxTimeMS(timeout).lean();
+      return documents;
+    }, "listDocuments()");
+  }
+};
+
 // models/intent.model.ts
-var import_mongoose3 = __toESM(require("mongoose"), 1);
-var IntentObjectSchema = new import_mongoose3.Schema(
+var import_mongoose5 = __toESM(require("mongoose"), 1);
+var IntentObjectSchema = new import_mongoose5.Schema(
   {
     id: {
       type: String,
@@ -187,7 +331,7 @@ var IntentObjectSchema = new import_mongoose3.Schema(
     }
   }
 );
-var IntentModel = import_mongoose3.default.model("Intent", IntentObjectSchema);
+var IntentModel = import_mongoose5.default.model("Intent", IntentObjectSchema);
 
 // implements/intent.memory.ts
 var MongoDBIntent = class {
@@ -241,9 +385,9 @@ var MongoDBIntent = class {
 
 // models/threads.model.ts
 var import_memory = require("@ainetwork/adk/types/memory");
-var import_mongoose4 = require("mongoose");
-var import_mongoose5 = __toESM(require("mongoose"), 1);
-var ThreadObjectSchema = new import_mongoose4.Schema(
+var import_mongoose6 = require("mongoose");
+var import_mongoose7 = __toESM(require("mongoose"), 1);
+var ThreadObjectSchema = new import_mongoose6.Schema(
   {
     type: {
       type: String,
@@ -279,20 +423,20 @@ var ThreadObjectSchema = new import_mongoose4.Schema(
     timestamps: true
   }
 );
-var ThreadModel = import_mongoose5.default.model("Thread", ThreadObjectSchema);
+var ThreadModel = import_mongoose7.default.model("Thread", ThreadObjectSchema);
 
 // models/messages.model.ts
 var import_memory2 = require("@ainetwork/adk/types/memory");
-var import_mongoose6 = require("mongoose");
-var import_mongoose7 = __toESM(require("mongoose"), 1);
-var MessageContentObjectSchema = new import_mongoose6.Schema(
+var import_mongoose8 = require("mongoose");
+var import_mongoose9 = __toESM(require("mongoose"), 1);
+var MessageContentObjectSchema = new import_mongoose8.Schema(
   {
     type: { type: String, required: true },
-    parts: { type: [import_mongoose6.Schema.Types.Mixed], required: true }
+    parts: { type: [import_mongoose8.Schema.Types.Mixed], required: true }
   },
   { _id: false }
 );
-var MessageObjectSchema = new import_mongoose6.Schema(
+var MessageObjectSchema = new import_mongoose8.Schema(
   {
     messageId: {
       type: String,
@@ -323,7 +467,7 @@ var MessageObjectSchema = new import_mongoose6.Schema(
       required: true
     },
     metadata: {
-      type: import_mongoose6.Schema.Types.Mixed,
+      type: import_mongoose8.Schema.Types.Mixed,
       default: {}
     }
   },
@@ -332,7 +476,7 @@ var MessageObjectSchema = new import_mongoose6.Schema(
   }
 );
 MessageObjectSchema.index({ threadId: 1, messageId: 1 }, { unique: true });
-var MessageModel = import_mongoose7.default.model("Message", MessageObjectSchema);
+var MessageModel = import_mongoose9.default.model("Message", MessageObjectSchema);
 
 // implements/thread.memory.ts
 var import_logger = require("@ainetwork/adk/utils/logger");
@@ -447,9 +591,9 @@ var MongoDBThread = class {
 };
 
 // models/user-workflow.model.ts
-var import_mongoose8 = require("mongoose");
-var import_mongoose9 = __toESM(require("mongoose"), 1);
-var UserWorkflowObjectSchema = new import_mongoose8.Schema(
+var import_mongoose10 = require("mongoose");
+var import_mongoose11 = __toESM(require("mongoose"), 1);
+var UserWorkflowObjectSchema = new import_mongoose10.Schema(
   {
     workflowId: {
       type: String,
@@ -484,13 +628,13 @@ var UserWorkflowObjectSchema = new import_mongoose8.Schema(
       required: true
     },
     definition: {
-      type: import_mongoose8.Schema.Types.Mixed
+      type: import_mongoose10.Schema.Types.Mixed
     },
     variables: {
-      type: import_mongoose8.Schema.Types.Mixed
+      type: import_mongoose10.Schema.Types.Mixed
     },
     variableValues: {
-      type: import_mongoose8.Schema.Types.Mixed
+      type: import_mongoose10.Schema.Types.Mixed
     },
     schedule: {
       type: String
@@ -512,7 +656,7 @@ var UserWorkflowObjectSchema = new import_mongoose8.Schema(
     timestamps: true
   }
 );
-var UserWorkflowModel = import_mongoose9.default.model(
+var UserWorkflowModel = import_mongoose11.default.model(
   "UserWorkflow",
   UserWorkflowObjectSchema
 );
@@ -578,9 +722,9 @@ var MongoDBUserWorkflow = class {
 };
 
 // models/workflow-template.model.ts
-var import_mongoose10 = require("mongoose");
-var import_mongoose11 = __toESM(require("mongoose"), 1);
-var WorkflowTemplateObjectSchema = new import_mongoose10.Schema(
+var import_mongoose12 = require("mongoose");
+var import_mongoose13 = __toESM(require("mongoose"), 1);
+var WorkflowTemplateObjectSchema = new import_mongoose12.Schema(
   {
     templateId: {
       type: String,
@@ -608,17 +752,17 @@ var WorkflowTemplateObjectSchema = new import_mongoose10.Schema(
       required: true
     },
     definition: {
-      type: import_mongoose10.Schema.Types.Mixed
+      type: import_mongoose12.Schema.Types.Mixed
     },
     variables: {
-      type: import_mongoose10.Schema.Types.Mixed
+      type: import_mongoose12.Schema.Types.Mixed
     }
   },
   {
     timestamps: true
   }
 );
-var WorkflowTemplateModel = import_mongoose11.default.model(
+var WorkflowTemplateModel = import_mongoose13.default.model(
   "WorkflowTemplate",
   WorkflowTemplateObjectSchema
 );
@@ -688,6 +832,7 @@ var MongoDBMemory = class _MongoDBMemory {
   threadMemory;
   workflowTemplateMemory;
   userWorkflowMemory;
+  documentMemory;
   constructor(config) {
     const cfg = typeof config === "string" ? { uri: config } : config;
     this.uri = cfg.uri;
@@ -733,6 +878,10 @@ var MongoDBMemory = class _MongoDBMemory {
       this.executeWithRetry.bind(this),
       this.getOperationTimeout.bind(this)
     );
+    this.documentMemory = new MongoDBDocument(
+      this.executeWithRetry.bind(this),
+      this.getOperationTimeout.bind(this)
+    );
   }
   getAgentMemory() {
     return this.agentMemory;
@@ -749,26 +898,29 @@ var MongoDBMemory = class _MongoDBMemory {
   getUserWorkflowMemory() {
     return this.userWorkflowMemory;
   }
+  getDocumentMemory() {
+    return this.documentMemory;
+  }
   setupMongooseEventListeners() {
     if (this.eventListenersSetup) return;
     this.eventListenersSetup = true;
-    import_mongoose12.default.connection.on("connected", () => {
+    import_mongoose14.default.connection.on("connected", () => {
       this.connected = true;
       this.reconnectAttempts = 0;
       this.reconnecting = false;
       import_logger2.loggers.agent.info("MongoDB connected successfully");
     });
-    import_mongoose12.default.connection.on("disconnected", () => {
+    import_mongoose14.default.connection.on("disconnected", () => {
       this.connected = false;
       import_logger2.loggers.agent.warn("MongoDB disconnected");
       this.handleDisconnection();
     });
-    import_mongoose12.default.connection.on("error", (error) => {
+    import_mongoose14.default.connection.on("error", (error) => {
       this.connected = false;
       import_logger2.loggers.agent.error("MongoDB connection error:", error);
       this.handleDisconnection();
     });
-    import_mongoose12.default.connection.on("reconnected", () => {
+    import_mongoose14.default.connection.on("reconnected", () => {
       this.connected = true;
       this.reconnectAttempts = 0;
       this.reconnecting = false;
@@ -786,7 +938,7 @@ var MongoDBMemory = class _MongoDBMemory {
         `Attempting to reconnect to MongoDB (${this.reconnectAttempts}/${this.maxReconnectAttempts})...`
       );
       try {
-        await import_mongoose12.default.connect(this.uri, this.connectionConfig);
+        await import_mongoose14.default.connect(this.uri, this.connectionConfig);
         this.connected = true;
         this.reconnectAttempts = 0;
         this.reconnecting = false;
@@ -816,7 +968,7 @@ var MongoDBMemory = class _MongoDBMemory {
       return;
     }
     try {
-      await import_mongoose12.default.connect(this.uri, this.connectionConfig);
+      await import_mongoose14.default.connect(this.uri, this.connectionConfig);
       this.connected = true;
       this.reconnectAttempts = 0;
       await this.setupTTLIndex();
@@ -835,7 +987,7 @@ var MongoDBMemory = class _MongoDBMemory {
         clearInterval(this.orphanCleanupTimer);
         this.orphanCleanupTimer = void 0;
       }
-      await import_mongoose12.default.disconnect();
+      await import_mongoose14.default.disconnect();
       this.connected = false;
     } catch (error) {
       import_logger2.loggers.agent.error("Failed to disconnect from MongoDB:", error);
@@ -861,7 +1013,7 @@ var MongoDBMemory = class _MongoDBMemory {
   async setupTTLIndex() {
     if (this.threadTTLSeconds === void 0) return;
     try {
-      const db = import_mongoose12.default.connection.db;
+      const db = import_mongoose14.default.connection.db;
       if (!db) return;
       const collection = db.collection("threads");
       const indexes = await collection.indexes();
@@ -901,7 +1053,7 @@ var MongoDBMemory = class _MongoDBMemory {
   async cleanupOrphanedMessages() {
     if (!this.connected) return;
     try {
-      const db = import_mongoose12.default.connection.db;
+      const db = import_mongoose14.default.connection.db;
       if (!db) return;
       const existingThreadIds = await db.collection("threads").distinct("threadId");
       const result = await MessageModel.deleteMany({
@@ -938,7 +1090,7 @@ var MongoDBMemory = class _MongoDBMemory {
           `${operationName} failed due to too many sessions, disconnecting to release sessions...`
         );
         try {
-          await import_mongoose12.default.disconnect();
+          await import_mongoose14.default.disconnect();
           this.connected = false;
         } catch (disconnectError) {
           import_logger2.loggers.agent.error("Failed to disconnect during session cleanup:", disconnectError);