@copilotkit/runtime 1.4.6 → 1.4.8-coagents-v0-3.1

package/dist/service-adapters/index.d.ts

@@ -1,8 +1,8 @@
-import { C as CopilotServiceAdapter, c as CopilotRuntimeChatCompletionRequest, d as CopilotRuntimeChatCompletionResponse } from '../langserve-f021ab9c.js';
-export { a as RemoteChain, R as RemoteChainParameters } from '../langserve-f021ab9c.js';
-export { G as GoogleGenerativeAIAdapter, f as GroqAdapter, e as GroqAdapterParams, L as LangChainAdapter, a as OpenAIAdapter, O as OpenAIAdapterParams, c as OpenAIAssistantAdapter, b as OpenAIAssistantAdapterParams, d as UnifyAdapter, U as UnifyAdapterParams } from '../groq-adapter-24abe931.js';
+import { C as CopilotServiceAdapter, c as CopilotRuntimeChatCompletionRequest, d as CopilotRuntimeChatCompletionResponse } from '../langserve-8ec29cba.js';
+export { a as RemoteChain, R as RemoteChainParameters } from '../langserve-8ec29cba.js';
+export { G as GoogleGenerativeAIAdapter, f as GroqAdapter, e as GroqAdapterParams, L as LangChainAdapter, a as OpenAIAdapter, O as OpenAIAdapterParams, c as OpenAIAssistantAdapter, b as OpenAIAssistantAdapterParams, d as UnifyAdapter, U as UnifyAdapterParams } from '../groq-adapter-50aa9621.js';
 import Anthropic from '@anthropic-ai/sdk';
-import '../index-10b1c870.js';
+import '../index-ff3fbc33.js';
 import '../graphql/types/base/index.js';
 import 'rxjs';
 import '@copilotkit/shared';

package/dist/service-adapters/index.js

