@talkpilot/core-db 1.2.0 → 1.2.1

package/src/talkpilot/clientAudioBuffers/clientAudioBuffer.getters.ts

@@ -1,117 +1,117 @@
-import { getDb } from "../index";
-import {
-  ClientAudioBufferDoc,
-  ClientAudioBufferType,
-  CreateAudioBufferInput,
-} from "./clientsAudioBuffers.types";
-import { Collection, InsertOneResult, DeleteResult } from "mongodb";
-
-const getClientAudioBuffersCollection =
-  (): Collection<ClientAudioBufferType> => {
-    return getDb().collection<ClientAudioBufferType>("clientAudioBuffers");
-  };
-
-/**
- * Deletes all audio buffers associated with a specific callSid.
- * Useful for clearing context between different tickets in the same call.
- */
-const deleteAudioBuffersByCallSid = async (
-  callSid: string,
-): Promise<DeleteResult> => {
-  return getClientAudioBuffersCollection().deleteMany({ callSid });
-};
-
-const insertClientAudioBuffer = async (
-  audioBufferInput: CreateAudioBufferInput,
-): Promise<InsertOneResult<ClientAudioBufferType>> => {
-  const collection = getClientAudioBuffersCollection();
-  const realtimeDate = new Date();
-  const ttlMs = audioBufferInput.ttlMs ?? 3600_000;
-  const expiresAt = new Date(realtimeDate.getTime() + ttlMs);
-
-  const inputFieldsForDocument = {
-    callSid: audioBufferInput.callSid,
-    clientId: audioBufferInput.clientId,
-    bucketStartMs: audioBufferInput.bucketStartMs,
-    bucketDurationMs: audioBufferInput.bucketDurationMs,
-    data: audioBufferInput.data,
-  };
-  const clientAudioBufferMongoDocument: ClientAudioBufferType = {
-    ...inputFieldsForDocument,
-    codec: "mulaw",
-    sampleRateHz: 8000,
-    channels: 1,
-    createdAt: realtimeDate,
-    expiresAt: expiresAt,
-  };
-  return collection.insertOne(clientAudioBufferMongoDocument);
-};
-
-const findClientAudioBuffersByCallSidAndRange = async (
-  callSid: string,
-  startMs?: number,
-  endMs?: number,
-): Promise<ClientAudioBufferDoc[]> => {
-  const collection = getClientAudioBuffersCollection();
-  if (startMs === undefined) {
-    return collection.find({ callSid }).toArray();
-  }
-  const endMsToUse = endMs ?? Date.now();
-  return collection
-    .find({
-      callSid,
-      bucketStartMs: {
-        $gte: startMs,
-        $lte: endMsToUse,
-      },
-    })
-    .toArray();
-};
-
-const getAudioChunksByCallSid = async (
-  callSid: string,
-  limit?: number,
-): Promise<Buffer[]> => {
-  const query = getClientAudioBuffersCollection().find({ callSid });
-  if (limit) {
-    query.sort({ bucketStartMs: -1 }).limit(limit);
-  } else {
-    query.sort({ bucketStartMs: 1 });
-  }
-
-  const docs = await query.toArray();
-
-  if (!docs.length) {
-    return [];
-  }
-  const orderedDocs = limit ? docs.reverse() : docs;
-
-  return orderedDocs.map((chunk) => {
-    const data: any = chunk.data;
-    if (Buffer.isBuffer(data)) return data;
-    if (data?.buffer) return Buffer.from(data.buffer);
-    return Buffer.from(data);
-  });
-};
-
-const getMergedAudioByCallSid = async (
-  callSid: string,
-  limit: number,
-): Promise<Buffer> => {
-  const buffers = await getAudioChunksByCallSid(callSid, limit);
-
-  if (!buffers.length) {
-    return Buffer.alloc(0);
-  }
-
-  return Buffer.concat(buffers);
-};
-
-export {
-  insertClientAudioBuffer,
-  getClientAudioBuffersCollection,
-  findClientAudioBuffersByCallSidAndRange,
-  getAudioChunksByCallSid,
-  getMergedAudioByCallSid,
-  deleteAudioBuffersByCallSid,
-};
+import { getDb } from "../index";
+import {
+  ClientAudioBufferDoc,
+  ClientAudioBufferType,
+  CreateAudioBufferInput,
+} from "./clientsAudioBuffers.types";
+import { Collection, InsertOneResult, DeleteResult } from "mongodb";
+
+const getClientAudioBuffersCollection =
+  (): Collection<ClientAudioBufferType> => {
+    return getDb().collection<ClientAudioBufferType>("clientAudioBuffers");
+  };
+
+/**
+ * Deletes all audio buffers associated with a specific callSid.
+ * Useful for clearing context between different tickets in the same call.
+ */
+const deleteAudioBuffersByCallSid = async (
+  callSid: string,
+): Promise<DeleteResult> => {
+  return getClientAudioBuffersCollection().deleteMany({ callSid });
+};
+
+const insertClientAudioBuffer = async (
+  audioBufferInput: CreateAudioBufferInput,
+): Promise<InsertOneResult<ClientAudioBufferType>> => {
+  const collection = getClientAudioBuffersCollection();
+  const realtimeDate = new Date();
+  const ttlMs = audioBufferInput.ttlMs ?? 3600_000;
+  const expiresAt = new Date(realtimeDate.getTime() + ttlMs);
+
+  const inputFieldsForDocument = {
+    callSid: audioBufferInput.callSid,
+    clientId: audioBufferInput.clientId,
+    bucketStartMs: audioBufferInput.bucketStartMs,
+    bucketDurationMs: audioBufferInput.bucketDurationMs,
+    data: audioBufferInput.data,
+  };
+  const clientAudioBufferMongoDocument: ClientAudioBufferType = {
+    ...inputFieldsForDocument,
+    codec: "mulaw",
+    sampleRateHz: 8000,
+    channels: 1,
+    createdAt: realtimeDate,
+    expiresAt: expiresAt,
+  };
+  return collection.insertOne(clientAudioBufferMongoDocument);
+};
+
+const findClientAudioBuffersByCallSidAndRange = async (
+  callSid: string,
+  startMs?: number,
+  endMs?: number,
+): Promise<ClientAudioBufferDoc[]> => {
+  const collection = getClientAudioBuffersCollection();
+  if (startMs === undefined) {
+    return collection.find({ callSid }).toArray();
+  }
+  const endMsToUse = endMs ?? Date.now();
+  return collection
+    .find({
+      callSid,
+      bucketStartMs: {
+        $gte: startMs,
+        $lte: endMsToUse,
+      },
+    })
+    .toArray();
+};
+
+const getAudioChunksByCallSid = async (
+  callSid: string,
+  limit?: number,
+): Promise<Buffer[]> => {
+  const query = getClientAudioBuffersCollection().find({ callSid });
+  if (limit) {
+    query.sort({ bucketStartMs: -1 }).limit(limit);
+  } else {
+    query.sort({ bucketStartMs: 1 });
+  }
+
+  const docs = await query.toArray();
+
+  if (!docs.length) {
+    return [];
+  }
+  const orderedDocs = limit ? docs.reverse() : docs;
+
+  return orderedDocs.map((chunk) => {
+    const data: any = chunk.data;
+    if (Buffer.isBuffer(data)) return data;
+    if (data?.buffer) return Buffer.from(data.buffer);
+    return Buffer.from(data);
+  });
+};
+
+const getMergedAudioByCallSid = async (
+  callSid: string,
+  limit: number,
+): Promise<Buffer> => {
+  const buffers = await getAudioChunksByCallSid(callSid, limit);
+
+  if (!buffers.length) {
+    return Buffer.alloc(0);
+  }
+
+  return Buffer.concat(buffers);
+};
+
+export {
+  insertClientAudioBuffer,
+  getClientAudioBuffersCollection,
+  findClientAudioBuffersByCallSidAndRange,
+  getAudioChunksByCallSid,
+  getMergedAudioByCallSid,
+  deleteAudioBuffersByCallSid,
+};

