@ragbits/api-client 0.0.3 → 1.4.0-dev.202512021005

package/dist/index.cjs

@@ -20,42 +20,40 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
-  ChatResponseType: () => ChatResponseType,
+  AuthType: () => AuthType,
   FeedbackType: () => FeedbackType,
   LiveUpdateType: () => LiveUpdateType,
   MessageRole: () => MessageRole,
-  RagbitsClient: () => RagbitsClient
+  RagbitsClient: () => RagbitsClient,
+  TaskStatus: () => TaskStatus
 });
 module.exports = __toCommonJS(index_exports);
 
-// src/types.ts
-var MessageRole = /* @__PURE__ */ ((MessageRole2) => {
-  MessageRole2["USER"] = "user";
-  MessageRole2["ASSISTANT"] = "assistant";
-  MessageRole2["SYSTEM"] = "system";
-  return MessageRole2;
-})(MessageRole || {});
-var ChatResponseType = /* @__PURE__ */ ((ChatResponseType2) => {
-  ChatResponseType2["MESSAGE"] = "message";
-  ChatResponseType2["REFERENCE"] = "reference";
-  ChatResponseType2["STATE_UPDATE"] = "state_update";
-  ChatResponseType2["TEXT"] = "text";
-  ChatResponseType2["MESSAGE_ID"] = "message_id";
-  ChatResponseType2["CONVERSATION_ID"] = "conversation_id";
-  ChatResponseType2["LIVE_UPDATE"] = "live_update";
-  ChatResponseType2["FOLLOWUP_MESSAGES"] = "followup_messages";
-  return ChatResponseType2;
-})(ChatResponseType || {});
-var FeedbackType = /* @__PURE__ */ ((FeedbackType2) => {
-  FeedbackType2["LIKE"] = "like";
-  FeedbackType2["DISLIKE"] = "dislike";
-  return FeedbackType2;
-})(FeedbackType || {});
-var LiveUpdateType = /* @__PURE__ */ ((LiveUpdateType2) => {
-  LiveUpdateType2["START"] = "START";
-  LiveUpdateType2["FINISH"] = "FINISH";
-  return LiveUpdateType2;
-})(LiveUpdateType || {});
+// src/autogen.types.ts
+var FeedbackType = {
+  Like: "like",
+  Dislike: "dislike"
+};
+var LiveUpdateType = {
+  Start: "START",
+  Finish: "FINISH"
+};
+var MessageRole = {
+  User: "user",
+  Assistant: "assistant",
+  System: "system"
+};
+var TaskStatus = {
+  Pending: "pending",
+  InProgress: "in_progress",
+  Completed: "completed",
+  Failed: "failed",
+  Cancelled: "cancelled",
+  Retrying: "retrying"
+};
+var AuthType = {
+  Credentials: "credentials"
+};
 
 // src/index.ts
 var RagbitsClient = class {
@@ -63,7 +61,15 @@ var RagbitsClient = class {
    * @param config - Configuration object
    */
   constructor(config = {}) {
-    this.baseUrl = config.baseUrl || "http://127.0.0.1:8000";
+    this.chunkQueue = /* @__PURE__ */ new Map();
+    this.baseUrl = config.baseUrl ?? "";
+    this.auth = config.auth;
+    if (this.baseUrl.endsWith("/")) {
+      this.baseUrl = this.baseUrl.slice(0, -1);
+    }
+    if (!this.baseUrl) {
+      return;
+    }
     try {
       new URL(this.baseUrl);
     } catch {
@@ -71,9 +77,6 @@ var RagbitsClient = class {
         `Invalid base URL: ${this.baseUrl}. Please provide a valid URL.`
       );
     }
-    if (this.baseUrl.endsWith("/")) {
-      this.baseUrl = this.baseUrl.slice(0, -1);
-    }
   }
   /**
    * Get the base URL used by this client
@@ -93,12 +96,20 @@ var RagbitsClient = class {
    * @private
    */
   async _makeRequest(url, options = {}) {
-    const defaultOptions = {
-      headers: {
-        "Content-Type": "application/json"
-      }
+    const defaultHeaders = {
+      "Content-Type": "application/json"
+    };
+    const headers = {
+      ...defaultHeaders,
+      ...this.normalizeHeaders(options.headers)
     };
-    const response = await fetch(url, { ...defaultOptions, ...options });
+    if (this.auth?.getToken) {
+      headers["Authorization"] = `Bearer ${this.auth.getToken()}`;
+    }
+    const response = await fetch(url, { ...options, headers });
+    if (response.status === 401) {
+      this.auth?.onUnauthorized?.();
+    }
     if (!response.ok) {
       throw new Error(`HTTP error! status: ${response.status}`);
     }
@@ -106,7 +117,7 @@ var RagbitsClient = class {
   }
   /**
    * Method to make API requests to known endpoints only
-   * @param endpoint - API endpoint path (must be predefined)
+   * @param endpoint - API endpoint path
    * @param options - Typed request options for the specific endpoint
    */
   async makeRequest(endpoint, options) {
@@ -126,25 +137,26 @@ var RagbitsClient = class {
       requestOptions.body = typeof body === "string" ? body : JSON.stringify(body);
     }
     const response = await this._makeRequest(
-      this._buildApiUrl(endpoint),
+      this._buildApiUrl(endpoint.toString()),
       requestOptions
     );
     return response.json();
   }
   /**
    * Method for streaming requests to known endpoints only
-   * @param endpoint - Streaming endpoint path (must be predefined)
+   * @param endpoint - Streaming endpoint path
    * @param data - Request data
    * @param callbacks - Stream callbacks
    * @param signal - Optional AbortSignal for cancelling the request
    */
-  makeStreamRequest(endpoint, data, callbacks, signal) {
+  makeStreamRequest(endpoint, data, callbacks, signal, customHeaders) {
     let isCancelled = false;
     const processStream = async (response) => {
       const reader = response.body?.pipeThrough(new TextDecoderStream()).getReader();
       if (!reader) {
         throw new Error("Response body is null");
       }
+      let buffer = "";
       while (!isCancelled && !signal?.aborted) {
         try {
           const { value, done } = await reader.read();
@@ -152,7 +164,9 @@ var RagbitsClient = class {
             callbacks.onClose?.();
             break;
           }
-          const lines = value.split("\n");
+          buffer += value;
+          const lines = buffer.split("\n");
+          buffer = lines.pop() ?? "";
           for (const line of lines) {
             if (!line.startsWith("data: ")) continue;
             try {
@@ -160,7 +174,14 @@ var RagbitsClient = class {
               const parsedData = JSON.parse(
                 jsonString
               );
+              if (parsedData.type === "chunked_content") {
+                this.handleChunkedContent(parsedData, callbacks);
+                continue;
+              }
               await callbacks.onMessage(parsedData);
+              await new Promise(
+                (resolve) => setTimeout(resolve, 0)
+              );
             } catch (parseError) {
               console.error("Error parsing JSON:", parseError);
               await callbacks.onError(
@@ -169,6 +190,9 @@ var RagbitsClient = class {
             }
           }
         } catch (streamError) {
+          if (signal?.aborted) {
+            return;
+          }
           console.error("Stream error:", streamError);
           await callbacks.onError(new Error("Error reading stream"));
           break;
@@ -177,15 +201,29 @@ var RagbitsClient = class {
     };
     const startStream = async () => {
       try {
-        const response = await fetch(this._buildApiUrl(endpoint), {
-          method: "POST",
-          headers: {
-            "Content-Type": "application/json",
-            Accept: "text/event-stream"
-          },
-          body: JSON.stringify(data),
-          signal
-        });
+        const defaultHeaders = {
+          "Content-Type": "application/json",
+          Accept: "text/event-stream"
+        };
+        const headers = {
+          ...defaultHeaders,
+          ...customHeaders
+        };
+        if (this.auth?.getToken) {
+          headers["Authorization"] = `Bearer ${this.auth.getToken()}`;
+        }
+        const response = await fetch(
+          this._buildApiUrl(endpoint.toString()),
+          {
+            method: "POST",
+            headers,
+            body: JSON.stringify(data),
+            signal
+          }
+        );
+        if (response.status === 401) {
+          this.auth?.onUnauthorized?.();
+        }
         if (!response.ok) {
           throw new Error(`HTTP error! status: ${response.status}`);
         }
@@ -209,12 +247,68 @@ var RagbitsClient = class {
       isCancelled = true;
     };
   }
+  normalizeHeaders(init) {
+    if (!init) return {};
+    if (init instanceof Headers) {
+      return Object.fromEntries(init.entries());
+    }
+    if (Array.isArray(init)) {
+      return Object.fromEntries(init);
+    }
+    return init;
+  }
+  async handleChunkedContent(data, callbacks) {
+    const response = data;
+    const content = response.content;
+    const {
+      content_type: contentType,
+      id,
+      chunk_index: chunkIndex,
+      total_chunks: totalChunks,
+      mime_type: mimeType,
+      data: chunkData
+    } = content;
+    if (!this.chunkQueue.has(id)) {
+      this.chunkQueue.set(id, {
+        chunks: /* @__PURE__ */ new Map(),
+        totalChunks,
+        mimeType
+      });
+    }
+    const imageInfo = this.chunkQueue.get(id);
+    imageInfo.chunks.set(chunkIndex, chunkData);
+    if (imageInfo.chunks.size !== totalChunks) return;
+    const sortedChunks = Array.from(
+      { length: totalChunks },
+      (_, i) => imageInfo.chunks.get(i)
+    );
+    const completeBase64 = sortedChunks.join("");
+    try {
+      atob(completeBase64);
+    } catch (e) {
+      this.chunkQueue.delete(id);
+      console.error("\u274C Invalid base64 data: ", e);
+      await callbacks.onError(new Error("Error reading stream"));
+    }
+    if (contentType === "image") {
+      const completeImageResponse = {
+        type: "image",
+        content: {
+          id,
+          url: `${imageInfo.mimeType},${completeBase64}`
+        }
+      };
+      await callbacks.onMessage(completeImageResponse);
+    }
+    this.chunkQueue.delete(id);
+  }
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  ChatResponseType,
+  AuthType,
   FeedbackType,
   LiveUpdateType,
   MessageRole,
-  RagbitsClient
+  RagbitsClient,
+  TaskStatus
 });

package/dist/index.d.ts

@@ -1,9 +1,11 @@
-import type { ClientConfig, StreamCallbacks, ApiEndpointPath, ApiEndpointResponse, TypedApiRequestOptions, StreamingEndpointPath, StreamingEndpointRequest, StreamingEndpointStream } from './types';
+import type { ClientConfig, StreamCallbacks, BaseApiEndpoints, EndpointDefinition, EndpointResponse, RequestOptions, BaseStreamingEndpoints, EndpointRequest } from './types';
 /**
  * Client for communicating with the Ragbits API
  */
 export declare class RagbitsClient {
     private readonly baseUrl;
+    private readonly auth;
+    private chunkQueue;
     /**
      * @param config - Configuration object
      */
@@ -24,17 +26,24 @@ export declare class RagbitsClient {
     private _makeRequest;
     /**
      * Method to make API requests to known endpoints only
-     * @param endpoint - API endpoint path (must be predefined)
+     * @param endpoint - API endpoint path
      * @param options - Typed request options for the specific endpoint
      */
-    makeRequest<T extends ApiEndpointPath>(endpoint: T, options?: TypedApiRequestOptions<T>): Promise<ApiEndpointResponse<T>>;
+    makeRequest<Endpoints extends {
+        [K in keyof Endpoints]: EndpointDefinition;
+    } = BaseApiEndpoints, URL extends keyof Endpoints = keyof Endpoints>(endpoint: URL, options?: RequestOptions<URL, Endpoints>): Promise<EndpointResponse<URL, Endpoints>>;
     /**
      * Method for streaming requests to known endpoints only
-     * @param endpoint - Streaming endpoint path (must be predefined)
+     * @param endpoint - Streaming endpoint path
      * @param data - Request data
      * @param callbacks - Stream callbacks
      * @param signal - Optional AbortSignal for cancelling the request
      */
-    makeStreamRequest<T extends StreamingEndpointPath>(endpoint: T, data: StreamingEndpointRequest<T>, callbacks: StreamCallbacks<StreamingEndpointStream<T>>, signal?: AbortSignal): () => void;
+    makeStreamRequest<Endpoints extends {
+        [K in keyof Endpoints]: EndpointDefinition;
+    } = BaseStreamingEndpoints, URL extends keyof Endpoints = keyof Endpoints>(endpoint: URL, data: EndpointRequest<URL, Endpoints>, callbacks: StreamCallbacks<EndpointResponse<URL, Endpoints>>, signal?: AbortSignal, customHeaders?: Record<string, string>): () => void;
+    private normalizeHeaders;
+    private handleChunkedContent;
 }
 export * from './types';
+export * from './autogen.types';

package/dist/index.js

@@ -1,31 +1,28 @@
-// src/types.ts
-var MessageRole = /* @__PURE__ */ ((MessageRole2) => {
-  MessageRole2["USER"] = "user";
-  MessageRole2["ASSISTANT"] = "assistant";
-  MessageRole2["SYSTEM"] = "system";
-  return MessageRole2;
-})(MessageRole || {});
-var ChatResponseType = /* @__PURE__ */ ((ChatResponseType2) => {
-  ChatResponseType2["MESSAGE"] = "message";
-  ChatResponseType2["REFERENCE"] = "reference";
-  ChatResponseType2["STATE_UPDATE"] = "state_update";
-  ChatResponseType2["TEXT"] = "text";
-  ChatResponseType2["MESSAGE_ID"] = "message_id";
-  ChatResponseType2["CONVERSATION_ID"] = "conversation_id";
-  ChatResponseType2["LIVE_UPDATE"] = "live_update";
-  ChatResponseType2["FOLLOWUP_MESSAGES"] = "followup_messages";
-  return ChatResponseType2;
-})(ChatResponseType || {});
-var FeedbackType = /* @__PURE__ */ ((FeedbackType2) => {
-  FeedbackType2["LIKE"] = "like";
-  FeedbackType2["DISLIKE"] = "dislike";
-  return FeedbackType2;
-})(FeedbackType || {});
-var LiveUpdateType = /* @__PURE__ */ ((LiveUpdateType2) => {
-  LiveUpdateType2["START"] = "START";
-  LiveUpdateType2["FINISH"] = "FINISH";
-  return LiveUpdateType2;
-})(LiveUpdateType || {});
+// src/autogen.types.ts
+var FeedbackType = {
+  Like: "like",
+  Dislike: "dislike"
+};
+var LiveUpdateType = {
+  Start: "START",
+  Finish: "FINISH"
+};
+var MessageRole = {
+  User: "user",
+  Assistant: "assistant",
+  System: "system"
+};
+var TaskStatus = {
+  Pending: "pending",
+  InProgress: "in_progress",
+  Completed: "completed",
+  Failed: "failed",
+  Cancelled: "cancelled",
+  Retrying: "retrying"
+};
+var AuthType = {
+  Credentials: "credentials"
+};
 
 // src/index.ts
 var RagbitsClient = class {
@@ -33,7 +30,15 @@ var RagbitsClient = class {
    * @param config - Configuration object
    */
   constructor(config = {}) {
-    this.baseUrl = config.baseUrl || "http://127.0.0.1:8000";
+    this.chunkQueue = /* @__PURE__ */ new Map();
+    this.baseUrl = config.baseUrl ?? "";
+    this.auth = config.auth;
+    if (this.baseUrl.endsWith("/")) {
+      this.baseUrl = this.baseUrl.slice(0, -1);
+    }
+    if (!this.baseUrl) {
+      return;
+    }
     try {
       new URL(this.baseUrl);
     } catch {
@@ -41,9 +46,6 @@ var RagbitsClient = class {
         `Invalid base URL: ${this.baseUrl}. Please provide a valid URL.`
       );
     }
-    if (this.baseUrl.endsWith("/")) {
-      this.baseUrl = this.baseUrl.slice(0, -1);
-    }
   }
   /**
    * Get the base URL used by this client
@@ -63,12 +65,20 @@ var RagbitsClient = class {
    * @private
    */
   async _makeRequest(url, options = {}) {
-    const defaultOptions = {
-      headers: {
-        "Content-Type": "application/json"
-      }
+    const defaultHeaders = {
+      "Content-Type": "application/json"
+    };
+    const headers = {
+      ...defaultHeaders,
+      ...this.normalizeHeaders(options.headers)
     };
-    const response = await fetch(url, { ...defaultOptions, ...options });
+    if (this.auth?.getToken) {
+      headers["Authorization"] = `Bearer ${this.auth.getToken()}`;
+    }
+    const response = await fetch(url, { ...options, headers });
+    if (response.status === 401) {
+      this.auth?.onUnauthorized?.();
+    }
     if (!response.ok) {
       throw new Error(`HTTP error! status: ${response.status}`);
     }
@@ -76,7 +86,7 @@ var RagbitsClient = class {
   }
   /**
    * Method to make API requests to known endpoints only
-   * @param endpoint - API endpoint path (must be predefined)
+   * @param endpoint - API endpoint path
    * @param options - Typed request options for the specific endpoint
    */
   async makeRequest(endpoint, options) {
@@ -96,25 +106,26 @@ var RagbitsClient = class {
       requestOptions.body = typeof body === "string" ? body : JSON.stringify(body);
     }
     const response = await this._makeRequest(
-      this._buildApiUrl(endpoint),
+      this._buildApiUrl(endpoint.toString()),
       requestOptions
     );
     return response.json();
   }
   /**
    * Method for streaming requests to known endpoints only
-   * @param endpoint - Streaming endpoint path (must be predefined)
+   * @param endpoint - Streaming endpoint path
    * @param data - Request data
    * @param callbacks - Stream callbacks
    * @param signal - Optional AbortSignal for cancelling the request
    */
-  makeStreamRequest(endpoint, data, callbacks, signal) {
+  makeStreamRequest(endpoint, data, callbacks, signal, customHeaders) {
     let isCancelled = false;
     const processStream = async (response) => {
       const reader = response.body?.pipeThrough(new TextDecoderStream()).getReader();
       if (!reader) {
         throw new Error("Response body is null");
       }
+      let buffer = "";
       while (!isCancelled && !signal?.aborted) {
         try {
           const { value, done } = await reader.read();
@@ -122,7 +133,9 @@ var RagbitsClient = class {
             callbacks.onClose?.();
             break;
           }
-          const lines = value.split("\n");
+          buffer += value;
+          const lines = buffer.split("\n");
+          buffer = lines.pop() ?? "";
           for (const line of lines) {
             if (!line.startsWith("data: ")) continue;
             try {
@@ -130,7 +143,14 @@ var RagbitsClient = class {
               const parsedData = JSON.parse(
                 jsonString
               );
+              if (parsedData.type === "chunked_content") {
+                this.handleChunkedContent(parsedData, callbacks);
+                continue;
+              }
               await callbacks.onMessage(parsedData);
+              await new Promise(
+                (resolve) => setTimeout(resolve, 0)
+              );
             } catch (parseError) {
               console.error("Error parsing JSON:", parseError);
               await callbacks.onError(
@@ -139,6 +159,9 @@ var RagbitsClient = class {
             }
           }
         } catch (streamError) {
+          if (signal?.aborted) {
+            return;
+          }
           console.error("Stream error:", streamError);
           await callbacks.onError(new Error("Error reading stream"));
           break;
@@ -147,15 +170,29 @@ var RagbitsClient = class {
     };
     const startStream = async () => {
       try {
-        const response = await fetch(this._buildApiUrl(endpoint), {
-          method: "POST",
-          headers: {
-            "Content-Type": "application/json",
-            Accept: "text/event-stream"
-          },
-          body: JSON.stringify(data),
-          signal
-        });
+        const defaultHeaders = {
+          "Content-Type": "application/json",
+          Accept: "text/event-stream"
+        };
+        const headers = {
+          ...defaultHeaders,
+          ...customHeaders
+        };
+        if (this.auth?.getToken) {
+          headers["Authorization"] = `Bearer ${this.auth.getToken()}`;
+        }
+        const response = await fetch(
+          this._buildApiUrl(endpoint.toString()),
+          {
+            method: "POST",
+            headers,
+            body: JSON.stringify(data),
+            signal
+          }
+        );
+        if (response.status === 401) {
+          this.auth?.onUnauthorized?.();
+        }
         if (!response.ok) {
           throw new Error(`HTTP error! status: ${response.status}`);
         }
@@ -179,11 +216,67 @@ var RagbitsClient = class {
       isCancelled = true;
     };
   }
+  normalizeHeaders(init) {
+    if (!init) return {};
+    if (init instanceof Headers) {
+      return Object.fromEntries(init.entries());
+    }
+    if (Array.isArray(init)) {
+      return Object.fromEntries(init);
+    }
+    return init;
+  }
+  async handleChunkedContent(data, callbacks) {
+    const response = data;
+    const content = response.content;
+    const {
+      content_type: contentType,
+      id,
+      chunk_index: chunkIndex,
+      total_chunks: totalChunks,
+      mime_type: mimeType,
+      data: chunkData
+    } = content;
+    if (!this.chunkQueue.has(id)) {
+      this.chunkQueue.set(id, {
+        chunks: /* @__PURE__ */ new Map(),
+        totalChunks,
+        mimeType
+      });
+    }
+    const imageInfo = this.chunkQueue.get(id);
+    imageInfo.chunks.set(chunkIndex, chunkData);
+    if (imageInfo.chunks.size !== totalChunks) return;
+    const sortedChunks = Array.from(
+      { length: totalChunks },
+      (_, i) => imageInfo.chunks.get(i)
+    );
+    const completeBase64 = sortedChunks.join("");
+    try {
+      atob(completeBase64);
+    } catch (e) {
+      this.chunkQueue.delete(id);
+      console.error("\u274C Invalid base64 data: ", e);
+      await callbacks.onError(new Error("Error reading stream"));
+    }
+    if (contentType === "image") {
+      const completeImageResponse = {
+        type: "image",
+        content: {
+          id,
+          url: `${imageInfo.mimeType},${completeBase64}`
+        }
+      };
+      await callbacks.onMessage(completeImageResponse);
+    }
+    this.chunkQueue.delete(id);
+  }
 };
 export {
-  ChatResponseType,
+  AuthType,
   FeedbackType,
   LiveUpdateType,
   MessageRole,
-  RagbitsClient
+  RagbitsClient,
+  TaskStatus
 };