@@ -315,6 +315,8 @@ var OpenAIAdapter = class {
     eventSource.stream(async (eventStream$) => {
       var _a, _b;
       let mode = null;
+      let currentMessageId;
+      let currentToolCallId;
       for await (const chunk of stream) {
         if (chunk.choices.length === 0) {
           continue;
@@ -323,30 +325,52 @@ var OpenAIAdapter = class {
         const content = chunk.choices[0].delta.content;
         if (mode === "message" && (toolCall == null ? void 0 : toolCall.id)) {
           mode = null;
-          eventStream$.sendTextMessageEnd();
+          eventStream$.sendTextMessageEnd({
+            messageId: currentMessageId
+          });
         } else if (mode === "function" && (toolCall === void 0 || (toolCall == null ? void 0 : toolCall.id))) {
           mode = null;
-          eventStream$.sendActionExecutionEnd();
+          eventStream$.sendActionExecutionEnd({
+            actionExecutionId: currentToolCallId
+          });
         }
         if (mode === null) {
           if (toolCall == null ? void 0 : toolCall.id) {
             mode = "function";
-            eventStream$.sendActionExecutionStart(toolCall.id, toolCall.function.name);
+            currentToolCallId = toolCall.id;
+            eventStream$.sendActionExecutionStart({
+              actionExecutionId: currentToolCallId,
+              parentMessageId: chunk.id,
+              actionName: toolCall.function.name
+            });
           } else if (content) {
             mode = "message";
-            eventStream$.sendTextMessageStart(chunk.id);
+            currentMessageId = chunk.id;
+            eventStream$.sendTextMessageStart({
+              messageId: currentMessageId
+            });
           }
         }
         if (mode === "message" && content) {
-          eventStream$.sendTextMessageContent(content);
+          eventStream$.sendTextMessageContent({
+            messageId: currentMessageId,
+            content
+          });
         } else if (mode === "function" && ((_b = toolCall == null ? void 0 : toolCall.function) == null ? void 0 : _b.arguments)) {
-          eventStream$.sendActionExecutionArgs(toolCall.function.arguments);
+          eventStream$.sendActionExecutionArgs({
+            actionExecutionId: currentToolCallId,
+            args: toolCall.function.arguments
+          });
         }
       }
       if (mode === "message") {
-        eventStream$.sendTextMessageEnd();
+        eventStream$.sendTextMessageEnd({
+          messageId: currentMessageId
+        });
       } else if (mode === "function") {
-        eventStream$.sendActionExecutionEnd();
+        eventStream$.sendActionExecutionEnd({
+          actionExecutionId: currentToolCallId
+        });
       }
       eventStream$.complete();
     });
@@ -414,17 +438,25 @@ function isBaseMessageChunk(message) {
 __name(isBaseMessageChunk, "isBaseMessageChunk");
 function maybeSendActionExecutionResultIsMessage(eventStream$, actionExecution) {
   if (actionExecution) {
-    eventStream$.sendActionExecutionResult(actionExecution.id, actionExecution.name, "Sending a message");
+    eventStream$.sendActionExecutionResult({
+      actionExecutionId: actionExecution.id,
+      actionName: actionExecution.name,
+      result: "Sending a message"
+    });
   }
 }
 __name(maybeSendActionExecutionResultIsMessage, "maybeSendActionExecutionResultIsMessage");
 async function streamLangChainResponse({ result, eventStream$, actionExecution }) {
-  var _a, _b, _c, _d, _e, _f, _g, _h, _i;
+  var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l;
   if (typeof result === "string") {
     if (!actionExecution) {
       eventStream$.sendTextMessage((0, import_shared2.randomId)(), result);
     } else {
-      eventStream$.sendActionExecutionResult(actionExecution.id, actionExecution.name, result);
+      eventStream$.sendActionExecutionResult({
+        actionExecutionId: actionExecution.id,
+        actionName: actionExecution.name,
+        result
+      });
     }
   } else if (isAIMessage(result)) {
     maybeSendActionExecutionResultIsMessage(eventStream$, actionExecution);
@@ -432,7 +464,11 @@ async function streamLangChainResponse({ result, eventStream$, actionExecution }
       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));
+      eventStream$.sendActionExecution({
+        actionExecutionId: toolCall.id || (0, import_shared2.randomId)(),
+        actionName: toolCall.name,
+        args: JSON.stringify(toolCall.args)
+      });
     }
   } else if (isBaseMessageChunk(result)) {
     maybeSendActionExecutionResultIsMessage(eventStream$, actionExecution);
@@ -441,13 +477,18 @@ async function streamLangChainResponse({ result, eventStream$, actionExecution }
     }
     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));
+        eventStream$.sendActionExecution({
+          actionExecutionId: toolCall.id || (0, import_shared2.randomId)(),
+          actionName: toolCall.name,
+          args: JSON.stringify(toolCall.args)
+        });
       }
     }
   } else if (result && "getReader" in result) {
     maybeSendActionExecutionResultIsMessage(eventStream$, actionExecution);
     let reader = result.getReader();
     let mode = null;
+    let currentMessageId;
     const toolCallDetails = {
       name: null,
       id: null,
@@ -486,10 +527,14 @@ async function streamLangChainResponse({ result, eventStream$, actionExecution }
         }
         if (mode === "message" && (toolCallId || done)) {
           mode = null;
-          eventStream$.sendTextMessageEnd();
+          eventStream$.sendTextMessageEnd({
+            messageId: currentMessageId
+          });
         } else if (mode === "function" && (!hasToolCall || done)) {
           mode = null;
-          eventStream$.sendActionExecutionEnd();
+          eventStream$.sendActionExecutionEnd({
+            actionExecutionId: toolCallId
+          });
         }
         if (done) {
           break;
@@ -497,21 +542,40 @@ async function streamLangChainResponse({ result, eventStream$, actionExecution }
         if (mode === null) {
           if (hasToolCall && toolCallId && toolCallName) {
             mode = "function";
-            eventStream$.sendActionExecutionStart(toolCallId, toolCallName);
+            eventStream$.sendActionExecutionStart({
+              actionExecutionId: toolCallId,
+              actionName: toolCallName,
+              parentMessageId: (_i = value.lc_kwargs) == null ? void 0 : _i.id
+            });
           } else if (content) {
             mode = "message";
-            eventStream$.sendTextMessageStart((0, import_shared2.randomId)());
+            currentMessageId = ((_j = value.lc_kwargs) == null ? void 0 : _j.id) || (0, import_shared2.randomId)();
+            eventStream$.sendTextMessageStart({
+              messageId: currentMessageId
+            });
           }
         }
         if (mode === "message" && content) {
-          eventStream$.sendTextMessageContent(Array.isArray(content) ? ((_i = content[0]) == null ? void 0 : _i.text) ?? "" : content);
+          eventStream$.sendTextMessageContent({
+            messageId: currentMessageId,
+            content: Array.isArray(content) ? ((_k = content[0]) == null ? void 0 : _k.text) ?? "" : content
+          });
         } else if (mode === "function" && toolCallArgs) {
           if (toolCallDetails.index !== toolCallDetails.prevIndex) {
-            eventStream$.sendActionExecutionEnd();
-            eventStream$.sendActionExecutionStart(toolCallId, toolCallName);
+            eventStream$.sendActionExecutionEnd({
+              actionExecutionId: toolCallId
+            });
+            eventStream$.sendActionExecutionStart({
+              actionExecutionId: toolCallId,
+              actionName: toolCallName,
+              parentMessageId: (_l = value.lc_kwargs) == null ? void 0 : _l.id
+            });
             toolCallDetails.prevIndex = toolCallDetails.index;
           }
-          eventStream$.sendActionExecutionArgs(toolCallArgs);
+          eventStream$.sendActionExecutionArgs({
+            actionExecutionId: toolCallId,
+            args: toolCallArgs
+          });
         }
       } catch (error) {
         console.error("Error reading from stream", error);
@@ -519,7 +583,11 @@ async function streamLangChainResponse({ result, eventStream$, actionExecution }
       }
     }
   } else if (actionExecution) {
-    eventStream$.sendActionExecutionResult(actionExecution.id, actionExecution.name, encodeResult(result));
+    eventStream$.sendActionExecutionResult({
+      actionExecutionId: actionExecution.id,
+      actionName: actionExecution.name,
+      result: encodeResult(result)
+    });
   } else {
     throw new Error("Invalid return type from LangChain function.");
   }
@@ -699,21 +767,33 @@ var OpenAIAssistantAdapter = class {
     eventSource.stream(async (eventStream$) => {
       var _a, _b, _c, _d, _e, _f;
       let inFunctionCall = false;
+      let currentMessageId;
+      let currentToolCallId;
       for await (const chunk of stream) {
         switch (chunk.event) {
           case "thread.message.created":
             if (inFunctionCall) {
-              eventStream$.sendActionExecutionEnd();
+              eventStream$.sendActionExecutionEnd({
+                actionExecutionId: currentToolCallId
+              });
             }
-            eventStream$.sendTextMessageStart(chunk.data.id);
+            currentMessageId = chunk.data.id;
+            eventStream$.sendTextMessageStart({
+              messageId: currentMessageId
+            });
             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);
+              eventStream$.sendTextMessageContent({
+                messageId: currentMessageId,
+                content: (_b = chunk.data.delta.content) == null ? void 0 : _b[0].text.value
+              });
             }
             break;
           case "thread.message.completed":
-            eventStream$.sendTextMessageEnd();
+            eventStream$.sendTextMessageEnd({
+              messageId: currentMessageId
+            });
             break;
           case "thread.run.step.delta":
             let toolCallId;
@@ -726,18 +806,30 @@ var OpenAIAssistantAdapter = class {
             }
             if (toolCallName && toolCallId) {
               if (inFunctionCall) {
-                eventStream$.sendActionExecutionEnd();
+                eventStream$.sendActionExecutionEnd({
+                  actionExecutionId: currentToolCallId
+                });
               }
               inFunctionCall = true;
-              eventStream$.sendActionExecutionStart(toolCallId, toolCallName);
+              currentToolCallId = toolCallId;
+              eventStream$.sendActionExecutionStart({
+                actionExecutionId: currentToolCallId,
+                parentMessageId: chunk.data.id,
+                actionName: toolCallName
+              });
             } else if (toolCallArgs) {
-              eventStream$.sendActionExecutionArgs(toolCallArgs);
+              eventStream$.sendActionExecutionArgs({
+                actionExecutionId: currentToolCallId,
+                args: toolCallArgs
+              });
             }
             break;
         }
       }
       if (inFunctionCall) {
-        eventStream$.sendActionExecutionEnd();
+        eventStream$.sendActionExecutionEnd({
+          actionExecutionId: currentToolCallId
+        });
       }
       eventStream$.complete();
     });
@@ -790,46 +882,77 @@ var UnifyAdapter = class {
       }
     });
     let model = null;
