@talkpilot/core-db 1.3.53 → 1.3.55

package/src/websitalk/knowledgeDocuments/__tests__/knowledgeDocuments.spec.ts

@@ -1,186 +1,238 @@
-import {
-  KNOWLEDGE_CONTENT_TYPES,
-  KNOWLEDGE_DOCUMENT_STATUSES,
-  KNOWLEDGE_FILE_TYPES,
-} from "../knowledgeDocuments.constants";
-import {
-  createKnowledgeDocument,
-  deleteKnowledgeDocument,
-  getKnowledgeDocumentById,
-  getKnowledgeDocumentsCollection,
-  listKnowledgeDocumentsByClientId,
-  listReadyKnowledgeDocumentIdsByClientId,
-  updateKnowledgeDocument,
-} from "../knowledgeDocuments.getters";
-import { createKnowledgeDocumentFixture } from "../../../test-utils/factories/websitalk/knowledgeDocuments";
-
-describe("db.websitalk.knowledgeDocuments", () => {
-  const clientId = "test-client";
-  const nonExistentId = "507f1f77bcf86cd799439011";
-
-  describe("createKnowledgeDocument", () => {
-    it("should create a knowledge document", async () => {
-      const id = await createKnowledgeDocument({
-        clientId,
-        filename: "faq.pdf",
-        contentType: KNOWLEDGE_CONTENT_TYPES.PDF,
-        fileType: KNOWLEDGE_FILE_TYPES.PDF,
-        storagePath: "websitalk/knowledge_files/test/faq.pdf",
-        byteSize: 2048,
-      });
-
-      const saved = await getKnowledgeDocumentById(id.toString());
-      expect(saved).toMatchObject({
-        clientId,
-        filename: "faq.pdf",
-        status: KNOWLEDGE_DOCUMENT_STATUSES.UPLOADING,
-        progressPercent: 0,
-        fileType: KNOWLEDGE_FILE_TYPES.PDF,
-      });
-      expect(saved?.createdAt).toBeInstanceOf(Date);
-      expect(saved?.updatedAt).toBeInstanceOf(Date);
-    });
-  });
-
-  describe("getKnowledgeDocumentById", () => {
-    it("should throw for an invalid id", async () => {
-      await expect(getKnowledgeDocumentById("invalid-id")).rejects.toThrow();
-    });
-  });
-
-  describe("listKnowledgeDocumentsByClientId", () => {
-    it("should return documents for a client only", async () => {
-      await getKnowledgeDocumentsCollection().insertMany([
-        createKnowledgeDocumentFixture({ clientId, filename: "a.pdf" }),
-        createKnowledgeDocumentFixture({ clientId, filename: "b.pdf" }),
-        createKnowledgeDocumentFixture({
-          clientId: "other-client",
-          filename: "c.pdf",
-        }),
-      ]);
-
-      const results = await listKnowledgeDocumentsByClientId(clientId);
-      expect(results).toHaveLength(2);
-      expect(results.map((doc) => doc.filename).sort()).toEqual([
-        "a.pdf",
-        "b.pdf",
-      ]);
-    });
-
-    it("should return documents sorted by createdAt descending", async () => {
-      await getKnowledgeDocumentsCollection().insertMany([
-        createKnowledgeDocumentFixture({
-          clientId,
-          filename: "older.pdf",
-          createdAt: new Date("2024-01-01"),
-          updatedAt: new Date("2024-01-01"),
-        }),
-        createKnowledgeDocumentFixture({
-          clientId,
-          filename: "newer.pdf",
-          createdAt: new Date("2024-06-01"),
-          updatedAt: new Date("2024-06-01"),
-        }),
-      ]);
-
-      const results = await listKnowledgeDocumentsByClientId(clientId);
-      expect(results.map((doc) => doc.filename)).toEqual([
-        "newer.pdf",
-        "older.pdf",
-      ]);
-    });
-  });
-
-  describe("listReadyKnowledgeDocumentIdsByClientId", () => {
-    it("should return only READY document ids sorted by createdAt descending", async () => {
-      await getKnowledgeDocumentsCollection().insertMany([
-        createKnowledgeDocumentFixture({
-          clientId,
-          status: KNOWLEDGE_DOCUMENT_STATUSES.READY,
-          createdAt: new Date("2024-01-01"),
-          updatedAt: new Date("2024-01-01"),
-        }),
-        createKnowledgeDocumentFixture({
-          clientId,
-          status: KNOWLEDGE_DOCUMENT_STATUSES.READY,
-          createdAt: new Date("2024-06-01"),
-          updatedAt: new Date("2024-06-01"),
-        }),
-        createKnowledgeDocumentFixture({
-          clientId,
-          status: KNOWLEDGE_DOCUMENT_STATUSES.PROCESSING,
-        }),
-        createKnowledgeDocumentFixture({
-          clientId: "other",
-          status: KNOWLEDGE_DOCUMENT_STATUSES.READY,
-        }),
-      ]);
-
-      const readyDocs = await getKnowledgeDocumentsCollection()
-        .find({ clientId, status: KNOWLEDGE_DOCUMENT_STATUSES.READY })
-        .sort({ createdAt: -1 })
-        .toArray();
-      const expectedIds = readyDocs.map((doc) => doc._id.toString());
-
-      const ids = await listReadyKnowledgeDocumentIdsByClientId(clientId);
-      expect(ids).toHaveLength(2);
-      expect(ids).toEqual(expectedIds);
-    });
-  });
-
-  describe("updateKnowledgeDocument", () => {
-    it("should update status and progress", async () => {
-      const id = await createKnowledgeDocument({
-        clientId,
-        filename: "doc.pdf",
-        contentType: KNOWLEDGE_CONTENT_TYPES.PDF,
-        fileType: KNOWLEDGE_FILE_TYPES.PDF,
-        storagePath: "websitalk/knowledge_files/test/doc.pdf",
-        byteSize: 100,
-      });
-
-      const before = await getKnowledgeDocumentById(id.toString());
-
-      const updated = await updateKnowledgeDocument(id.toString(), {
-        status: KNOWLEDGE_DOCUMENT_STATUSES.READY,
-        progressPercent: 100,
-        pageCount: 12,
-      });
-
-      expect(updated?.status).toBe(KNOWLEDGE_DOCUMENT_STATUSES.READY);
-      expect(updated?.progressPercent).toBe(100);
-      expect(updated?.pageCount).toBe(12);
-      expect(updated?.updatedAt.getTime()).toBeGreaterThanOrEqual(
-        before!.updatedAt.getTime(),
-      );
-    });
-
-    it("should return null when the document does not exist", async () => {
-      expect(
-        await updateKnowledgeDocument(nonExistentId, {
-          status: KNOWLEDGE_DOCUMENT_STATUSES.READY,
-        }),
-      ).toBeNull();
-    });
-  });
-
-  describe("deleteKnowledgeDocument", () => {
-    it("should delete and return true", async () => {
-      const id = await createKnowledgeDocument({
-        clientId: "delete-client",
-        filename: "gone.pdf",
-        contentType: KNOWLEDGE_CONTENT_TYPES.PDF,
-        fileType: KNOWLEDGE_FILE_TYPES.PDF,
-        storagePath: "websitalk/knowledge_files/delete/gone.pdf",
-        byteSize: 10,
-      });
-
-      expect(await deleteKnowledgeDocument(id.toString())).toBe(true);
-      expect(await getKnowledgeDocumentById(id.toString())).toBeNull();
-    });
-
-    it("should return false when the document does not exist", async () => {
-      expect(await deleteKnowledgeDocument(nonExistentId)).toBe(false);
-    });
-  });
-});
+import {
+  KNOWLEDGE_CONTENT_TYPES,
+  KNOWLEDGE_DOCUMENT_STATUSES,
+  KNOWLEDGE_FILE_TYPES,
+} from "../knowledgeDocuments.constants";
+import {
+  createKnowledgeDocument,
+  deleteKnowledgeDocument,
+  getKnowledgeDocumentById,
+  getKnowledgeDocumentsCollection,
+  listKnowledgeDocumentsByClientId,
+  listReadyKnowledgeDocumentIdsByClientId,
+  listReadyKnowledgeDocumentsByClientId,
+  updateKnowledgeDocument,
+} from "../knowledgeDocuments.getters";
+import { createKnowledgeDocumentFixture } from "../../../test-utils/factories/websitalk/knowledgeDocuments";
+
+describe("db.websitalk.knowledgeDocuments", () => {
+  const clientId = "test-client";
+  const nonExistentId = "507f1f77bcf86cd799439011";
+
+  describe("createKnowledgeDocument", () => {
+    it("should create a knowledge document", async () => {
+      const id = await createKnowledgeDocument({
+        clientId,
+        filename: "faq.pdf",
+        contentType: KNOWLEDGE_CONTENT_TYPES.PDF,
+        fileType: KNOWLEDGE_FILE_TYPES.PDF,
+        storagePath: "websitalk/knowledge_files/test/faq.pdf",
+        byteSize: 2048,
+      });
+
+      const saved = await getKnowledgeDocumentById(id.toString());
+      expect(saved).toMatchObject({
+        clientId,
+        filename: "faq.pdf",
+        status: KNOWLEDGE_DOCUMENT_STATUSES.UPLOADING,
+        progressPercent: 0,
+        fileType: KNOWLEDGE_FILE_TYPES.PDF,
+      });
+      expect(saved?.createdAt).toBeInstanceOf(Date);
+      expect(saved?.updatedAt).toBeInstanceOf(Date);
+    });
+  });
+
+  describe("getKnowledgeDocumentById", () => {
+    it("should throw for an invalid id", async () => {
+      await expect(getKnowledgeDocumentById("invalid-id")).rejects.toThrow();
+    });
+  });
+
+  describe("listKnowledgeDocumentsByClientId", () => {
+    it("should return documents for a client only", async () => {
+      await getKnowledgeDocumentsCollection().insertMany([
+        createKnowledgeDocumentFixture({ clientId, filename: "a.pdf" }),
+        createKnowledgeDocumentFixture({ clientId, filename: "b.pdf" }),
+        createKnowledgeDocumentFixture({
+          clientId: "other-client",
+          filename: "c.pdf",
+        }),
+      ]);
+
+      const results = await listKnowledgeDocumentsByClientId(clientId);
+      expect(results).toHaveLength(2);
+      expect(results.map((doc) => doc.filename).sort()).toEqual([
+        "a.pdf",
+        "b.pdf",
+      ]);
+    });
+
+    it("should return documents sorted by createdAt descending", async () => {
+      await getKnowledgeDocumentsCollection().insertMany([
+        createKnowledgeDocumentFixture({
+          clientId,
+          filename: "older.pdf",
+          createdAt: new Date("2024-01-01"),
+          updatedAt: new Date("2024-01-01"),
+        }),
+        createKnowledgeDocumentFixture({
+          clientId,
+          filename: "newer.pdf",
+          createdAt: new Date("2024-06-01"),
+          updatedAt: new Date("2024-06-01"),
+        }),
+      ]);
+
+      const results = await listKnowledgeDocumentsByClientId(clientId);
+      expect(results.map((doc) => doc.filename)).toEqual([
+        "newer.pdf",
+        "older.pdf",
+      ]);
+    });
+  });
+
+  describe("listReadyKnowledgeDocumentIdsByClientId", () => {
+    it("should return only READY document ids sorted by createdAt descending", async () => {
+      await getKnowledgeDocumentsCollection().insertMany([
+        createKnowledgeDocumentFixture({
+          clientId,
+          status: KNOWLEDGE_DOCUMENT_STATUSES.READY,
+          createdAt: new Date("2024-01-01"),
+          updatedAt: new Date("2024-01-01"),
+        }),
+        createKnowledgeDocumentFixture({
+          clientId,
+          status: KNOWLEDGE_DOCUMENT_STATUSES.READY,
+          createdAt: new Date("2024-06-01"),
+          updatedAt: new Date("2024-06-01"),
+        }),
+        createKnowledgeDocumentFixture({
+          clientId,
+          status: KNOWLEDGE_DOCUMENT_STATUSES.PROCESSING,
+        }),
+        createKnowledgeDocumentFixture({
+          clientId: "other",
+          status: KNOWLEDGE_DOCUMENT_STATUSES.READY,
+        }),
+      ]);
+
+      const readyDocs = await getKnowledgeDocumentsCollection()
+        .find({ clientId, status: KNOWLEDGE_DOCUMENT_STATUSES.READY })
+        .sort({ createdAt: -1 })
+        .toArray();
+      const expectedIds = readyDocs.map((doc) => doc._id.toString());
+
+      const ids = await listReadyKnowledgeDocumentIdsByClientId(clientId);
+      expect(ids).toHaveLength(2);
+      expect(ids).toEqual(expectedIds);
+    });
+  });
+
+  describe("listReadyKnowledgeDocumentsByClientId", () => {
+    it("should return only READY documents sorted by createdAt descending", async () => {
+      await getKnowledgeDocumentsCollection().insertMany([
+        createKnowledgeDocumentFixture({
+          clientId,
+          filename: "older.pdf",
+          status: KNOWLEDGE_DOCUMENT_STATUSES.READY,
+          markdownStoragePath: "path/older.md",
+          charCount: 100,
+          totalTokens: 20,
+          createdAt: new Date("2024-01-01"),
+          updatedAt: new Date("2024-01-01"),
+        }),
+        createKnowledgeDocumentFixture({
+          clientId,
+          filename: "newer.pdf",
+          status: KNOWLEDGE_DOCUMENT_STATUSES.READY,
+          markdownStoragePath: "path/newer.md",
+          charCount: 200,
+          totalTokens: 40,
+          createdAt: new Date("2024-06-01"),
+          updatedAt: new Date("2024-06-01"),
+        }),
+        createKnowledgeDocumentFixture({
+          clientId,
+          filename: "processing.pdf",
+          status: KNOWLEDGE_DOCUMENT_STATUSES.PROCESSING,
+        }),
+        createKnowledgeDocumentFixture({
+          clientId: "other",
+          filename: "other.pdf",
+          status: KNOWLEDGE_DOCUMENT_STATUSES.READY,
+        }),
+      ]);
+
+      const docs = await listReadyKnowledgeDocumentsByClientId(clientId);
+      expect(docs.map((doc) => doc.filename)).toEqual([
+        "newer.pdf",
+        "older.pdf",
+      ]);
+      expect(docs[0]).toMatchObject({
+        clientId,
+        filename: "newer.pdf",
+        markdownStoragePath: "path/newer.md",
+        charCount: 200,
+        totalTokens: 40,
+        status: KNOWLEDGE_DOCUMENT_STATUSES.READY,
+      });
+    });
+  });
+
+  describe("updateKnowledgeDocument", () => {
+    it("should update status and progress", async () => {
+      const id = await createKnowledgeDocument({
+        clientId,
+        filename: "doc.pdf",
+        contentType: KNOWLEDGE_CONTENT_TYPES.PDF,
+        fileType: KNOWLEDGE_FILE_TYPES.PDF,
+        storagePath: "websitalk/knowledge_files/test/doc.pdf",
+        byteSize: 100,
+      });
+
+      const before = await getKnowledgeDocumentById(id.toString());
+
+      const updated = await updateKnowledgeDocument(id.toString(), {
+        status: KNOWLEDGE_DOCUMENT_STATUSES.READY,
+        progressPercent: 100,
+        pageCount: 12,
+      });
+
+      expect(updated?.status).toBe(KNOWLEDGE_DOCUMENT_STATUSES.READY);
+      expect(updated?.progressPercent).toBe(100);
+      expect(updated?.pageCount).toBe(12);
+      expect(updated?.updatedAt.getTime()).toBeGreaterThanOrEqual(
+        before!.updatedAt.getTime(),
+      );
+    });
+
+    it("should return null when the document does not exist", async () => {
+      expect(
+        await updateKnowledgeDocument(nonExistentId, {
+          status: KNOWLEDGE_DOCUMENT_STATUSES.READY,
+        }),
+      ).toBeNull();
+    });
+  });
+
+  describe("deleteKnowledgeDocument", () => {
+    it("should delete and return true", async () => {
+      const id = await createKnowledgeDocument({
+        clientId: "delete-client",
+        filename: "gone.pdf",
+        contentType: KNOWLEDGE_CONTENT_TYPES.PDF,
+        fileType: KNOWLEDGE_FILE_TYPES.PDF,
+        storagePath: "websitalk/knowledge_files/delete/gone.pdf",
+        byteSize: 10,
+      });
+
+      expect(await deleteKnowledgeDocument(id.toString())).toBe(true);
+      expect(await getKnowledgeDocumentById(id.toString())).toBeNull();
+    });
+
+    it("should return false when the document does not exist", async () => {
+      expect(await deleteKnowledgeDocument(nonExistentId)).toBe(false);
+    });
+  });
+});

