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

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

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

package/dist/models/threads.model.d.cts

@@ -8,31 +8,31 @@ declare const ThreadObjectSchema: Schema<any, mongoose.Model<any, any, any, any,
     updatedAt: NativeDate;
 } & {
     type: ThreadType;
-    threadId: string;
     userId: string;
+    threadId: string;
     title?: string | null | undefined;
-    isPinned?: boolean | null | undefined;
     workflowId?: string | null | undefined;
+    isPinned?: boolean | null | undefined;
 }, Document<unknown, {}, mongoose.FlatRecord<{
     createdAt: NativeDate;
     updatedAt: NativeDate;
 } & {
     type: ThreadType;
-    threadId: string;
     userId: string;
+    threadId: string;
     title?: string | null | undefined;
-    isPinned?: boolean | null | undefined;
     workflowId?: string | null | undefined;
+    isPinned?: boolean | null | undefined;
 }>, {}> & mongoose.FlatRecord<{
     createdAt: NativeDate;
     updatedAt: NativeDate;
 } & {
     type: ThreadType;
-    threadId: string;
     userId: string;
+    threadId: string;
     title?: string | null | undefined;
-    isPinned?: boolean | null | undefined;
     workflowId?: string | null | undefined;
+    isPinned?: boolean | null | undefined;
 }> & {
     _id: mongoose.Types.ObjectId;
 } & {

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

@@ -8,31 +8,31 @@ declare const ThreadObjectSchema: Schema<any, mongoose.Model<any, any, any, any,
     updatedAt: NativeDate;
 } & {
     type: ThreadType;
-    threadId: string;
     userId: string;
+    threadId: string;
     title?: string | null | undefined;
-    isPinned?: boolean | null | undefined;
     workflowId?: string | null | undefined;
+    isPinned?: boolean | null | undefined;
 }, Document<unknown, {}, mongoose.FlatRecord<{
     createdAt: NativeDate;
     updatedAt: NativeDate;
 } & {
     type: ThreadType;
-    threadId: string;
     userId: string;
+    threadId: string;
     title?: string | null | undefined;
-    isPinned?: boolean | null | undefined;
     workflowId?: string | null | undefined;
+    isPinned?: boolean | null | undefined;
 }>, {}> & mongoose.FlatRecord<{
     createdAt: NativeDate;
     updatedAt: NativeDate;
 } & {
     type: ThreadType;
-    threadId: string;
     userId: string;
+    threadId: string;
     title?: string | null | undefined;
-    isPinned?: boolean | null | undefined;
     workflowId?: string | null | undefined;
+    isPinned?: boolean | null | undefined;
 }> & {
     _id: mongoose.Types.ObjectId;
 } & {

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

@@ -9,8 +9,8 @@ declare const UserWorkflowObjectSchema: Schema<any, mongoose.Model<any, any, any
 } & {
     userId: string;
     title: string;
-    workflowId: string;
     content: string;
+    workflowId: string;
     active: boolean;
     description?: string | null | undefined;
     category?: string | null | undefined;
@@ -29,8 +29,8 @@ declare const UserWorkflowObjectSchema: Schema<any, mongoose.Model<any, any, any
 } & {
     userId: string;
     title: string;
-    workflowId: string;
     content: string;
+    workflowId: string;
     active: boolean;
     description?: string | null | undefined;
     category?: string | null | undefined;
@@ -49,8 +49,8 @@ declare const UserWorkflowObjectSchema: Schema<any, mongoose.Model<any, any, any
 } & {
     userId: string;
     title: string;
-    workflowId: string;
     content: string;
+    workflowId: string;
     active: boolean;
     description?: string | null | undefined;
     category?: string | null | undefined;

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

@@ -9,8 +9,8 @@ declare const UserWorkflowObjectSchema: Schema<any, mongoose.Model<any, any, any
 } & {
     userId: string;
     title: string;
-    workflowId: string;
     content: string;
+    workflowId: string;
     active: boolean;
     description?: string | null | undefined;
     category?: string | null | undefined;
@@ -29,8 +29,8 @@ declare const UserWorkflowObjectSchema: Schema<any, mongoose.Model<any, any, any
 } & {
     userId: string;
     title: string;
-    workflowId: string;
     content: string;
+    workflowId: string;
     active: boolean;
     description?: string | null | undefined;
     category?: string | null | undefined;
@@ -49,8 +49,8 @@ declare const UserWorkflowObjectSchema: Schema<any, mongoose.Model<any, any, any
 } & {
     userId: string;
     title: string;
-    workflowId: string;
     content: string;
+    workflowId: string;
     active: boolean;
     description?: string | null | undefined;
     category?: string | null | undefined;

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,88 @@
+import type { IDocumentMemory } from "@ainetwork/adk/modules";
+import type { Document, DocumentFilter } 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 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.0",
   "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.0"
   },
   "devDependencies": {
     "typescript": "^5.0.0"
@@ -40,5 +40,5 @@
   "publishConfig": {
     "access": "public"
   },
-  "gitHead": "ae4661ecacd1bb3c8f23cc903a62bebf71db7e5d"
+  "gitHead": "39b90b146668f11b0b6a15a3ba0ab5bf80eb026f"
 }