package/src/talkpilot/clientAudioBuffers/clientsAudioBuffers.types.ts

@@ -1,25 +1,25 @@
-import { WithId } from "mongodb";
-
-export type ClientAudioBufferType = {
-  callSid: string;
-  clientId: string;
-  bucketStartMs: number;
-  bucketDurationMs: number;
-  codec: "mulaw";
-  sampleRateHz: 8000;
-  channels: 1;
-  data: Buffer;
-  expiresAt: Date;
-  createdAt: Date;
-};
-
-export type CreateAudioBufferInput = {
-  callSid: string;
-  clientId: string;
-  bucketStartMs: number;
-  bucketDurationMs: number;
-  data: Buffer;
-  ttlMs?: number;
-};
-
-export type ClientAudioBufferDoc = WithId<ClientAudioBufferType>;
+import { WithId } from "mongodb";
+
+export type ClientAudioBufferType = {
+  callSid: string;
+  clientId: string;
+  bucketStartMs: number;
+  bucketDurationMs: number;
+  codec: "mulaw";
+  sampleRateHz: 8000;
+  channels: 1;
+  data: Buffer;
+  expiresAt: Date;
+  createdAt: Date;
+};
+
+export type CreateAudioBufferInput = {
+  callSid: string;
+  clientId: string;
+  bucketStartMs: number;
+  bucketDurationMs: number;
+  data: Buffer;
+  ttlMs?: number;
+};
+
+export type ClientAudioBufferDoc = WithId<ClientAudioBufferType>;