package/src/websitalk/knowledgeDocuments/index.ts

@@ -1,3 +1,3 @@
-export * from "./knowledgeDocuments.constants";
-export * from "./knowledgeDocuments.getters";
-export * from "./knowledgeDocuments.types";
+export * from "./knowledgeDocuments.constants";
+export * from "./knowledgeDocuments.getters";
+export * from "./knowledgeDocuments.types";

package/src/websitalk/knowledgeDocuments/knowledgeDocuments.constants.ts

@@ -1,17 +1,26 @@
-export const KNOWLEDGE_DOCUMENTS_COLLECTION = "knowledgeDocuments";
-
-export const KNOWLEDGE_DOCUMENT_STATUSES = {
-  UPLOADING: "UPLOADING",
-  EXTRACTING: "EXTRACTING",
-  PROCESSING: "PROCESSING",
-  READY: "READY",
-  FAILED: "FAILED",
-} as const;
-
-export const KNOWLEDGE_FILE_TYPES = {
-  PDF: "pdf",
-} as const;
-
-export const KNOWLEDGE_CONTENT_TYPES = {
-  PDF: "application/pdf",
-} as const;
+export const KNOWLEDGE_DOCUMENTS_COLLECTION = "knowledgeDocuments";
+
+export const KNOWLEDGE_DOCUMENT_STATUSES = {
+  UPLOADING: "UPLOADING",
+  EXTRACTING: "EXTRACTING",
+  PROCESSING: "PROCESSING",
+  READY: "READY",
+  FAILED: "FAILED",
+} as const;
+
+export const KNOWLEDGE_FILE_TYPES = {
+  PDF: "pdf",
+} as const;
+
+export const KNOWLEDGE_CONTENT_TYPES = {
+  PDF: "application/pdf",
+} as const;
+
+export const KNOWLEDGE_DOCUMENT_READY_PROJECTION = {
+  clientId: 1,
+  filename: 1,
+  markdownStoragePath: 1,
+  charCount: 1,
+  totalTokens: 1,
+  status: 1,
+} as const;

