@copilotkit/runtime 1.8.5-next.3 → 1.8.5-next.5

package/dist/lib/integrations/node-http/index.mjs

@@ -1,6 +1,6 @@
 import {
   copilotRuntimeNodeHttpEndpoint
-} from "../../../chunk-UTUGZIQI.mjs";
+} from "../../../chunk-Q4KJZACA.mjs";
 import "../../../chunk-DNI7KA7Y.mjs";
 import "../../../chunk-5BIEM2UU.mjs";
 import "../../../chunk-RTFJTJMA.mjs";

package/dist/{shared-86ec42e7.d.ts → shared-93687488.d.ts}

@@ -5,6 +5,7 @@ import { Parameter, Action } from '@copilotkit/shared';
 import { b as CopilotServiceAdapter, A as ActionInput, d as AgentSessionInput, e as AgentStateInput, F as ForwardedParametersInput, E as ExtensionsInput, R as RemoteChainParameters, f as RuntimeEventSource, g as ExtensionsResponse } from './langserve-6f7af8d3.js';
 import { M as MessageInput, a as Message } from './index-5bec5424.js';
 import { CopilotCloudOptions } from './lib/cloud/index.js';
+import { AbstractAgent } from '@agentwire/client';
 
 declare enum MetaEventName {
     LangGraphInterruptEvent = "LangGraphInterruptEvent",
@@ -274,6 +275,7 @@ interface CopilotRuntimeConstructorParams<T extends Parameter[] | [] = []> {
     remoteActions?: CopilotKitEndpoint[];
     remoteEndpoints?: EndpointDefinition[];
     langserve?: RemoteChainParameters[];
+    agents?: Record<string, AbstractAgent>;
     delegateAgentProcessingToServiceAdapter?: boolean;
     /**
      * Configuration for LLM request/response logging.
@@ -333,6 +335,7 @@ interface CopilotRuntimeConstructorParams<T extends Parameter[] | [] = []> {
 }
 declare class CopilotRuntime<const T extends Parameter[] | [] = []> {
     actions: ActionsConfiguration<T>;
+    agents: Record<string, AbstractAgent>;
     remoteEndpointDefinitions: EndpointDefinition[];
     private langserve;
     private onBeforeRequest?;

package/package.json

@@ -9,7 +9,7 @@
   "publishConfig": {
     "access": "public"
   },
-  "version": "1.8.5-next.3",
+  "version": "1.8.5-next.5",
   "sideEffects": false,
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",
@@ -36,6 +36,10 @@
     "tsconfig": "1.4.6"
   },
   "dependencies": {
+    "@agentwire/client": "0.0.26",
+    "@agentwire/proto": "0.0.26",
+    "@agentwire/core": "0.0.26",
+    "@agentwire/encoder": "0.0.26",
     "@anthropic-ai/sdk": "^0.27.3",
     "@graphql-yoga/plugin-defer-stream": "^3.3.1",
     "@langchain/community": "^0.3.29",
@@ -59,7 +63,7 @@
     "rxjs": "^7.8.1",
     "type-graphql": "2.0.0-rc.1",
     "zod": "^3.23.3",
-    "@copilotkit/shared": "1.8.5-next.3"
+    "@copilotkit/shared": "1.8.5-next.5"
   },
   "keywords": [
     "copilotkit",

package/src/lib/runtime/agentwire-action.ts

@@ -0,0 +1,123 @@
+import { createHash } from "node:crypto";
+import { CopilotKitEndpoint, RemoteAgentHandlerParams } from "./remote-actions";
+import { GraphQLContext } from "../integrations";
+import { Logger } from "pino";
+import { Message } from "../../graphql/types/converted";
+import { AgentStateInput } from "../../graphql/inputs/agent-state.input";
+import { Observable } from "rxjs";
+import { RuntimeEvent } from "../../service-adapters/events";
+import telemetry from "../telemetry-client";
+
+import { ToolCall } from "@agentwire/client";
+import {
+  Message as AgentWireMessage,
+  AssistantMessage as AgentWireAssistantMessage,
+} from "@agentwire/client";
+
+import { parseJson } from "@copilotkit/shared";
+import { AbstractAgent } from "@agentwire/client";
+
+export function constructAgentWireRemoteAction({
+  logger,
+  messages,
+  agentStates,
+  agent,
+}: {
+  logger: Logger;
+  messages: Message[];
+  agentStates?: AgentStateInput[];
+  agent: AbstractAgent;
+}) {
+  const action = {
+    name: agent.agentId,
+    description: agent.description,
+    parameters: [],
+    handler: async (_args: any) => {},
+    remoteAgentHandler: async ({
+      actionInputsWithoutAgents,
+      threadId,
+    }: RemoteAgentHandlerParams): Promise<Observable<RuntimeEvent>> => {
+      logger.debug({ actionName: agent.agentId }, "Executing remote agent");
+
+      const agentWireMessages = convertMessagesToAgentWire(messages);
+      agent.messages = agentWireMessages;
+      agent.threadId = threadId;
+
+      telemetry.capture("oss.runtime.remote_action_executed", {
+        agentExecution: true,
+        type: "self-hosted",
+        agentsAmount: 1,
+      });
+
+      let state = {};
+      if (agentStates) {
+        const jsonState = agentStates.find((state) => state.agentName === agent.agentId);
+        if (jsonState) {
+          state = parseJson(jsonState.state, {});
+        }
+      }
+      agent.state = state;
+
+      const tools = actionInputsWithoutAgents.map((input) => {
+        return {
+          name: input.name,
+          description: input.description,
+          parameters: JSON.parse(input.jsonSchema),
+        };
+      });
+
+      return agent.legacy_to_be_removed_runAgentBridged({
+        tools,
+      }) as Observable<RuntimeEvent>;
+    },
+  };
+  return [action];
+}
+
+export function convertMessagesToAgentWire(messages: Message[]): AgentWireMessage[] {
+  const result: AgentWireMessage[] = [];
+
+  for (const message of messages) {
+    if (message.isTextMessage()) {
+      result.push({
+        id: message.id,
+        role: message.role as any,
+        content: message.content,
+      });
+    } else if (message.isActionExecutionMessage()) {
+      const toolCall: ToolCall = {
+        id: message.id,
+        type: "function",
+        function: {
+          name: message.name,
+          arguments: JSON.stringify(message.arguments),
+        },
+      };
+
+      if (message.parentMessageId && result.some((m) => m.id === message.parentMessageId)) {
+        const parentMessage: AgentWireAssistantMessage | undefined = result.find(
+          (m) => m.id === message.parentMessageId,
+        ) as AgentWireAssistantMessage;
+        if (parentMessage.toolCalls === undefined) {
+          parentMessage.toolCalls = [];
+        }
+        parentMessage.toolCalls.push(toolCall);
+      } else {
+        result.push({
+          id: message.parentMessageId ?? message.id,
+          role: "assistant",
+          toolCalls: [toolCall],
+        });
+      }
+    } else if (message.isResultMessage()) {
+      result.push({
+        id: message.id,
+        role: "tool",
+        content: message.result,
+        toolCallId: message.actionExecutionId,
+      });
+    }
+  }
+
+  return result;
+}

package/src/lib/runtime/copilot-runtime.ts

@@ -67,6 +67,7 @@ import {
   LLMResponseData,
   LLMErrorData,
 } from "../observability";
+import { AbstractAgent } from "@agentwire/client";
 import { MessageRole } from "../../graphql/types/enums";
 
 // +++ MCP Imports +++
@@ -185,6 +186,11 @@ export interface CopilotRuntimeConstructorParams<T extends Parameter[] | [] = []
    */
   langserve?: RemoteChainParameters[];
 
+  /*
+   * A map of agent names to AgentWire agents.
+   */
+  agents?: Record<string, AbstractAgent>;
+
   /*
    * Delegates agent state processing to the service adapter.
    *
@@ -254,6 +260,7 @@ export interface CopilotRuntimeConstructorParams<T extends Parameter[] | [] = []
 
 export class CopilotRuntime<const T extends Parameter[] | [] = []> {
   public actions: ActionsConfiguration<T>;
+  public agents: Record<string, AbstractAgent>;
   public remoteEndpointDefinitions: EndpointDefinition[];
   private langserve: Promise<Action<any>>[] = [];
   private onBeforeRequest?: OnBeforeRequestHandler;
@@ -294,7 +301,7 @@ export class CopilotRuntime<const T extends Parameter[] | [] = []> {
     this.delegateAgentProcessingToServiceAdapter =
       params?.delegateAgentProcessingToServiceAdapter || false;
     this.observability = params?.observability_c;
-
+    this.agents = params?.agents ?? {};
     // +++ MCP Initialization +++
     this.mcpEndpointsConfig = params?.mcpEndpoints;
     this.createMCPClientImpl = params?.createMCPClient;
@@ -1090,6 +1097,7 @@ please use an LLM adapter instead.`,
       messages: inputMessages,
       agentStates,
       frontendUrl: url,
+      agents: this.agents,
     });
 
     const configuredActions =

package/src/lib/runtime/remote-action-constructors.ts

@@ -9,16 +9,11 @@ import { GraphQLContext } from "../integrations";
 import { Logger } from "pino";
 import { Message } from "../../graphql/types/converted";
 import { AgentStateInput } from "../../graphql/inputs/agent-state.input";
-import { Observable, ReplaySubject } from "rxjs";
-import {
-  RuntimeEvent,
-  RuntimeEventSource,
-  RuntimeEventSubject,
-} from "../../service-adapters/events";
+import { Observable } from "rxjs";
+import { RuntimeEvent, RuntimeEventSubject } from "../../service-adapters/events";
 import telemetry from "../telemetry-client";
 import { RemoteLangGraphEventSource } from "../../agents/langgraph/event-source";
 import { Action } from "@copilotkit/shared";
-import { LangGraphEvent } from "../../agents/langgraph/events";
 import { execute } from "./remote-lg-action";
 import { CopilotKitError, CopilotKitLowLevelError } from "@copilotkit/shared";
 import { writeJsonLineResponseToEventStream } from "../streaming";

package/src/lib/runtime/remote-actions.ts

@@ -1,4 +1,4 @@
-import { Action } from "@copilotkit/shared";
+import { Action, CopilotKitErrorCode } from "@copilotkit/shared";
 import { GraphQLContext } from "../integrations/shared";
 import { Logger } from "pino";
 import { Message } from "../../graphql/types/converted";
@@ -17,6 +17,8 @@ import {
   CopilotKitError,
 } from "@copilotkit/shared";
 import { MetaEventInput } from "../../graphql/inputs/meta-event.input";
+import { AbstractAgent } from "@agentwire/client";
+import { constructAgentWireRemoteAction } from "./agentwire-action";
 
 export type EndpointDefinition = CopilotKitEndpoint | LangGraphPlatformEndpoint;
 
@@ -127,12 +129,14 @@ export async function setupRemoteActions({
   messages,
   agentStates,
   frontendUrl,
+  agents,
 }: {
   remoteEndpointDefinitions: EndpointDefinition[];
   graphqlContext: GraphQLContext;
   messages: Message[];
   agentStates?: AgentStateInput[];
   frontendUrl?: string;
+  agents: Record<string, AbstractAgent>;
 }): Promise<Action[]> {
   const logger = graphqlContext.logger.child({ component: "remote-actions.fetchRemoteActions" });
   logger.debug({ remoteEndpointDefinitions }, "Fetching from remote endpoints");
@@ -181,5 +185,25 @@ export async function setupRemoteActions({
     }),
   );
 
+  for (const [key, agent] of Object.entries(agents)) {
+    if (agent.agentId !== undefined && agent.agentId !== key) {
+      throw new CopilotKitError({
+        message: `Agent ${key} has agentId ${agent.agentId} which does not match the key ${key}`,
+        code: CopilotKitErrorCode.UNKNOWN,
+      });
+    } else if (agent.agentId === undefined) {
+      agent.agentId = key;
+    }
+
+    result.push(
+      constructAgentWireRemoteAction({
+        logger,
+        messages,
+        agentStates,
+        agent: agent,
+      }),
+    );
+  }
+
   return result.flat();
 }

package/src/lib/runtime/remote-lg-action.ts

@@ -13,7 +13,7 @@ import telemetry from "../telemetry-client";
 import { MetaEventInput } from "../../graphql/inputs/meta-event.input";
 import { MetaEventName } from "../../graphql/types/meta-events.type";
 import { RunsStreamPayload } from "@langchain/langgraph-sdk/dist/types";
-import { parseJson } from "@copilotkit/shared";
+import { parseJson, CopilotKitMisuseError } from "@copilotkit/shared";
 import { RemoveMessage } from "@langchain/core/messages";
 
 type State = Record<string, any>;
@@ -83,7 +83,29 @@ export async function execute(args: ExecutionArgs): Promise<ReadableStream<Uint8
       try {
         await streamEvents(controller, args);
         controller.close();
-      } catch (err) {}
+      } catch (err) {
+        // Unwrap the possible cause
+        const cause = err?.cause;
+
+        // Check code directly if it exists
+        const errorCode = cause?.code || err?.code;
+
+        if (errorCode === "ECONNREFUSED") {
+          throw new CopilotKitMisuseError({
+            message: `
+              The LangGraph client could not connect to the graph. Please further check previous logs, which includes further details.
+              
+              See more: https://docs.copilotkit.ai/troubleshooting/common-issues`,
+          });
+        } else {
+          throw new CopilotKitMisuseError({
+            message: `
+              The LangGraph client threw unhandled error ${err}.
+              
+              See more: https://docs.copilotkit.ai/troubleshooting/common-issues`,
+          });
+        }
+      }
     },
   });
 }