@ekodb/ekodb-client 0.13.0 → 0.15.0

package/dist/websocket.test.js

@@ -190,6 +190,30 @@ function waitForMessage(ws) {
             (0, vitest_1.expect)(events[2].executionTimeMs).toBe(150);
             client.close();
         });
+        (0, vitest_1.it)("includes context_window in end event", async () => {
+            const client = new client_1.WebSocketClient(`ws://localhost:${port}/api/ws`, "test-token");
+            const streamPromise = client.chatSend("chat-cw", "test");
+            await new Promise((r) => wss.once("connection", r));
+            const ws = getLastConnection();
+            await waitForMessage(ws);
+            const stream = await streamPromise;
+            const events = [];
+            stream.on("event", (e) => events.push(e));
+            ws.send(JSON.stringify({
+                type: "ChatStreamEnd",
+                payload: {
+                    chat_id: "chat-cw",
+                    message_id: "msg-cw",
+                    execution_time_ms: 100,
+                    context_window: 128000,
+                },
+            }));
+            await new Promise((r) => stream.on("close", r));
+            (0, vitest_1.expect)(events).toHaveLength(1);
+            (0, vitest_1.expect)(events[0].type).toBe("end");
+            (0, vitest_1.expect)(events[0].contextWindow).toBe(128000);
+            client.close();
+        });
         (0, vitest_1.it)("handles chat stream error", async () => {
             const client = new client_1.WebSocketClient(`ws://localhost:${port}/api/ws`, "test-token");
             const streamPromise = client.chatSend("chat-2", "test");
@@ -404,4 +428,69 @@ function waitForMessage(ws) {
             (0, vitest_1.expect)(closed).toBe(true);
         });
     });
+    // --------------------------------------------------------------------------
+    // rawCompletion
+    // --------------------------------------------------------------------------
+    (0, vitest_1.describe)("rawCompletion", () => {
+        (0, vitest_1.it)("sends RawComplete request and returns content", async () => {
+            const client = new client_1.WebSocketClient(`ws://localhost:${port}/api/ws`, "test-token");
+            const resultPromise = client.rawCompletion({
+                system_prompt: "You are helpful.",
+                message: "Say hello.",
+            });
+            await new Promise((r) => wss.once("connection", r));
+            const ws = getLastConnection();
+            const msg = await waitForMessage(ws);
+            (0, vitest_1.expect)(msg.type).toBe("RawComplete");
+            (0, vitest_1.expect)(msg.payload.system_prompt).toBe("You are helpful.");
+            (0, vitest_1.expect)(msg.payload.message).toBe("Say hello.");
+            ws.send(JSON.stringify({
+                type: "Success",
+                payload: {
+                    data: { content: "Hello! How can I help?" },
+                },
+            }));
+            const result = await resultPromise;
+            (0, vitest_1.expect)(result.content).toBe("Hello! How can I help?");
+            client.close();
+        });
+        (0, vitest_1.it)("includes optional fields when provided", async () => {
+            const client = new client_1.WebSocketClient(`ws://localhost:${port}/api/ws`, "test-token");
+            const resultPromise = client.rawCompletion({
+                system_prompt: "System.",
+                message: "User.",
+                provider: "openai",
+                model: "gpt-4o-mini",
+                max_tokens: 512,
+            });
+            await new Promise((r) => wss.once("connection", r));
+            const ws = getLastConnection();
+            const msg = await waitForMessage(ws);
+            (0, vitest_1.expect)(msg.payload.provider).toBe("openai");
+            (0, vitest_1.expect)(msg.payload.model).toBe("gpt-4o-mini");
+            (0, vitest_1.expect)(msg.payload.max_tokens).toBe(512);
+            ws.send(JSON.stringify({
+                type: "Success",
+                payload: { data: { content: "Done." } },
+            }));
+            await resultPromise;
+            client.close();
+        });
+        (0, vitest_1.it)("rejects on error response", async () => {
+            const client = new client_1.WebSocketClient(`ws://localhost:${port}/api/ws`, "test-token");
+            const resultPromise = client.rawCompletion({
+                system_prompt: "System.",
+                message: "User.",
+            });
+            await new Promise((r) => wss.once("connection", r));
+            const ws = getLastConnection();
+            await waitForMessage(ws);
+            ws.send(JSON.stringify({
+                type: "Error",
+                message: "Model not found",
+            }));
+            await (0, vitest_1.expect)(resultPromise).rejects.toThrow("Model not found");
+            client.close();
+        });
+    });
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ekodb/ekodb-client",
-  "version": "0.13.0",
+  "version": "0.15.0",
   "description": "Official TypeScript/JavaScript client for ekoDB",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",