@micdrop/server 1.0.1 → 1.0.3

package/README.md

@@ -41,6 +41,7 @@ const config: CallConfig = {
   systemPrompt: 'You are a helpful assistant',
 
   // Optional first message from assistant
+  // Omit to generate the first message
   firstMessage: 'Hello!',
 
   // Function to generate assistant responses
@@ -109,6 +110,12 @@ The main class for managing WebSocket connections and audio streaming.
 ```typescript
 class CallSocket {
   constructor(socket: WebSocket, config: CallConfig)
+
+  // Add assistant message and send to client with audio (TTS)
+  answer(message: string): Promise<void>
+
+  // Reset conversation (including system prompt)
+  resetConversation(conversation: Conversation): void
 }
 ```
 
@@ -124,6 +131,15 @@ interface CallConfig {
   // Optional first message from assistant
   firstMessage?: string
 
+  // Enable debug logging with timestamps
+  debugLog?: boolean
+
+  // Save last speech audio file for debugging (speech.ogg)
+  debugSaveSpeech?: boolean
+
+  // Disable text-to-speech conversion
+  disableTTS?: boolean
+
   // Generate assistant's response
   generateAnswer(conversation: Conversation): Promise<string>
 
@@ -188,6 +204,24 @@ enum CallServerCommands {
 4. Server processes audio and responds with text/audio
 5. Process continues until interview ends
 
+## Ending the call
+
+The call has two ways to end:
+
+- When the client closes the websocket connection.
+- When the generated answer contains the keyword "END_INTERVIEW".
+
+You can prompt it like this:
+
+```typescript
+import { END_INTERVIEW } from '@micdrop/server'
+
+const systemPrompt = `
+You are a voice assistant interviewing the user.
+To end the interview, briefly thank the user and say good bye, then say "${END_INTERVIEW}".
+`
+```
+
 ## Error Handling
 
 The server implements standardized error handling with specific codes:
@@ -254,4 +288,6 @@ MIT
 
 ## Author
 
-[Lonestone](https://www.lonestone.io) ([GitHub](https://github.com/lonestone))
+Originally developed for [Raconte.ai](https://www.raconte.ai)
+
+by [Lonestone](https://www.lonestone.io) ([GitHub](https://github.com/lonestone))

package/dist/index.d.mts

@@ -14,6 +14,9 @@ declare enum CallServerCommands {
 interface CallConfig {
     systemPrompt: string;
     firstMessage?: string;
+    debugLog?: boolean;
+    debugSaveSpeech?: boolean;
+    disableTTS?: boolean;
     generateAnswer(conversation: Conversation): Promise<string>;
     speech2Text(blob: Blob, prompt?: string): Promise<string>;
     text2Speech(text: string): Promise<ArrayBuffer>;
@@ -41,12 +44,13 @@ declare class CallSocket {
     private chunks;
     private conversation;
     constructor(socket: WebSocket, config: CallConfig);
+    resetConversation(conversation: Conversation): void;
     private addMessage;
     private onClose;
     private onMessage;
     private onStopSpeaking;
-    private answer;
-    private debug;
+    answer(message: string): Promise<void>;
+    private log;
 }
 
 declare enum CallErrorCode {

package/dist/index.d.ts

@@ -14,6 +14,9 @@ declare enum CallServerCommands {
 interface CallConfig {
     systemPrompt: string;
     firstMessage?: string;
+    debugLog?: boolean;
+    debugSaveSpeech?: boolean;
+    disableTTS?: boolean;
     generateAnswer(conversation: Conversation): Promise<string>;
     speech2Text(blob: Blob, prompt?: string): Promise<string>;
     text2Speech(text: string): Promise<ArrayBuffer>;
@@ -41,12 +44,13 @@ declare class CallSocket {
     private chunks;
     private conversation;
     constructor(socket: WebSocket, config: CallConfig);
+    resetConversation(conversation: Conversation): void;
     private addMessage;
     private onClose;
     private onMessage;
     private onStopSpeaking;
-    private answer;
-    private debug;
+    answer(message: string): Promise<void>;
+    private log;
 }
 
 declare enum CallErrorCode {

package/dist/index.js

@@ -61,9 +61,6 @@ var CallServerCommands = /* @__PURE__ */ ((CallServerCommands2) => {
 })(CallServerCommands || {});
 
 // src/CallSocket.ts
-var debugEnabled = false;
-var debugSaveFile = false;
-var disableTTS = false;
 var END_INTERVIEW = "END_INTERVIEW";
 var CallSocket = class {
   constructor(socket, config) {
@@ -78,7 +75,7 @@ var CallSocket = class {
     // Chunks of user speech since user started speaking
     this.chunks = [];
     this.conversation = [{ role: "system", content: config.systemPrompt }];
-    this.debug(`Call started`);
+    this.log(`Call started`);
     if (config.firstMessage) {
       this.answer(config.firstMessage);
     } else {
@@ -90,6 +87,11 @@ var CallSocket = class {
     this.socket.on("close", this.onClose.bind(this));
     this.socket.on("message", this.onMessage.bind(this));
   }
+  // Reset conversation
+  resetConversation(conversation) {
+    this.log("Reset conversation");
+    this.conversation = conversation;
+  }
   addMessage(message) {
     this.conversation.push(message);
     this.socket.send(
@@ -98,7 +100,7 @@ var CallSocket = class {
     this.config.onMessage?.(message);
   }
   onClose() {
-    this.debug("Connection closed");
+    this.log("Connection closed");
     this.abortAnswer = true;
     const duration = Math.round((Date.now() - this.startTime) / 1e3);
     this.config.onEnd?.({
@@ -116,7 +118,7 @@ var CallSocket = class {
     }
     if (message.byteLength < 15) {
       const cmd = message.toString();
-      this.debug(`Command: ${cmd}`);
+      this.log(`Command: ${cmd}`);
       if (cmd === "startSpeaking" /* StartSpeaking */) {
         this.isSpeaking = true;
         this.abortAnswer = true;
@@ -127,7 +129,7 @@ var CallSocket = class {
         await this.onStopSpeaking();
       }
     } else if (Buffer.isBuffer(message) && this.isSpeaking) {
-      this.debug(`Received chunk (${message.byteLength} bytes)`);
+      this.log(`Received chunk (${message.byteLength} bytes)`);
       this.chunks.push(message);
     }
   }
@@ -137,7 +139,7 @@ var CallSocket = class {
     const blob = new Blob(this.chunks, { type: "audio/ogg" });
     this.chunks.length = 0;
     try {
-      if (debugSaveFile) {
+      if (this.config.debugSaveSpeech) {
         const filePath = path.join(__dirname, "speech.ogg");
         fs.writeFileSync(filePath, Buffer.from(await blob.arrayBuffer()));
         return;
@@ -147,18 +149,18 @@ var CallSocket = class {
         this.conversation[this.conversation.length - 1]?.content
       );
       if (!transcript) {
-        this.debug("Ignoring empty transcript");
+        this.log("Ignoring empty transcript");
         return;
       }
-      this.debug("User transcript:", transcript);
+      this.log("User transcript:", transcript);
       this.addMessage({ role: "user", content: transcript });
       if (this.abortAnswer) {
-        this.debug("Answer aborted, no answer generated");
+        this.log("Answer aborted, no answer generated");
         return;
       }
       const answer = await this.config.generateAnswer(this.conversation);
       if (this.abortAnswer) {
-        this.debug("Answer aborted, ignoring answer");
+        this.log("Answer aborted, ignoring answer");
         return;
       }
       await this.answer(answer);
@@ -167,6 +169,7 @@ var CallSocket = class {
       this.socket.close();
     }
   }
+  // Add assistant message and send to client with audio (TTS)
   async answer(message) {
     let isEnd = false;
     if (message.includes(END_INTERVIEW)) {
@@ -174,12 +177,12 @@ var CallSocket = class {
       isEnd = true;
     }
     if (message.length) {
-      this.debug("Assistant message:", message);
+      this.log("Assistant message:", message);
       this.addMessage({ role: "assistant", content: message });
-      if (!disableTTS) {
+      if (!this.config.disableTTS) {
         const audio = await this.config.text2Speech(message);
         if (this.abortAnswer) {
-          this.debug("Answer aborted, removing last assistant message");
+          this.log("Answer aborted, removing last assistant message");
           const lastMessage = this.conversation[this.conversation.length - 1];
           if (lastMessage?.role === "assistant") {
             this.conversation.pop();
@@ -187,23 +190,21 @@ var CallSocket = class {
           }
           return;
         }
-        this.debug(`Send audio: (${audio.byteLength} bytes)`);
+        this.log(`Send audio: (${audio.byteLength} bytes)`);
         this.socket.send(audio);
       }
     }
     if (isEnd) {
-      this.debug("Interview ended");
+      this.log("Interview ended");
       this.socket.send("endInterview" /* EndInterview */);
     }
   }
-  debug(...message) {
-    if (!debugEnabled) return;
-    const nowTime = Date.now();
-    console.log(
-      `[WS] [${nowTime - this.startTime} | ${nowTime - this.lastDebug}ms]`,
-      ...message
-    );
-    this.lastDebug = nowTime;
+  log(...message) {
+    if (!this.config.debugLog) return;
+    const now = Date.now();
+    const delta = now - this.lastDebug;
+    this.lastDebug = now;
+    console.log(`[Debug +${delta}ms]`, ...message);
   }
 };
 

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/index.ts","../src/CallSocket.ts","../src/types.ts","../src/errors.ts","../src/waitForParams.ts"],"sourcesContent":["export * from './CallSocket'\nexport * from './errors'\nexport * from './types'\nexport * from './waitForParams'\n","import * as fs from 'fs'\nimport * as path from 'path'\nimport { WebSocket } from 'ws'\nimport {\n  CallClientCommands,\n  CallConfig,\n  CallServerCommands,\n  Conversation,\n  ConversationMessage,\n} from './types'\n\nconst debugEnabled = false\nconst debugSaveFile = false\nconst disableTTS = false\n\nexport const END_INTERVIEW = 'END_INTERVIEW'\n\nexport class CallSocket {\n  private startTime = Date.now()\n  private lastDebug = Date.now()\n\n  // An answer can be aborted if user is speaking\n  private abortAnswer = false\n\n  // When user is speaking, we're waiting to chunks or to stop\n  private isSpeaking = false\n\n  // Chunks of user speech since user started speaking\n  private chunks: Buffer[] = []\n\n  // Conversation history\n  private conversation: Conversation\n\n  constructor(\n    public socket: WebSocket,\n    public config: CallConfig\n  ) {\n    this.conversation = [{ role: 'system', content: config.systemPrompt }]\n    this.debug(`Call started`)\n\n    // Assistant speaks first\n\n    // LLM: Generate answer\n    if (config.firstMessage) {\n      this.answer(config.firstMessage)\n    } else {\n      this.config\n        .generateAnswer(this.conversation)\n        .then((answer) => this.answer(answer))\n        .catch((error) => {\n          console.error('[WS]', error)\n          this.socket.close()\n        })\n    }\n\n    // Listen to events\n    this.socket.on('close', this.onClose.bind(this))\n    this.socket.on('message', this.onMessage.bind(this))\n  }\n\n  private addMessage(message: ConversationMessage) {\n    this.conversation.push(message)\n    this.socket.send(\n      `${\n        message.role === 'user'\n          ? CallServerCommands.UserMessage\n          : CallServerCommands.AssistantMessage\n      } ${message.content}`\n    )\n    this.config.onMessage?.(message)\n  }\n\n  private onClose() {\n    this.debug('Connection closed')\n    this.abortAnswer = true\n    const duration = Math.round((Date.now() - this.startTime) / 1000)\n\n    // End call callback\n    this.config.onEnd?.({\n      conversation: this.conversation.slice(1), // Remove system message\n      duration,\n    })\n\n    // Unset params\n    // @ts-ignore\n    this.socket = null\n    // @ts-ignore\n    this.config = null\n  }\n\n  private async onMessage(message: Buffer) {\n    if (!Buffer.isBuffer(message)) {\n      console.warn(`[WS] Message is not a buffer`)\n      return\n    }\n\n    // Commands\n    if (message.byteLength < 15) {\n      const cmd = message.toString()\n      this.debug(`Command: ${cmd}`)\n\n      if (cmd === CallClientCommands.StartSpeaking) {\n        // User started speaking\n        this.isSpeaking = true\n        // Abort answer if there is generation in progress\n        this.abortAnswer = true\n      } else if (cmd === CallClientCommands.Mute) {\n        // User muted the call\n        // Abort answer if there is generation in progress\n        this.abortAnswer = true\n      } else if (cmd === CallClientCommands.StopSpeaking) {\n        // User stopped speaking\n        this.isSpeaking = false\n        await this.onStopSpeaking()\n      }\n    }\n\n    // Audio chunk\n    else if (Buffer.isBuffer(message) && this.isSpeaking) {\n      this.debug(`Received chunk (${message.byteLength} bytes)`)\n      this.chunks.push(message)\n    }\n  }\n\n  private async onStopSpeaking() {\n    // Do nothing if there is no chunk\n    if (this.chunks.length === 0) return\n\n    this.abortAnswer = false\n\n    // Combine audio blob\n    const blob = new Blob(this.chunks, { type: 'audio/ogg' })\n\n    // Reset chunks for next user speech\n    this.chunks.length = 0\n\n    try {\n      // Save file to disk\n      if (debugSaveFile) {\n        const filePath = path.join(__dirname, 'speech.ogg')\n        fs.writeFileSync(filePath, Buffer.from(await blob.arrayBuffer()))\n        return\n      }\n\n      // STT: Get transcript and send to client\n      const transcript = await this.config.speech2Text(\n        blob,\n        this.conversation[this.conversation.length - 1]?.content\n      )\n      if (!transcript) {\n        this.debug('Ignoring empty transcript')\n        return\n      }\n\n      this.debug('User transcript:', transcript)\n\n      // Send transcript to client\n      this.addMessage({ role: 'user', content: transcript })\n\n      if (this.abortAnswer) {\n        this.debug('Answer aborted, no answer generated')\n        return\n      }\n\n      // LLM: Generate answer\n      const answer = await this.config.generateAnswer(this.conversation)\n      if (this.abortAnswer) {\n        this.debug('Answer aborted, ignoring answer')\n        return\n      }\n\n      await this.answer(answer)\n    } catch (error) {\n      console.error('[WS]', error)\n      this.socket.close()\n    }\n  }\n\n  private async answer(message: string) {\n    let isEnd = false\n\n    // Detect end of interview\n    if (message.includes(END_INTERVIEW)) {\n      message = message.replace(END_INTERVIEW, '').trim()\n      isEnd = true\n    }\n\n    if (message.length) {\n      // Send answer to client\n      this.debug('Assistant message:', message)\n      this.addMessage({ role: 'assistant', content: message })\n\n      // TTS: Generate answer audio\n      if (!disableTTS) {\n        const audio = await this.config.text2Speech(message)\n        if (this.abortAnswer) {\n          // Remove last assistant message if aborted\n          this.debug('Answer aborted, removing last assistant message')\n          const lastMessage = this.conversation[this.conversation.length - 1]\n          if (lastMessage?.role === 'assistant') {\n            this.conversation.pop()\n            this.socket.send(CallServerCommands.CancelLastAssistantMessage)\n          }\n          return\n        }\n\n        // Send audio to client\n        this.debug(`Send audio: (${audio.byteLength} bytes)`)\n        this.socket.send(audio)\n      }\n    }\n\n    // End of call\n    if (isEnd) {\n      this.debug('Interview ended')\n      this.socket.send(CallServerCommands.EndInterview)\n    }\n  }\n\n  private debug(...message: any[]) {\n    if (!debugEnabled) return\n    const nowTime = Date.now()\n    console.log(\n      `[WS] [${nowTime - this.startTime} | ${nowTime - this.lastDebug}ms]`,\n      ...message\n    )\n    this.lastDebug = nowTime\n  }\n}\n","export enum CallClientCommands {\n  StartSpeaking = 'startSpeaking',\n  StopSpeaking = 'stopSpeaking',\n  Mute = 'mute',\n}\n\nexport enum CallServerCommands {\n  UserMessage = 'userMessage',\n  AssistantMessage = 'assistantMessage',\n  CancelLastAssistantMessage = 'cancelLastAssistantMessage',\n  EndInterview = 'endInterview',\n}\n\nexport interface CallConfig {\n  systemPrompt: string\n  firstMessage?: string\n  generateAnswer(conversation: Conversation): Promise<string>\n  speech2Text(blob: Blob, prompt?: string): Promise<string>\n  text2Speech(text: string): Promise<ArrayBuffer>\n  onMessage?(message: ConversationMessage): void\n  onEnd?(call: CallSummary): void\n}\n\nexport interface CallSummary {\n  conversation: Conversation\n  duration: number\n}\n\nexport type Conversation = ConversationMessage[]\n\nexport interface ConversationMessage {\n  role: 'system' | 'user' | 'assistant'\n  content: string\n}\n","import WebSocket from 'ws'\n\nexport enum CallErrorCode {\n  BadRequest = 4400,\n  Unauthorized = 4401,\n  NotFound = 4404,\n}\n\nexport class CallError extends Error {\n  code: number\n\n  constructor(code: number, message: string) {\n    super(message)\n    this.code = code\n  }\n}\n\nexport function handleError(socket: WebSocket, error: unknown) {\n  if (error instanceof CallError) {\n    socket.close(error.code, error.message)\n  } else {\n    console.error(error)\n    socket.close(1011)\n  }\n  socket.terminate()\n}\n","import { WebSocket } from 'ws'\nimport { CallError, CallErrorCode } from './errors'\n\nexport async function waitForParams<CallParams>(\n  socket: WebSocket,\n  validate: (params: any) => CallParams\n): Promise<CallParams> {\n  return new Promise<CallParams>((resolve, reject) => {\n    // Handle timeout\n    const timeout = setTimeout(() => {\n      reject(new CallError(CallErrorCode.BadRequest, 'Missing params'))\n    }, 3000)\n\n    const onParams = (payload: string) => {\n      // Clear timeout and listener\n      clearTimeout(timeout)\n      socket.off('message', onParams)\n\n      try {\n        // Parse JSON payload\n        const params = validate(JSON.parse(payload))\n        resolve(params)\n      } catch (error) {\n        reject(new CallError(CallErrorCode.BadRequest, 'Invalid params'))\n      }\n    }\n\n    // Listen for params\n    socket.on('message', onParams)\n  })\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,SAAoB;AACpB,WAAsB;;;ACDf,IAAK,qBAAL,kBAAKA,wBAAL;AACL,EAAAA,oBAAA,mBAAgB;AAChB,EAAAA,oBAAA,kBAAe;AACf,EAAAA,oBAAA,UAAO;AAHG,SAAAA;AAAA,GAAA;AAML,IAAK,qBAAL,kBAAKC,wBAAL;AACL,EAAAA,oBAAA,iBAAc;AACd,EAAAA,oBAAA,sBAAmB;AACnB,EAAAA,oBAAA,gCAA6B;AAC7B,EAAAA,oBAAA,kBAAe;AAJL,SAAAA;AAAA,GAAA;;;ADKZ,IAAM,eAAe;AACrB,IAAM,gBAAgB;AACtB,IAAM,aAAa;AAEZ,IAAM,gBAAgB;AAEtB,IAAM,aAAN,MAAiB;AAAA,EAgBtB,YACS,QACA,QACP;AAFO;AACA;AAjBT,SAAQ,YAAY,KAAK,IAAI;AAC7B,SAAQ,YAAY,KAAK,IAAI;AAG7B;AAAA,SAAQ,cAAc;AAGtB;AAAA,SAAQ,aAAa;AAGrB;AAAA,SAAQ,SAAmB,CAAC;AAS1B,SAAK,eAAe,CAAC,EAAE,MAAM,UAAU,SAAS,OAAO,aAAa,CAAC;AACrE,SAAK,MAAM,cAAc;AAKzB,QAAI,OAAO,cAAc;AACvB,WAAK,OAAO,OAAO,YAAY;AAAA,IACjC,OAAO;AACL,WAAK,OACF,eAAe,KAAK,YAAY,EAChC,KAAK,CAAC,WAAW,KAAK,OAAO,MAAM,CAAC,EACpC,MAAM,CAAC,UAAU;AAChB,gBAAQ,MAAM,QAAQ,KAAK;AAC3B,aAAK,OAAO,MAAM;AAAA,MACpB,CAAC;AAAA,IACL;AAGA,SAAK,OAAO,GAAG,SAAS,KAAK,QAAQ,KAAK,IAAI,CAAC;AAC/C,SAAK,OAAO,GAAG,WAAW,KAAK,UAAU,KAAK,IAAI,CAAC;AAAA,EACrD;AAAA,EAEQ,WAAW,SAA8B;AAC/C,SAAK,aAAa,KAAK,OAAO;AAC9B,SAAK,OAAO;AAAA,MACV,GACE,QAAQ,SAAS,oFAGnB,IAAI,QAAQ,OAAO;AAAA,IACrB;AACA,SAAK,OAAO,YAAY,OAAO;AAAA,EACjC;AAAA,EAEQ,UAAU;AAChB,SAAK,MAAM,mBAAmB;AAC9B,SAAK,cAAc;AACnB,UAAM,WAAW,KAAK,OAAO,KAAK,IAAI,IAAI,KAAK,aAAa,GAAI;AAGhE,SAAK,OAAO,QAAQ;AAAA,MAClB,cAAc,KAAK,aAAa,MAAM,CAAC;AAAA;AAAA,MACvC;AAAA,IACF,CAAC;AAID,SAAK,SAAS;AAEd,SAAK,SAAS;AAAA,EAChB;AAAA,EAEA,MAAc,UAAU,SAAiB;AACvC,QAAI,CAAC,OAAO,SAAS,OAAO,GAAG;AAC7B,cAAQ,KAAK,8BAA8B;AAC3C;AAAA,IACF;AAGA,QAAI,QAAQ,aAAa,IAAI;AAC3B,YAAM,MAAM,QAAQ,SAAS;AAC7B,WAAK,MAAM,YAAY,GAAG,EAAE;AAE5B,UAAI,6CAA0C;AAE5C,aAAK,aAAa;AAElB,aAAK,cAAc;AAAA,MACrB,WAAW,2BAAiC;AAG1C,aAAK,cAAc;AAAA,MACrB,WAAW,2CAAyC;AAElD,aAAK,aAAa;AAClB,cAAM,KAAK,eAAe;AAAA,MAC5B;AAAA,IACF,WAGS,OAAO,SAAS,OAAO,KAAK,KAAK,YAAY;AACpD,WAAK,MAAM,mBAAmB,QAAQ,UAAU,SAAS;AACzD,WAAK,OAAO,KAAK,OAAO;AAAA,IAC1B;AAAA,EACF;AAAA,EAEA,MAAc,iBAAiB;AAE7B,QAAI,KAAK,OAAO,WAAW,EAAG;AAE9B,SAAK,cAAc;AAGnB,UAAM,OAAO,IAAI,KAAK,KAAK,QAAQ,EAAE,MAAM,YAAY,CAAC;AAGxD,SAAK,OAAO,SAAS;AAErB,QAAI;AAEF,UAAI,eAAe;AACjB,cAAM,WAAgB,UAAK,WAAW,YAAY;AAClD,QAAG,iBAAc,UAAU,OAAO,KAAK,MAAM,KAAK,YAAY,CAAC,CAAC;AAChE;AAAA,MACF;AAGA,YAAM,aAAa,MAAM,KAAK,OAAO;AAAA,QACnC;AAAA,QACA,KAAK,aAAa,KAAK,aAAa,SAAS,CAAC,GAAG;AAAA,MACnD;AACA,UAAI,CAAC,YAAY;AACf,aAAK,MAAM,2BAA2B;AACtC;AAAA,MACF;AAEA,WAAK,MAAM,oBAAoB,UAAU;AAGzC,WAAK,WAAW,EAAE,MAAM,QAAQ,SAAS,WAAW,CAAC;AAErD,UAAI,KAAK,aAAa;AACpB,aAAK,MAAM,qCAAqC;AAChD;AAAA,MACF;AAGA,YAAM,SAAS,MAAM,KAAK,OAAO,eAAe,KAAK,YAAY;AACjE,UAAI,KAAK,aAAa;AACpB,aAAK,MAAM,iCAAiC;AAC5C;AAAA,MACF;AAEA,YAAM,KAAK,OAAO,MAAM;AAAA,IAC1B,SAAS,OAAO;AACd,cAAQ,MAAM,QAAQ,KAAK;AAC3B,WAAK,OAAO,MAAM;AAAA,IACpB;AAAA,EACF;AAAA,EAEA,MAAc,OAAO,SAAiB;AACpC,QAAI,QAAQ;AAGZ,QAAI,QAAQ,SAAS,aAAa,GAAG;AACnC,gBAAU,QAAQ,QAAQ,eAAe,EAAE,EAAE,KAAK;AAClD,cAAQ;AAAA,IACV;AAEA,QAAI,QAAQ,QAAQ;AAElB,WAAK,MAAM,sBAAsB,OAAO;AACxC,WAAK,WAAW,EAAE,MAAM,aAAa,SAAS,QAAQ,CAAC;AAGvD,UAAI,CAAC,YAAY;AACf,cAAM,QAAQ,MAAM,KAAK,OAAO,YAAY,OAAO;AACnD,YAAI,KAAK,aAAa;AAEpB,eAAK,MAAM,iDAAiD;AAC5D,gBAAM,cAAc,KAAK,aAAa,KAAK,aAAa,SAAS,CAAC;AAClE,cAAI,aAAa,SAAS,aAAa;AACrC,iBAAK,aAAa,IAAI;AACtB,iBAAK,OAAO,kEAAkD;AAAA,UAChE;AACA;AAAA,QACF;AAGA,aAAK,MAAM,gBAAgB,MAAM,UAAU,SAAS;AACpD,aAAK,OAAO,KAAK,KAAK;AAAA,MACxB;AAAA,IACF;AAGA,QAAI,OAAO;AACT,WAAK,MAAM,iBAAiB;AAC5B,WAAK,OAAO,sCAAoC;AAAA,IAClD;AAAA,EACF;AAAA,EAEQ,SAAS,SAAgB;AAC/B,QAAI,CAAC,aAAc;AACnB,UAAM,UAAU,KAAK,IAAI;AACzB,YAAQ;AAAA,MACN,SAAS,UAAU,KAAK,SAAS,MAAM,UAAU,KAAK,SAAS;AAAA,MAC/D,GAAG;AAAA,IACL;AACA,SAAK,YAAY;AAAA,EACnB;AACF;;;AElOO,IAAK,gBAAL,kBAAKC,mBAAL;AACL,EAAAA,8BAAA,gBAAa,QAAb;AACA,EAAAA,8BAAA,kBAAe,QAAf;AACA,EAAAA,8BAAA,cAAW,QAAX;AAHU,SAAAA;AAAA,GAAA;AAML,IAAM,YAAN,cAAwB,MAAM;AAAA,EAGnC,YAAY,MAAc,SAAiB;AACzC,UAAM,OAAO;AACb,SAAK,OAAO;AAAA,EACd;AACF;AAEO,SAAS,YAAY,QAAmB,OAAgB;AAC7D,MAAI,iBAAiB,WAAW;AAC9B,WAAO,MAAM,MAAM,MAAM,MAAM,OAAO;AAAA,EACxC,OAAO;AACL,YAAQ,MAAM,KAAK;AACnB,WAAO,MAAM,IAAI;AAAA,EACnB;AACA,SAAO,UAAU;AACnB;;;ACtBA,eAAsB,cACpB,QACA,UACqB;AACrB,SAAO,IAAI,QAAoB,CAAC,SAAS,WAAW;AAElD,UAAM,UAAU,WAAW,MAAM;AAC/B,aAAO,IAAI,iCAAoC,gBAAgB,CAAC;AAAA,IAClE,GAAG,GAAI;AAEP,UAAM,WAAW,CAAC,YAAoB;AAEpC,mBAAa,OAAO;AACpB,aAAO,IAAI,WAAW,QAAQ;AAE9B,UAAI;AAEF,cAAM,SAAS,SAAS,KAAK,MAAM,OAAO,CAAC;AAC3C,gBAAQ,MAAM;AAAA,MAChB,SAAS,OAAO;AACd,eAAO,IAAI,iCAAoC,gBAAgB,CAAC;AAAA,MAClE;AAAA,IACF;AAGA,WAAO,GAAG,WAAW,QAAQ;AAAA,EAC/B,CAAC;AACH;","names":["CallClientCommands","CallServerCommands","CallErrorCode"]}
+{"version":3,"sources":["../src/index.ts","../src/CallSocket.ts","../src/types.ts","../src/errors.ts","../src/waitForParams.ts"],"sourcesContent":["export * from './CallSocket'\nexport * from './errors'\nexport * from './types'\nexport * from './waitForParams'\n","import * as fs from 'fs'\nimport * as path from 'path'\nimport { WebSocket } from 'ws'\nimport {\n  CallClientCommands,\n  CallConfig,\n  CallServerCommands,\n  Conversation,\n  ConversationMessage,\n} from './types'\n\nexport const END_INTERVIEW = 'END_INTERVIEW'\n\nexport class CallSocket {\n  private startTime = Date.now()\n  private lastDebug = Date.now()\n\n  // An answer can be aborted if user is speaking\n  private abortAnswer = false\n\n  // When user is speaking, we're waiting to chunks or to stop\n  private isSpeaking = false\n\n  // Chunks of user speech since user started speaking\n  private chunks: Buffer[] = []\n\n  // Conversation history\n  private conversation: Conversation\n\n  constructor(\n    public socket: WebSocket,\n    public config: CallConfig\n  ) {\n    this.conversation = [{ role: 'system', content: config.systemPrompt }]\n    this.log(`Call started`)\n\n    // Assistant speaks first\n\n    // LLM: Generate answer\n    if (config.firstMessage) {\n      this.answer(config.firstMessage)\n    } else {\n      this.config\n        .generateAnswer(this.conversation)\n        .then((answer) => this.answer(answer))\n        .catch((error) => {\n          console.error('[WS]', error)\n          this.socket.close()\n        })\n    }\n\n    // Listen to events\n    this.socket.on('close', this.onClose.bind(this))\n    this.socket.on('message', this.onMessage.bind(this))\n  }\n\n  // Reset conversation\n  public resetConversation(conversation: Conversation) {\n    this.log('Reset conversation')\n    this.conversation = conversation\n  }\n\n  private addMessage(message: ConversationMessage) {\n    this.conversation.push(message)\n    this.socket.send(\n      `${\n        message.role === 'user'\n          ? CallServerCommands.UserMessage\n          : CallServerCommands.AssistantMessage\n      } ${message.content}`\n    )\n    this.config.onMessage?.(message)\n  }\n\n  private onClose() {\n    this.log('Connection closed')\n    this.abortAnswer = true\n    const duration = Math.round((Date.now() - this.startTime) / 1000)\n\n    // End call callback\n    this.config.onEnd?.({\n      conversation: this.conversation.slice(1), // Remove system message\n      duration,\n    })\n\n    // Unset params\n    // @ts-ignore\n    this.socket = null\n    // @ts-ignore\n    this.config = null\n  }\n\n  private async onMessage(message: Buffer) {\n    if (!Buffer.isBuffer(message)) {\n      console.warn(`[WS] Message is not a buffer`)\n      return\n    }\n\n    // Commands\n    if (message.byteLength < 15) {\n      const cmd = message.toString()\n      this.log(`Command: ${cmd}`)\n\n      if (cmd === CallClientCommands.StartSpeaking) {\n        // User started speaking\n        this.isSpeaking = true\n        // Abort answer if there is generation in progress\n        this.abortAnswer = true\n      } else if (cmd === CallClientCommands.Mute) {\n        // User muted the call\n        // Abort answer if there is generation in progress\n        this.abortAnswer = true\n      } else if (cmd === CallClientCommands.StopSpeaking) {\n        // User stopped speaking\n        this.isSpeaking = false\n        await this.onStopSpeaking()\n      }\n    }\n\n    // Audio chunk\n    else if (Buffer.isBuffer(message) && this.isSpeaking) {\n      this.log(`Received chunk (${message.byteLength} bytes)`)\n      this.chunks.push(message)\n    }\n  }\n\n  private async onStopSpeaking() {\n    // Do nothing if there is no chunk\n    if (this.chunks.length === 0) return\n\n    this.abortAnswer = false\n\n    // Combine audio blob\n    const blob = new Blob(this.chunks, { type: 'audio/ogg' })\n\n    // Reset chunks for next user speech\n    this.chunks.length = 0\n\n    try {\n      // Save file to disk\n      if (this.config.debugSaveSpeech) {\n        const filePath = path.join(__dirname, 'speech.ogg')\n        fs.writeFileSync(filePath, Buffer.from(await blob.arrayBuffer()))\n        return\n      }\n\n      // STT: Get transcript and send to client\n      const transcript = await this.config.speech2Text(\n        blob,\n        this.conversation[this.conversation.length - 1]?.content\n      )\n      if (!transcript) {\n        this.log('Ignoring empty transcript')\n        return\n      }\n\n      this.log('User transcript:', transcript)\n\n      // Send transcript to client\n      this.addMessage({ role: 'user', content: transcript })\n\n      if (this.abortAnswer) {\n        this.log('Answer aborted, no answer generated')\n        return\n      }\n\n      // LLM: Generate answer\n      const answer = await this.config.generateAnswer(this.conversation)\n      if (this.abortAnswer) {\n        this.log('Answer aborted, ignoring answer')\n        return\n      }\n\n      await this.answer(answer)\n    } catch (error) {\n      console.error('[WS]', error)\n      this.socket.close()\n    }\n  }\n\n  // Add assistant message and send to client with audio (TTS)\n  public async answer(message: string) {\n    let isEnd = false\n\n    // Detect end of interview\n    if (message.includes(END_INTERVIEW)) {\n      message = message.replace(END_INTERVIEW, '').trim()\n      isEnd = true\n    }\n\n    if (message.length) {\n      // Send answer to client\n      this.log('Assistant message:', message)\n      this.addMessage({ role: 'assistant', content: message })\n\n      // TTS: Generate answer audio\n      if (!this.config.disableTTS) {\n        const audio = await this.config.text2Speech(message)\n        if (this.abortAnswer) {\n          // Remove last assistant message if aborted\n          this.log('Answer aborted, removing last assistant message')\n          const lastMessage = this.conversation[this.conversation.length - 1]\n          if (lastMessage?.role === 'assistant') {\n            this.conversation.pop()\n            this.socket.send(CallServerCommands.CancelLastAssistantMessage)\n          }\n          return\n        }\n\n        // Send audio to client\n        this.log(`Send audio: (${audio.byteLength} bytes)`)\n        this.socket.send(audio)\n      }\n    }\n\n    // End of call\n    if (isEnd) {\n      this.log('Interview ended')\n      this.socket.send(CallServerCommands.EndInterview)\n    }\n  }\n\n  private log(...message: any[]) {\n    if (!this.config.debugLog) return\n    const now = Date.now()\n    const delta = now - this.lastDebug\n    this.lastDebug = now\n    console.log(`[Debug +${delta}ms]`, ...message)\n  }\n}\n","export enum CallClientCommands {\n  StartSpeaking = 'startSpeaking',\n  StopSpeaking = 'stopSpeaking',\n  Mute = 'mute',\n}\n\nexport enum CallServerCommands {\n  UserMessage = 'userMessage',\n  AssistantMessage = 'assistantMessage',\n  CancelLastAssistantMessage = 'cancelLastAssistantMessage',\n  EndInterview = 'endInterview',\n}\n\nexport interface CallConfig {\n  systemPrompt: string\n  firstMessage?: string\n  debugLog?: boolean\n  debugSaveSpeech?: boolean\n  disableTTS?: boolean\n  generateAnswer(conversation: Conversation): Promise<string>\n  speech2Text(blob: Blob, prompt?: string): Promise<string>\n  text2Speech(text: string): Promise<ArrayBuffer>\n  onMessage?(message: ConversationMessage): void\n  onEnd?(call: CallSummary): void\n}\n\nexport interface CallSummary {\n  conversation: Conversation\n  duration: number\n}\n\nexport type Conversation = ConversationMessage[]\n\nexport interface ConversationMessage {\n  role: 'system' | 'user' | 'assistant'\n  content: string\n}\n","import WebSocket from 'ws'\n\nexport enum CallErrorCode {\n  BadRequest = 4400,\n  Unauthorized = 4401,\n  NotFound = 4404,\n}\n\nexport class CallError extends Error {\n  code: number\n\n  constructor(code: number, message: string) {\n    super(message)\n    this.code = code\n  }\n}\n\nexport function handleError(socket: WebSocket, error: unknown) {\n  if (error instanceof CallError) {\n    socket.close(error.code, error.message)\n  } else {\n    console.error(error)\n    socket.close(1011)\n  }\n  socket.terminate()\n}\n","import { WebSocket } from 'ws'\nimport { CallError, CallErrorCode } from './errors'\n\nexport async function waitForParams<CallParams>(\n  socket: WebSocket,\n  validate: (params: any) => CallParams\n): Promise<CallParams> {\n  return new Promise<CallParams>((resolve, reject) => {\n    // Handle timeout\n    const timeout = setTimeout(() => {\n      reject(new CallError(CallErrorCode.BadRequest, 'Missing params'))\n    }, 3000)\n\n    const onParams = (payload: string) => {\n      // Clear timeout and listener\n      clearTimeout(timeout)\n      socket.off('message', onParams)\n\n      try {\n        // Parse JSON payload\n        const params = validate(JSON.parse(payload))\n        resolve(params)\n      } catch (error) {\n        reject(new CallError(CallErrorCode.BadRequest, 'Invalid params'))\n      }\n    }\n\n    // Listen for params\n    socket.on('message', onParams)\n  })\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,SAAoB;AACpB,WAAsB;;;ACDf,IAAK,qBAAL,kBAAKA,wBAAL;AACL,EAAAA,oBAAA,mBAAgB;AAChB,EAAAA,oBAAA,kBAAe;AACf,EAAAA,oBAAA,UAAO;AAHG,SAAAA;AAAA,GAAA;AAML,IAAK,qBAAL,kBAAKC,wBAAL;AACL,EAAAA,oBAAA,iBAAc;AACd,EAAAA,oBAAA,sBAAmB;AACnB,EAAAA,oBAAA,gCAA6B;AAC7B,EAAAA,oBAAA,kBAAe;AAJL,SAAAA;AAAA,GAAA;;;ADKL,IAAM,gBAAgB;AAEtB,IAAM,aAAN,MAAiB;AAAA,EAgBtB,YACS,QACA,QACP;AAFO;AACA;AAjBT,SAAQ,YAAY,KAAK,IAAI;AAC7B,SAAQ,YAAY,KAAK,IAAI;AAG7B;AAAA,SAAQ,cAAc;AAGtB;AAAA,SAAQ,aAAa;AAGrB;AAAA,SAAQ,SAAmB,CAAC;AAS1B,SAAK,eAAe,CAAC,EAAE,MAAM,UAAU,SAAS,OAAO,aAAa,CAAC;AACrE,SAAK,IAAI,cAAc;AAKvB,QAAI,OAAO,cAAc;AACvB,WAAK,OAAO,OAAO,YAAY;AAAA,IACjC,OAAO;AACL,WAAK,OACF,eAAe,KAAK,YAAY,EAChC,KAAK,CAAC,WAAW,KAAK,OAAO,MAAM,CAAC,EACpC,MAAM,CAAC,UAAU;AAChB,gBAAQ,MAAM,QAAQ,KAAK;AAC3B,aAAK,OAAO,MAAM;AAAA,MACpB,CAAC;AAAA,IACL;AAGA,SAAK,OAAO,GAAG,SAAS,KAAK,QAAQ,KAAK,IAAI,CAAC;AAC/C,SAAK,OAAO,GAAG,WAAW,KAAK,UAAU,KAAK,IAAI,CAAC;AAAA,EACrD;AAAA;AAAA,EAGO,kBAAkB,cAA4B;AACnD,SAAK,IAAI,oBAAoB;AAC7B,SAAK,eAAe;AAAA,EACtB;AAAA,EAEQ,WAAW,SAA8B;AAC/C,SAAK,aAAa,KAAK,OAAO;AAC9B,SAAK,OAAO;AAAA,MACV,GACE,QAAQ,SAAS,oFAGnB,IAAI,QAAQ,OAAO;AAAA,IACrB;AACA,SAAK,OAAO,YAAY,OAAO;AAAA,EACjC;AAAA,EAEQ,UAAU;AAChB,SAAK,IAAI,mBAAmB;AAC5B,SAAK,cAAc;AACnB,UAAM,WAAW,KAAK,OAAO,KAAK,IAAI,IAAI,KAAK,aAAa,GAAI;AAGhE,SAAK,OAAO,QAAQ;AAAA,MAClB,cAAc,KAAK,aAAa,MAAM,CAAC;AAAA;AAAA,MACvC;AAAA,IACF,CAAC;AAID,SAAK,SAAS;AAEd,SAAK,SAAS;AAAA,EAChB;AAAA,EAEA,MAAc,UAAU,SAAiB;AACvC,QAAI,CAAC,OAAO,SAAS,OAAO,GAAG;AAC7B,cAAQ,KAAK,8BAA8B;AAC3C;AAAA,IACF;AAGA,QAAI,QAAQ,aAAa,IAAI;AAC3B,YAAM,MAAM,QAAQ,SAAS;AAC7B,WAAK,IAAI,YAAY,GAAG,EAAE;AAE1B,UAAI,6CAA0C;AAE5C,aAAK,aAAa;AAElB,aAAK,cAAc;AAAA,MACrB,WAAW,2BAAiC;AAG1C,aAAK,cAAc;AAAA,MACrB,WAAW,2CAAyC;AAElD,aAAK,aAAa;AAClB,cAAM,KAAK,eAAe;AAAA,MAC5B;AAAA,IACF,WAGS,OAAO,SAAS,OAAO,KAAK,KAAK,YAAY;AACpD,WAAK,IAAI,mBAAmB,QAAQ,UAAU,SAAS;AACvD,WAAK,OAAO,KAAK,OAAO;AAAA,IAC1B;AAAA,EACF;AAAA,EAEA,MAAc,iBAAiB;AAE7B,QAAI,KAAK,OAAO,WAAW,EAAG;AAE9B,SAAK,cAAc;AAGnB,UAAM,OAAO,IAAI,KAAK,KAAK,QAAQ,EAAE,MAAM,YAAY,CAAC;AAGxD,SAAK,OAAO,SAAS;AAErB,QAAI;AAEF,UAAI,KAAK,OAAO,iBAAiB;AAC/B,cAAM,WAAgB,UAAK,WAAW,YAAY;AAClD,QAAG,iBAAc,UAAU,OAAO,KAAK,MAAM,KAAK,YAAY,CAAC,CAAC;AAChE;AAAA,MACF;AAGA,YAAM,aAAa,MAAM,KAAK,OAAO;AAAA,QACnC;AAAA,QACA,KAAK,aAAa,KAAK,aAAa,SAAS,CAAC,GAAG;AAAA,MACnD;AACA,UAAI,CAAC,YAAY;AACf,aAAK,IAAI,2BAA2B;AACpC;AAAA,MACF;AAEA,WAAK,IAAI,oBAAoB,UAAU;AAGvC,WAAK,WAAW,EAAE,MAAM,QAAQ,SAAS,WAAW,CAAC;AAErD,UAAI,KAAK,aAAa;AACpB,aAAK,IAAI,qCAAqC;AAC9C;AAAA,MACF;AAGA,YAAM,SAAS,MAAM,KAAK,OAAO,eAAe,KAAK,YAAY;AACjE,UAAI,KAAK,aAAa;AACpB,aAAK,IAAI,iCAAiC;AAC1C;AAAA,MACF;AAEA,YAAM,KAAK,OAAO,MAAM;AAAA,IAC1B,SAAS,OAAO;AACd,cAAQ,MAAM,QAAQ,KAAK;AAC3B,WAAK,OAAO,MAAM;AAAA,IACpB;AAAA,EACF;AAAA;AAAA,EAGA,MAAa,OAAO,SAAiB;AACnC,QAAI,QAAQ;AAGZ,QAAI,QAAQ,SAAS,aAAa,GAAG;AACnC,gBAAU,QAAQ,QAAQ,eAAe,EAAE,EAAE,KAAK;AAClD,cAAQ;AAAA,IACV;AAEA,QAAI,QAAQ,QAAQ;AAElB,WAAK,IAAI,sBAAsB,OAAO;AACtC,WAAK,WAAW,EAAE,MAAM,aAAa,SAAS,QAAQ,CAAC;AAGvD,UAAI,CAAC,KAAK,OAAO,YAAY;AAC3B,cAAM,QAAQ,MAAM,KAAK,OAAO,YAAY,OAAO;AACnD,YAAI,KAAK,aAAa;AAEpB,eAAK,IAAI,iDAAiD;AAC1D,gBAAM,cAAc,KAAK,aAAa,KAAK,aAAa,SAAS,CAAC;AAClE,cAAI,aAAa,SAAS,aAAa;AACrC,iBAAK,aAAa,IAAI;AACtB,iBAAK,OAAO,kEAAkD;AAAA,UAChE;AACA;AAAA,QACF;AAGA,aAAK,IAAI,gBAAgB,MAAM,UAAU,SAAS;AAClD,aAAK,OAAO,KAAK,KAAK;AAAA,MACxB;AAAA,IACF;AAGA,QAAI,OAAO;AACT,WAAK,IAAI,iBAAiB;AAC1B,WAAK,OAAO,sCAAoC;AAAA,IAClD;AAAA,EACF;AAAA,EAEQ,OAAO,SAAgB;AAC7B,QAAI,CAAC,KAAK,OAAO,SAAU;AAC3B,UAAM,MAAM,KAAK,IAAI;AACrB,UAAM,QAAQ,MAAM,KAAK;AACzB,SAAK,YAAY;AACjB,YAAQ,IAAI,WAAW,KAAK,OAAO,GAAG,OAAO;AAAA,EAC/C;AACF;;;AEnOO,IAAK,gBAAL,kBAAKC,mBAAL;AACL,EAAAA,8BAAA,gBAAa,QAAb;AACA,EAAAA,8BAAA,kBAAe,QAAf;AACA,EAAAA,8BAAA,cAAW,QAAX;AAHU,SAAAA;AAAA,GAAA;AAML,IAAM,YAAN,cAAwB,MAAM;AAAA,EAGnC,YAAY,MAAc,SAAiB;AACzC,UAAM,OAAO;AACb,SAAK,OAAO;AAAA,EACd;AACF;AAEO,SAAS,YAAY,QAAmB,OAAgB;AAC7D,MAAI,iBAAiB,WAAW;AAC9B,WAAO,MAAM,MAAM,MAAM,MAAM,OAAO;AAAA,EACxC,OAAO;AACL,YAAQ,MAAM,KAAK;AACnB,WAAO,MAAM,IAAI;AAAA,EACnB;AACA,SAAO,UAAU;AACnB;;;ACtBA,eAAsB,cACpB,QACA,UACqB;AACrB,SAAO,IAAI,QAAoB,CAAC,SAAS,WAAW;AAElD,UAAM,UAAU,WAAW,MAAM;AAC/B,aAAO,IAAI,iCAAoC,gBAAgB,CAAC;AAAA,IAClE,GAAG,GAAI;AAEP,UAAM,WAAW,CAAC,YAAoB;AAEpC,mBAAa,OAAO;AACpB,aAAO,IAAI,WAAW,QAAQ;AAE9B,UAAI;AAEF,cAAM,SAAS,SAAS,KAAK,MAAM,OAAO,CAAC;AAC3C,gBAAQ,MAAM;AAAA,MAChB,SAAS,OAAO;AACd,eAAO,IAAI,iCAAoC,gBAAgB,CAAC;AAAA,MAClE;AAAA,IACF;AAGA,WAAO,GAAG,WAAW,QAAQ;AAAA,EAC/B,CAAC;AACH;","names":["CallClientCommands","CallServerCommands","CallErrorCode"]}

package/dist/index.mjs

@@ -18,9 +18,6 @@ var CallServerCommands = /* @__PURE__ */ ((CallServerCommands2) => {
 })(CallServerCommands || {});
 
 // src/CallSocket.ts
-var debugEnabled = false;
-var debugSaveFile = false;
-var disableTTS = false;
 var END_INTERVIEW = "END_INTERVIEW";
 var CallSocket = class {
   constructor(socket, config) {
@@ -35,7 +32,7 @@ var CallSocket = class {
     // Chunks of user speech since user started speaking
     this.chunks = [];
     this.conversation = [{ role: "system", content: config.systemPrompt }];
-    this.debug(`Call started`);
+    this.log(`Call started`);
     if (config.firstMessage) {
       this.answer(config.firstMessage);
     } else {
@@ -47,6 +44,11 @@ var CallSocket = class {
     this.socket.on("close", this.onClose.bind(this));
     this.socket.on("message", this.onMessage.bind(this));
   }
+  // Reset conversation
+  resetConversation(conversation) {
+    this.log("Reset conversation");
+    this.conversation = conversation;
+  }
   addMessage(message) {
     this.conversation.push(message);
     this.socket.send(
@@ -55,7 +57,7 @@ var CallSocket = class {
     this.config.onMessage?.(message);
   }
   onClose() {
-    this.debug("Connection closed");
+    this.log("Connection closed");
     this.abortAnswer = true;
     const duration = Math.round((Date.now() - this.startTime) / 1e3);
     this.config.onEnd?.({
@@ -73,7 +75,7 @@ var CallSocket = class {
     }
     if (message.byteLength < 15) {
       const cmd = message.toString();
-      this.debug(`Command: ${cmd}`);
+      this.log(`Command: ${cmd}`);
       if (cmd === "startSpeaking" /* StartSpeaking */) {
         this.isSpeaking = true;
         this.abortAnswer = true;
@@ -84,7 +86,7 @@ var CallSocket = class {
         await this.onStopSpeaking();
       }
     } else if (Buffer.isBuffer(message) && this.isSpeaking) {
-      this.debug(`Received chunk (${message.byteLength} bytes)`);
+      this.log(`Received chunk (${message.byteLength} bytes)`);
       this.chunks.push(message);
     }
   }
@@ -94,7 +96,7 @@ var CallSocket = class {
     const blob = new Blob(this.chunks, { type: "audio/ogg" });
     this.chunks.length = 0;
     try {
-      if (debugSaveFile) {
+      if (this.config.debugSaveSpeech) {
         const filePath = path.join(__dirname, "speech.ogg");
         fs.writeFileSync(filePath, Buffer.from(await blob.arrayBuffer()));
         return;
@@ -104,18 +106,18 @@ var CallSocket = class {
         this.conversation[this.conversation.length - 1]?.content
       );
       if (!transcript) {
-        this.debug("Ignoring empty transcript");
+        this.log("Ignoring empty transcript");
         return;
       }
-      this.debug("User transcript:", transcript);
+      this.log("User transcript:", transcript);
       this.addMessage({ role: "user", content: transcript });
       if (this.abortAnswer) {
-        this.debug("Answer aborted, no answer generated");
+        this.log("Answer aborted, no answer generated");
         return;
       }
       const answer = await this.config.generateAnswer(this.conversation);
       if (this.abortAnswer) {
-        this.debug("Answer aborted, ignoring answer");
+        this.log("Answer aborted, ignoring answer");
         return;
       }
       await this.answer(answer);
@@ -124,6 +126,7 @@ var CallSocket = class {
       this.socket.close();
     }
   }
+  // Add assistant message and send to client with audio (TTS)
   async answer(message) {
     let isEnd = false;
     if (message.includes(END_INTERVIEW)) {
@@ -131,12 +134,12 @@ var CallSocket = class {
       isEnd = true;
     }
     if (message.length) {
-      this.debug("Assistant message:", message);
+      this.log("Assistant message:", message);
       this.addMessage({ role: "assistant", content: message });
-      if (!disableTTS) {
+      if (!this.config.disableTTS) {
         const audio = await this.config.text2Speech(message);
         if (this.abortAnswer) {
-          this.debug("Answer aborted, removing last assistant message");
+          this.log("Answer aborted, removing last assistant message");
           const lastMessage = this.conversation[this.conversation.length - 1];
           if (lastMessage?.role === "assistant") {
             this.conversation.pop();
@@ -144,23 +147,21 @@ var CallSocket = class {
           }
           return;
         }
-        this.debug(`Send audio: (${audio.byteLength} bytes)`);
+        this.log(`Send audio: (${audio.byteLength} bytes)`);
         this.socket.send(audio);
       }
     }
     if (isEnd) {
-      this.debug("Interview ended");
+      this.log("Interview ended");
       this.socket.send("endInterview" /* EndInterview */);
     }
   }
-  debug(...message) {
-    if (!debugEnabled) return;
-    const nowTime = Date.now();
-    console.log(
-      `[WS] [${nowTime - this.startTime} | ${nowTime - this.lastDebug}ms]`,
-      ...message
-    );
-    this.lastDebug = nowTime;
+  log(...message) {
+    if (!this.config.debugLog) return;
+    const now = Date.now();
+    const delta = now - this.lastDebug;
+    this.lastDebug = now;
+    console.log(`[Debug +${delta}ms]`, ...message);
   }
 };
 

package/dist/index.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/CallSocket.ts","../src/types.ts","../src/errors.ts","../src/waitForParams.ts"],"sourcesContent":["import * as fs from 'fs'\nimport * as path from 'path'\nimport { WebSocket } from 'ws'\nimport {\n  CallClientCommands,\n  CallConfig,\n  CallServerCommands,\n  Conversation,\n  ConversationMessage,\n} from './types'\n\nconst debugEnabled = false\nconst debugSaveFile = false\nconst disableTTS = false\n\nexport const END_INTERVIEW = 'END_INTERVIEW'\n\nexport class CallSocket {\n  private startTime = Date.now()\n  private lastDebug = Date.now()\n\n  // An answer can be aborted if user is speaking\n  private abortAnswer = false\n\n  // When user is speaking, we're waiting to chunks or to stop\n  private isSpeaking = false\n\n  // Chunks of user speech since user started speaking\n  private chunks: Buffer[] = []\n\n  // Conversation history\n  private conversation: Conversation\n\n  constructor(\n    public socket: WebSocket,\n    public config: CallConfig\n  ) {\n    this.conversation = [{ role: 'system', content: config.systemPrompt }]\n    this.debug(`Call started`)\n\n    // Assistant speaks first\n\n    // LLM: Generate answer\n    if (config.firstMessage) {\n      this.answer(config.firstMessage)\n    } else {\n      this.config\n        .generateAnswer(this.conversation)\n        .then((answer) => this.answer(answer))\n        .catch((error) => {\n          console.error('[WS]', error)\n          this.socket.close()\n        })\n    }\n\n    // Listen to events\n    this.socket.on('close', this.onClose.bind(this))\n    this.socket.on('message', this.onMessage.bind(this))\n  }\n\n  private addMessage(message: ConversationMessage) {\n    this.conversation.push(message)\n    this.socket.send(\n      `${\n        message.role === 'user'\n          ? CallServerCommands.UserMessage\n          : CallServerCommands.AssistantMessage\n      } ${message.content}`\n    )\n    this.config.onMessage?.(message)\n  }\n\n  private onClose() {\n    this.debug('Connection closed')\n    this.abortAnswer = true\n    const duration = Math.round((Date.now() - this.startTime) / 1000)\n\n    // End call callback\n    this.config.onEnd?.({\n      conversation: this.conversation.slice(1), // Remove system message\n      duration,\n    })\n\n    // Unset params\n    // @ts-ignore\n    this.socket = null\n    // @ts-ignore\n    this.config = null\n  }\n\n  private async onMessage(message: Buffer) {\n    if (!Buffer.isBuffer(message)) {\n      console.warn(`[WS] Message is not a buffer`)\n      return\n    }\n\n    // Commands\n    if (message.byteLength < 15) {\n      const cmd = message.toString()\n      this.debug(`Command: ${cmd}`)\n\n      if (cmd === CallClientCommands.StartSpeaking) {\n        // User started speaking\n        this.isSpeaking = true\n        // Abort answer if there is generation in progress\n        this.abortAnswer = true\n      } else if (cmd === CallClientCommands.Mute) {\n        // User muted the call\n        // Abort answer if there is generation in progress\n        this.abortAnswer = true\n      } else if (cmd === CallClientCommands.StopSpeaking) {\n        // User stopped speaking\n        this.isSpeaking = false\n        await this.onStopSpeaking()\n      }\n    }\n\n    // Audio chunk\n    else if (Buffer.isBuffer(message) && this.isSpeaking) {\n      this.debug(`Received chunk (${message.byteLength} bytes)`)\n      this.chunks.push(message)\n    }\n  }\n\n  private async onStopSpeaking() {\n    // Do nothing if there is no chunk\n    if (this.chunks.length === 0) return\n\n    this.abortAnswer = false\n\n    // Combine audio blob\n    const blob = new Blob(this.chunks, { type: 'audio/ogg' })\n\n    // Reset chunks for next user speech\n    this.chunks.length = 0\n\n    try {\n      // Save file to disk\n      if (debugSaveFile) {\n        const filePath = path.join(__dirname, 'speech.ogg')\n        fs.writeFileSync(filePath, Buffer.from(await blob.arrayBuffer()))\n        return\n      }\n\n      // STT: Get transcript and send to client\n      const transcript = await this.config.speech2Text(\n        blob,\n        this.conversation[this.conversation.length - 1]?.content\n      )\n      if (!transcript) {\n        this.debug('Ignoring empty transcript')\n        return\n      }\n\n      this.debug('User transcript:', transcript)\n\n      // Send transcript to client\n      this.addMessage({ role: 'user', content: transcript })\n\n      if (this.abortAnswer) {\n        this.debug('Answer aborted, no answer generated')\n        return\n      }\n\n      // LLM: Generate answer\n      const answer = await this.config.generateAnswer(this.conversation)\n      if (this.abortAnswer) {\n        this.debug('Answer aborted, ignoring answer')\n        return\n      }\n\n      await this.answer(answer)\n    } catch (error) {\n      console.error('[WS]', error)\n      this.socket.close()\n    }\n  }\n\n  private async answer(message: string) {\n    let isEnd = false\n\n    // Detect end of interview\n    if (message.includes(END_INTERVIEW)) {\n      message = message.replace(END_INTERVIEW, '').trim()\n      isEnd = true\n    }\n\n    if (message.length) {\n      // Send answer to client\n      this.debug('Assistant message:', message)\n      this.addMessage({ role: 'assistant', content: message })\n\n      // TTS: Generate answer audio\n      if (!disableTTS) {\n        const audio = await this.config.text2Speech(message)\n        if (this.abortAnswer) {\n          // Remove last assistant message if aborted\n          this.debug('Answer aborted, removing last assistant message')\n          const lastMessage = this.conversation[this.conversation.length - 1]\n          if (lastMessage?.role === 'assistant') {\n            this.conversation.pop()\n            this.socket.send(CallServerCommands.CancelLastAssistantMessage)\n          }\n          return\n        }\n\n        // Send audio to client\n        this.debug(`Send audio: (${audio.byteLength} bytes)`)\n        this.socket.send(audio)\n      }\n    }\n\n    // End of call\n    if (isEnd) {\n      this.debug('Interview ended')\n      this.socket.send(CallServerCommands.EndInterview)\n    }\n  }\n\n  private debug(...message: any[]) {\n    if (!debugEnabled) return\n    const nowTime = Date.now()\n    console.log(\n      `[WS] [${nowTime - this.startTime} | ${nowTime - this.lastDebug}ms]`,\n      ...message\n    )\n    this.lastDebug = nowTime\n  }\n}\n","export enum CallClientCommands {\n  StartSpeaking = 'startSpeaking',\n  StopSpeaking = 'stopSpeaking',\n  Mute = 'mute',\n}\n\nexport enum CallServerCommands {\n  UserMessage = 'userMessage',\n  AssistantMessage = 'assistantMessage',\n  CancelLastAssistantMessage = 'cancelLastAssistantMessage',\n  EndInterview = 'endInterview',\n}\n\nexport interface CallConfig {\n  systemPrompt: string\n  firstMessage?: string\n  generateAnswer(conversation: Conversation): Promise<string>\n  speech2Text(blob: Blob, prompt?: string): Promise<string>\n  text2Speech(text: string): Promise<ArrayBuffer>\n  onMessage?(message: ConversationMessage): void\n  onEnd?(call: CallSummary): void\n}\n\nexport interface CallSummary {\n  conversation: Conversation\n  duration: number\n}\n\nexport type Conversation = ConversationMessage[]\n\nexport interface ConversationMessage {\n  role: 'system' | 'user' | 'assistant'\n  content: string\n}\n","import WebSocket from 'ws'\n\nexport enum CallErrorCode {\n  BadRequest = 4400,\n  Unauthorized = 4401,\n  NotFound = 4404,\n}\n\nexport class CallError extends Error {\n  code: number\n\n  constructor(code: number, message: string) {\n    super(message)\n    this.code = code\n  }\n}\n\nexport function handleError(socket: WebSocket, error: unknown) {\n  if (error instanceof CallError) {\n    socket.close(error.code, error.message)\n  } else {\n    console.error(error)\n    socket.close(1011)\n  }\n  socket.terminate()\n}\n","import { WebSocket } from 'ws'\nimport { CallError, CallErrorCode } from './errors'\n\nexport async function waitForParams<CallParams>(\n  socket: WebSocket,\n  validate: (params: any) => CallParams\n): Promise<CallParams> {\n  return new Promise<CallParams>((resolve, reject) => {\n    // Handle timeout\n    const timeout = setTimeout(() => {\n      reject(new CallError(CallErrorCode.BadRequest, 'Missing params'))\n    }, 3000)\n\n    const onParams = (payload: string) => {\n      // Clear timeout and listener\n      clearTimeout(timeout)\n      socket.off('message', onParams)\n\n      try {\n        // Parse JSON payload\n        const params = validate(JSON.parse(payload))\n        resolve(params)\n      } catch (error) {\n        reject(new CallError(CallErrorCode.BadRequest, 'Invalid params'))\n      }\n    }\n\n    // Listen for params\n    socket.on('message', onParams)\n  })\n}\n"],"mappings":";AAAA,YAAY,QAAQ;AACpB,YAAY,UAAU;;;ACDf,IAAK,qBAAL,kBAAKA,wBAAL;AACL,EAAAA,oBAAA,mBAAgB;AAChB,EAAAA,oBAAA,kBAAe;AACf,EAAAA,oBAAA,UAAO;AAHG,SAAAA;AAAA,GAAA;AAML,IAAK,qBAAL,kBAAKC,wBAAL;AACL,EAAAA,oBAAA,iBAAc;AACd,EAAAA,oBAAA,sBAAmB;AACnB,EAAAA,oBAAA,gCAA6B;AAC7B,EAAAA,oBAAA,kBAAe;AAJL,SAAAA;AAAA,GAAA;;;ADKZ,IAAM,eAAe;AACrB,IAAM,gBAAgB;AACtB,IAAM,aAAa;AAEZ,IAAM,gBAAgB;AAEtB,IAAM,aAAN,MAAiB;AAAA,EAgBtB,YACS,QACA,QACP;AAFO;AACA;AAjBT,SAAQ,YAAY,KAAK,IAAI;AAC7B,SAAQ,YAAY,KAAK,IAAI;AAG7B;AAAA,SAAQ,cAAc;AAGtB;AAAA,SAAQ,aAAa;AAGrB;AAAA,SAAQ,SAAmB,CAAC;AAS1B,SAAK,eAAe,CAAC,EAAE,MAAM,UAAU,SAAS,OAAO,aAAa,CAAC;AACrE,SAAK,MAAM,cAAc;AAKzB,QAAI,OAAO,cAAc;AACvB,WAAK,OAAO,OAAO,YAAY;AAAA,IACjC,OAAO;AACL,WAAK,OACF,eAAe,KAAK,YAAY,EAChC,KAAK,CAAC,WAAW,KAAK,OAAO,MAAM,CAAC,EACpC,MAAM,CAAC,UAAU;AAChB,gBAAQ,MAAM,QAAQ,KAAK;AAC3B,aAAK,OAAO,MAAM;AAAA,MACpB,CAAC;AAAA,IACL;AAGA,SAAK,OAAO,GAAG,SAAS,KAAK,QAAQ,KAAK,IAAI,CAAC;AAC/C,SAAK,OAAO,GAAG,WAAW,KAAK,UAAU,KAAK,IAAI,CAAC;AAAA,EACrD;AAAA,EAEQ,WAAW,SAA8B;AAC/C,SAAK,aAAa,KAAK,OAAO;AAC9B,SAAK,OAAO;AAAA,MACV,GACE,QAAQ,SAAS,oFAGnB,IAAI,QAAQ,OAAO;AAAA,IACrB;AACA,SAAK,OAAO,YAAY,OAAO;AAAA,EACjC;AAAA,EAEQ,UAAU;AAChB,SAAK,MAAM,mBAAmB;AAC9B,SAAK,cAAc;AACnB,UAAM,WAAW,KAAK,OAAO,KAAK,IAAI,IAAI,KAAK,aAAa,GAAI;AAGhE,SAAK,OAAO,QAAQ;AAAA,MAClB,cAAc,KAAK,aAAa,MAAM,CAAC;AAAA;AAAA,MACvC;AAAA,IACF,CAAC;AAID,SAAK,SAAS;AAEd,SAAK,SAAS;AAAA,EAChB;AAAA,EAEA,MAAc,UAAU,SAAiB;AACvC,QAAI,CAAC,OAAO,SAAS,OAAO,GAAG;AAC7B,cAAQ,KAAK,8BAA8B;AAC3C;AAAA,IACF;AAGA,QAAI,QAAQ,aAAa,IAAI;AAC3B,YAAM,MAAM,QAAQ,SAAS;AAC7B,WAAK,MAAM,YAAY,GAAG,EAAE;AAE5B,UAAI,6CAA0C;AAE5C,aAAK,aAAa;AAElB,aAAK,cAAc;AAAA,MACrB,WAAW,2BAAiC;AAG1C,aAAK,cAAc;AAAA,MACrB,WAAW,2CAAyC;AAElD,aAAK,aAAa;AAClB,cAAM,KAAK,eAAe;AAAA,MAC5B;AAAA,IACF,WAGS,OAAO,SAAS,OAAO,KAAK,KAAK,YAAY;AACpD,WAAK,MAAM,mBAAmB,QAAQ,UAAU,SAAS;AACzD,WAAK,OAAO,KAAK,OAAO;AAAA,IAC1B;AAAA,EACF;AAAA,EAEA,MAAc,iBAAiB;AAE7B,QAAI,KAAK,OAAO,WAAW,EAAG;AAE9B,SAAK,cAAc;AAGnB,UAAM,OAAO,IAAI,KAAK,KAAK,QAAQ,EAAE,MAAM,YAAY,CAAC;AAGxD,SAAK,OAAO,SAAS;AAErB,QAAI;AAEF,UAAI,eAAe;AACjB,cAAM,WAAgB,UAAK,WAAW,YAAY;AAClD,QAAG,iBAAc,UAAU,OAAO,KAAK,MAAM,KAAK,YAAY,CAAC,CAAC;AAChE;AAAA,MACF;AAGA,YAAM,aAAa,MAAM,KAAK,OAAO;AAAA,QACnC;AAAA,QACA,KAAK,aAAa,KAAK,aAAa,SAAS,CAAC,GAAG;AAAA,MACnD;AACA,UAAI,CAAC,YAAY;AACf,aAAK,MAAM,2BAA2B;AACtC;AAAA,MACF;AAEA,WAAK,MAAM,oBAAoB,UAAU;AAGzC,WAAK,WAAW,EAAE,MAAM,QAAQ,SAAS,WAAW,CAAC;AAErD,UAAI,KAAK,aAAa;AACpB,aAAK,MAAM,qCAAqC;AAChD;AAAA,MACF;AAGA,YAAM,SAAS,MAAM,KAAK,OAAO,eAAe,KAAK,YAAY;AACjE,UAAI,KAAK,aAAa;AACpB,aAAK,MAAM,iCAAiC;AAC5C;AAAA,MACF;AAEA,YAAM,KAAK,OAAO,MAAM;AAAA,IAC1B,SAAS,OAAO;AACd,cAAQ,MAAM,QAAQ,KAAK;AAC3B,WAAK,OAAO,MAAM;AAAA,IACpB;AAAA,EACF;AAAA,EAEA,MAAc,OAAO,SAAiB;AACpC,QAAI,QAAQ;AAGZ,QAAI,QAAQ,SAAS,aAAa,GAAG;AACnC,gBAAU,QAAQ,QAAQ,eAAe,EAAE,EAAE,KAAK;AAClD,cAAQ;AAAA,IACV;AAEA,QAAI,QAAQ,QAAQ;AAElB,WAAK,MAAM,sBAAsB,OAAO;AACxC,WAAK,WAAW,EAAE,MAAM,aAAa,SAAS,QAAQ,CAAC;AAGvD,UAAI,CAAC,YAAY;AACf,cAAM,QAAQ,MAAM,KAAK,OAAO,YAAY,OAAO;AACnD,YAAI,KAAK,aAAa;AAEpB,eAAK,MAAM,iDAAiD;AAC5D,gBAAM,cAAc,KAAK,aAAa,KAAK,aAAa,SAAS,CAAC;AAClE,cAAI,aAAa,SAAS,aAAa;AACrC,iBAAK,aAAa,IAAI;AACtB,iBAAK,OAAO,kEAAkD;AAAA,UAChE;AACA;AAAA,QACF;AAGA,aAAK,MAAM,gBAAgB,MAAM,UAAU,SAAS;AACpD,aAAK,OAAO,KAAK,KAAK;AAAA,MACxB;AAAA,IACF;AAGA,QAAI,OAAO;AACT,WAAK,MAAM,iBAAiB;AAC5B,WAAK,OAAO,sCAAoC;AAAA,IAClD;AAAA,EACF;AAAA,EAEQ,SAAS,SAAgB;AAC/B,QAAI,CAAC,aAAc;AACnB,UAAM,UAAU,KAAK,IAAI;AACzB,YAAQ;AAAA,MACN,SAAS,UAAU,KAAK,SAAS,MAAM,UAAU,KAAK,SAAS;AAAA,MAC/D,GAAG;AAAA,IACL;AACA,SAAK,YAAY;AAAA,EACnB;AACF;;;AElOO,IAAK,gBAAL,kBAAKC,mBAAL;AACL,EAAAA,8BAAA,gBAAa,QAAb;AACA,EAAAA,8BAAA,kBAAe,QAAf;AACA,EAAAA,8BAAA,cAAW,QAAX;AAHU,SAAAA;AAAA,GAAA;AAML,IAAM,YAAN,cAAwB,MAAM;AAAA,EAGnC,YAAY,MAAc,SAAiB;AACzC,UAAM,OAAO;AACb,SAAK,OAAO;AAAA,EACd;AACF;AAEO,SAAS,YAAY,QAAmB,OAAgB;AAC7D,MAAI,iBAAiB,WAAW;AAC9B,WAAO,MAAM,MAAM,MAAM,MAAM,OAAO;AAAA,EACxC,OAAO;AACL,YAAQ,MAAM,KAAK;AACnB,WAAO,MAAM,IAAI;AAAA,EACnB;AACA,SAAO,UAAU;AACnB;;;ACtBA,eAAsB,cACpB,QACA,UACqB;AACrB,SAAO,IAAI,QAAoB,CAAC,SAAS,WAAW;AAElD,UAAM,UAAU,WAAW,MAAM;AAC/B,aAAO,IAAI,iCAAoC,gBAAgB,CAAC;AAAA,IAClE,GAAG,GAAI;AAEP,UAAM,WAAW,CAAC,YAAoB;AAEpC,mBAAa,OAAO;AACpB,aAAO,IAAI,WAAW,QAAQ;AAE9B,UAAI;AAEF,cAAM,SAAS,SAAS,KAAK,MAAM,OAAO,CAAC;AAC3C,gBAAQ,MAAM;AAAA,MAChB,SAAS,OAAO;AACd,eAAO,IAAI,iCAAoC,gBAAgB,CAAC;AAAA,MAClE;AAAA,IACF;AAGA,WAAO,GAAG,WAAW,QAAQ;AAAA,EAC/B,CAAC;AACH;","names":["CallClientCommands","CallServerCommands","CallErrorCode"]}
+{"version":3,"sources":["../src/CallSocket.ts","../src/types.ts","../src/errors.ts","../src/waitForParams.ts"],"sourcesContent":["import * as fs from 'fs'\nimport * as path from 'path'\nimport { WebSocket } from 'ws'\nimport {\n  CallClientCommands,\n  CallConfig,\n  CallServerCommands,\n  Conversation,\n  ConversationMessage,\n} from './types'\n\nexport const END_INTERVIEW = 'END_INTERVIEW'\n\nexport class CallSocket {\n  private startTime = Date.now()\n  private lastDebug = Date.now()\n\n  // An answer can be aborted if user is speaking\n  private abortAnswer = false\n\n  // When user is speaking, we're waiting to chunks or to stop\n  private isSpeaking = false\n\n  // Chunks of user speech since user started speaking\n  private chunks: Buffer[] = []\n\n  // Conversation history\n  private conversation: Conversation\n\n  constructor(\n    public socket: WebSocket,\n    public config: CallConfig\n  ) {\n    this.conversation = [{ role: 'system', content: config.systemPrompt }]\n    this.log(`Call started`)\n\n    // Assistant speaks first\n\n    // LLM: Generate answer\n    if (config.firstMessage) {\n      this.answer(config.firstMessage)\n    } else {\n      this.config\n        .generateAnswer(this.conversation)\n        .then((answer) => this.answer(answer))\n        .catch((error) => {\n          console.error('[WS]', error)\n          this.socket.close()\n        })\n    }\n\n    // Listen to events\n    this.socket.on('close', this.onClose.bind(this))\n    this.socket.on('message', this.onMessage.bind(this))\n  }\n\n  // Reset conversation\n  public resetConversation(conversation: Conversation) {\n    this.log('Reset conversation')\n    this.conversation = conversation\n  }\n\n  private addMessage(message: ConversationMessage) {\n    this.conversation.push(message)\n    this.socket.send(\n      `${\n        message.role === 'user'\n          ? CallServerCommands.UserMessage\n          : CallServerCommands.AssistantMessage\n      } ${message.content}`\n    )\n    this.config.onMessage?.(message)\n  }\n\n  private onClose() {\n    this.log('Connection closed')\n    this.abortAnswer = true\n    const duration = Math.round((Date.now() - this.startTime) / 1000)\n\n    // End call callback\n    this.config.onEnd?.({\n      conversation: this.conversation.slice(1), // Remove system message\n      duration,\n    })\n\n    // Unset params\n    // @ts-ignore\n    this.socket = null\n    // @ts-ignore\n    this.config = null\n  }\n\n  private async onMessage(message: Buffer) {\n    if (!Buffer.isBuffer(message)) {\n      console.warn(`[WS] Message is not a buffer`)\n      return\n    }\n\n    // Commands\n    if (message.byteLength < 15) {\n      const cmd = message.toString()\n      this.log(`Command: ${cmd}`)\n\n      if (cmd === CallClientCommands.StartSpeaking) {\n        // User started speaking\n        this.isSpeaking = true\n        // Abort answer if there is generation in progress\n        this.abortAnswer = true\n      } else if (cmd === CallClientCommands.Mute) {\n        // User muted the call\n        // Abort answer if there is generation in progress\n        this.abortAnswer = true\n      } else if (cmd === CallClientCommands.StopSpeaking) {\n        // User stopped speaking\n        this.isSpeaking = false\n        await this.onStopSpeaking()\n      }\n    }\n\n    // Audio chunk\n    else if (Buffer.isBuffer(message) && this.isSpeaking) {\n      this.log(`Received chunk (${message.byteLength} bytes)`)\n      this.chunks.push(message)\n    }\n  }\n\n  private async onStopSpeaking() {\n    // Do nothing if there is no chunk\n    if (this.chunks.length === 0) return\n\n    this.abortAnswer = false\n\n    // Combine audio blob\n    const blob = new Blob(this.chunks, { type: 'audio/ogg' })\n\n    // Reset chunks for next user speech\n    this.chunks.length = 0\n\n    try {\n      // Save file to disk\n      if (this.config.debugSaveSpeech) {\n        const filePath = path.join(__dirname, 'speech.ogg')\n        fs.writeFileSync(filePath, Buffer.from(await blob.arrayBuffer()))\n        return\n      }\n\n      // STT: Get transcript and send to client\n      const transcript = await this.config.speech2Text(\n        blob,\n        this.conversation[this.conversation.length - 1]?.content\n      )\n      if (!transcript) {\n        this.log('Ignoring empty transcript')\n        return\n      }\n\n      this.log('User transcript:', transcript)\n\n      // Send transcript to client\n      this.addMessage({ role: 'user', content: transcript })\n\n      if (this.abortAnswer) {\n        this.log('Answer aborted, no answer generated')\n        return\n      }\n\n      // LLM: Generate answer\n      const answer = await this.config.generateAnswer(this.conversation)\n      if (this.abortAnswer) {\n        this.log('Answer aborted, ignoring answer')\n        return\n      }\n\n      await this.answer(answer)\n    } catch (error) {\n      console.error('[WS]', error)\n      this.socket.close()\n    }\n  }\n\n  // Add assistant message and send to client with audio (TTS)\n  public async answer(message: string) {\n    let isEnd = false\n\n    // Detect end of interview\n    if (message.includes(END_INTERVIEW)) {\n      message = message.replace(END_INTERVIEW, '').trim()\n      isEnd = true\n    }\n\n    if (message.length) {\n      // Send answer to client\n      this.log('Assistant message:', message)\n      this.addMessage({ role: 'assistant', content: message })\n\n      // TTS: Generate answer audio\n      if (!this.config.disableTTS) {\n        const audio = await this.config.text2Speech(message)\n        if (this.abortAnswer) {\n          // Remove last assistant message if aborted\n          this.log('Answer aborted, removing last assistant message')\n          const lastMessage = this.conversation[this.conversation.length - 1]\n          if (lastMessage?.role === 'assistant') {\n            this.conversation.pop()\n            this.socket.send(CallServerCommands.CancelLastAssistantMessage)\n          }\n          return\n        }\n\n        // Send audio to client\n        this.log(`Send audio: (${audio.byteLength} bytes)`)\n        this.socket.send(audio)\n      }\n    }\n\n    // End of call\n    if (isEnd) {\n      this.log('Interview ended')\n      this.socket.send(CallServerCommands.EndInterview)\n    }\n  }\n\n  private log(...message: any[]) {\n    if (!this.config.debugLog) return\n    const now = Date.now()\n    const delta = now - this.lastDebug\n    this.lastDebug = now\n    console.log(`[Debug +${delta}ms]`, ...message)\n  }\n}\n","export enum CallClientCommands {\n  StartSpeaking = 'startSpeaking',\n  StopSpeaking = 'stopSpeaking',\n  Mute = 'mute',\n}\n\nexport enum CallServerCommands {\n  UserMessage = 'userMessage',\n  AssistantMessage = 'assistantMessage',\n  CancelLastAssistantMessage = 'cancelLastAssistantMessage',\n  EndInterview = 'endInterview',\n}\n\nexport interface CallConfig {\n  systemPrompt: string\n  firstMessage?: string\n  debugLog?: boolean\n  debugSaveSpeech?: boolean\n  disableTTS?: boolean\n  generateAnswer(conversation: Conversation): Promise<string>\n  speech2Text(blob: Blob, prompt?: string): Promise<string>\n  text2Speech(text: string): Promise<ArrayBuffer>\n  onMessage?(message: ConversationMessage): void\n  onEnd?(call: CallSummary): void\n}\n\nexport interface CallSummary {\n  conversation: Conversation\n  duration: number\n}\n\nexport type Conversation = ConversationMessage[]\n\nexport interface ConversationMessage {\n  role: 'system' | 'user' | 'assistant'\n  content: string\n}\n","import WebSocket from 'ws'\n\nexport enum CallErrorCode {\n  BadRequest = 4400,\n  Unauthorized = 4401,\n  NotFound = 4404,\n}\n\nexport class CallError extends Error {\n  code: number\n\n  constructor(code: number, message: string) {\n    super(message)\n    this.code = code\n  }\n}\n\nexport function handleError(socket: WebSocket, error: unknown) {\n  if (error instanceof CallError) {\n    socket.close(error.code, error.message)\n  } else {\n    console.error(error)\n    socket.close(1011)\n  }\n  socket.terminate()\n}\n","import { WebSocket } from 'ws'\nimport { CallError, CallErrorCode } from './errors'\n\nexport async function waitForParams<CallParams>(\n  socket: WebSocket,\n  validate: (params: any) => CallParams\n): Promise<CallParams> {\n  return new Promise<CallParams>((resolve, reject) => {\n    // Handle timeout\n    const timeout = setTimeout(() => {\n      reject(new CallError(CallErrorCode.BadRequest, 'Missing params'))\n    }, 3000)\n\n    const onParams = (payload: string) => {\n      // Clear timeout and listener\n      clearTimeout(timeout)\n      socket.off('message', onParams)\n\n      try {\n        // Parse JSON payload\n        const params = validate(JSON.parse(payload))\n        resolve(params)\n      } catch (error) {\n        reject(new CallError(CallErrorCode.BadRequest, 'Invalid params'))\n      }\n    }\n\n    // Listen for params\n    socket.on('message', onParams)\n  })\n}\n"],"mappings":";AAAA,YAAY,QAAQ;AACpB,YAAY,UAAU;;;ACDf,IAAK,qBAAL,kBAAKA,wBAAL;AACL,EAAAA,oBAAA,mBAAgB;AAChB,EAAAA,oBAAA,kBAAe;AACf,EAAAA,oBAAA,UAAO;AAHG,SAAAA;AAAA,GAAA;AAML,IAAK,qBAAL,kBAAKC,wBAAL;AACL,EAAAA,oBAAA,iBAAc;AACd,EAAAA,oBAAA,sBAAmB;AACnB,EAAAA,oBAAA,gCAA6B;AAC7B,EAAAA,oBAAA,kBAAe;AAJL,SAAAA;AAAA,GAAA;;;ADKL,IAAM,gBAAgB;AAEtB,IAAM,aAAN,MAAiB;AAAA,EAgBtB,YACS,QACA,QACP;AAFO;AACA;AAjBT,SAAQ,YAAY,KAAK,IAAI;AAC7B,SAAQ,YAAY,KAAK,IAAI;AAG7B;AAAA,SAAQ,cAAc;AAGtB;AAAA,SAAQ,aAAa;AAGrB;AAAA,SAAQ,SAAmB,CAAC;AAS1B,SAAK,eAAe,CAAC,EAAE,MAAM,UAAU,SAAS,OAAO,aAAa,CAAC;AACrE,SAAK,IAAI,cAAc;AAKvB,QAAI,OAAO,cAAc;AACvB,WAAK,OAAO,OAAO,YAAY;AAAA,IACjC,OAAO;AACL,WAAK,OACF,eAAe,KAAK,YAAY,EAChC,KAAK,CAAC,WAAW,KAAK,OAAO,MAAM,CAAC,EACpC,MAAM,CAAC,UAAU;AAChB,gBAAQ,MAAM,QAAQ,KAAK;AAC3B,aAAK,OAAO,MAAM;AAAA,MACpB,CAAC;AAAA,IACL;AAGA,SAAK,OAAO,GAAG,SAAS,KAAK,QAAQ,KAAK,IAAI,CAAC;AAC/C,SAAK,OAAO,GAAG,WAAW,KAAK,UAAU,KAAK,IAAI,CAAC;AAAA,EACrD;AAAA;AAAA,EAGO,kBAAkB,cAA4B;AACnD,SAAK,IAAI,oBAAoB;AAC7B,SAAK,eAAe;AAAA,EACtB;AAAA,EAEQ,WAAW,SAA8B;AAC/C,SAAK,aAAa,KAAK,OAAO;AAC9B,SAAK,OAAO;AAAA,MACV,GACE,QAAQ,SAAS,oFAGnB,IAAI,QAAQ,OAAO;AAAA,IACrB;AACA,SAAK,OAAO,YAAY,OAAO;AAAA,EACjC;AAAA,EAEQ,UAAU;AAChB,SAAK,IAAI,mBAAmB;AAC5B,SAAK,cAAc;AACnB,UAAM,WAAW,KAAK,OAAO,KAAK,IAAI,IAAI,KAAK,aAAa,GAAI;AAGhE,SAAK,OAAO,QAAQ;AAAA,MAClB,cAAc,KAAK,aAAa,MAAM,CAAC;AAAA;AAAA,MACvC;AAAA,IACF,CAAC;AAID,SAAK,SAAS;AAEd,SAAK,SAAS;AAAA,EAChB;AAAA,EAEA,MAAc,UAAU,SAAiB;AACvC,QAAI,CAAC,OAAO,SAAS,OAAO,GAAG;AAC7B,cAAQ,KAAK,8BAA8B;AAC3C;AAAA,IACF;AAGA,QAAI,QAAQ,aAAa,IAAI;AAC3B,YAAM,MAAM,QAAQ,SAAS;AAC7B,WAAK,IAAI,YAAY,GAAG,EAAE;AAE1B,UAAI,6CAA0C;AAE5C,aAAK,aAAa;AAElB,aAAK,cAAc;AAAA,MACrB,WAAW,2BAAiC;AAG1C,aAAK,cAAc;AAAA,MACrB,WAAW,2CAAyC;AAElD,aAAK,aAAa;AAClB,cAAM,KAAK,eAAe;AAAA,MAC5B;AAAA,IACF,WAGS,OAAO,SAAS,OAAO,KAAK,KAAK,YAAY;AACpD,WAAK,IAAI,mBAAmB,QAAQ,UAAU,SAAS;AACvD,WAAK,OAAO,KAAK,OAAO;AAAA,IAC1B;AAAA,EACF;AAAA,EAEA,MAAc,iBAAiB;AAE7B,QAAI,KAAK,OAAO,WAAW,EAAG;AAE9B,SAAK,cAAc;AAGnB,UAAM,OAAO,IAAI,KAAK,KAAK,QAAQ,EAAE,MAAM,YAAY,CAAC;AAGxD,SAAK,OAAO,SAAS;AAErB,QAAI;AAEF,UAAI,KAAK,OAAO,iBAAiB;AAC/B,cAAM,WAAgB,UAAK,WAAW,YAAY;AAClD,QAAG,iBAAc,UAAU,OAAO,KAAK,MAAM,KAAK,YAAY,CAAC,CAAC;AAChE;AAAA,MACF;AAGA,YAAM,aAAa,MAAM,KAAK,OAAO;AAAA,QACnC;AAAA,QACA,KAAK,aAAa,KAAK,aAAa,SAAS,CAAC,GAAG;AAAA,MACnD;AACA,UAAI,CAAC,YAAY;AACf,aAAK,IAAI,2BAA2B;AACpC;AAAA,MACF;AAEA,WAAK,IAAI,oBAAoB,UAAU;AAGvC,WAAK,WAAW,EAAE,MAAM,QAAQ,SAAS,WAAW,CAAC;AAErD,UAAI,KAAK,aAAa;AACpB,aAAK,IAAI,qCAAqC;AAC9C;AAAA,MACF;AAGA,YAAM,SAAS,MAAM,KAAK,OAAO,eAAe,KAAK,YAAY;AACjE,UAAI,KAAK,aAAa;AACpB,aAAK,IAAI,iCAAiC;AAC1C;AAAA,MACF;AAEA,YAAM,KAAK,OAAO,MAAM;AAAA,IAC1B,SAAS,OAAO;AACd,cAAQ,MAAM,QAAQ,KAAK;AAC3B,WAAK,OAAO,MAAM;AAAA,IACpB;AAAA,EACF;AAAA;AAAA,EAGA,MAAa,OAAO,SAAiB;AACnC,QAAI,QAAQ;AAGZ,QAAI,QAAQ,SAAS,aAAa,GAAG;AACnC,gBAAU,QAAQ,QAAQ,eAAe,EAAE,EAAE,KAAK;AAClD,cAAQ;AAAA,IACV;AAEA,QAAI,QAAQ,QAAQ;AAElB,WAAK,IAAI,sBAAsB,OAAO;AACtC,WAAK,WAAW,EAAE,MAAM,aAAa,SAAS,QAAQ,CAAC;AAGvD,UAAI,CAAC,KAAK,OAAO,YAAY;AAC3B,cAAM,QAAQ,MAAM,KAAK,OAAO,YAAY,OAAO;AACnD,YAAI,KAAK,aAAa;AAEpB,eAAK,IAAI,iDAAiD;AAC1D,gBAAM,cAAc,KAAK,aAAa,KAAK,aAAa,SAAS,CAAC;AAClE,cAAI,aAAa,SAAS,aAAa;AACrC,iBAAK,aAAa,IAAI;AACtB,iBAAK,OAAO,kEAAkD;AAAA,UAChE;AACA;AAAA,QACF;AAGA,aAAK,IAAI,gBAAgB,MAAM,UAAU,SAAS;AAClD,aAAK,OAAO,KAAK,KAAK;AAAA,MACxB;AAAA,IACF;AAGA,QAAI,OAAO;AACT,WAAK,IAAI,iBAAiB;AAC1B,WAAK,OAAO,sCAAoC;AAAA,IAClD;AAAA,EACF;AAAA,EAEQ,OAAO,SAAgB;AAC7B,QAAI,CAAC,KAAK,OAAO,SAAU;AAC3B,UAAM,MAAM,KAAK,IAAI;AACrB,UAAM,QAAQ,MAAM,KAAK;AACzB,SAAK,YAAY;AACjB,YAAQ,IAAI,WAAW,KAAK,OAAO,GAAG,OAAO;AAAA,EAC/C;AACF;;;AEnOO,IAAK,gBAAL,kBAAKC,mBAAL;AACL,EAAAA,8BAAA,gBAAa,QAAb;AACA,EAAAA,8BAAA,kBAAe,QAAf;AACA,EAAAA,8BAAA,cAAW,QAAX;AAHU,SAAAA;AAAA,GAAA;AAML,IAAM,YAAN,cAAwB,MAAM;AAAA,EAGnC,YAAY,MAAc,SAAiB;AACzC,UAAM,OAAO;AACb,SAAK,OAAO;AAAA,EACd;AACF;AAEO,SAAS,YAAY,QAAmB,OAAgB;AAC7D,MAAI,iBAAiB,WAAW;AAC9B,WAAO,MAAM,MAAM,MAAM,MAAM,OAAO;AAAA,EACxC,OAAO;AACL,YAAQ,MAAM,KAAK;AACnB,WAAO,MAAM,IAAI;AAAA,EACnB;AACA,SAAO,UAAU;AACnB;;;ACtBA,eAAsB,cACpB,QACA,UACqB;AACrB,SAAO,IAAI,QAAoB,CAAC,SAAS,WAAW;AAElD,UAAM,UAAU,WAAW,MAAM;AAC/B,aAAO,IAAI,iCAAoC,gBAAgB,CAAC;AAAA,IAClE,GAAG,GAAI;AAEP,UAAM,WAAW,CAAC,YAAoB;AAEpC,mBAAa,OAAO;AACpB,aAAO,IAAI,WAAW,QAAQ;AAE9B,UAAI;AAEF,cAAM,SAAS,SAAS,KAAK,MAAM,OAAO,CAAC;AAC3C,gBAAQ,MAAM;AAAA,MAChB,SAAS,OAAO;AACd,eAAO,IAAI,iCAAoC,gBAAgB,CAAC;AAAA,MAClE;AAAA,IACF;AAGA,WAAO,GAAG,WAAW,QAAQ;AAAA,EAC/B,CAAC;AACH;","names":["CallClientCommands","CallServerCommands","CallErrorCode"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@micdrop/server",
-  "version": "1.0.1",
+  "version": "1.0.3",
   "description": "A lib for Node.js that helps to use the mic and speaker for voice conversation",
   "author": "Lonestone",
   "license": "MIT",