@talkpilot/core-db 1.3.7 → 1.3.8

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

@@ -1,68 +1,68 @@
-import { mongodbClient } from "../mongodb-client";
-import { MongoClient } from "mongodb";
-
-describe("MongoDBClient Validation", () => {
-  const originalEnv = process.env;
-
-  beforeEach(() => {
-    process.env = { ...originalEnv };
-    delete process.env.MONGO_URI;
-    delete process.env.MONGODB_URI;
-    delete process.env.MONGODB_DB_NAME;
-
-    // Force reset singleton state
-    (mongodbClient as any).client = null;
-    (mongodbClient as any).db = null;
-  });
-
-  afterAll(async () => {
-    process.env = originalEnv;
-    // Ensure we are disconnected
-    await mongodbClient.disconnect();
-  });
-
-  it("should throw error if no URI is provided", async () => {
-    await expect(mongodbClient.connect()).rejects.toThrow(
-      "[core-db] Configuration Error: 'MONGO_URI' is missing",
-    );
-  });
-
-  it("should throw error if an invalid URI format is provided", async () => {
-    await expect(mongodbClient.connect("not-a-mongo-uri")).rejects.toThrow(
-      "[core-db] Connection Error: The provided MongoDB URI is invalid",
-    );
-  });
-
-  it("should throw error if URI is valid but database name is missing", async () => {
-    // We mock the actual MongoClient connection so it doesn't try to hit a real network
-    const connectSpy = jest
-      .spyOn(MongoClient.prototype, "connect")
-      .mockResolvedValue({} as any);
-
-    const uriWithoutDb = "mongodb://127.0.0.1:27017";
-    await expect(mongodbClient.connect(uriWithoutDb)).rejects.toThrow(
-      "[core-db] Database name not specified",
-    );
-
-    connectSpy.mockRestore();
-  });
-
-  it("should succeed when valid URI and DB name are provided", async () => {
-    const mockDb = { databaseName: "my-db" };
-    const connectSpy = jest
-      .spyOn(MongoClient.prototype, "connect")
-      .mockResolvedValue({} as any);
-    const dbSpy = jest
-      .spyOn(MongoClient.prototype, "db")
-      .mockReturnValue(mockDb as any);
-
-    const validUri = "mongodb://127.0.0.1:27017/my-db";
-    await mongodbClient.connect(validUri);
-
-    expect(mongodbClient.getDb()).toBeDefined();
-    expect(mongodbClient.getDb().databaseName).toBe("my-db");
-
-    connectSpy.mockRestore();
-    dbSpy.mockRestore();
-  });
-});
+import { mongodbClient } from "../mongodb-client";
+import { MongoClient } from "mongodb";
+
+describe("MongoDBClient Validation", () => {
+  const originalEnv = process.env;
+
+  beforeEach(() => {
+    process.env = { ...originalEnv };
+    delete process.env.MONGO_URI;
+    delete process.env.MONGODB_URI;
+    delete process.env.MONGODB_DB_NAME;
+
+    // Force reset singleton state
+    (mongodbClient as any).client = null;
+    (mongodbClient as any).db = null;
+  });
+
+  afterAll(async () => {
+    process.env = originalEnv;
+    // Ensure we are disconnected
+    await mongodbClient.disconnect();
+  });
+
+  it("should throw error if no URI is provided", async () => {
+    await expect(mongodbClient.connect()).rejects.toThrow(
+      "[core-db] Configuration Error: 'MONGO_URI' is missing",
+    );
+  });
+
+  it("should throw error if an invalid URI format is provided", async () => {
+    await expect(mongodbClient.connect("not-a-mongo-uri")).rejects.toThrow(
+      "[core-db] Connection Error: The provided MongoDB URI is invalid",
+    );
+  });
+
+  it("should throw error if URI is valid but database name is missing", async () => {
+    // We mock the actual MongoClient connection so it doesn't try to hit a real network
+    const connectSpy = jest
+      .spyOn(MongoClient.prototype, "connect")
+      .mockResolvedValue({} as any);
+
+    const uriWithoutDb = "mongodb://127.0.0.1:27017";
+    await expect(mongodbClient.connect(uriWithoutDb)).rejects.toThrow(
+      "[core-db] Database name not specified",
+    );
+
+    connectSpy.mockRestore();
+  });
+
+  it("should succeed when valid URI and DB name are provided", async () => {
+    const mockDb = { databaseName: "my-db" };
+    const connectSpy = jest
+      .spyOn(MongoClient.prototype, "connect")
+      .mockResolvedValue({} as any);
+    const dbSpy = jest
+      .spyOn(MongoClient.prototype, "db")
+      .mockReturnValue(mockDb as any);
+
+    const validUri = "mongodb://127.0.0.1:27017/my-db";
+    await mongodbClient.connect(validUri);
+
+    expect(mongodbClient.getDb()).toBeDefined();
+    expect(mongodbClient.getDb().databaseName).toBe("my-db");
+
+    connectSpy.mockRestore();
+    dbSpy.mockRestore();
+  });
+});

