@talkpilot/core-db 1.3.54 → 1.3.55

package/src/websitalk/knowledgeDocuments/knowledgeDocuments.types.ts

@@ -1,62 +1,75 @@
-import { WithId, ObjectId as MongoObjectId } from "mongodb";
-
-import type {
-  KNOWLEDGE_CONTENT_TYPES,
-  KNOWLEDGE_DOCUMENT_STATUSES,
-  KNOWLEDGE_FILE_TYPES,
-} from "./knowledgeDocuments.constants";
-
-export type KnowledgeDocumentStatus =
-  (typeof KNOWLEDGE_DOCUMENT_STATUSES)[keyof typeof KNOWLEDGE_DOCUMENT_STATUSES];
-
-export type KnowledgeFileType =
-  (typeof KNOWLEDGE_FILE_TYPES)[keyof typeof KNOWLEDGE_FILE_TYPES];
-
-export type KnowledgeContentType =
-  (typeof KNOWLEDGE_CONTENT_TYPES)[keyof typeof KNOWLEDGE_CONTENT_TYPES];
-
-export type KnowledgeDocument = {
-  clientId: string;
-  filename: string;
-  contentType: KnowledgeContentType;
-  fileType: KnowledgeFileType;
-  storagePath: string;
-  markdownStoragePath?: string;
-  byteSize: number;
-  pageCount?: number;
-  pagesExtracted?: number;
-  charCount?: number;
-  contentHash?: string;
-  status: KnowledgeDocumentStatus;
-  errorMessage?: string;
-  progressPercent: number;
-  createdAt: Date;
-  updatedAt: Date;
-};
-
-export type KnowledgeDocumentDoc = WithId<KnowledgeDocument>;
-
-export type CreateKnowledgeDocumentInput = Omit<
-  KnowledgeDocument,
-  "createdAt" | "updatedAt" | "progressPercent" | "status"
-> & {
-  status?: KnowledgeDocumentStatus;
-  progressPercent?: number;
-  _id?: MongoObjectId;
-};
-
-export type UpdateKnowledgeDocumentInput = Partial<
-  Pick<
-    KnowledgeDocument,
-    | "storagePath"
-    | "markdownStoragePath"
-    | "byteSize"
-    | "pageCount"
-    | "pagesExtracted"
-    | "charCount"
-    | "contentHash"
-    | "status"
-    | "errorMessage"
-    | "progressPercent"
-  >
->;
+import { WithId, ObjectId as MongoObjectId } from "mongodb";
+
+import type {
+  KNOWLEDGE_CONTENT_TYPES,
+  KNOWLEDGE_DOCUMENT_STATUSES,
+  KNOWLEDGE_FILE_TYPES,
+} from "./knowledgeDocuments.constants";
+
+export type KnowledgeDocumentStatus =
+  (typeof KNOWLEDGE_DOCUMENT_STATUSES)[keyof typeof KNOWLEDGE_DOCUMENT_STATUSES];
+
+export type KnowledgeFileType =
+  (typeof KNOWLEDGE_FILE_TYPES)[keyof typeof KNOWLEDGE_FILE_TYPES];
+
+export type KnowledgeContentType =
+  (typeof KNOWLEDGE_CONTENT_TYPES)[keyof typeof KNOWLEDGE_CONTENT_TYPES];
+
+export type KnowledgeDocument = {
+  clientId: string;
+  filename: string;
+  contentType: KnowledgeContentType;
+  fileType: KnowledgeFileType;
+  storagePath: string;
+  markdownStoragePath?: string;
+  byteSize: number;
+  pageCount?: number;
+  pagesExtracted?: number;
+  charCount?: number;
+  totalTokens?: number;
+  contentHash?: string;
+  status: KnowledgeDocumentStatus;
+  errorMessage?: string;
+  progressPercent: number;
+  createdAt: Date;
+  updatedAt: Date;
+};
+
+export type KnowledgeDocumentDoc = WithId<KnowledgeDocument>;
+
+export type CreateKnowledgeDocumentInput = Omit<
+  KnowledgeDocument,
+  "createdAt" | "updatedAt" | "progressPercent" | "status"
+> & {
+  status?: KnowledgeDocumentStatus;
+  progressPercent?: number;
+  _id?: MongoObjectId;
+};
+
+export type UpdateKnowledgeDocumentInput = Partial<
+  Pick<
+    KnowledgeDocument,
+    | "storagePath"
+    | "markdownStoragePath"
+    | "byteSize"
+    | "pageCount"
+    | "pagesExtracted"
+    | "charCount"
+    | "totalTokens"
+    | "contentHash"
+    | "status"
+    | "errorMessage"
+    | "progressPercent"
+  >
+>;
+
+export type KnowledgeDocumentReadyProjection = Pick<
+  KnowledgeDocumentDoc,
+  | "_id"
+  | "clientId"
+  | "filename"
+  | "markdownStoragePath"
+  | "charCount"
+  | "totalTokens"
+  | "status"
+>;