+    let currentMessageId;
+    let currentToolCallId;
     request.eventSource.stream(async (eventStream$) => {
       var _a, _b;
       let mode = null;
       for await (const chunk of stream) {
         if (this.start) {
           model = chunk.model;
-          eventStream$.sendTextMessageStart((0, import_shared4.randomId)());
-          eventStream$.sendTextMessageContent(`Model used: ${model}
-`);
-          eventStream$.sendTextMessageEnd();
+          currentMessageId = (0, import_shared4.randomId)();
+          eventStream$.sendTextMessageStart({
+            messageId: currentMessageId
+          });
+          eventStream$.sendTextMessageContent({
+            messageId: currentMessageId,
+            content: `Model used: ${model}
+`
+          });
+          eventStream$.sendTextMessageEnd({
+            messageId: currentMessageId
+          });
           this.start = false;
         }
         const toolCall = (_a = chunk.choices[0].delta.tool_calls) == null ? void 0 : _a[0];
         const content = chunk.choices[0].delta.content;
         if (mode === "message" && (toolCall == null ? void 0 : toolCall.id)) {
           mode = null;
-          eventStream$.sendTextMessageEnd();
+          eventStream$.sendTextMessageEnd({
+            messageId: currentMessageId
+          });
         } else if (mode === "function" && (toolCall === void 0 || (toolCall == null ? void 0 : toolCall.id))) {
           mode = null;
-          eventStream$.sendActionExecutionEnd();
+          eventStream$.sendActionExecutionEnd({
+            actionExecutionId: currentToolCallId
+          });
         }
         if (mode === null) {
           if (toolCall == null ? void 0 : toolCall.id) {
             mode = "function";
-            eventStream$.sendActionExecutionStart(toolCall.id, toolCall.function.name);
+            currentToolCallId = toolCall.id;
+            eventStream$.sendActionExecutionStart({
+              actionExecutionId: currentToolCallId,
+              actionName: toolCall.function.name
+            });
           } else if (content) {
             mode = "message";
-            eventStream$.sendTextMessageStart(chunk.id);
+            currentMessageId = chunk.id;
+            eventStream$.sendTextMessageStart({
+              messageId: currentMessageId
+            });
           }
         }
         if (mode === "message" && content) {
-          eventStream$.sendTextMessageContent(content);
+          eventStream$.sendTextMessageContent({
+            messageId: currentMessageId,
+            content
+          });
         } else if (mode === "function" && ((_b = toolCall == null ? void 0 : toolCall.function) == null ? void 0 : _b.arguments)) {
-          eventStream$.sendActionExecutionArgs(toolCall.function.arguments);
+          eventStream$.sendActionExecutionArgs({
+            actionExecutionId: currentToolCallId,
+            args: toolCall.function.arguments
+          });
         }
       }
       if (mode === "message") {
-        eventStream$.sendTextMessageEnd();
+        eventStream$.sendTextMessageEnd({
+          messageId: currentMessageId
+        });
       } else if (mode === "function") {
-        eventStream$.sendActionExecutionEnd();
+        eventStream$.sendActionExecutionEnd({
+          actionExecutionId: currentToolCallId
+        });
       }
       eventStream$.complete();
     });
