@decartai/sdk 0.0.18 → 0.0.19

package/dist/realtime/client.d.ts

@@ -39,12 +39,12 @@ type RealTimeClient = {
     enhance
   }?: {
     enhance?: boolean;
-  }) => void;
+  }) => Promise<void>;
   isConnected: () => boolean;
   getConnectionState: () => "connected" | "connecting" | "disconnected";
   disconnect: () => void;
-  on: (event: keyof Events, listener: (...args: Events[keyof Events][]) => void) => void;
-  off: (event: keyof Events, listener: (...args: Events[keyof Events][]) => void) => void;
+  on: <K extends keyof Events>(event: K, listener: (data: Events[K]) => void) => void;
+  off: <K extends keyof Events>(event: K, listener: (data: Events[K]) => void) => void;
   sessionId: string;
 };
 //#endregion

package/dist/realtime/methods.js

@@ -1,8 +1,9 @@
 import { z } from "zod";
 
 //#region src/realtime/methods.ts
+const PROMPT_TIMEOUT_MS = 15 * 1e3;
 const realtimeMethods = (webrtcManager) => {
-	const setPrompt = (prompt, { enhance } = {}) => {
+	const setPrompt = async (prompt, { enhance } = {}) => {
 		const parsedInput = z.object({
 			prompt: z.string().min(1),
 			enhance: z.boolean().optional().default(true)
@@ -11,11 +12,30 @@ const realtimeMethods = (webrtcManager) => {
 			enhance
 		});
 		if (!parsedInput.success) throw parsedInput.error;
-		webrtcManager.sendMessage({
-			type: "prompt",
-			prompt: parsedInput.data.prompt,
-			enhance_prompt: parsedInput.data.enhance
-		});
+		const emitter = webrtcManager.getWebsocketMessageEmitter();
+		let promptAckListener;
+		let timeoutId;
+		try {
+			const ackPromise = new Promise((resolve, reject) => {
+				promptAckListener = (promptAckMessage) => {
+					if (promptAckMessage.prompt === parsedInput.data.prompt) if (promptAckMessage.success) resolve();
+					else reject(promptAckMessage.error);
+				};
+				emitter.on("promptAck", promptAckListener);
+			});
+			webrtcManager.sendMessage({
+				type: "prompt",
+				prompt: parsedInput.data.prompt,
+				enhance_prompt: parsedInput.data.enhance
+			});
+			const timeoutPromise = new Promise((_, reject) => {
+				timeoutId = setTimeout(() => reject(/* @__PURE__ */ new Error("Prompt timed out")), PROMPT_TIMEOUT_MS);
+			});
+			await Promise.race([ackPromise, timeoutPromise]);
+		} finally {
+			if (promptAckListener) emitter.off("promptAck", promptAckListener);
+			if (timeoutId) clearTimeout(timeoutId);
+		}
 	};
 	return { setPrompt };
 };

package/dist/realtime/webrtc-connection.js

@@ -1,4 +1,5 @@
 import { buildUserAgent } from "../utils/user-agent.js";
+import mitt from "mitt";
 
 //#region src/realtime/webrtc-connection.ts
 const ICE_SERVERS = [{ urls: "stun:stun.l.google.com:19302" }];
@@ -8,6 +9,7 @@ var WebRTCConnection = class {
 	localStream = null;
 	connectionReject = null;
 	state = "disconnected";
+	websocketMessagesEmitter = mitt();
 	constructor(callbacks = {}) {
 		this.callbacks = callbacks;
 	}
@@ -105,6 +107,9 @@ var WebRTCConnection = class {
 					if (turnConfig) await this.setupNewPeerConnection(turnConfig);
 					break;
 				}
+				case "prompt_ack":
+					this.websocketMessagesEmitter.emit("promptAck", msg);
+					break;
 			}
 		} catch (error) {
 			console.error("[WebRTC] Error:", error);

package/dist/realtime/webrtc-manager.js

@@ -55,6 +55,9 @@ var WebRTCManager = class {
 	getConnectionState() {
 		return this.connection.state;
 	}
+	getWebsocketMessageEmitter() {
+		return this.connection.websocketMessagesEmitter;
+	}
 };
 
 //#endregion

package/dist/version.js

@@ -4,7 +4,7 @@
 * Injected at build time from package.json.
 * Falls back to '0.0.0-dev' in development.
 */
-const VERSION = "0.0.18";
+const VERSION = "0.0.19";
 
 //#endregion
 export { VERSION };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@decartai/sdk",
-  "version": "0.0.18",
+  "version": "0.0.19",
   "description": "Decart's JavaScript SDK",
   "type": "module",
   "license": "MIT",