@sonzai-labs/agents 1.0.13 → 1.1.0

package/dist/index.d.cts

@@ -3,12 +3,15 @@ interface HTTPClientOptions {
     apiKey: string;
     timeout: number;
     maxRetries: number;
+    defaultHeaders?: Record<string, string>;
+    customFetch?: typeof fetch;
 }
 declare class HTTPClient {
     private readonly baseUrl;
     private readonly headers;
     private readonly timeout;
     private readonly maxRetries;
+    private readonly fetchFn;
     constructor(options: HTTPClientOptions);
     request<T = unknown>(method: string, path: string, options?: {
         body?: Record<string, unknown>;
@@ -1297,6 +1300,8 @@ interface SonzaiConfig {
     baseUrl?: string;
     timeout?: number;
     maxRetries?: number;
+    defaultHeaders?: Record<string, string>;
+    customFetch?: typeof fetch;
 }
 interface AgentListOptions {
     pageSize?: number;

package/dist/index.d.ts

@@ -3,12 +3,15 @@ interface HTTPClientOptions {
     apiKey: string;
     timeout: number;
     maxRetries: number;
+    defaultHeaders?: Record<string, string>;
+    customFetch?: typeof fetch;
 }
 declare class HTTPClient {
     private readonly baseUrl;
     private readonly headers;
     private readonly timeout;
     private readonly maxRetries;
+    private readonly fetchFn;
     constructor(options: HTTPClientOptions);
     request<T = unknown>(method: string, path: string, options?: {
         body?: Record<string, unknown>;
@@ -1297,6 +1300,8 @@ interface SonzaiConfig {
     baseUrl?: string;
     timeout?: number;
     maxRetries?: number;
+    defaultHeaders?: Record<string, string>;
+    customFetch?: typeof fetch;
 }
 interface AgentListOptions {
     pageSize?: number;

package/dist/index.js

@@ -70,15 +70,18 @@ var HTTPClient = class {
   headers;
   timeout;
   maxRetries;
+  fetchFn;
   constructor(options) {
     this.baseUrl = options.baseUrl.replace(/\/$/, "");
     this.headers = {
       Authorization: `Bearer ${options.apiKey}`,
       "Content-Type": "application/json",
-      "User-Agent": "sonzai-typescript/1.0.13"
+      "User-Agent": "sonzai-typescript/1.1.0",
+      ...options.defaultHeaders
     };
     this.timeout = options.timeout;
     this.maxRetries = options.maxRetries;
+    this.fetchFn = options.customFetch ?? fetch;
   }
   async request(method, path, options) {
     const url = this.buildUrl(path, options?.params);
@@ -88,7 +91,7 @@ var HTTPClient = class {
       const controller = new AbortController();
       const timer = setTimeout(() => controller.abort(), this.timeout);
       try {
-        const response = await fetch(url, {
+        const response = await this.fetchFn(url, {
           method,
           headers: this.headers,
           body: options?.body ? JSON.stringify(options.body) : void 0,
@@ -153,7 +156,7 @@ var HTTPClient = class {
       blob = new Blob([data], { type: contentType });
     }
     formData.append("file", blob, fileName);
-    const response = await fetch(url, {
+    const response = await this.fetchFn(url, {
       method: "POST",
       headers: {
         Authorization: this.headers["Authorization"],
@@ -173,7 +176,7 @@ var HTTPClient = class {
     const controller = new AbortController();
     const timer = setTimeout(() => controller.abort(), this.timeout * 10);
     try {
-      const response = await fetch(url, {
+      const response = await this.fetchFn(url, {
         method,
         headers: {
           ...this.headers,
@@ -2333,7 +2336,9 @@ var Sonzai = class {
       baseUrl,
       apiKey,
       timeout: config?.timeout ?? DEFAULT_TIMEOUT,
-      maxRetries: config?.maxRetries ?? DEFAULT_MAX_RETRIES
+      maxRetries: config?.maxRetries ?? DEFAULT_MAX_RETRIES,
+      defaultHeaders: config?.defaultHeaders,
+      customFetch: config?.customFetch
     });
     this.agents = new Agents(this.http);
     this.knowledge = new Knowledge(this.http);