package/src/talkpilot/agents/__tests__/agents.getters.spec.ts

@@ -1,29 +1,29 @@
-import { findAgents } from "../agents.getters";
-import { setDb } from "../../index";
-import { Db, Collection } from "mongodb";
-import { createAgent } from "../../../test-utils/factories";
-
-describe("agents getters", () => {
-  let mockDb: Partial<Db>;
-  let mockCollection: Partial<Collection>;
-
-  beforeEach(() => {
-    const mockAgent = createAgent({ name: "Agent Smith" });
-    mockCollection = {
-      find: jest.fn().mockReturnValue({
-        toArray: jest.fn().mockResolvedValue([mockAgent]),
-      }),
-    };
-    mockDb = {
-      collection: jest.fn().mockReturnValue(mockCollection),
-    };
-    setDb(mockDb as Db);
-  });
-
-  it("findAgents should return agents", async () => {
-    const result = await findAgents();
-
-    expect(mockDb.collection).toHaveBeenCalledWith("agents");
-    expect(result[0].name).toBe("Agent Smith");
-  });
-});
+import { findAgents } from "../agents.getters";
+import { setDb } from "../../index";
+import { Db, Collection } from "mongodb";
+import { createAgent } from "../../../test-utils/factories";
+
+describe("agents getters", () => {
+  let mockDb: Partial<Db>;
+  let mockCollection: Partial<Collection>;
+
+  beforeEach(() => {
+    const mockAgent = createAgent({ name: "Agent Smith" });
+    mockCollection = {
+      find: jest.fn().mockReturnValue({
+        toArray: jest.fn().mockResolvedValue([mockAgent]),
+      }),
+    };
+    mockDb = {
+      collection: jest.fn().mockReturnValue(mockCollection),
+    };
+    setDb(mockDb as Db);
+  });
+
+  it("findAgents should return agents", async () => {
+    const result = await findAgents();
+
+    expect(mockDb.collection).toHaveBeenCalledWith("agents");
+    expect(result[0].name).toBe("Agent Smith");
+  });
+});

package/src/talkpilot/agents/agents.getters.ts

@@ -1,34 +1,34 @@
-import { getDb, Agent, ObjectId } from "../index";
-
-export const getAgentsCollection = () => {
-  return getDb().collection<Agent>("agents");
-};
-
-export const findAgents = async (filter: any = {}) => {
-  return getAgentsCollection().find(filter).toArray();
-};
-
-export const findAgentById = (agentId: string) => {
-  return getAgentsCollection().findOne({ _id: new ObjectId(agentId) });
-};
-
-export const findAgentsByClientId = (clientId: string) => {
-  return getAgentsCollection().find({ clientId }).toArray();
-};
-
-export const createAgent = async (agent: Agent) => {
-  const { insertedId } = await getAgentsCollection().insertOne(agent);
-  return findAgentById(String(insertedId));
-};
-
-export const updateAgent = (agentId: string, data: Partial<Agent>) => {
-  return getAgentsCollection().findOneAndUpdate(
-    { _id: new ObjectId(agentId) },
-    { $set: data },
-    { returnDocument: "after" },
-  );
-};
-
-export const removeAgent = (agentId: string) => {
-  return getAgentsCollection().deleteOne({ _id: new ObjectId(agentId) });
-};
+import { getDb, Agent, ObjectId } from "../index";
+
+export const getAgentsCollection = () => {
+  return getDb().collection<Agent>("agents");
+};
+
+export const findAgents = async (filter: any = {}) => {
+  return getAgentsCollection().find(filter).toArray();
+};
+
+export const findAgentById = (agentId: string) => {
+  return getAgentsCollection().findOne({ _id: new ObjectId(agentId) });
+};
+
+export const findAgentsByClientId = (clientId: string) => {
+  return getAgentsCollection().find({ clientId }).toArray();
+};
+
+export const createAgent = async (agent: Agent) => {
+  const { insertedId } = await getAgentsCollection().insertOne(agent);
+  return findAgentById(String(insertedId));
+};
+
+export const updateAgent = (agentId: string, data: Partial<Agent>) => {
+  return getAgentsCollection().findOneAndUpdate(
+    { _id: new ObjectId(agentId) },
+    { $set: data },
+    { returnDocument: "after" },
+  );
+};
+
+export const removeAgent = (agentId: string) => {
+  return getAgentsCollection().deleteOne({ _id: new ObjectId(agentId) });
+};

