@axiom-lattice/client-sdk 4.6.0 → 4.7.0

package/dist/index.mjs

@@ -1856,6 +1856,7 @@ var ScheduledTaskStatus = /* @__PURE__ */ ((ScheduledTaskStatus2) => {
 })(ScheduledTaskStatus || {});
 
 // src/abstract-client.ts
+import { threadAffinityHeaders } from "@axiom-lattice/protocols";
 var AbstractClient = class {
   /**
    * Creates a new AbstractClient instance
@@ -1986,7 +1987,8 @@ var AbstractClient = class {
         return await this.makeRequest(
           `/api/assistants/${params.assistantId}/threads/${params.threadId}/abort`,
           {
-            method: "POST"
+            method: "POST",
+            headers: threadAffinityHeaders(params.threadId)
           }
         );
       }
@@ -3366,6 +3368,9 @@ var AbstractClient = class {
   }
 };
 
+// src/client.ts
+import { threadAffinityHeaders as threadAffinityHeaders2 } from "@axiom-lattice/protocols";
+
 // src/resources-client.ts
 var ResourcesClient = class {
   constructor(baseURL, getHeaders) {
@@ -4288,6 +4293,7 @@ var _Client = class extends AbstractClient {
     const requestHeaders = {
       ...this.headers,
       ...workspaceHeaders,
+      ...this.bodyAffinityHeaders(options?.body),
       ...options?.headers || {}
     };
     if (method === "DELETE" && !options?.body) {
@@ -4325,7 +4331,8 @@ var _Client = class extends AbstractClient {
     const headers = {
       "Content-Type": "application/json",
       Accept: "text/event-stream",
-      ...this.getAllHeaders()
+      ...this.getAllHeaders(),
+      ...threadAffinityHeaders2(options.threadId)
     };
     const controller = new AbortController();
     const { signal } = controller;
@@ -4423,11 +4430,18 @@ var _Client = class extends AbstractClient {
       ...workspaceHeaders
     };
   }
+  bodyAffinityHeaders(body) {
+    if (!body || typeof body !== "object" || body instanceof FormData)
+      return {};
+    const threadId = body.thread_id;
+    return typeof threadId === "string" ? threadAffinityHeaders2(threadId) : {};
+  }
   streamRun(options, onEvent, onComplete, onError) {
     const headers = {
       "Content-Type": "application/json",
       Accept: "text/event-stream",
-      ...this.getAllHeaders()
+      ...this.getAllHeaders(),
+      ...threadAffinityHeaders2(options.threadId)
     };
     const controller = new AbortController();
     const { signal } = controller;
@@ -4537,9 +4551,11 @@ var RuntimeAxiomClient = class extends Client {
   async makeRequest(url, options) {
     await this.primeToken();
     const method = options?.method ?? "GET";
+    const affinity = this.bodyAffinityHeaders(options?.body);
     const headers = {
       "Content-Type": "application/json",
       ...this.identityHeaders(),
+      ...affinity,
       ...options?.headers ?? {}
     };
     const requestOptions = { method, headers };
@@ -4593,6 +4609,7 @@ var RuntimeAxiomClient = class extends Client {
 };
 
 // src/wechat-client.ts
+import { threadAffinityHeaders as threadAffinityHeaders3 } from "@axiom-lattice/protocols";
 var import_encoding = __toESM(require_encoding());
 var WeChatClient = class extends AbstractClient {
   /**
@@ -4690,7 +4707,8 @@ var WeChatClient = class extends AbstractClient {
     const headers = {
       "Content-Type": "application/json",
       Authorization: `Bearer ${this.config.apiKey}`,
-      ...this.config.headers
+      ...this.config.headers,
+      ...threadAffinityHeaders3(options.threadId)
     };
     if (this.tenantId) {
       headers["x-tenant-id"] = this.tenantId;
@@ -4804,7 +4822,8 @@ var WeChatClient = class extends AbstractClient {
     const headers = {
       "Content-Type": "application/json",
       Authorization: `Bearer ${this.config.apiKey}`,
-      ...this.config.headers
+      ...this.config.headers,
+      ...threadAffinityHeaders3(options.threadId)
     };
     if (this.tenantId) {
       headers["x-tenant-id"] = this.tenantId;