package/src/talkpilot/clientAudioBuffers/index.ts

@@ -1,2 +1,2 @@
-export * from "./clientAudioBuffer.getters";
-export * from "./clientsAudioBuffers.types";
+export * from "./clientAudioBuffer.getters";
+export * from "./clientsAudioBuffers.types";

package/src/talkpilot/clients/clients.getters.ts

@@ -1,16 +1,16 @@
-import { getDb, Client } from "../index";
-import { Collection } from "mongodb";
-
-export const getClientsCollection = (): Collection<Client> => {
-  return getDb().collection<Client>("clients");
-};
-
-export const getClientByAPIKey = async (
-  apiKey: string,
-): Promise<Client | null> => {
-  const client = await getClientsCollection().findOne({
-    apiKey,
-    isActive: true,
-  });
-  return client ? client : null;
-};
+import { getDb, Client } from "../index";
+import { Collection } from "mongodb";
+
+export const getClientsCollection = (): Collection<Client> => {
+  return getDb().collection<Client>("clients");
+};
+
+export const getClientByAPIKey = async (
+  apiKey: string,
+): Promise<Client | null> => {
+  const client = await getClientsCollection().findOne({
+    apiKey,
+    isActive: true,
+  });
+  return client ? client : null;
+};

package/src/talkpilot/clients/clients.types.ts

@@ -1,14 +1,14 @@
-import { ObjectId } from "mongodb";
-
-export type Client = {
-  _id?: ObjectId;
-  createdAt: Date;
-  updatedAt: Date;
-  clientId: string;
-  isActive?: boolean | null;
-  apiKey?: string | null;
-  credits: number;
-  managedBy?: string;
-};
-
-export type ClientDoc = Client & { _id: ObjectId };
+import { ObjectId } from "mongodb";
+
+export type Client = {
+  _id?: ObjectId;
+  createdAt: Date;
+  updatedAt: Date;
+  clientId: string;
+  isActive?: boolean | null;
+  apiKey?: string | null;
+  credits: number;
+  managedBy?: string;
+};
+
+export type ClientDoc = Client & { _id: ObjectId };

package/src/talkpilot/clients/index.ts

@@ -1,2 +1,2 @@
-export * from "./clients.getters";
-export * from "./clients.types";
+export * from "./clients.getters";
+export * from "./clients.types";

package/src/talkpilot/clientsConfig/__tests__/clientsConfig.spec.ts