package/src/websitalk/knowledgeDocuments/knowledgeDocuments.getters.ts

@@ -1,78 +1,95 @@
-import { Collection, ObjectId as MongoObjectId } from "mongodb";
-import { getDb, ObjectId } from "../index";
-import {
-  KNOWLEDGE_DOCUMENTS_COLLECTION,
-  KNOWLEDGE_DOCUMENT_STATUSES,
-} from "./knowledgeDocuments.constants";
-import type {
-  CreateKnowledgeDocumentInput,
-  KnowledgeDocument,
-  KnowledgeDocumentDoc,
-  UpdateKnowledgeDocumentInput,
-} from "./knowledgeDocuments.types";
-
-export const getKnowledgeDocumentsCollection = (): Collection<KnowledgeDocument> => {
-  return getDb().collection<KnowledgeDocument>(KNOWLEDGE_DOCUMENTS_COLLECTION);
-};
-
-export const createKnowledgeDocument = async (
-  input: CreateKnowledgeDocumentInput,
-): Promise<MongoObjectId> => {
-  const now = new Date();
-  const { status, progressPercent, _id, ...rest } = input;
-  const { insertedId } = await getKnowledgeDocumentsCollection().insertOne({
-    ...(_id ? { _id } : {}),
-    ...rest,
-    status: status ?? KNOWLEDGE_DOCUMENT_STATUSES.UPLOADING,
-    progressPercent: progressPercent ?? 0,
-    createdAt: now,
-    updatedAt: now,
-  });
-  return insertedId;
-};
-
-export const getKnowledgeDocumentById = async (
-  id: string,
-): Promise<KnowledgeDocumentDoc | null> => {
-  return getKnowledgeDocumentsCollection().findOne({ _id: new ObjectId(id) });
-};
-
-export const listKnowledgeDocumentsByClientId = async (
-  clientId: string,
-): Promise<KnowledgeDocumentDoc[]> => {
-  return getKnowledgeDocumentsCollection()
-    .find({ clientId })
-    .sort({ createdAt: -1 })
-    .toArray();
-};
-
-export const listReadyKnowledgeDocumentIdsByClientId = async (
-  clientId: string,
-): Promise<string[]> => {
-  const docs = await getKnowledgeDocumentsCollection()
-    .find(
-      { clientId, status: KNOWLEDGE_DOCUMENT_STATUSES.READY },
-      { projection: { _id: 1 } },
-    )
-    .sort({ createdAt: -1 })
-    .toArray();
-  return docs.map((doc) => doc._id.toString());
-};
-
-export const updateKnowledgeDocument = async (
-  id: string,
-  updates: UpdateKnowledgeDocumentInput,
-): Promise<KnowledgeDocumentDoc | null> => {
-  return getKnowledgeDocumentsCollection().findOneAndUpdate(
-    { _id: new ObjectId(id) },
-    { $set: { ...updates, updatedAt: new Date() } },
-    { returnDocument: "after" },
-  );
-};
-
-export const deleteKnowledgeDocument = async (id: string): Promise<boolean> => {
-  const result = await getKnowledgeDocumentsCollection().deleteOne({
-    _id: new ObjectId(id),
-  });
-  return result.deletedCount === 1;
-};
+import { Collection, ObjectId as MongoObjectId } from "mongodb";
+import { getDb, ObjectId } from "../index";
+import {
+  KNOWLEDGE_DOCUMENTS_COLLECTION,
+  KNOWLEDGE_DOCUMENT_READY_PROJECTION,
+  KNOWLEDGE_DOCUMENT_STATUSES,
+} from "./knowledgeDocuments.constants";
+import type {
+  CreateKnowledgeDocumentInput,
+  KnowledgeDocument,
+  KnowledgeDocumentDoc,
+  KnowledgeDocumentReadyProjection,
+  UpdateKnowledgeDocumentInput,
+} from "./knowledgeDocuments.types";
+
+export const getKnowledgeDocumentsCollection =
+  (): Collection<KnowledgeDocument> => {
+    return getDb().collection<KnowledgeDocument>(
+      KNOWLEDGE_DOCUMENTS_COLLECTION,
+    );
+  };
+
+export const createKnowledgeDocument = async (
+  input: CreateKnowledgeDocumentInput,
+): Promise<MongoObjectId> => {
+  const now = new Date();
+  const { status, progressPercent, _id, ...rest } = input;
+  const { insertedId } = await getKnowledgeDocumentsCollection().insertOne({
+    ...(_id ? { _id } : {}),
+    ...rest,
+    status: status ?? KNOWLEDGE_DOCUMENT_STATUSES.UPLOADING,
+    progressPercent: progressPercent ?? 0,
+    createdAt: now,
+    updatedAt: now,
+  });
+  return insertedId;
+};
+
+export const getKnowledgeDocumentById = async (
+  id: string,
+): Promise<KnowledgeDocumentDoc | null> => {
+  return getKnowledgeDocumentsCollection().findOne({ _id: new ObjectId(id) });
+};
+
+export const listKnowledgeDocumentsByClientId = async (
+  clientId: string,
+): Promise<KnowledgeDocumentDoc[]> => {
+  return getKnowledgeDocumentsCollection()
+    .find({ clientId })
+    .sort({ createdAt: -1 })
+    .toArray();
+};
+
+export const listReadyKnowledgeDocumentIdsByClientId = async (
+  clientId: string,
+): Promise<string[]> => {
+  const docs = await getKnowledgeDocumentsCollection()
+    .find(
+      { clientId, status: KNOWLEDGE_DOCUMENT_STATUSES.READY },
+      { projection: { _id: 1 } },
+    )
+    .sort({ createdAt: -1 })
+    .toArray();
+  return docs.map((doc) => doc._id.toString());
+};
+
+export const listReadyKnowledgeDocumentsByClientId = async (
+  clientId: string,
+): Promise<KnowledgeDocumentReadyProjection[]> => {
+  return getKnowledgeDocumentsCollection()
+    .find(
+      { clientId, status: KNOWLEDGE_DOCUMENT_STATUSES.READY },
+      { projection: KNOWLEDGE_DOCUMENT_READY_PROJECTION },
+    )
+    .sort({ createdAt: -1 })
+    .toArray() as Promise<KnowledgeDocumentReadyProjection[]>;
+};
+
+export const updateKnowledgeDocument = async (
+  id: string,
+  updates: UpdateKnowledgeDocumentInput,
+): Promise<KnowledgeDocumentDoc | null> => {
+  return getKnowledgeDocumentsCollection().findOneAndUpdate(
+    { _id: new ObjectId(id) },
+    { $set: { ...updates, updatedAt: new Date() } },
+    { returnDocument: "after" },
+  );
+};
+
+export const deleteKnowledgeDocument = async (id: string): Promise<boolean> => {
+  const result = await getKnowledgeDocumentsCollection().deleteOne({
+    _id: new ObjectId(id),
+  });
+  return result.deletedCount === 1;
+};