@sonzai-labs/agents 1.0.11 → 1.1.0

package/dist/index.cjs

@@ -72,15 +72,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.11"
+      "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);
@@ -90,7 +93,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,
@@ -144,12 +147,38 @@ var HTTPClient = class {
   async delete(path, params) {
     return this.request("DELETE", path, { params });
   }
+  async uploadFile(path, fileName, fileData, contentType) {
+    const url = this.buildUrl(path);
+    const formData = new FormData();
+    let blob;
+    if (fileData instanceof Blob) {
+      blob = fileData;
+    } else {
+      const data = fileData instanceof ArrayBuffer ? fileData : new Uint8Array(fileData).buffer;
+      blob = new Blob([data], { type: contentType });
+    }
+    formData.append("file", blob, fileName);
+    const response = await this.fetchFn(url, {
+      method: "POST",
+      headers: {
+        Authorization: this.headers["Authorization"],
+        "User-Agent": this.headers["User-Agent"]
+      },
+      body: formData
+    });
+    await this.throwOnError(response);
+    const contentTypeResp = response.headers.get("content-type") ?? "";
+    if (contentTypeResp.includes("application/json")) {
+      return await response.json();
+    }
+    return await response.text();
+  }
   async *streamSSE(method, path, body) {
     const url = this.buildUrl(path);
     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,
@@ -755,9 +784,14 @@ var Inventory = class {
     if (options.item_type) params.item_type = options.item_type;
     if (options.query) params.query = options.query;
     if (options.project_id) params.project_id = options.project_id;
+    if (options.filters) params.filters = options.filters;
+    if (options.sort_by) params.sort_by = options.sort_by;
+    if (options.sort_order) params.sort_order = options.sort_order;
     if (options.aggregations) params.aggregations = options.aggregations;
     if (options.group_by) params.group_by = options.group_by;
     if (options.limit) params.limit = String(options.limit);
+    if (options.offset) params.offset = String(options.offset);
+    if (options.cursor) params.cursor = options.cursor;
     if (options.instanceId) params.instance_id = options.instanceId;
     return this.http.get(
       `/api/v1/agents/${agentId}/users/${encodeURIComponent(userId)}/inventory`,
@@ -1181,8 +1215,7 @@ var Agents = class {
       body.tool_capabilities = options.toolCapabilities;
     if (options.language) body.language = options.language;
     if (options.seedMemories) body.seed_memories = options.seedMemories;
-    if (options.loreContext)
-      body.lore_generation_context = options.loreContext;
+    if (options.loreContext) body.lore_generation_context = options.loreContext;
     if (options.generateOriginStory != null)
       body.generate_origin_story = options.generateOriginStory;
     if (options.generatePersonalizedMemories != null)
@@ -1344,10 +1377,7 @@ var Agents = class {
     if (options.config) body.config = options.config;
     if (options.model) body.model = options.model;
     if (options.configOverride) body.config_override = options.configOverride;
-    return this.http.post(
-      `/api/v1/agents/${agentId}/simulate`,
-      body
-    );
+    return this.http.post(`/api/v1/agents/${agentId}/simulate`, body);
   }
   /** Run simulation + evaluation combined (two-step: POST to start, then stream SSE). */
   async *runEval(agentId, options) {
@@ -1373,10 +1403,7 @@ var Agents = class {
     if (options.adaptationTemplateId)
       body.adaptation_template_id = options.adaptationTemplateId;
     if (options.qualityOnly != null) body.quality_only = options.qualityOnly;
-    return this.http.post(
-      `/api/v1/agents/${agentId}/run-eval`,
-      body
-    );
+    return this.http.post(`/api/v1/agents/${agentId}/run-eval`, body);
   }
   /** Re-evaluate an existing run (two-step: POST to start, then stream SSE). */
   async *evalOnly(agentId, options) {
@@ -1397,10 +1424,7 @@ var Agents = class {
     if (options.adaptationTemplateId)
       body.adaptation_template_id = options.adaptationTemplateId;
     if (options.qualityOnly != null) body.quality_only = options.qualityOnly;
-    return this.http.post(
-      `/api/v1/agents/${agentId}/eval-only`,
-      body
-    );
+    return this.http.post(`/api/v1/agents/${agentId}/eval-only`, body);
   }
   // -- Context Engine convenience accessors --
   async getMood(agentId, options = {}) {
@@ -1444,10 +1468,7 @@ var Agents = class {
     if (options.description) body.description = options.description;
     if (options.displayName) body.display_name = options.displayName;
     if (options.strength != null) body.strength = options.strength;
-    return this.http.post(
-      `/api/v1/agents/${agentId}/habits`,
-      body
-    );
+    return this.http.post(`/api/v1/agents/${agentId}/habits`, body);
   }
   /** Update an existing habit by name. */
   async updateHabit(agentId, habitName, options) {
@@ -1487,10 +1508,7 @@ var Agents = class {
     if (options.type) body.type = options.type;
     if (options.priority != null) body.priority = options.priority;
     if (options.relatedTraits) body.related_traits = options.relatedTraits;
-    return this.http.post(
-      `/api/v1/agents/${agentId}/goals`,
-      body
-    );
+    return this.http.post(`/api/v1/agents/${agentId}/goals`, body);
   }
   /** Update an existing goal. Set userId for per-user goals. */
   async updateGoal(agentId, goalId, options) {
@@ -1510,10 +1528,7 @@ var Agents = class {
   async deleteGoal(agentId, goalId, options = {}) {
     const params = {};
     if (options.userId) params.user_id = options.userId;
-    await this.http.delete(
-      `/api/v1/agents/${agentId}/goals/${goalId}`,
-      params
-    );
+    await this.http.delete(`/api/v1/agents/${agentId}/goals/${goalId}`, params);
   }
   async getInterests(agentId, options = {}) {
     return this.http.get(`/api/v1/agents/${agentId}/interests`, {
@@ -1530,6 +1545,12 @@ var Agents = class {
   async getUsers(agentId) {
     return this.http.get(`/api/v1/agents/${agentId}/users`);
   }
+  async respondToToolCall(agentId, options) {
+    return this.http.post(
+      `/api/v1/agents/${agentId}/tools/respond`,
+      options
+    );
+  }
   /** Get constellation data for an agent. */
   async getConstellation(agentId, options = {}) {
     return this.http.get(`/api/v1/agents/${agentId}/constellation`, {
@@ -1589,38 +1610,59 @@ var Agents = class {
   // -- Agent Status --
   /** Set the active status of an agent. */
   async setStatus(agentId, options) {
-    return this.http.patch(`/api/v1/agents/${agentId}/status`, options);
+    return this.http.patch(
+      `/api/v1/agents/${agentId}/status`,
+      options
+    );
   }
   // -- Project Association --
   /** Update an agent's project association. */
   async updateProject(agentId, options) {
-    return this.http.patch(`/api/v1/agents/${agentId}/project`, options);
+    return this.http.patch(
+      `/api/v1/agents/${agentId}/project`,
+      options
+    );
   }
   // -- Capabilities --
   /** Get an agent's capabilities. */
   async getCapabilities(agentId) {
-    return this.http.get(`/api/v1/agents/${agentId}/capabilities`);
+    return this.http.get(
+      `/api/v1/agents/${agentId}/capabilities`
+    );
   }
   /** Update an agent's capabilities. */
   async updateCapabilities(agentId, options) {
-    return this.http.patch(`/api/v1/agents/${agentId}/capabilities`, options);
+    return this.http.patch(
+      `/api/v1/agents/${agentId}/capabilities`,
+      options
+    );
   }
   // -- Custom Tools --
   /** List custom tools for an agent. */
   async listCustomTools(agentId) {
-    return this.http.get(`/api/v1/agents/${agentId}/tools`);
+    return this.http.get(
+      `/api/v1/agents/${agentId}/tools`
+    );
   }
   /** Create a custom tool for an agent. */
   async createCustomTool(agentId, options) {
-    return this.http.post(`/api/v1/agents/${agentId}/tools`, options);
+    return this.http.post(
+      `/api/v1/agents/${agentId}/tools`,
+      options
+    );
   }
   /** Update a custom tool for an agent. */
   async updateCustomTool(agentId, toolName, options) {
-    return this.http.put(`/api/v1/agents/${agentId}/tools/${toolName}`, options);
+    return this.http.put(
+      `/api/v1/agents/${agentId}/tools/${toolName}`,
+      options
+    );
   }
   /** Delete a custom tool from an agent. */
   async deleteCustomTool(agentId, toolName) {
-    return this.http.delete(`/api/v1/agents/${agentId}/tools/${toolName}`);
+    return this.http.delete(
+      `/api/v1/agents/${agentId}/tools/${toolName}`
+    );
   }
   // -- Avatar Generation --
   /** Trigger avatar generation for an agent. */
@@ -1648,8 +1690,12 @@ var Agents = class {
     if (options.instanceId) body.instanceId = options.instanceId;
     if (options.provider) body.provider = options.provider;
     if (options.model) body.model = options.model;
-    if (options.includeExtractions) body.include_extractions = options.includeExtractions;
-    return this.http.post(`/api/v1/agents/${agentId}/process`, body);
+    if (options.includeExtractions)
+      body.include_extractions = options.includeExtractions;
+    return this.http.post(
+      `/api/v1/agents/${agentId}/process`,
+      body
+    );
   }
   /** Get available LLM providers and models for the /process endpoint. */
   async getModels(agentId) {
@@ -1678,14 +1724,20 @@ var Agents = class {
   // -- Consolidation --
   /** Trigger memory consolidation for an agent. */
   async consolidate(agentId, options) {
-    return this.http.post(`/api/v1/agents/${agentId}/consolidate`, options ?? {});
+    return this.http.post(
+      `/api/v1/agents/${agentId}/consolidate`,
+      options ?? {}
+    );
   }
   /** Get memory summaries for an agent. */
   async getSummaries(agentId, options) {
     const params = {};
     if (options?.period) params.period = options.period;
     if (options?.limit) params.limit = String(options.limit);
-    return this.http.get(`/api/v1/agents/${agentId}/summaries`, params);
+    return this.http.get(
+      `/api/v1/agents/${agentId}/summaries`,
+      params
+    );
   }
   // -- Time Machine --
   /** Get a point-in-time snapshot of an agent's personality and mood. */
@@ -1693,7 +1745,10 @@ var Agents = class {
     const params = { at: options.at };
     if (options.userId) params.user_id = options.userId;
     if (options.instanceId) params.instance_id = options.instanceId;
-    return this.http.get(`/api/v1/agents/${agentId}/timemachine`, params);
+    return this.http.get(
+      `/api/v1/agents/${agentId}/timemachine`,
+      params
+    );
   }
   // -- Knowledge Search (tool endpoint) --
   /** Search the knowledge base for an agent. */
@@ -1876,6 +1931,14 @@ var Knowledge = class {
       `/api/v1/projects/${projectId}/knowledge/documents/${docId}`
     );
   }
+  async uploadDocument(projectId, fileName, fileData, contentType = "application/octet-stream") {
+    return this.http.uploadFile(
+      `/api/v1/projects/${projectId}/knowledge/documents`,
+      fileName,
+      fileData,
+      contentType
+    );
+  }
   // -- Facts / Graph --
   /** Insert entities and relationships into the knowledge graph. */
   async insertFacts(projectId, options) {
@@ -1889,6 +1952,14 @@ var Knowledge = class {
     const params = {};
     if (options.type) params.type = options.type;
     if (options.limit) params.limit = String(options.limit);
+    if (options.offset) params.offset = String(options.offset);
+    if (options.sort_by) params.sort_by = options.sort_by;
+    if (options.sort_order) params.sort_order = options.sort_order;
+    if (options.properties) {
+      for (const [k, v] of Object.entries(options.properties)) {
+        params[`properties.${k}`] = v;
+      }
+    }
     return this.http.get(
       `/api/v1/projects/${projectId}/knowledge/nodes`,
       params
@@ -1926,6 +1997,7 @@ var Knowledge = class {
     if (options.includeHistory) params.history = "true";
     if (options.entityTypes) params.type = options.entityTypes;
     if (options.filters) params.filters = options.filters;
+    if (options.hops) params.hops = String(options.hops);
     return this.http.get(
       `/api/v1/projects/${projectId}/knowledge/search`,
       params
@@ -2266,7 +2338,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);