Adds batch tracing support check

langchain-ai · Feb 8, 2024 · 72055c6 · 72055c6
1 parent d8a68da
commit 72055c6
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/js/src/client.ts b/js/src/client.ts
@@ -647,7 +647,22 @@ export class Client {
         signal: AbortSignal.timeout(this.timeout_ms),
       }
     );
-    await raiseForStatus(response, "batch create run");
+    if (response.status === 404) {
+      this.autoBatchTracing = false;
+      for (const preparedCreateParam of body.post) {
+        await this.createRun(preparedCreateParam as CreateRunParams);
+      }
+      for (const preparedUpdateParam of body.patch) {
+        if (preparedUpdateParam.id !== undefined) {
+          await this.updateRun(
+            preparedUpdateParam.id,
+            preparedUpdateParam as UpdateRunParams
+          );
+        }
+      }
+    } else {
+      await raiseForStatus(response, "batch create run");
+    }
   }
 
   public async updateRun(runId: string, run: RunUpdate): Promise<void> {