@talkpilot/core-db 1.0.2 → 1.0.3

package/dist/talkpilot/clientAudioBuffers/clientAudioBuffer.getters.d.ts

@@ -3,7 +3,7 @@ import { Collection, InsertOneResult } from 'mongodb';
 declare const getClientAudioBuffersCollection: () => Collection<ClientAudioBufferType>;
 declare const insertClientAudioBuffer: (audioBufferInput: CreateAudioBufferInput) => Promise<InsertOneResult<ClientAudioBufferType>>;
 declare const findClientAudioBuffersByCallSidAndRange: (callSid: string, startMs?: number, endMs?: number) => Promise<ClientAudioBufferDoc[]>;
-declare const getAudioChunksByCallSid: (callSid: string) => Promise<Buffer[]>;
-declare const getMergedAudioByCallSid: (callSid: string) => Promise<Buffer>;
+declare const getAudioChunksByCallSid: (callSid: string, limit?: number) => Promise<Buffer[]>;
+declare const getMergedAudioByCallSid: (callSid: string, limit: number) => Promise<Buffer>;
 export { insertClientAudioBuffer, getClientAudioBuffersCollection, findClientAudioBuffersByCallSidAndRange, getAudioChunksByCallSid, getMergedAudioByCallSid, };
 //# sourceMappingURL=clientAudioBuffer.getters.d.ts.map

package/dist/talkpilot/clientAudioBuffers/clientAudioBuffer.getters.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"clientAudioBuffer.getters.d.ts","sourceRoot":"","sources":["../../../src/talkpilot/clientAudioBuffers/clientAudioBuffer.getters.ts"],"names":[],"mappings":"AACA,OAAO,EACL,oBAAoB,EACpB,qBAAqB,EACrB,sBAAsB,EACvB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAEtD,QAAA,MAAM,+BAA+B,QAAO,UAAU,CAAC,qBAAqB,CAE3E,CAAC;AAEF,QAAA,MAAM,uBAAuB,GAC3B,kBAAkB,sBAAsB,KACvC,OAAO,CAAC,eAAe,CAAC,qBAAqB,CAAC,CAsBhD,CAAC;AAEF,QAAA,MAAM,uCAAuC,GAC3C,SAAS,MAAM,EACf,UAAU,MAAM,EAChB,QAAQ,MAAM,KACb,OAAO,CAAC,oBAAoB,EAAE,CAehC,CAAC;AAEF,QAAA,MAAM,uBAAuB,GAAU,SAAS,MAAM,KAAG,OAAO,CAAC,MAAM,EAAE,CAuBxE,CAAC;AAEF,QAAA,MAAM,uBAAuB,GAAU,SAAS,MAAM,KAAG,OAAO,CAAC,MAAM,CAQtE,CAAC;AAEF,OAAO,EACL,uBAAuB,EACvB,+BAA+B,EAC/B,uCAAuC,EACvC,uBAAuB,EACvB,uBAAuB,GACxB,CAAC"}
+{"version":3,"file":"clientAudioBuffer.getters.d.ts","sourceRoot":"","sources":["../../../src/talkpilot/clientAudioBuffers/clientAudioBuffer.getters.ts"],"names":[],"mappings":"AACA,OAAO,EACL,oBAAoB,EACpB,qBAAqB,EACrB,sBAAsB,EACvB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAEtD,QAAA,MAAM,+BAA+B,QAAO,UAAU,CAAC,qBAAqB,CAE3E,CAAC;AAEF,QAAA,MAAM,uBAAuB,GAC3B,kBAAkB,sBAAsB,KACvC,OAAO,CAAC,eAAe,CAAC,qBAAqB,CAAC,CAsBhD,CAAC;AAEF,QAAA,MAAM,uCAAuC,GAC3C,SAAS,MAAM,EACf,UAAU,MAAM,EAChB,QAAQ,MAAM,KACb,OAAO,CAAC,oBAAoB,EAAE,CAehC,CAAC;AAEF,QAAA,MAAM,uBAAuB,GAAU,SAAS,MAAM,EAAE,QAAQ,MAAM,KAAG,OAAO,CAAC,MAAM,EAAE,CAqBxF,CAAC;AAEF,QAAA,MAAM,uBAAuB,GAAU,SAAS,MAAM,EAAE,OAAO,MAAM,KAAG,OAAO,CAAC,MAAM,CAQrF,CAAC;AAEF,OAAO,EACL,uBAAuB,EACvB,+BAA+B,EAC/B,uCAAuC,EACvC,uBAAuB,EACvB,uBAAuB,GACxB,CAAC"}

