@ekodb/ekodb-client 0.24.0 → 0.26.0

package/src/index.ts

@@ -7,7 +7,13 @@ export {
   RateLimitError,
   SchemaCache,
   extractRecordId,
+  HealthStatus,
+  parseHealthStatus,
+  HealthOK,
+  HealthDegraded,
+  HealthUnknown,
 } from "./client";
+export type { HealthState } from "./client";
 export { QueryBuilder, SortOrder } from "./query-builder";
 export { SearchQueryBuilder } from "./search";
 export {
@@ -82,6 +88,8 @@ export type {
   UpdateSessionRequest,
   MergeSessionsRequest,
   ChatModels,
+  ChatProviderState,
+  ChatProviderStatus,
   CompactChatRequest,
   CompactChatResponse,
   EmbedRequest,

package/src/websocket.test.ts

@@ -432,6 +432,96 @@ describe("WebSocketClient", () => {
       client.close();
     });
 
+    // The WebSocket route carries the provider-failure classification like
+    // the SSE route does, in the event's camelCase shape.
+    it("carries the provider failure classification on a chat stream error", async () => {
+      const client = new WebSocketClient(
+        `ws://localhost:${port}/api/ws`,
+        "test-token",
+      );
+
+      const streamPromise = client.chatSend("chat-4", "test");
+
+      await new Promise((r) => wss.once("connection", r));
+      const ws = getLastConnection();
+      await waitForMessage(ws);
+
+      const stream = await streamPromise;
+      const events: any[] = [];
+      stream.on("event", (e) => events.push(e));
+
+      ws.send(
+        JSON.stringify({
+          type: "ChatStreamError",
+          payload: {
+            chat_id: "chat-4",
+            error: "OpenAI API error 429 Too Many Requests",
+            error_kind: "provider_rate_limited",
+            provider: "openai",
+            provider_status: 429,
+            retry_after_secs: 7,
+          },
+        }),
+      );
+
+      await new Promise((r) => stream.on("close", r));
+      expect(events).toEqual([
+        {
+          type: "error",
+          error: "OpenAI API error 429 Too Many Requests",
+          errorKind: "provider_rate_limited",
+          provider: "openai",
+          providerStatus: 429,
+          retryAfterSecs: 7,
+        },
+      ]);
+
+      client.close();
+    });
+
+    // A structured `error` value is still an error, with string text — the
+    // WebSocket route guards the shape exactly as the SSE route does.
+    it("keeps the error text a string when the payload's error is an object", async () => {
+      const client = new WebSocketClient(
+        `ws://localhost:${port}/api/ws`,
+        "test-token",
+      );
+
+      const streamPromise = client.chatSend("chat-5", "test");
+
+      await new Promise((r) => wss.once("connection", r));
+      const ws = getLastConnection();
+      await waitForMessage(ws);
+
+      const stream = await streamPromise;
+      const events: any[] = [];
+      stream.on("event", (e) => events.push(e));
+
+      ws.send(
+        JSON.stringify({
+          type: "ChatStreamError",
+          payload: {
+            chat_id: "chat-5",
+            error: { code: "upstream_down", status: 503 },
+            error_kind: "provider_unavailable",
+            provider: "gemini",
+          },
+        }),
+      );
+
+      await new Promise((r) => stream.on("close", r));
+      expect(events).toEqual([
+        {
+          type: "error",
+          error: "Unknown error",
+          errorKind: "provider_unavailable",
+          provider: "gemini",
+        },
+      ]);
+
+      client.close();
+    });
+
     it("sends options with ChatSend", async () => {
       const client = new WebSocketClient(
         `ws://localhost:${port}/api/ws`,

package/tsconfig.json

@@ -1,7 +1,7 @@
 {
   "compilerOptions": {
     "target": "ES2020",
-    "module": "commonjs",
+    "module": "nodenext",
     "lib": ["ES2020"],
     "declaration": true,
     "outDir": "./dist",
@@ -10,7 +10,7 @@
     "esModuleInterop": true,
     "skipLibCheck": true,
     "forceConsistentCasingInFileNames": true,
-    "moduleResolution": "node",
+    "moduleResolution": "nodenext",
     "resolveJsonModule": true,
     "noUnusedLocals": true,
     "noUnusedParameters": true