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

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

@@ -25,12 +25,12 @@ declare const MessageObjectSchema: Schema<any, mongoose.Model<any, any, any, any
 } & {
     threadId: string;
     userId: string;
-    messageId: string;
-    role: MessageRole;
     content: {
         type: string;
         parts: any[];
     };
+    messageId: string;
+    role: MessageRole;
     timestamp: number;
     metadata: any;
 }, Document<unknown, {}, mongoose.FlatRecord<{
@@ -39,12 +39,12 @@ declare const MessageObjectSchema: Schema<any, mongoose.Model<any, any, any, any
 } & {
     threadId: string;
     userId: string;
-    messageId: string;
-    role: MessageRole;
     content: {
         type: string;
         parts: any[];
     };
+    messageId: string;
+    role: MessageRole;
     timestamp: number;
     metadata: any;
 }>, {}> & mongoose.FlatRecord<{
@@ -53,12 +53,12 @@ declare const MessageObjectSchema: Schema<any, mongoose.Model<any, any, any, any
 } & {
     threadId: string;
     userId: string;
-    messageId: string;
-    role: MessageRole;
     content: {
         type: string;
         parts: any[];
     };
+    messageId: string;
+    role: MessageRole;
     timestamp: number;
     metadata: any;
 }> & {

package/dist/models/workflow-template.model.d.cts

@@ -7,8 +7,8 @@ declare const WorkflowTemplateObjectSchema: Schema<any, mongoose.Model<any, any,
     createdAt: NativeDate;
     updatedAt: NativeDate;
 } & {
-    description: string;
     title: string;
+    description: string;
     content: string;
     active: boolean;
     templateId: string;
@@ -19,8 +19,8 @@ declare const WorkflowTemplateObjectSchema: Schema<any, mongoose.Model<any, any,
     createdAt: NativeDate;
     updatedAt: NativeDate;
 } & {
-    description: string;
     title: string;
+    description: string;
     content: string;
     active: boolean;
     templateId: string;
@@ -31,8 +31,8 @@ declare const WorkflowTemplateObjectSchema: Schema<any, mongoose.Model<any, any,
     createdAt: NativeDate;
     updatedAt: NativeDate;
 } & {
-    description: string;
     title: string;
+    description: string;
     content: string;
     active: boolean;
     templateId: string;

package/dist/models/workflow-template.model.d.ts

@@ -7,8 +7,8 @@ declare const WorkflowTemplateObjectSchema: Schema<any, mongoose.Model<any, any,
     createdAt: NativeDate;
     updatedAt: NativeDate;
 } & {
-    description: string;
     title: string;
+    description: string;
     content: string;
     active: boolean;
     templateId: string;
@@ -19,8 +19,8 @@ declare const WorkflowTemplateObjectSchema: Schema<any, mongoose.Model<any, any,
     createdAt: NativeDate;
     updatedAt: NativeDate;
 } & {
-    description: string;
     title: string;
+    description: string;
     content: string;
     active: boolean;
     templateId: string;
@@ -31,8 +31,8 @@ declare const WorkflowTemplateObjectSchema: Schema<any, mongoose.Model<any, any,
     createdAt: NativeDate;
     updatedAt: NativeDate;
 } & {
-    description: string;
     title: string;
+    description: string;
     content: string;
     active: boolean;
     templateId: string;

package/implements/base.memory.ts

@@ -1,7 +1,8 @@
-import { IAgentMemory, IIntentMemory, IMemory, IUserWorkflowMemory, IThreadMemory, IWorkflowTemplateMemory } from "@ainetwork/adk/modules";
+import { IAgentMemory, IDocumentMemory, IIntentMemory, IMemory, IUserWorkflowMemory, IThreadMemory, IWorkflowTemplateMemory } from "@ainetwork/adk/modules";
 import mongoose from "mongoose";
 import { loggers } from "@ainetwork/adk/utils/logger";
 import { MongoDBAgent } from "./agent.memory";
+import { MongoDBDocument } from "./document.memory";
 import { MongoDBIntent } from "./intent.memory";
 import { MongoDBThread } from "./thread.memory";
 import { MongoDBUserWorkflow } from "./user-workflow.memory";
@@ -39,6 +40,7 @@ export class MongoDBMemory implements IMemory {
   private threadMemory: MongoDBThread;
   private workflowTemplateMemory: MongoDBWorkflowTemplate;
   private userWorkflowMemory: MongoDBUserWorkflow;
+  private documentMemory: MongoDBDocument;
 
   constructor(config: string | MongoDBMemoryConfig) {
     const cfg = typeof config === 'string' ? { uri: config } : config;
@@ -93,6 +95,11 @@ export class MongoDBMemory implements IMemory {
 			this.executeWithRetry.bind(this),
 			this.getOperationTimeout.bind(this)
 		);
+
+		this.documentMemory = new MongoDBDocument(
+			this.executeWithRetry.bind(this),
+			this.getOperationTimeout.bind(this)
+		);
   }
 
   public getAgentMemory(): IAgentMemory {
@@ -115,6 +122,10 @@ export class MongoDBMemory implements IMemory {
     return this.userWorkflowMemory;
   }
 
+  public getDocumentMemory(): IDocumentMemory {
+    return this.documentMemory;
+  }
+
   private setupMongooseEventListeners(): void {
     if (this.eventListenersSetup) return;
 

package/implements/document.memory.ts

@@ -0,0 +1,125 @@
+import type { IDocumentMemory } from "@ainetwork/adk/modules";
+import type {
+  Document,
+  DocumentFilter,
+  DocumentSlot,
+} from "@ainetwork/adk/types/document";
+import { DocumentModel } from "../models/document.model";
+
+export type ExecuteWithRetryFn = <T>(
+  operation: () => Promise<T>,
+  operationName?: string
+) => Promise<T>;
+
+export type GetOperationTimeoutFn = () => number;
+
+export class MongoDBDocument implements IDocumentMemory {
+  private executeWithRetry: ExecuteWithRetryFn;
+  private getOperationTimeout: GetOperationTimeoutFn;
+
+  constructor(
+    executeWithRetry: ExecuteWithRetryFn,
+    getOperationTimeout: GetOperationTimeoutFn
+  ) {
+    this.executeWithRetry = executeWithRetry;
+    this.getOperationTimeout = getOperationTimeout;
+  }
+
+  public async getDocument(documentId: string): Promise<Document | undefined> {
+    return this.executeWithRetry(async () => {
+      const timeout = this.getOperationTimeout();
+      const document = await DocumentModel.findOne({ documentId })
+        .maxTimeMS(timeout)
+        .lean<Document>();
+      return document || undefined;
+    }, "getDocument()");
+  }
+
+  public async createDocument(document: Document): Promise<Document> {
+    return this.executeWithRetry(async () => {
+      const created = await DocumentModel.create(document);
+      return created.toObject() as Document;
+    }, "createDocument()");
+  }
+
+  public async updateDocument(
+    documentId: string,
+    document: Partial<Document>
+  ): Promise<void> {
+    // `documentId` is immutable; never let a payload reassign it.
+    const { documentId: _documentId, ...mutableUpdates } = document;
+
+    return this.executeWithRetry(async () => {
+      const timeout = this.getOperationTimeout();
+      await DocumentModel.updateOne(
+        { documentId },
+        { $set: mutableUpdates }
+      ).maxTimeMS(timeout);
+    }, "updateDocument()");
+  }
+
+  public async updateDocumentSlot(
+    documentId: string,
+    slotId: string,
+    patch: Partial<DocumentSlot>
+  ): Promise<void> {
+    // Target only the matched slot via the positional `$` operator so
+    // concurrent fills of other slots are never overwritten. `slotId` is the
+    // slot's identity and must not be patched.
+    const { slotId: _slotId, ...fields } = patch;
+    const set: Record<string, unknown> = { updatedAt: new Date().toISOString() };
+    const unset: Record<string, unknown> = {};
+    for (const [key, value] of Object.entries(fields)) {
+      if (value === undefined) {
+        unset[`slots.$.${key}`] = "";
+      } else {
+        set[`slots.$.${key}`] = value;
+      }
+    }
+
+    const update: Record<string, unknown> = { $set: set, $inc: { version: 1 } };
+    if (Object.keys(unset).length > 0) {
+      update.$unset = unset;
+    }
+
+    return this.executeWithRetry(async () => {
+      const timeout = this.getOperationTimeout();
+      await DocumentModel.updateOne(
+        { documentId, "slots.slotId": slotId },
+        update
+      ).maxTimeMS(timeout);
+    }, "updateDocumentSlot()");
+  }
+
+  public async deleteDocument(documentId: string): Promise<void> {
+    return this.executeWithRetry(async () => {
+      const timeout = this.getOperationTimeout();
+      await DocumentModel.deleteOne({ documentId }).maxTimeMS(timeout);
+    }, "deleteDocument()");
+  }
+
+  public async listDocuments(
+    userId?: string,
+    filter?: DocumentFilter
+  ): Promise<Document[]> {
+    return this.executeWithRetry(async () => {
+      const timeout = this.getOperationTimeout();
+      const query: Record<string, unknown> = {};
+      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) {
+        // Subset match: every provided label must equal the stored value.
+        for (const [key, value] of Object.entries(filter.labels)) {
+          query[`labels.${key}`] = value;
+        }
+      }
+
+      const documents = await DocumentModel.find(query)
+        .maxTimeMS(timeout)
+        .lean<Document[]>();
+      return documents;
+    }, "listDocuments()");
+  }
+}

package/models/document.model.ts

@@ -0,0 +1,113 @@
+import {
+	type DocumentAdvice,
+	DocumentFormat,
+	type DocumentSlot,
+	DocumentSource,
+} from "@ainetwork/adk/types/document";
+import type { WorkflowRenderedBlock } from "@ainetwork/adk/types/memory";
+import { type Document, Schema } from "mongoose";
+import mongoose from "mongoose";
+
+export const 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,
+	}
+);
+
+export interface DocumentDocument extends Document {
+	documentId: string;
+	userId: string;
+	title: string;
+	format: DocumentFormat;
+	content: string;
+	blocks?: WorkflowRenderedBlock[];
+	slots?: DocumentSlot[];
+	advice?: DocumentAdvice;
+	labels?: Record<string, string>;
+	source: DocumentSource;
+	workflowId?: string;
+	threadId?: string;
+	version: number;
+	editedManually?: boolean;
+	createdAt: string;
+	updatedAt: string;
+}
+
+export const DocumentModel = mongoose.model<DocumentDocument>(
+	"Document",
+	DocumentObjectSchema
+);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ainetwork/adk-provider-memory-mongodb",
-  "version": "0.6.2",
+  "version": "0.7.1",
   "author": "AI Network (https://ainetwork.ai)",
   "type": "module",
   "engines": {
@@ -31,7 +31,7 @@
     "mongoose": "^8.16.5"
   },
   "peerDependencies": {
-    "@ainetwork/adk": "^0.6.3"
+    "@ainetwork/adk": "^0.7.1"
   },
   "devDependencies": {
     "typescript": "^5.0.0"
@@ -40,5 +40,5 @@
   "publishConfig": {
     "access": "public"
   },
-  "gitHead": "ae4661ecacd1bb3c8f23cc903a62bebf71db7e5d"
+  "gitHead": "b5988a3d1333127538f41a55cee36fc96632b69c"
 }