@axiom-lattice/client-sdk 4.6.0 → 4.7.0

package/dist/index.js

@@ -1887,6 +1887,7 @@ var ScheduledTaskStatus = /* @__PURE__ */ ((ScheduledTaskStatus2) => {
 })(ScheduledTaskStatus || {});
 
 // src/abstract-client.ts
+var import_protocols = require("@axiom-lattice/protocols");
 var AbstractClient = class {
   /**
    * Creates a new AbstractClient instance
@@ -2017,7 +2018,8 @@ var AbstractClient = class {
         return await this.makeRequest(
           `/api/assistants/${params.assistantId}/threads/${params.threadId}/abort`,
           {
-            method: "POST"
+            method: "POST",
+            headers: (0, import_protocols.threadAffinityHeaders)(params.threadId)
           }
         );
       }
@@ -3397,6 +3399,9 @@ var AbstractClient = class {
   }
 };
 
+// src/client.ts
+var import_protocols2 = require("@axiom-lattice/protocols");
+
 // src/resources-client.ts
 var ResourcesClient = class {
   constructor(baseURL, getHeaders) {
@@ -4319,6 +4324,7 @@ var _Client = class extends AbstractClient {
     const requestHeaders = {
       ...this.headers,
       ...workspaceHeaders,
+      ...this.bodyAffinityHeaders(options?.body),
       ...options?.headers || {}
     };
     if (method === "DELETE" && !options?.body) {
@@ -4356,7 +4362,8 @@ var _Client = class extends AbstractClient {
     const headers = {
       "Content-Type": "application/json",
       Accept: "text/event-stream",
-      ...this.getAllHeaders()
+      ...this.getAllHeaders(),
+      ...(0, import_protocols2.threadAffinityHeaders)(options.threadId)
     };
     const controller = new AbortController();
     const { signal } = controller;
@@ -4454,11 +4461,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" ? (0, import_protocols2.threadAffinityHeaders)(threadId) : {};
+  }
   streamRun(options, onEvent, onComplete, onError) {
     const headers = {
       "Content-Type": "application/json",
       Accept: "text/event-stream",
-      ...this.getAllHeaders()
+      ...this.getAllHeaders(),
+      ...(0, import_protocols2.threadAffinityHeaders)(options.threadId)
     };
     const controller = new AbortController();
     const { signal } = controller;
@@ -4568,9 +4582,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 };
@@ -4624,6 +4640,7 @@ var RuntimeAxiomClient = class extends Client {
 };
 
 // src/wechat-client.ts
+var import_protocols3 = require("@axiom-lattice/protocols");
 var import_encoding = __toESM(require_encoding());
 var WeChatClient = class extends AbstractClient {
   /**
@@ -4721,7 +4738,8 @@ var WeChatClient = class extends AbstractClient {
     const headers = {
       "Content-Type": "application/json",
       Authorization: `Bearer ${this.config.apiKey}`,
-      ...this.config.headers
+      ...this.config.headers,
+      ...(0, import_protocols3.threadAffinityHeaders)(options.threadId)
     };
     if (this.tenantId) {
       headers["x-tenant-id"] = this.tenantId;
@@ -4835,7 +4853,8 @@ var WeChatClient = class extends AbstractClient {
     const headers = {
       "Content-Type": "application/json",
       Authorization: `Bearer ${this.config.apiKey}`,
-      ...this.config.headers
+      ...this.config.headers,
+      ...(0, import_protocols3.threadAffinityHeaders)(options.threadId)
     };
     if (this.tenantId) {
       headers["x-tenant-id"] = this.tenantId;