@copilotkit/runtime 1.2.2-feat-runtime-remote-actions.1 → 1.3.0

package/dist/lib/index.js

@@ -44,7 +44,7 @@ var require_package = __commonJS({
       publishConfig: {
         access: "public"
       },
-      version: "1.2.2-feat-runtime-remote-actions.1",
+      version: "1.3.0",
       sideEffects: false,
       main: "./dist/index.js",
       module: "./dist/index.mjs",
@@ -59,9 +59,7 @@ var require_package = __commonJS({
         test: "jest --passWithNoTests",
         "check-types": "tsc --noEmit",
         clean: "rm -rf .turbo && rm -rf node_modules && rm -rf dist && rm -rf .next && rm -rf __snapshots__",
-        "generate-graphql-schema": "rm -rf __snapshots__ && ts-node ./scripts/generate-gql-schema.ts",
-        "link:global": "pnpm link --global",
-        "unlink:global": "pnpm unlink --global"
+        "generate-graphql-schema": "rm -rf __snapshots__ && ts-node ./scripts/generate-gql-schema.ts"
       },
       devDependencies: {
         "@swc/core": "1.5.28",
@@ -80,6 +78,7 @@ var require_package = __commonJS({
         typescript: "^5.2.3"
       },
       dependencies: {
+        "@anthropic-ai/sdk": "^0.27.3",
         "@copilotkit/shared": "workspace:*",
         "@google/generative-ai": "^0.11.2",
         "@graphql-yoga/plugin-defer-stream": "^3.3.1",
@@ -143,6 +142,84 @@ module.exports = __toCommonJS(lib_exports);
 // src/lib/runtime/copilot-runtime.ts
 var import_shared8 = require("@copilotkit/shared");
 
+// src/service-adapters/langchain/langserve.ts
+var import_remote = require("langchain/runnables/remote");
+var RemoteChain = class {
+  name;
+  description;
+  chainUrl;
+  parameters;
+  parameterType;
+  constructor(options) {
+    this.name = options.name;
+    this.description = options.description;
+    this.chainUrl = options.chainUrl;
+    this.parameters = options.parameters;
+    this.parameterType = options.parameterType || "multi";
+  }
+  async toAction() {
+    if (!this.parameters) {
+      await this.inferLangServeParameters();
+    }
+    return {
+      name: this.name,
+      description: this.description,
+      parameters: this.parameters,
+      handler: async (args) => {
+        const runnable = new import_remote.RemoteRunnable({
+          url: this.chainUrl
+        });
+        let input;
+        if (this.parameterType === "single") {
+          input = args[Object.keys(args)[0]];
+        } else {
+          input = args;
+        }
+        return await runnable.invoke(input);
+      }
+    };
+  }
+  async inferLangServeParameters() {
+    const supportedTypes = [
+      "string",
+      "number",
+      "boolean"
+    ];
+    let schemaUrl = this.chainUrl.replace(/\/+$/, "") + "/input_schema";
+    let schema = await fetch(schemaUrl).then((res) => res.json()).catch(() => {
+      throw new Error("Failed to fetch langserve schema at " + schemaUrl);
+    });
+    if (supportedTypes.includes(schema.type)) {
+      this.parameterType = "single";
+      this.parameters = [
+        {
+          name: "input",
+          type: schema.type,
+          description: "The input to the chain"
+        }
+      ];
+    } else if (schema.type === "object") {
+      this.parameterType = "multi";
+      this.parameters = Object.keys(schema.properties).map((key) => {
+        var _a;
+        let property = schema.properties[key];
+        if (!supportedTypes.includes(property.type)) {
+          throw new Error("Unsupported schema type");
+        }
+        return {
+          name: key,
+          type: property.type,
+          description: property.description || "",
+          required: ((_a = schema.required) == null ? void 0 : _a.includes(key)) || false
+        };
+      });
+    } else {
+      throw new Error("Unsupported schema type");
+    }
+  }
+};
+__name(RemoteChain, "RemoteChain");
+
 // src/service-adapters/openai/openai-adapter.ts
 var import_openai = __toESM(require("openai"));
 
@@ -443,167 +520,229 @@ var OpenAIAdapter = class {
 };
 __name(OpenAIAdapter, "OpenAIAdapter");
 
-// src/service-adapters/openai/openai-assistant-adapter.ts
-var import_openai2 = __toESM(require("openai"));
-var OpenAIAssistantAdapter = class {
-  openai;
-  codeInterpreterEnabled;
-  assistantId;
-  fileSearchEnabled;
-  disableParallelToolCalls;
-  constructor(params) {
-    this.openai = params.openai || new import_openai2.default({});
-    this.codeInterpreterEnabled = params.codeInterpreterEnabled === false || true;
-    this.fileSearchEnabled = params.fileSearchEnabled === false || true;
-    this.assistantId = params.assistantId;
-    this.disableParallelToolCalls = (params == null ? void 0 : params.disableParallelToolCalls) || false;
-  }
-  async process(request) {
-    const { messages, actions, eventSource, runId, forwardedParameters } = request;
-    let threadId = request.threadId || (await this.openai.beta.threads.create()).id;
-    const lastMessage = messages.at(-1);
-    let nextRunId = void 0;
-    if (lastMessage instanceof ResultMessage && runId) {
-      nextRunId = await this.submitToolOutputs(threadId, runId, messages, eventSource);
-    } else if (lastMessage instanceof TextMessage) {
-      nextRunId = await this.submitUserMessage(threadId, messages, actions, eventSource, forwardedParameters);
-    } else {
-      throw new Error("No actionable message found in the messages");
-    }
-    return {
-      threadId,
-      runId: nextRunId
-    };
-  }
-  async submitToolOutputs(threadId, runId, messages, eventSource) {
-    let run = await this.openai.beta.threads.runs.retrieve(threadId, runId);
-    if (!run.required_action) {
-      throw new Error("No tool outputs required");
-    }
-    const toolCallsIds = run.required_action.submit_tool_outputs.tool_calls.map((toolCall) => toolCall.id);
-    const resultMessages = messages.filter((message) => message instanceof ResultMessage && toolCallsIds.includes(message.actionExecutionId));
-    if (toolCallsIds.length != resultMessages.length) {
-      throw new Error("Number of function results does not match the number of tool calls");
-    }
-    const toolOutputs = resultMessages.map((message) => {
-      return {
-        tool_call_id: message.actionExecutionId,
-        output: message.result
-      };
-    });
-    const stream = this.openai.beta.threads.runs.submitToolOutputsStream(threadId, runId, {
-      tool_outputs: toolOutputs,
-      ...this.disableParallelToolCalls && {
-        parallel_tool_calls: false
-      }
-    });
-    await this.streamResponse(stream, eventSource);
-    return runId;
-  }
-  async submitUserMessage(threadId, messages, actions, eventSource, forwardedParameters) {
-    messages = [
-      ...messages
-    ];
-    const instructionsMessage = messages.shift();
-    const instructions = instructionsMessage instanceof TextMessage ? instructionsMessage.content : "";
-    const userMessage = messages.map(convertMessageToOpenAIMessage).map(convertSystemMessageToAssistantAPI).at(-1);
-    if (userMessage.role !== "user") {
-      throw new Error("No user message found");
+// src/service-adapters/langchain/utils.ts
+var import_messages = require("@langchain/core/messages");
+var import_tools = require("@langchain/core/tools");
+var import_zod = require("zod");
+var import_shared2 = require("@copilotkit/shared");
+function convertMessageToLangChainMessage(message) {
+  if (message instanceof TextMessage) {
+    if (message.role == "user") {
+      return new import_messages.HumanMessage(message.content);
+    } else if (message.role == "assistant") {
+      return new import_messages.AIMessage(message.content);
+    } else if (message.role === "system") {
+      return new import_messages.SystemMessage(message.content);
     }
-    await this.openai.beta.threads.messages.create(threadId, {
-      role: "user",
-      content: userMessage.content
-    });
-    const openaiTools = actions.map(convertActionInputToOpenAITool);
-    const tools = [
-      ...openaiTools,
-      ...this.codeInterpreterEnabled ? [
-        {
-          type: "code_interpreter"
-        }
-      ] : [],
-      ...this.fileSearchEnabled ? [
+  } else if (message instanceof ActionExecutionMessage) {
+    return new import_messages.AIMessage({
+      content: "",
+      tool_calls: [
         {
-          type: "file_search"
+          id: message.id,
+          args: message.arguments,
+          name: message.name
         }
-      ] : []
-    ];
-    let stream = this.openai.beta.threads.runs.stream(threadId, {
-      assistant_id: this.assistantId,
-      instructions,
-      tools,
-      ...(forwardedParameters == null ? void 0 : forwardedParameters.maxTokens) && {
-        max_completion_tokens: forwardedParameters.maxTokens
-      },
-      ...this.disableParallelToolCalls && {
-        parallel_tool_calls: false
-      }
+      ]
     });
-    await this.streamResponse(stream, eventSource);
-    return getRunIdFromStream(stream);
-  }
-  async streamResponse(stream, eventSource) {
-    eventSource.stream(async (eventStream$) => {
-      var _a, _b, _c, _d, _e, _f;
-      let inFunctionCall = false;
-      for await (const chunk of stream) {
-        switch (chunk.event) {
-          case "thread.message.created":
-            if (inFunctionCall) {
-              eventStream$.sendActionExecutionEnd();
-            }
-            eventStream$.sendTextMessageStart(chunk.data.id);
-            break;
-          case "thread.message.delta":
-            if (((_a = chunk.data.delta.content) == null ? void 0 : _a[0].type) === "text") {
-              eventStream$.sendTextMessageContent((_b = chunk.data.delta.content) == null ? void 0 : _b[0].text.value);
-            }
-            break;
-          case "thread.message.completed":
-            eventStream$.sendTextMessageEnd();
-            break;
-          case "thread.run.step.delta":
-            let toolCallId;
-            let toolCallName;
-            let toolCallArgs;
-            if (chunk.data.delta.step_details.type === "tool_calls" && ((_c = chunk.data.delta.step_details.tool_calls) == null ? void 0 : _c[0].type) === "function") {
-              toolCallId = (_d = chunk.data.delta.step_details.tool_calls) == null ? void 0 : _d[0].id;
-              toolCallName = (_e = chunk.data.delta.step_details.tool_calls) == null ? void 0 : _e[0].function.name;
-              toolCallArgs = (_f = chunk.data.delta.step_details.tool_calls) == null ? void 0 : _f[0].function.arguments;
-            }
-            if (toolCallName && toolCallId) {
-              if (inFunctionCall) {
-                eventStream$.sendActionExecutionEnd();
-              }
-              inFunctionCall = true;
-              eventStream$.sendActionExecutionStart(toolCallId, toolCallName);
-            } else if (toolCallArgs) {
-              eventStream$.sendActionExecutionArgs(toolCallArgs);
-            }
-            break;
-        }
-      }
-      if (inFunctionCall) {
-        eventStream$.sendActionExecutionEnd();
-      }
-      eventStream$.complete();
+  } else if (message instanceof ResultMessage) {
+    return new import_messages.ToolMessage({
+      content: message.result,
+      tool_call_id: message.actionExecutionId
     });
   }
-};
-__name(OpenAIAssistantAdapter, "OpenAIAssistantAdapter");
-function getRunIdFromStream(stream) {
-  return new Promise((resolve, reject) => {
-    let runIdGetter = /* @__PURE__ */ __name((event) => {
-      if (event.event === "thread.run.created") {
-        const runId = event.data.id;
-        stream.off("event", runIdGetter);
-        resolve(runId);
-      }
-    }, "runIdGetter");
-    stream.on("event", runIdGetter);
-  });
 }
-__name(getRunIdFromStream, "getRunIdFromStream");
+__name(convertMessageToLangChainMessage, "convertMessageToLangChainMessage");
+function convertJsonSchemaToZodSchema(jsonSchema, required) {
+  if (jsonSchema.type === "object") {
+    const spec = {};
+    for (const [key, value] of Object.entries(jsonSchema.properties)) {
+      spec[key] = convertJsonSchemaToZodSchema(value, jsonSchema.required ? jsonSchema.required.includes(key) : false);
+    }
+    let schema = import_zod.z.object(spec);
+    return !required ? schema.optional() : schema;
+  } else if (jsonSchema.type === "string") {
+    let schema = import_zod.z.string().describe(jsonSchema.description);
+    return !required ? schema.optional() : schema;
+  } else if (jsonSchema.type === "number") {
+    let schema = import_zod.z.number().describe(jsonSchema.description);
+    return !required ? schema.optional() : schema;
+  } else if (jsonSchema.type === "boolean") {
+    let schema = import_zod.z.boolean().describe(jsonSchema.description);
+    return !required ? schema.optional() : schema;
+  } else if (jsonSchema.type === "array") {
+    let itemSchema = convertJsonSchemaToZodSchema(jsonSchema.items, false);
+    let schema = import_zod.z.array(itemSchema);
+    return !required ? schema.optional() : schema;
+  }
+}
+__name(convertJsonSchemaToZodSchema, "convertJsonSchemaToZodSchema");
+function convertActionInputToLangChainTool(actionInput) {
+  return new import_tools.DynamicStructuredTool({
+    name: actionInput.name,
+    description: actionInput.description,
+    schema: convertJsonSchemaToZodSchema(JSON.parse(actionInput.jsonSchema), true),
+    func: async () => {
+      return "";
+    }
+  });
+}
+__name(convertActionInputToLangChainTool, "convertActionInputToLangChainTool");
+function getConstructorName(object) {
+  if (object && typeof object === "object" && object.constructor && object.constructor.name) {
+    return object.constructor.name;
+  }
+  return "";
+}
+__name(getConstructorName, "getConstructorName");
+function isAIMessage(message) {
+  return getConstructorName(message) === "AIMessage";
+}
+__name(isAIMessage, "isAIMessage");
+function isAIMessageChunk(message) {
+  return getConstructorName(message) === "AIMessageChunk";
+}
+__name(isAIMessageChunk, "isAIMessageChunk");
+function isBaseMessageChunk(message) {
+  return getConstructorName(message) === "BaseMessageChunk";
+}
+__name(isBaseMessageChunk, "isBaseMessageChunk");
+function maybeSendActionExecutionResultIsMessage(eventStream$, actionExecution) {
+  if (actionExecution) {
+    eventStream$.sendActionExecutionResult(actionExecution.id, actionExecution.name, "Sending a message");
+  }
+}
+__name(maybeSendActionExecutionResultIsMessage, "maybeSendActionExecutionResultIsMessage");
+async function streamLangChainResponse({ result, eventStream$, actionExecution }) {
+  var _a, _b, _c, _d, _e, _f, _g, _h;
+  if (typeof result === "string") {
+    if (!actionExecution) {
+      eventStream$.sendTextMessage((0, import_shared2.randomId)(), result);
+    } else {
+      eventStream$.sendActionExecutionResult(actionExecution.id, actionExecution.name, result);
+    }
+  } else if (isAIMessage(result)) {
+    maybeSendActionExecutionResultIsMessage(eventStream$, actionExecution);
+    if (result.content) {
+      eventStream$.sendTextMessage((0, import_shared2.randomId)(), result.content);
+    }
+    for (const toolCall of result.tool_calls) {
+      eventStream$.sendActionExecution(toolCall.id || (0, import_shared2.randomId)(), toolCall.name, JSON.stringify(toolCall.args));
+    }
+  } else if (isBaseMessageChunk(result)) {
+    maybeSendActionExecutionResultIsMessage(eventStream$, actionExecution);
+    if ((_a = result.lc_kwargs) == null ? void 0 : _a.content) {
+      eventStream$.sendTextMessage((0, import_shared2.randomId)(), result.content);
+    }
+    if ((_b = result.lc_kwargs) == null ? void 0 : _b.tool_calls) {
+      for (const toolCall of (_c = result.lc_kwargs) == null ? void 0 : _c.tool_calls) {
+        eventStream$.sendActionExecution(toolCall.id || (0, import_shared2.randomId)(), toolCall.name, JSON.stringify(toolCall.args));
+      }
+    }
+  } else if (result && "getReader" in result) {
+    maybeSendActionExecutionResultIsMessage(eventStream$, actionExecution);
+    let reader = result.getReader();
+    let mode = null;
+    while (true) {
+      try {
+        const { done, value } = await reader.read();
+        let toolCallName = void 0;
+        let toolCallId = void 0;
+        let toolCallArgs = void 0;
+        let hasToolCall = false;
+        let content = value == null ? void 0 : value.content;
+        if (isAIMessageChunk(value)) {
+          let chunk = (_d = value.tool_call_chunks) == null ? void 0 : _d[0];
+          toolCallName = chunk == null ? void 0 : chunk.name;
+          toolCallId = chunk == null ? void 0 : chunk.id;
+          toolCallArgs = chunk == null ? void 0 : chunk.args;
+          hasToolCall = chunk != void 0;
+        } else if (isBaseMessageChunk(value)) {
+          let chunk = (_f = (_e = value.additional_kwargs) == null ? void 0 : _e.tool_calls) == null ? void 0 : _f[0];
+          toolCallName = (_g = chunk == null ? void 0 : chunk.function) == null ? void 0 : _g.name;
+          toolCallId = chunk == null ? void 0 : chunk.id;
+          toolCallArgs = (_h = chunk == null ? void 0 : chunk.function) == null ? void 0 : _h.arguments;
+          hasToolCall = (chunk == null ? void 0 : chunk.function) != void 0;
+        }
+        if (mode === "message" && (toolCallId || done)) {
+          mode = null;
+          eventStream$.sendTextMessageEnd();
+        } else if (mode === "function" && (!hasToolCall || done)) {
+          mode = null;
+          eventStream$.sendActionExecutionEnd();
+        }
+        if (done) {
+          break;
+        }
+        if (mode === null) {
+          if (hasToolCall) {
+            mode = "function";
+            eventStream$.sendActionExecutionStart(toolCallId, toolCallName);
+          } else if (content) {
+            mode = "message";
+            eventStream$.sendTextMessageStart((0, import_shared2.randomId)());
+          }
+        }
+        if (mode === "message" && content) {
+          eventStream$.sendTextMessageContent(content);
+        } else if (mode === "function" && toolCallArgs) {
+          eventStream$.sendActionExecutionArgs(toolCallArgs);
+        }
+      } catch (error) {
+        console.error("Error reading from stream", error);
+        break;
+      }
+    }
+  } else if (actionExecution) {
+    eventStream$.sendActionExecutionResult(actionExecution.id, actionExecution.name, encodeResult(result));
+  } else {
+    throw new Error("Invalid return type from LangChain function.");
+  }
+  eventStream$.complete();
+}
+__name(streamLangChainResponse, "streamLangChainResponse");
+function encodeResult(result) {
+  if (result === void 0) {
+    return "";
+  } else if (typeof result === "string") {
+    return result;
+  } else {
+    return JSON.stringify(result);
+  }
+}
+__name(encodeResult, "encodeResult");
+
+// src/service-adapters/langchain/langchain-adapter.ts
+var import_shared3 = require("@copilotkit/shared");
+var LangChainAdapter = class {
+  options;
+  /**
+  * To use LangChain as a backend, provide a handler function to the adapter with your custom LangChain logic.
+  */
+  constructor(options) {
+    this.options = options;
+  }
+  async process(request) {
+    const { eventSource, model, actions, messages, threadId, runId } = request;
+    const result = await this.options.chainFn({
+      messages: messages.map(convertMessageToLangChainMessage),
+      tools: actions.map(convertActionInputToLangChainTool),
+      model,
+      threadId,
+      runId
+    });
+    eventSource.stream(async (eventStream$) => {
+      await streamLangChainResponse({
+        result,
+        eventStream$
+      });
+    });
+    return {
+      threadId: threadId || (0, import_shared3.randomId)()
+    };
+  }
+};
+__name(LangChainAdapter, "LangChainAdapter");
 
 // src/service-adapters/google/google-genai-adapter.ts
 var import_generative_ai = require("@google/generative-ai");
@@ -694,7 +833,7 @@ function tryParseJson(str) {
 __name(tryParseJson, "tryParseJson");
 
 // src/service-adapters/google/google-genai-adapter.ts
-var import_shared2 = require("@copilotkit/shared");
+var import_shared4 = require("@copilotkit/shared");
 var GoogleGenerativeAIAdapter = class {
   model;
   constructor(options) {
@@ -761,7 +900,7 @@ var GoogleGenerativeAIAdapter = class {
         }
         if (!isTextMessage) {
           isTextMessage = true;
-          eventStream$.sendTextMessageStart((0, import_shared2.randomId)());
+          eventStream$.sendTextMessageStart((0, import_shared4.randomId)());
         }
         eventStream$.sendTextMessageContent(chunkText);
       }
@@ -771,13 +910,13 @@ var GoogleGenerativeAIAdapter = class {
       let calls = (await result.response).functionCalls();
       if (calls) {
         for (let call of calls) {
-          eventStream$.sendActionExecution((0, import_shared2.randomId)(), call.name, JSON.stringify(replaceNewlinesInObject(call.args)));
+          eventStream$.sendActionExecution((0, import_shared4.randomId)(), call.name, JSON.stringify(replaceNewlinesInObject(call.args)));
         }
       }
       eventStream$.complete();
     });
     return {
-      threadId: request.threadId || (0, import_shared2.randomId)()
+      threadId: request.threadId || (0, import_shared4.randomId)()
     };
   }
 };
@@ -800,307 +939,167 @@ function replaceNewlinesInObject(obj) {
 }
 __name(replaceNewlinesInObject, "replaceNewlinesInObject");
 
-// src/service-adapters/langchain/utils.ts
-var import_messages = require("@langchain/core/messages");
-var import_tools = require("@langchain/core/tools");
-var import_zod = require("zod");
-var import_shared3 = require("@copilotkit/shared");
-function convertMessageToLangChainMessage(message) {
-  if (message instanceof TextMessage) {
-    if (message.role == "user") {
-      return new import_messages.HumanMessage(message.content);
-    } else if (message.role == "assistant") {
-      return new import_messages.AIMessage(message.content);
-    } else if (message.role === "system") {
-      return new import_messages.SystemMessage(message.content);
-    }
-  } else if (message instanceof ActionExecutionMessage) {
-    return new import_messages.AIMessage({
-      content: "",
-      tool_calls: [
-        {
-          id: message.id,
-          args: message.arguments,
-          name: message.name
-        }
-      ]
-    });
-  } else if (message instanceof ResultMessage) {
-    return new import_messages.ToolMessage({
-      content: message.result,
-      tool_call_id: message.actionExecutionId
-    });
+// src/service-adapters/openai/openai-assistant-adapter.ts
+var import_openai2 = __toESM(require("openai"));
+var OpenAIAssistantAdapter = class {
+  openai;
+  codeInterpreterEnabled;
+  assistantId;
+  fileSearchEnabled;
+  disableParallelToolCalls;
+  constructor(params) {
+    this.openai = params.openai || new import_openai2.default({});
+    this.codeInterpreterEnabled = params.codeInterpreterEnabled === false || true;
+    this.fileSearchEnabled = params.fileSearchEnabled === false || true;
+    this.assistantId = params.assistantId;
+    this.disableParallelToolCalls = (params == null ? void 0 : params.disableParallelToolCalls) || false;
   }
-}
-__name(convertMessageToLangChainMessage, "convertMessageToLangChainMessage");
-function convertJsonSchemaToZodSchema(jsonSchema, required) {
-  if (jsonSchema.type === "object") {
-    const spec = {};
-    for (const [key, value] of Object.entries(jsonSchema.properties)) {
-      spec[key] = convertJsonSchemaToZodSchema(value, jsonSchema.required ? jsonSchema.required.includes(key) : false);
+  async process(request) {
+    const { messages, actions, eventSource, runId, forwardedParameters } = request;
+    let threadId = request.threadId || (await this.openai.beta.threads.create()).id;
+    const lastMessage = messages.at(-1);
+    let nextRunId = void 0;
+    if (lastMessage instanceof ResultMessage && runId) {
+      nextRunId = await this.submitToolOutputs(threadId, runId, messages, eventSource);
+    } else if (lastMessage instanceof TextMessage) {
+      nextRunId = await this.submitUserMessage(threadId, messages, actions, eventSource, forwardedParameters);
+    } else {
+      throw new Error("No actionable message found in the messages");
     }
-    let schema = import_zod.z.object(spec);
-    return !required ? schema.optional() : schema;
-  } else if (jsonSchema.type === "string") {
-    let schema = import_zod.z.string().describe(jsonSchema.description);
-    return !required ? schema.optional() : schema;
-  } else if (jsonSchema.type === "number") {
-    let schema = import_zod.z.number().describe(jsonSchema.description);
-    return !required ? schema.optional() : schema;
-  } else if (jsonSchema.type === "boolean") {
-    let schema = import_zod.z.boolean().describe(jsonSchema.description);
-    return !required ? schema.optional() : schema;
-  } else if (jsonSchema.type === "array") {
-    let itemSchema = convertJsonSchemaToZodSchema(jsonSchema.items, false);
-    let schema = import_zod.z.array(itemSchema);
-    return !required ? schema.optional() : schema;
-  }
-}
-__name(convertJsonSchemaToZodSchema, "convertJsonSchemaToZodSchema");
-function convertActionInputToLangChainTool(actionInput) {
-  return new import_tools.DynamicStructuredTool({
-    name: actionInput.name,
-    description: actionInput.description,
-    schema: convertJsonSchemaToZodSchema(JSON.parse(actionInput.jsonSchema), true),
-    func: async () => {
-      return "";
-    }
-  });
-}
-__name(convertActionInputToLangChainTool, "convertActionInputToLangChainTool");
-function getConstructorName(object) {
-  if (object && typeof object === "object" && object.constructor && object.constructor.name) {
-    return object.constructor.name;
-  }
-  return "";
-}
-__name(getConstructorName, "getConstructorName");
-function isAIMessage(message) {
-  return getConstructorName(message) === "AIMessage";
-}
-__name(isAIMessage, "isAIMessage");
-function isAIMessageChunk(message) {
-  return getConstructorName(message) === "AIMessageChunk";
-}
-__name(isAIMessageChunk, "isAIMessageChunk");
-function isBaseMessageChunk(message) {
-  return getConstructorName(message) === "BaseMessageChunk";
-}
-__name(isBaseMessageChunk, "isBaseMessageChunk");
-function maybeSendActionExecutionResultIsMessage(eventStream$, actionExecution) {
-  if (actionExecution) {
-    eventStream$.sendActionExecutionResult(actionExecution.id, actionExecution.name, "Sending a message");
+    return {
+      threadId,
+      runId: nextRunId
+    };
   }
-}
-__name(maybeSendActionExecutionResultIsMessage, "maybeSendActionExecutionResultIsMessage");
-async function streamLangChainResponse({ result, eventStream$, actionExecution }) {
-  var _a, _b, _c, _d, _e, _f, _g, _h;
-  if (typeof result === "string") {
-    if (!actionExecution) {
-      eventStream$.sendTextMessage((0, import_shared3.randomId)(), result);
-    } else {
-      eventStream$.sendActionExecutionResult(actionExecution.id, actionExecution.name, result);
-    }
-  } else if (isAIMessage(result)) {
-    maybeSendActionExecutionResultIsMessage(eventStream$, actionExecution);
-    if (result.content) {
-      eventStream$.sendTextMessage((0, import_shared3.randomId)(), result.content);
-    }
-    for (const toolCall of result.tool_calls) {
-      eventStream$.sendActionExecution(toolCall.id || (0, import_shared3.randomId)(), toolCall.name, JSON.stringify(toolCall.args));
+  async submitToolOutputs(threadId, runId, messages, eventSource) {
+    let run = await this.openai.beta.threads.runs.retrieve(threadId, runId);
+    if (!run.required_action) {
+      throw new Error("No tool outputs required");
     }
-  } else if (isBaseMessageChunk(result)) {
-    maybeSendActionExecutionResultIsMessage(eventStream$, actionExecution);
-    if ((_a = result.lc_kwargs) == null ? void 0 : _a.content) {
-      eventStream$.sendTextMessage((0, import_shared3.randomId)(), result.content);
+    const toolCallsIds = run.required_action.submit_tool_outputs.tool_calls.map((toolCall) => toolCall.id);
+    const resultMessages = messages.filter((message) => message instanceof ResultMessage && toolCallsIds.includes(message.actionExecutionId));
+    if (toolCallsIds.length != resultMessages.length) {
+      throw new Error("Number of function results does not match the number of tool calls");
     }
-    if ((_b = result.lc_kwargs) == null ? void 0 : _b.tool_calls) {
-      for (const toolCall of (_c = result.lc_kwargs) == null ? void 0 : _c.tool_calls) {
-        eventStream$.sendActionExecution(toolCall.id || (0, import_shared3.randomId)(), toolCall.name, JSON.stringify(toolCall.args));
+    const toolOutputs = resultMessages.map((message) => {
+      return {
+        tool_call_id: message.actionExecutionId,
+        output: message.result
+      };
+    });
+    const stream = this.openai.beta.threads.runs.submitToolOutputsStream(threadId, runId, {
+      tool_outputs: toolOutputs,
+      ...this.disableParallelToolCalls && {
+        parallel_tool_calls: false
       }
+    });
+    await this.streamResponse(stream, eventSource);
+    return runId;
+  }
+  async submitUserMessage(threadId, messages, actions, eventSource, forwardedParameters) {
+    messages = [
+      ...messages
+    ];
+    const instructionsMessage = messages.shift();
+    const instructions = instructionsMessage instanceof TextMessage ? instructionsMessage.content : "";
+    const userMessage = messages.map(convertMessageToOpenAIMessage).map(convertSystemMessageToAssistantAPI).at(-1);
+    if (userMessage.role !== "user") {
+      throw new Error("No user message found");
     }
-  } else if (result && "getReader" in result) {
-    maybeSendActionExecutionResultIsMessage(eventStream$, actionExecution);
-    let reader = result.getReader();
-    let mode = null;
-    while (true) {
-      try {
-        const { done, value } = await reader.read();
-        let toolCallName = void 0;
-        let toolCallId = void 0;
-        let toolCallArgs = void 0;
-        let hasToolCall = false;
-        let content = value == null ? void 0 : value.content;
-        if (isAIMessageChunk(value)) {
-          let chunk = (_d = value.tool_call_chunks) == null ? void 0 : _d[0];
-          toolCallName = chunk == null ? void 0 : chunk.name;
-          toolCallId = chunk == null ? void 0 : chunk.id;
-          toolCallArgs = chunk == null ? void 0 : chunk.args;
-          hasToolCall = chunk != void 0;
-        } else if (isBaseMessageChunk(value)) {
-          let chunk = (_f = (_e = value.additional_kwargs) == null ? void 0 : _e.tool_calls) == null ? void 0 : _f[0];
-          toolCallName = (_g = chunk == null ? void 0 : chunk.function) == null ? void 0 : _g.name;
-          toolCallId = chunk == null ? void 0 : chunk.id;
-          toolCallArgs = (_h = chunk == null ? void 0 : chunk.function) == null ? void 0 : _h.arguments;
-          hasToolCall = (chunk == null ? void 0 : chunk.function) != void 0;
-        }
-        if (mode === "message" && (toolCallId || done)) {
-          mode = null;
-          eventStream$.sendTextMessageEnd();
-        } else if (mode === "function" && (!hasToolCall || done)) {
-          mode = null;
-          eventStream$.sendActionExecutionEnd();
-        }
-        if (done) {
-          break;
-        }
-        if (mode === null) {
-          if (hasToolCall) {
-            mode = "function";
-            eventStream$.sendActionExecutionStart(toolCallId, toolCallName);
-          } else if (content) {
-            mode = "message";
-            eventStream$.sendTextMessageStart((0, import_shared3.randomId)());
-          }
+    await this.openai.beta.threads.messages.create(threadId, {
+      role: "user",
+      content: userMessage.content
+    });
+    const openaiTools = actions.map(convertActionInputToOpenAITool);
+    const tools = [
+      ...openaiTools,
+      ...this.codeInterpreterEnabled ? [
+        {
+          type: "code_interpreter"
         }
-        if (mode === "message" && content) {
-          eventStream$.sendTextMessageContent(content);
-        } else if (mode === "function" && toolCallArgs) {
-          eventStream$.sendActionExecutionArgs(toolCallArgs);
+      ] : [],
+      ...this.fileSearchEnabled ? [
+        {
+          type: "file_search"
         }
-      } catch (error) {
-        console.error("Error reading from stream", error);
-        break;
+      ] : []
+    ];
+    let stream = this.openai.beta.threads.runs.stream(threadId, {
+      assistant_id: this.assistantId,
+      instructions,
+      tools,
+      ...(forwardedParameters == null ? void 0 : forwardedParameters.maxTokens) && {
+        max_completion_tokens: forwardedParameters.maxTokens
+      },
+      ...this.disableParallelToolCalls && {
+        parallel_tool_calls: false
       }
-    }
-  } else if (actionExecution) {
-    eventStream$.sendActionExecutionResult(actionExecution.id, actionExecution.name, encodeResult(result));
-  } else {
-    throw new Error("Invalid return type from LangChain function.");
-  }
-  eventStream$.complete();
-}
-__name(streamLangChainResponse, "streamLangChainResponse");
-function encodeResult(result) {
-  if (result === void 0) {
-    return "";
-  } else if (typeof result === "string") {
-    return result;
-  } else {
-    return JSON.stringify(result);
-  }
-}
-__name(encodeResult, "encodeResult");
-
-// src/service-adapters/langchain/langchain-adapter.ts
-var import_shared4 = require("@copilotkit/shared");
-var LangChainAdapter = class {
-  options;
-  /**
-  * To use LangChain as a backend, provide a handler function to the adapter with your custom LangChain logic.
-  */
-  constructor(options) {
-    this.options = options;
-  }
-  async process(request) {
-    const { eventSource, model, actions, messages, threadId, runId } = request;
-    const result = await this.options.chainFn({
-      messages: messages.map(convertMessageToLangChainMessage),
-      tools: actions.map(convertActionInputToLangChainTool),
-      model,
-      threadId,
-      runId
     });
-    eventSource.stream(async (eventStream$) => {
-      await streamLangChainResponse({
-        result,
-        eventStream$
-      });
-    });
-    return {
-      threadId: threadId || (0, import_shared4.randomId)()
-    };
-  }
-};
-__name(LangChainAdapter, "LangChainAdapter");
-
-// src/service-adapters/langchain/langserve.ts
-var import_remote = require("langchain/runnables/remote");
-var RemoteChain = class {
-  name;
-  description;
-  chainUrl;
-  parameters;
-  parameterType;
-  constructor(options) {
-    this.name = options.name;
-    this.description = options.description;
-    this.chainUrl = options.chainUrl;
-    this.parameters = options.parameters;
-    this.parameterType = options.parameterType || "multi";
+    await this.streamResponse(stream, eventSource);
+    return getRunIdFromStream(stream);
   }
-  async toAction() {
-    if (!this.parameters) {
-      await this.inferLangServeParameters();
-    }
-    return {
-      name: this.name,
-      description: this.description,
-      parameters: this.parameters,
-      handler: async (args) => {
-        const runnable = new import_remote.RemoteRunnable({
-          url: this.chainUrl
-        });
-        let input;
-        if (this.parameterType === "single") {
-          input = args[Object.keys(args)[0]];
-        } else {
-          input = args;
+  async streamResponse(stream, eventSource) {
+    eventSource.stream(async (eventStream$) => {
+      var _a, _b, _c, _d, _e, _f;
+      let inFunctionCall = false;
+      for await (const chunk of stream) {
+        switch (chunk.event) {
+          case "thread.message.created":
+            if (inFunctionCall) {
+              eventStream$.sendActionExecutionEnd();
+            }
+            eventStream$.sendTextMessageStart(chunk.data.id);
+            break;
+          case "thread.message.delta":
+            if (((_a = chunk.data.delta.content) == null ? void 0 : _a[0].type) === "text") {
+              eventStream$.sendTextMessageContent((_b = chunk.data.delta.content) == null ? void 0 : _b[0].text.value);
+            }
+            break;
+          case "thread.message.completed":
+            eventStream$.sendTextMessageEnd();
+            break;
+          case "thread.run.step.delta":
+            let toolCallId;
+            let toolCallName;
+            let toolCallArgs;
+            if (chunk.data.delta.step_details.type === "tool_calls" && ((_c = chunk.data.delta.step_details.tool_calls) == null ? void 0 : _c[0].type) === "function") {
+              toolCallId = (_d = chunk.data.delta.step_details.tool_calls) == null ? void 0 : _d[0].id;
+              toolCallName = (_e = chunk.data.delta.step_details.tool_calls) == null ? void 0 : _e[0].function.name;
+              toolCallArgs = (_f = chunk.data.delta.step_details.tool_calls) == null ? void 0 : _f[0].function.arguments;
+            }
+            if (toolCallName && toolCallId) {
+              if (inFunctionCall) {
+                eventStream$.sendActionExecutionEnd();
+              }
+              inFunctionCall = true;
+              eventStream$.sendActionExecutionStart(toolCallId, toolCallName);
+            } else if (toolCallArgs) {
+              eventStream$.sendActionExecutionArgs(toolCallArgs);
+            }
+            break;
         }
-        return await runnable.invoke(input);
       }
-    };
-  }
-  async inferLangServeParameters() {
-    const supportedTypes = [
-      "string",
-      "number",
-      "boolean"
-    ];
-    let schemaUrl = this.chainUrl.replace(/\/+$/, "") + "/input_schema";
-    let schema = await fetch(schemaUrl).then((res) => res.json()).catch(() => {
-      throw new Error("Failed to fetch langserve schema at " + schemaUrl);
+      if (inFunctionCall) {
+        eventStream$.sendActionExecutionEnd();
+      }
+      eventStream$.complete();
     });
-    if (supportedTypes.includes(schema.type)) {
-      this.parameterType = "single";
-      this.parameters = [
-        {
-          name: "input",
-          type: schema.type,
-          description: "The input to the chain"
-        }
-      ];
-    } else if (schema.type === "object") {
-      this.parameterType = "multi";
-      this.parameters = Object.keys(schema.properties).map((key) => {
-        var _a;
-        let property = schema.properties[key];
-        if (!supportedTypes.includes(property.type)) {
-          throw new Error("Unsupported schema type");
-        }
-        return {
-          name: key,
-          type: property.type,
-          description: property.description || "",
-          required: ((_a = schema.required) == null ? void 0 : _a.includes(key)) || false
-        };
-      });
-    } else {
-      throw new Error("Unsupported schema type");
-    }
   }
 };
-__name(RemoteChain, "RemoteChain");
+__name(OpenAIAssistantAdapter, "OpenAIAssistantAdapter");
+function getRunIdFromStream(stream) {
+  return new Promise((resolve, reject) => {
+    let runIdGetter = /* @__PURE__ */ __name((event) => {
+      if (event.event === "thread.run.created") {
+        const runId = event.data.id;
+        stream.off("event", runIdGetter);
+        resolve(runId);
+      }
+    }, "runIdGetter");
+    stream.on("event", runIdGetter);
+  });
+}
+__name(getRunIdFromStream, "getRunIdFromStream");
 
 // src/service-adapters/unify/unify-adapter.ts
 var import_openai3 = __toESM(require("openai"));