package/dist/talkpilot/clientAudioBuffers/clientAudioBuffer.getters.js

@@ -46,27 +46,31 @@ const findClientAudioBuffersByCallSidAndRange = async (callSid, startMs, endMs)
         .toArray();
 };
 exports.findClientAudioBuffersByCallSidAndRange = findClientAudioBuffersByCallSidAndRange;
-const getAudioChunksByCallSid = async (callSid) => {
-    const collection = getClientAudioBuffersCollection();
-    const chunks = await collection.find({ callSid }).toArray();
-    if (!chunks.length) {
+const getAudioChunksByCallSid = async (callSid, limit) => {
+    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 [];
     }
-    chunks.sort((a, b) => a.bucketStartMs - b.bucketStartMs);
-    return chunks.map((chunk) => {
+    const orderedDocs = limit ? docs.reverse() : docs;
+    return orderedDocs.map((chunk) => {
         const data = chunk.data;
-        if (Buffer.isBuffer(data)) {
+        if (Buffer.isBuffer(data))
             return data;
-        }
-        if (data?.buffer) {
+        if (data?.buffer)
             return Buffer.from(data.buffer);
-        }
         return Buffer.from(data);
     });
 };
 exports.getAudioChunksByCallSid = getAudioChunksByCallSid;
-const getMergedAudioByCallSid = async (callSid) => {
-    const buffers = await getAudioChunksByCallSid(callSid);
+const getMergedAudioByCallSid = async (callSid, limit) => {
+    const buffers = await getAudioChunksByCallSid(callSid, limit);
     if (!buffers.length) {
         return Buffer.alloc(0);
     }

package/dist/talkpilot/clientAudioBuffers/clientAudioBuffer.getters.js.map

@@ -1 +1 @@
-{"version":3,"file":"clientAudioBuffer.getters.js","sourceRoot":"","sources":["../../../src/talkpilot/clientAudioBuffers/clientAudioBuffer.getters.ts"],"names":[],"mappings":";;;AAAA,iCAAiC;AAQjC,MAAM,+BAA+B,GAAG,GAAsC,EAAE;IAC9E,OAAO,IAAA,UAAK,GAAE,CAAC,UAAU,CAAwB,oBAAoB,CAAC,CAAC;AACzE,CAAC,CAAC;AAsFA,0EAA+B;AApFjC,MAAM,uBAAuB,GAAG,KAAK,EACnC,gBAAwC,EACS,EAAE;IACnD,MAAM,UAAU,GAAG,+BAA+B,EAAE,CAAC;IACrD,MAAM,YAAY,GAAG,IAAI,IAAI,EAAE,CAAC;IAChC,MAAM,KAAK,GAAG,gBAAgB,CAAC,KAAK,IAAI,QAAQ,CAAC;IACjD,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,GAAG,KAAK,CAAC,CAAC;IAE3D,MAAM,sBAAsB,GAAG;QAC7B,OAAO,EAAE,gBAAgB,CAAC,OAAO;QACjC,QAAQ,EAAE,gBAAgB,CAAC,QAAQ;QACnC,aAAa,EAAE,gBAAgB,CAAC,aAAa;QAC7C,gBAAgB,EAAE,gBAAgB,CAAC,gBAAgB;QACnD,IAAI,EAAE,gBAAgB,CAAC,IAAI;KAC5B,CAAC;IACF,MAAM,8BAA8B,GAA0B;QAC5D,GAAG,sBAAsB;QACzB,KAAK,EAAE,OAAO;QACd,YAAY,EAAE,IAAI;QAClB,QAAQ,EAAE,CAAC;QACX,SAAS,EAAE,YAAY;QACvB,SAAS,EAAE,SAAS;KACrB,CAAC;IACF,OAAO,UAAU,CAAC,SAAS,CAAC,8BAA8B,CAAC,CAAC;AAC9D,CAAC,CAAC;AA2DA,0DAAuB;AAzDzB,MAAM,uCAAuC,GAAG,KAAK,EACnD,OAAe,EACf,OAAgB,EAChB,KAAc,EACmB,EAAE;IACnC,MAAM,UAAU,GAAG,+BAA+B,EAAE,CAAC;IACrD,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QAC1B,OAAO,UAAU,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;IAChD,CAAC;IACD,MAAM,UAAU,GAAG,KAAK,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;IACvC,OAAO,UAAU;SACd,IAAI,CAAC;QACJ,OAAO;QACP,aAAa,EAAE;YACb,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,UAAU;SACjB;KACF,CAAC;SACD,OAAO,EAAE,CAAC;AACf,CAAC,CAAC;AAwCA,0FAAuC;AAtCzC,MAAM,uBAAuB,GAAG,KAAK,EAAE,OAAe,EAAqB,EAAE;IAC3E,MAAM,UAAU,GAAG,+BAA+B,EAAE,CAAC;IACrD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;IAE5D,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QACnB,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,CAAC,IAAI,CAAC,CAAC,CAAM,EAAE,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,GAAG,CAAC,CAAC,aAAa,CAAC,CAAC;IAEnE,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,KAAU,EAAE,EAAE;QAC/B,MAAM,IAAI,GAAQ,KAAK,CAAC,IAAI,CAAC;QAE7B,IAAI,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YAC1B,OAAO,IAAI,CAAC;QACd,CAAC;QAED,IAAI,IAAI,EAAE,MAAM,EAAE,CAAC;YACjB,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAClC,CAAC;QAED,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC3B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AAgBA,0DAAuB;AAdzB,MAAM,uBAAuB,GAAG,KAAK,EAAE,OAAe,EAAmB,EAAE;IACzE,MAAM,OAAO,GAAG,MAAM,uBAAuB,CAAC,OAAO,CAAC,CAAC;IAEvD,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;QACpB,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACzB,CAAC;IAED,OAAO,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;AAChC,CAAC,CAAC;AAOA,0DAAuB"}
+{"version":3,"file":"clientAudioBuffer.getters.js","sourceRoot":"","sources":["../../../src/talkpilot/clientAudioBuffers/clientAudioBuffer.getters.ts"],"names":[],"mappings":";;;AAAA,iCAAiC;AAQjC,MAAM,+BAA+B,GAAG,GAAsC,EAAE;IAC9E,OAAO,IAAA,UAAK,GAAE,CAAC,UAAU,CAAwB,oBAAoB,CAAC,CAAC;AACzE,CAAC,CAAC;AAoFA,0EAA+B;AAlFjC,MAAM,uBAAuB,GAAG,KAAK,EACnC,gBAAwC,EACS,EAAE;IACnD,MAAM,UAAU,GAAG,+BAA+B,EAAE,CAAC;IACrD,MAAM,YAAY,GAAG,IAAI,IAAI,EAAE,CAAC;IAChC,MAAM,KAAK,GAAG,gBAAgB,CAAC,KAAK,IAAI,QAAQ,CAAC;IACjD,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,GAAG,KAAK,CAAC,CAAC;IAE3D,MAAM,sBAAsB,GAAG;QAC7B,OAAO,EAAE,gBAAgB,CAAC,OAAO;QACjC,QAAQ,EAAE,gBAAgB,CAAC,QAAQ;QACnC,aAAa,EAAE,gBAAgB,CAAC,aAAa;QAC7C,gBAAgB,EAAE,gBAAgB,CAAC,gBAAgB;QACnD,IAAI,EAAE,gBAAgB,CAAC,IAAI;KAC5B,CAAC;IACF,MAAM,8BAA8B,GAA0B;QAC5D,GAAG,sBAAsB;QACzB,KAAK,EAAE,OAAO;QACd,YAAY,EAAE,IAAI;QAClB,QAAQ,EAAE,CAAC;QACX,SAAS,EAAE,YAAY;QACvB,SAAS,EAAE,SAAS;KACrB,CAAC;IACF,OAAO,UAAU,CAAC,SAAS,CAAC,8BAA8B,CAAC,CAAC;AAC9D,CAAC,CAAC;AAyDA,0DAAuB;AAvDzB,MAAM,uCAAuC,GAAG,KAAK,EACnD,OAAe,EACf,OAAgB,EAChB,KAAc,EACmB,EAAE;IACnC,MAAM,UAAU,GAAG,+BAA+B,EAAE,CAAC;IACrD,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QAC1B,OAAO,UAAU,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;IAChD,CAAC;IACD,MAAM,UAAU,GAAG,KAAK,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;IACvC,OAAO,UAAU;SACd,IAAI,CAAC;QACJ,OAAO;QACP,aAAa,EAAE;YACb,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,UAAU;SACjB;KACF,CAAC;SACD,OAAO,EAAE,CAAC;AACf,CAAC,CAAC;AAsCA,0FAAuC;AApCzC,MAAM,uBAAuB,GAAG,KAAK,EAAE,OAAe,EAAE,KAAc,EAAqB,EAAE;IAC3F,MAAM,KAAK,GAAG,+BAA+B,EAAE,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;IAClE,IAAI,KAAK,EAAE,CAAC;QACV,KAAK,CAAC,IAAI,CAAC,EAAE,aAAa,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACjD,CAAC;SAAM,CAAC;QACN,KAAK,CAAC,IAAI,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC;IACnC,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;IAEnC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;QACjB,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,MAAM,WAAW,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;IAElD,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;QAC/B,MAAM,IAAI,GAAQ,KAAK,CAAC,IAAI,CAAC;QAC7B,IAAI,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC;YAAE,OAAO,IAAI,CAAC;QACvC,IAAI,IAAI,EAAE,MAAM;YAAE,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAClD,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC3B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AAgBA,0DAAuB;AAdzB,MAAM,uBAAuB,GAAG,KAAK,EAAE,OAAe,EAAE,KAAa,EAAmB,EAAE;IACxF,MAAM,OAAO,GAAG,MAAM,uBAAuB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;IAE9D,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;QACpB,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACzB,CAAC;IAED,OAAO,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;AAChC,CAAC,CAAC;AAOA,0DAAuB"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@talkpilot/core-db",
-  "version": "1.0.2",
+  "version": "1.0.3",
   "description": "Core database package for centralized connections and ORM integration.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

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

@@ -3,8 +3,9 @@ import {
   insertClientAudioBuffer,
   findClientAudioBuffersByCallSidAndRange,
   getMergedAudioByCallSid,
+  getAudioChunksByCallSid,
 } from '../clientAudioBuffer.getters';
-import { ObjectId } from 'mongodb';
+import { ObjectId } from '@core/db';
 import { createAudioBufferInput } from '../../../test-utils/factories';
 
 describe('clientAudioBuffers getters', () => {
@@ -59,20 +60,67 @@ describe('clientAudioBuffers getters', () => {
     const thirdBuffer = Buffer.from([4, 5]);
 
     await insertClientAudioBuffer(
-      createAudioBufferInput({ callSid, bucketStartMs: 1000, data: secondBuffer })
+        createAudioBufferInput({ callSid, bucketStartMs: 1000, data: secondBuffer })
     );
     await insertClientAudioBuffer(
-      createAudioBufferInput({ callSid, bucketStartMs: 500, data: firstBuffer })
+        createAudioBufferInput({ callSid, bucketStartMs: 500, data: firstBuffer })
     );
     await insertClientAudioBuffer(
-      createAudioBufferInput({ callSid, bucketStartMs: 3000, data: thirdBuffer })
+        createAudioBufferInput({ callSid, bucketStartMs: 3000, data: thirdBuffer })
     );
 
-    const merged = await getMergedAudioByCallSid(callSid);
+    const merged = await getMergedAudioByCallSid(callSid, 10);
     const expected = Buffer.concat([firstBuffer, secondBuffer, thirdBuffer]);
 
     expect(merged).toBeInstanceOf(Buffer);
     expect(merged.length).toBe(expected.length);
     expect(merged.equals(expected)).toBe(true);
   });
+
+  it('should return all audio chunks when less than or equal to limit buffers exist', async () => {
+    const callSid = 'CA1234567890abcdef1234567890abclt';
+
+    const limit = 30;
+    const buffers: Buffer[] = [];
+    for (let i = 0; i < limit; i++) {
+      const buf = Buffer.from([i]);
+      buffers.push(buf);
+      await insertClientAudioBuffer(
+          createAudioBufferInput({
+            callSid,
+            bucketStartMs: 1000 + i * 1000,
+            data: buf,
+          })
+      );
+    }
+
+    const chunks = await getAudioChunksByCallSid(callSid, limit);
+
+    expect(chunks).toHaveLength(buffers.length);
+    expect(chunks.every((chunk) => chunk instanceof Buffer)).toBe(true);
+    expect(Buffer.concat(chunks).equals(Buffer.concat(buffers))).toBe(true);
+  });
+
+  it('should return only the last N audio chunks when more than limit buffers exist', async () => {
+    const callSid = 'CA1234567890abcdef1234567890abclm';
+
+    const limit = 30;
+    const buffers: Buffer[] = [];
+    for (let i = 0; i < 35; i++) {
+      const buf = Buffer.from([i]);
+      buffers.push(buf);
+      await insertClientAudioBuffer(
+          createAudioBufferInput({
+            callSid,
+            bucketStartMs: 1000 + i * 1000,
+            data: buf,
+          })
+      );
+    }
+
+    const chunks = await getAudioChunksByCallSid(callSid, limit);
+    const expectedBuffers = buffers.slice(-limit);
+    expect(chunks).toHaveLength(expectedBuffers.length);
+    expect(Buffer.concat(chunks).equals(Buffer.concat(expectedBuffers))).toBe(true);
+  });
 });

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

@@ -57,33 +57,31 @@ const findClientAudioBuffersByCallSidAndRange = async (
     .toArray();
 };
 
-const getAudioChunksByCallSid = async (callSid: string): Promise<Buffer[]> => {
-  const collection = getClientAudioBuffersCollection();
-  const chunks = await collection.find({ callSid }).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 });
+  }
 
-  if (!chunks.length) {
+  const docs = await query.toArray();
+
+  if (!docs.length) {
     return [];
   }
+  const orderedDocs = limit ? docs.reverse() : docs;
 
-  chunks.sort((a: any, b: any) => a.bucketStartMs - b.bucketStartMs);
-
-  return chunks.map((chunk: any) => {
+  return orderedDocs.map((chunk) => {
     const data: any = chunk.data;
-
-    if (Buffer.isBuffer(data)) {
-      return data;
-    }
-
-    if (data?.buffer) {
-      return Buffer.from(data.buffer);
-    }
-
+    if (Buffer.isBuffer(data)) return data;
+    if (data?.buffer) return Buffer.from(data.buffer);
     return Buffer.from(data);
   });
 };
 
-const getMergedAudioByCallSid = async (callSid: string): Promise<Buffer> => {
-  const buffers = await getAudioChunksByCallSid(callSid);
+const getMergedAudioByCallSid = async (callSid: string, limit: number): Promise<Buffer> => {
+  const buffers = await getAudioChunksByCallSid(callSid, limit);
 
   if (!buffers.length) {
     return Buffer.alloc(0);