package/src/talkpilot/agents/agents.types.ts

@@ -1,14 +1,14 @@
-import { WithId } from "mongodb";
-
-export type Agent = {
-  name: string;
-  language: string;
-  voice: string;
-  instructions: string;
-  labels: string[];
-  clientId: string;
-  role: string;
-  gender: "male" | "female";
-};
-
-export type AgentDoc = WithId<Agent>;
+import { WithId } from "mongodb";
+
+export type Agent = {
+  name: string;
+  language: string;
+  voice: string;
+  instructions: string;
+  labels: string[];
+  clientId: string;
+  role: string;
+  gender: "male" | "female";
+};
+
+export type AgentDoc = WithId<Agent>;

package/src/talkpilot/agents/index.ts

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

package/src/talkpilot/backgroundToolResults/__tests__/backgroundToolResults.getters.spec.ts

@@ -1,147 +1,147 @@
-import {
-  getBackgroundToolResultsCollection,
-  upsertBackgroundToolResult,
-  findBackgroundToolResult,
-  findBackgroundToolResultsByCallSid,
-} from "../backgroundToolResults.getters";
-import { BackgroundToolStatus } from "../backgroundToolResults.types";
-
-describe("backgroundToolResults getters", () => {
-  it('should return the "backgroundToolResults" collection', () => {
-    const collection = getBackgroundToolResultsCollection();
-    expect(collection.collectionName).toBe("backgroundToolResults");
-  });
-
-  it("should create a new background tool result with correct fields and default TTL", async () => {
-    const callSid = "CA-background-call-1";
-    const toolName = "exampleTool";
-    const status: BackgroundToolStatus = "pending";
-    const result = null;
-
-    const before = new Date();
-    const doc = await upsertBackgroundToolResult({
-      callSid,
-      toolName,
-      status,
-      result,
-    });
-    const after = new Date();
-
-    expect(doc).not.toBeNull();
-    expect(doc?.callSid).toBe(callSid);
-    expect(doc?.toolName).toBe(toolName);
-    expect(doc?.status).toBe(status);
-    expect(doc?.result).toBe(result);
-    expect(doc?.createdAt).toBeInstanceOf(Date);
-    expect(doc?.updatedAt).toBeInstanceOf(Date);
-    expect(doc?.expiredAt).toBeInstanceOf(Date);
-
-    const createdAt = doc!.createdAt.getTime();
-    const expiredAt = doc!.expiredAt.getTime();
-    const ttlMs = expiredAt - createdAt;
-
-    expect(ttlMs).toBeGreaterThanOrEqual(60 * 60_000 - 1000);
-    expect(ttlMs).toBeLessThanOrEqual(60 * 60_000 + 1000);
-
-    expect(doc!.createdAt.getTime()).toBeGreaterThanOrEqual(before.getTime());
-    expect(doc!.updatedAt.getTime()).toBeLessThanOrEqual(after.getTime());
-  });
-
-  it("should upsert by callSid + toolName and keep createdAt while updating other fields", async () => {
-    const callSid = "CA-background-call-2";
-    const toolName = "aggregationTool";
-
-    const first = await upsertBackgroundToolResult({
-      callSid,
-      toolName,
-      status: "pending",
-      result: null,
-    });
-
-    expect(first).not.toBeNull();
-
-    const firstCreatedAt = first!.createdAt;
-    const firstUpdatedAt = first!.updatedAt;
-    const firstExpiredAt = first!.expiredAt;
-
-    const customTtlMs = 5 * 60_000;
-    const secondResultPayload = { value: 42 };
-
-    const second = await upsertBackgroundToolResult({
-      callSid,
-      toolName,
-      status: "success",
-      result: secondResultPayload,
-      ttlMs: customTtlMs,
-    });
-
-    expect(second).not.toBeNull();
-    expect(second!._id.toHexString()).toBe(first!._id.toHexString());
-    expect(second!.callSid).toBe(callSid);
-    expect(second!.toolName).toBe(toolName);
-    expect(second!.status).toBe("success");
-    expect(second!.result).toEqual(secondResultPayload);
-
-    expect(second!.createdAt.getTime()).toBe(firstCreatedAt.getTime());
-
-    expect(second!.updatedAt.getTime()).toBeGreaterThanOrEqual(
-      firstUpdatedAt.getTime(),
-    );
-    expect(second!.expiredAt.getTime()).not.toBe(firstExpiredAt.getTime());
-
-    const ttlMs = second!.expiredAt.getTime() - second!.updatedAt.getTime();
-    expect(ttlMs).toBeGreaterThanOrEqual(customTtlMs - 1000);
-    expect(ttlMs).toBeLessThanOrEqual(customTtlMs + 1000);
-  });
-
-  it("should find background tool result by callSid and toolName", async () => {
-    const callSid = "CA-background-call-3";
-    const toolName = "fetchUserData";
-
-    await upsertBackgroundToolResult({
-      callSid,
-      toolName,
-      status: "success",
-      result: { foo: "bar" },
-    });
-
-    const found = await findBackgroundToolResult(callSid, toolName);
-    expect(found).not.toBeNull();
-    expect(found?.callSid).toBe(callSid);
-    expect(found?.toolName).toBe(toolName);
-  });
-
-  it("should return null when background tool result does not exist", async () => {
-    const found = await findBackgroundToolResult(
-      "non-existing-callSid",
-      "nonExistingTool",
-    );
-    expect(found).toBeNull();
-  });
-
-  it("should return all background tool results for a callSid ordered by createdAt", async () => {
-    const callSid = "CA-background-call-4";
-
-    await upsertBackgroundToolResult({
-      callSid,
-      toolName: "toolA",
-      status: "pending",
-      result: null,
-    });
-
-    await new Promise((resolve) => setTimeout(resolve, 10));
-
-    await upsertBackgroundToolResult({
-      callSid,
-      toolName: "toolB",
-      status: "success",
-      result: { ok: true },
-    });
-
-    const results = await findBackgroundToolResultsByCallSid(callSid);
-    expect(results).toHaveLength(2);
-    expect(results[0].createdAt.getTime()).toBeLessThanOrEqual(
-      results[1].createdAt.getTime(),
-    );
-  });
-});
+import {
+  getBackgroundToolResultsCollection,
+  upsertBackgroundToolResult,
+  findBackgroundToolResult,
+  findBackgroundToolResultsByCallSid,
+} from "../backgroundToolResults.getters";
+import { BackgroundToolStatus } from "../backgroundToolResults.types";
+
+describe("backgroundToolResults getters", () => {
+  it('should return the "backgroundToolResults" collection', () => {
+    const collection = getBackgroundToolResultsCollection();
+    expect(collection.collectionName).toBe("backgroundToolResults");
+  });
+
+  it("should create a new background tool result with correct fields and default TTL", async () => {
+    const callSid = "CA-background-call-1";
+    const toolName = "exampleTool";
+    const status: BackgroundToolStatus = "pending";
+    const result = null;
+
+    const before = new Date();
+    const doc = await upsertBackgroundToolResult({
+      callSid,
+      toolName,
+      status,
+      result,
+    });
+    const after = new Date();
+
+    expect(doc).not.toBeNull();
+    expect(doc?.callSid).toBe(callSid);
+    expect(doc?.toolName).toBe(toolName);
+    expect(doc?.status).toBe(status);
+    expect(doc?.result).toBe(result);
+    expect(doc?.createdAt).toBeInstanceOf(Date);
+    expect(doc?.updatedAt).toBeInstanceOf(Date);
+    expect(doc?.expiredAt).toBeInstanceOf(Date);
+
+    const createdAt = doc!.createdAt.getTime();
+    const expiredAt = doc!.expiredAt.getTime();
+    const ttlMs = expiredAt - createdAt;
+
+    expect(ttlMs).toBeGreaterThanOrEqual(60 * 60_000 - 1000);
+    expect(ttlMs).toBeLessThanOrEqual(60 * 60_000 + 1000);
+
+    expect(doc!.createdAt.getTime()).toBeGreaterThanOrEqual(before.getTime());
+    expect(doc!.updatedAt.getTime()).toBeLessThanOrEqual(after.getTime());
+  });
+
+  it("should upsert by callSid + toolName and keep createdAt while updating other fields", async () => {
+    const callSid = "CA-background-call-2";
+    const toolName = "aggregationTool";
+
+    const first = await upsertBackgroundToolResult({
+      callSid,
+      toolName,
+      status: "pending",
+      result: null,
+    });
+
+    expect(first).not.toBeNull();
+
+    const firstCreatedAt = first!.createdAt;
+    const firstUpdatedAt = first!.updatedAt;
+    const firstExpiredAt = first!.expiredAt;
+
+    const customTtlMs = 5 * 60_000;
+    const secondResultPayload = { value: 42 };
+
+    const second = await upsertBackgroundToolResult({
+      callSid,
+      toolName,
+      status: "success",
+      result: secondResultPayload,
+      ttlMs: customTtlMs,
+    });
+
+    expect(second).not.toBeNull();
+    expect(second!._id.toHexString()).toBe(first!._id.toHexString());
+    expect(second!.callSid).toBe(callSid);
+    expect(second!.toolName).toBe(toolName);
+    expect(second!.status).toBe("success");
+    expect(second!.result).toEqual(secondResultPayload);
+
+    expect(second!.createdAt.getTime()).toBe(firstCreatedAt.getTime());
+
+    expect(second!.updatedAt.getTime()).toBeGreaterThanOrEqual(
+      firstUpdatedAt.getTime(),
+    );
+    expect(second!.expiredAt.getTime()).not.toBe(firstExpiredAt.getTime());
+
+    const ttlMs = second!.expiredAt.getTime() - second!.updatedAt.getTime();
+    expect(ttlMs).toBeGreaterThanOrEqual(customTtlMs - 1000);
+    expect(ttlMs).toBeLessThanOrEqual(customTtlMs + 1000);
+  });
+
+  it("should find background tool result by callSid and toolName", async () => {
+    const callSid = "CA-background-call-3";
+    const toolName = "fetchUserData";
+
+    await upsertBackgroundToolResult({
+      callSid,
+      toolName,
+      status: "success",
+      result: { foo: "bar" },
+    });
+
+    const found = await findBackgroundToolResult(callSid, toolName);
+    expect(found).not.toBeNull();
+    expect(found?.callSid).toBe(callSid);
+    expect(found?.toolName).toBe(toolName);
+  });
+
+  it("should return null when background tool result does not exist", async () => {
+    const found = await findBackgroundToolResult(
+      "non-existing-callSid",
+      "nonExistingTool",
+    );
+    expect(found).toBeNull();
+  });
+
+  it("should return all background tool results for a callSid ordered by createdAt", async () => {
+    const callSid = "CA-background-call-4";
+
+    await upsertBackgroundToolResult({
+      callSid,
+      toolName: "toolA",
+      status: "pending",
+      result: null,
+    });
+
+    await new Promise((resolve) => setTimeout(resolve, 10));
+
+    await upsertBackgroundToolResult({
+      callSid,
+      toolName: "toolB",
+      status: "success",
+      result: { ok: true },
+    });
+
+    const results = await findBackgroundToolResultsByCallSid(callSid);
+    expect(results).toHaveLength(2);
+    expect(results[0].createdAt.getTime()).toBeLessThanOrEqual(
+      results[1].createdAt.getTime(),
+    );
+  });
+});