@@ -895,35 +1018,59 @@ var GroqAdapter = class {
     eventSource.stream(async (eventStream$) => {
       var _a, _b;
       let mode = null;
+      let currentMessageId;
+      let currentToolCallId;
       for await (const chunk of stream) {
         const toolCall = (_a = chunk.choices[0].delta.tool_calls) == null ? void 0 : _a[0];
         const content = chunk.choices[0].delta.content;
         if (mode === "message" && (toolCall == null ? void 0 : toolCall.id)) {
           mode = null;
-          eventStream$.sendTextMessageEnd();
+          eventStream$.sendTextMessageEnd({
+            messageId: currentMessageId
+          });
         } else if (mode === "function" && (toolCall === void 0 || (toolCall == null ? void 0 : toolCall.id))) {
           mode = null;
-          eventStream$.sendActionExecutionEnd();
+          eventStream$.sendActionExecutionEnd({
+            actionExecutionId: currentToolCallId
+          });
         }
         if (mode === null) {
           if (toolCall == null ? void 0 : toolCall.id) {
             mode = "function";
-            eventStream$.sendActionExecutionStart(toolCall.id, toolCall.function.name);
+            currentToolCallId = toolCall.id;
+            eventStream$.sendActionExecutionStart({
+              actionExecutionId: currentToolCallId,
+              actionName: toolCall.function.name,
+              parentMessageId: chunk.id
+            });
           } else if (content) {
             mode = "message";
-            eventStream$.sendTextMessageStart(chunk.id);
+            currentMessageId = chunk.id;
+            eventStream$.sendTextMessageStart({
+              messageId: currentMessageId
+            });
           }
         }
         if (mode === "message" && content) {
-          eventStream$.sendTextMessageContent(content);
+          eventStream$.sendTextMessageContent({
+            messageId: currentMessageId,
+            content
+          });
         } else if (mode === "function" && ((_b = toolCall == null ? void 0 : toolCall.function) == null ? void 0 : _b.arguments)) {
-          eventStream$.sendActionExecutionArgs(toolCall.function.arguments);
+          eventStream$.sendActionExecutionArgs({
+            actionExecutionId: currentToolCallId,
+            args: toolCall.function.arguments
+          });
         }
       }
       if (mode === "message") {
-        eventStream$.sendTextMessageEnd();
+        eventStream$.sendTextMessageEnd({
+          messageId: currentMessageId
+        });
       } else if (mode === "function") {
-        eventStream$.sendActionExecutionEnd();
+        eventStream$.sendActionExecutionEnd({
+          actionExecutionId: currentToolCallId
+        });
       }
       eventStream$.complete();
     });
@@ -1131,7 +1278,11 @@ var AnthropicAdapter = class {
             mode = "message";
           } else if (chunk.content_block.type === "tool_use") {
             currentToolCallId = chunk.content_block.id;
-            eventStream$.sendActionExecutionStart(currentToolCallId, chunk.content_block.name);
+            eventStream$.sendActionExecutionStart({
+              actionExecutionId: currentToolCallId,
+              actionName: chunk.content_block.name,
+              parentMessageId: currentMessageId
+            });
             mode = "function";
           }
         } else if (chunk.type === "content_block_delta") {
@@ -1139,21 +1290,33 @@ var AnthropicAdapter = class {
             const text = filterThinkingTextBuffer.onTextChunk(chunk.delta.text);
             if (text.length > 0) {
               if (!didOutputText) {
-                eventStream$.sendTextMessageStart(currentMessageId);
+                eventStream$.sendTextMessageStart({
+                  messageId: currentMessageId
+                });
                 didOutputText = true;
               }
-              eventStream$.sendTextMessageContent(text);
+              eventStream$.sendTextMessageContent({
+                messageId: currentMessageId,
+                content: text
+              });
             }
           } else if (chunk.delta.type === "input_json_delta") {
-            eventStream$.sendActionExecutionArgs(chunk.delta.partial_json);
+            eventStream$.sendActionExecutionArgs({
+              actionExecutionId: currentToolCallId,
+              args: chunk.delta.partial_json
+            });
           }
         } else if (chunk.type === "content_block_stop") {
           if (mode === "message") {
             if (didOutputText) {
-              eventStream$.sendTextMessageEnd();
+              eventStream$.sendTextMessageEnd({
+                messageId: currentMessageId
+              });
             }
           } else if (mode === "function") {
-            eventStream$.sendActionExecutionEnd();
+            eventStream$.sendActionExecutionEnd({
+              actionExecutionId: currentToolCallId
+            });
           }
         }
       }