package/src/websitalk/mongodb-client.ts

@@ -1,61 +1,61 @@
-import { MongoClient, Db } from "mongodb";
-import { setDb } from "./index";
-import { validateConfig, validateMongoUri } from "../utils/validation";
-
-class WebsitalkMongoDBClient {
-  private client: MongoClient | null = null;
-  private db: Db | null = null;
-  private readonly defaultDbName = "website-talk";
-
-  async connect(uri?: string, dbName?: string): Promise<void> {
-    if (this.client) {
-      return;
-    }
-
-    const mongodbUri = uri || process.env.MONGO_URI || process.env.MONGODB_URI;
-    validateConfig("MONGO_URI", mongodbUri);
-    validateMongoUri(mongodbUri!);
-
-    try {
-      this.client = new MongoClient(mongodbUri!);
-      await this.client.connect();
-      const targetDbName =
-        dbName || process.env.WEBSITALK_DB_NAME || this.defaultDbName;
-      this.db = this.client.db(targetDbName);
-      setDb(this.db);
-      console.info(`[core-db] Website Talk MongoDB connected: ${targetDbName}`);
-    } catch (error) {
-      console.error("[core-db] Website Talk connection failed", error);
-      throw error;
-    }
-  }
-
-  async disconnect(): Promise<void> {
-    if (this.client) {
-      try {
-        await this.client.close();
-        this.client = null;
-        this.db = null;
-        console.info("Website Talk MongoDB disconnected successfully");
-      } catch (error) {
-        console.error("[core-db] Website Talk disconnection failed", error);
-        throw error;
-      }
-    }
-  }
-
-  getDb(): Db {
-    if (!this.db) {
-      throw new Error(
-        "Website Talk database not initialized. Call connect() first.",
-      );
-    }
-    return this.db;
-  }
-
-  isConnected(): boolean {
-    return this.client !== null && this.client !== undefined;
-  }
-}
-
-export const websitalkMongodbClient = new WebsitalkMongoDBClient();
+import { MongoClient, Db } from "mongodb";
+import { setDb } from "./index";
+import { validateConfig, validateMongoUri } from "../utils/validation";
+
+class WebsitalkMongoDBClient {
+  private client: MongoClient | null = null;
+  private db: Db | null = null;
+  private readonly defaultDbName = "website-talk";
+
+  async connect(uri?: string, dbName?: string): Promise<void> {
+    if (this.client) {
+      return;
+    }
+
+    const mongodbUri = uri || process.env.MONGO_URI || process.env.MONGODB_URI;
+    validateConfig("MONGO_URI", mongodbUri);
+    validateMongoUri(mongodbUri!);
+
+    try {
+      this.client = new MongoClient(mongodbUri!);
+      await this.client.connect();
+      const targetDbName =
+        dbName || process.env.WEBSITALK_DB_NAME || this.defaultDbName;
+      this.db = this.client.db(targetDbName);
+      setDb(this.db);
+      console.info(`[core-db] Website Talk MongoDB connected: ${targetDbName}`);
+    } catch (error) {
+      console.error("[core-db] Website Talk connection failed", error);
+      throw error;
+    }
+  }
+
+  async disconnect(): Promise<void> {
+    if (this.client) {
+      try {
+        await this.client.close();
+        this.client = null;
+        this.db = null;
+        console.info("Website Talk MongoDB disconnected successfully");
+      } catch (error) {
+        console.error("[core-db] Website Talk disconnection failed", error);
+        throw error;
+      }
+    }
+  }
+
+  getDb(): Db {
+    if (!this.db) {
+      throw new Error(
+        "Website Talk database not initialized. Call connect() first.",
+      );
+    }
+    return this.db;
+  }
+
+  isConnected(): boolean {
+    return this.client !== null && this.client !== undefined;
+  }
+}
+
+export const websitalkMongodbClient = new WebsitalkMongoDBClient();