@@ -1,106 +1,187 @@
-import {
-  createClientConfigDoc,
-  getClientConfig,
-} from "../clientsConfig.getters";
-import { createClientConfig } from "../../../test-utils/factories";
-
-describe("db.clientsConfig", () => {
-  describe("getClientConfig", () => {
-    it("should return client config by clientId", async () => {
-      const clientId = "test-client-id";
-      const clientConfig = createClientConfig({
-        clientId,
-        language: "Hebrew",
-        products: { feature1: true, feature2: false } as any,
-      });
-
-      await createClientConfigDoc(clientConfig);
-
-      const result = await getClientConfig(clientId);
-
-      expect(result).toMatchObject({
-        clientId,
-        language: "Hebrew",
-        products: { feature1: true, feature2: false },
-      });
-    });
-
-    it("should support websiteTalk product config", async () => {
-      const clientId = "websiteTalk-test";
-      const clientConfig = createClientConfig({
-        clientId,
-        products: {
-          websiteTalk: {
-            logoUrl: "https://cdn.example.com/branding/acme/logo.png",
-            companyName: "Acme Ltd",
-            timezone: "Asia/Jerusalem",
-            language: "he",
-            defaultBaseUrl: "https://acme.example",
-            scheduler: {
-              type: "cron",
-              schedule: "0 9 * * 1-5"
-           },
-          },
-        },
-      });
-
-      await createClientConfigDoc(clientConfig);
-
-      const result = await getClientConfig(clientId);
-
-      expect(result).toMatchObject({
-        clientId,
-        products: {
-          websiteTalk: {
-            logoUrl: "https://cdn.example.com/branding/acme/logo.png",
-            companyName: "Acme Ltd",
-            timezone: "Asia/Jerusalem",
-            language: "he",
-            defaultBaseUrl: "https://acme.example",
-            scheduler: {
-              type: "cron",
-            },
-          },
-        },
-      });
-    });
-
-    it("should return null for non-existent clientId", async () => {
-      const result = await getClientConfig("non-existent-client");
-      expect(result).toBeNull();
-    });
-
-    it("should support communications config", async () => {
-      const clientId = "communications-test";
-      const clientConfig = createClientConfig({
-        clientId,
-        communications: {
-          sendGrid: {
-            emailTo: ["test@example.com"],
-            templateId: "temp-123",
-            senderId: "sender-123",
-          },
-          sms: {},
-          whatsapp: {},
-        },
-      });
-
-      await createClientConfigDoc(clientConfig);
-
-      const result = await getClientConfig(clientId);
-
-      expect(result).toMatchObject({
-        clientId,
-        communications: {
-          sendGrid: {
-            emailTo: ["test@example.com"],
-            templateId: "temp-123",
-            senderId: "sender-123",
-          },
-          sms: {},
-          whatsapp: {},
-        },
-      });
-    });
-  });
-});
+import {
+  createClientConfigDoc,
+  getClientConfig,
+} from "../clientsConfig.getters";
+import { createClientConfig } from "../../../test-utils/factories";
+
+describe("db.clientsConfig", () => {
+  describe("getClientConfig", () => {
+    it("should return client config by clientId", async () => {
+      const clientId = "test-client-id";
+      const clientConfig = createClientConfig({
+        clientId,
+        language: "Hebrew",
+        products: { feature1: true, feature2: false } as any,
+      });
+
+      await createClientConfigDoc(clientConfig);
+
+      const result = await getClientConfig(clientId);
+
+      expect(result).toMatchObject({
+        clientId,
+        language: "Hebrew",
+        products: { feature1: true, feature2: false },
+      });
+    });
+
+    it("should support websiteTalk product config", async () => {
+      const clientId = "websiteTalk-test";
+      const clientConfig = createClientConfig({
+        clientId,
+        products: {
+          websiteTalk: {
+            logoUrl: "https://cdn.example.com/branding/acme/logo.png",
+            companyName: "Acme Ltd",
+            timezone: "Asia/Jerusalem",
+            language: "he",
+            defaultBaseUrl: "https://acme.example",
+            scheduler: {
+              type: "cron",
+              schedule: "0 9 * * 1-5"
+           },
+          },
+        },
+      });
+
+      await createClientConfigDoc(clientConfig);
+
+      const result = await getClientConfig(clientId);
+
+      expect(result).toMatchObject({
+        clientId,
+        products: {
+          websiteTalk: {
+            logoUrl: "https://cdn.example.com/branding/acme/logo.png",
+            companyName: "Acme Ltd",
+            timezone: "Asia/Jerusalem",
+            language: "he",
+            defaultBaseUrl: "https://acme.example",
+            scheduler: {
+              type: "cron",
+            },
+          },
+        },
+      });
+    });
+
+    it("should return null for non-existent clientId", async () => {
+      const result = await getClientConfig("non-existent-client");
+      expect(result).toBeNull();
+    });
+
+    it("given moked_106 with both toolsPrompts when saved and retrieved then both prompts are persisted", async () => {
+      // Given
+      const clientId = "tools-prompts-both";
+      const clientConfig = createClientConfig({
+        clientId,
+        products: {
+          municipal: {
+            cityName: "ashdod",
+            timezone: "Asia/Jerusalem",
+            moked_106: {
+              toolsPrompts: {
+                findSubjectsSystemPrompt: "Custom subjects prompt",
+                findStreetsSystemPrompt: "Custom streets prompt",
+              },
+            },
+          },
+        },
+      });
+
+      // When
+      await createClientConfigDoc(clientConfig);
+      const result = await getClientConfig(clientId);
+
+      // Then
+      expect(result?.products.municipal?.moked_106?.toolsPrompts).toMatchObject({
+        findSubjectsSystemPrompt: "Custom subjects prompt",
+        findStreetsSystemPrompt: "Custom streets prompt",
+      });
+    });
+
+    it("given moked_106 with findSubjectsSystemPrompt only when saved and retrieved then findStreetsSystemPrompt is undefined", async () => {
+      // Given
+      const clientId = "tools-prompts-subjects-only";
+      const clientConfig = createClientConfig({
+        clientId,
+        products: {
+          municipal: {
+            cityName: "ashdod",
+            timezone: "Asia/Jerusalem",
+            moked_106: {
+              toolsPrompts: {
+                findSubjectsSystemPrompt: "Subjects only prompt",
+              },
+            },
+          },
+        },
+      });
+
+      // When
+      await createClientConfigDoc(clientConfig);
+      const result = await getClientConfig(clientId);
+
+      // Then
+      expect(result?.products.municipal?.moked_106?.toolsPrompts?.findSubjectsSystemPrompt).toBe("Subjects only prompt");
+      expect(result?.products.municipal?.moked_106?.toolsPrompts?.findStreetsSystemPrompt).toBeUndefined();
+    });
+
+    it("given moked_106 without toolsPrompts when saved and retrieved then toolsPrompts is undefined", async () => {
+      // Given
+      const clientId = "tools-prompts-absent";
+      const clientConfig = createClientConfig({
+        clientId,
+        products: {
+          municipal: {
+            cityName: "ashdod",
+            timezone: "Asia/Jerusalem",
+            moked_106: {
+              avoidSaveTickets: true,
+            },
+          },
+        },
+      });
+
+      // When
+      await createClientConfigDoc(clientConfig);
+      const result = await getClientConfig(clientId);
+
+      // Then
+      expect(result?.products.municipal?.moked_106?.toolsPrompts).toBeUndefined();
+    });
+
+    it("should support communications config", async () => {
+      const clientId = "communications-test";
+      const clientConfig = createClientConfig({
+        clientId,
+        communications: {
+          sendGrid: {
+            emailTo: ["test@example.com"],
+            templateId: "temp-123",
+            senderId: "sender-123",
+          },
+          sms: {},
+          whatsapp: {},
+        },
+      });
+
+      await createClientConfigDoc(clientConfig);
+
+      const result = await getClientConfig(clientId);
+
+      expect(result).toMatchObject({
+        clientId,
+        communications: {
+          sendGrid: {
+            emailTo: ["test@example.com"],
+            templateId: "temp-123",
+            senderId: "sender-123",
+          },
+          sms: {},
+          whatsapp: {},
+        },
+      });
+    });
+  });
+});