@copilotkit/runtime 1.56.2 → 1.56.4-canary.1777529757
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/agent/converters/tanstack.cjs +121 -25
- package/dist/agent/converters/tanstack.cjs.map +1 -1
- package/dist/agent/converters/tanstack.d.cts.map +1 -1
- package/dist/agent/converters/tanstack.d.mts.map +1 -1
- package/dist/agent/converters/tanstack.mjs +121 -25
- package/dist/agent/converters/tanstack.mjs.map +1 -1
- package/dist/graphql/resolvers/copilot.resolver.cjs +2 -1
- package/dist/graphql/resolvers/copilot.resolver.cjs.map +1 -1
- package/dist/graphql/resolvers/copilot.resolver.mjs +2 -1
- package/dist/graphql/resolvers/copilot.resolver.mjs.map +1 -1
- package/dist/graphql/resolvers/resolve-message-id.cjs +19 -0
- package/dist/graphql/resolvers/resolve-message-id.cjs.map +1 -0
- package/dist/graphql/resolvers/resolve-message-id.mjs +18 -0
- package/dist/graphql/resolvers/resolve-message-id.mjs.map +1 -0
- package/dist/lib/runtime/agent-integrations/langgraph/agent.cjs +8 -1
- package/dist/lib/runtime/agent-integrations/langgraph/agent.cjs.map +1 -1
- package/dist/lib/runtime/agent-integrations/langgraph/agent.d.cts.map +1 -1
- package/dist/lib/runtime/agent-integrations/langgraph/agent.d.mts.map +1 -1
- package/dist/lib/runtime/agent-integrations/langgraph/agent.mjs +8 -1
- package/dist/lib/runtime/agent-integrations/langgraph/agent.mjs.map +1 -1
- package/dist/lib/runtime/copilot-runtime.cjs +4 -2
- package/dist/lib/runtime/copilot-runtime.cjs.map +1 -1
- package/dist/lib/runtime/copilot-runtime.d.cts.map +1 -1
- package/dist/lib/runtime/copilot-runtime.d.mts.map +1 -1
- package/dist/lib/runtime/copilot-runtime.mjs +4 -2
- package/dist/lib/runtime/copilot-runtime.mjs.map +1 -1
- package/dist/package.cjs +7 -7
- package/dist/package.mjs +7 -7
- package/dist/v2/index.d.cts +2 -2
- package/dist/v2/index.d.mts +2 -2
- package/dist/v2/runtime/core/debug-event-bus.cjs +36 -0
- package/dist/v2/runtime/core/debug-event-bus.cjs.map +1 -0
- package/dist/v2/runtime/core/debug-event-bus.d.cts +19 -0
- package/dist/v2/runtime/core/debug-event-bus.d.cts.map +1 -0
- package/dist/v2/runtime/core/debug-event-bus.d.mts +19 -0
- package/dist/v2/runtime/core/debug-event-bus.d.mts.map +1 -0
- package/dist/v2/runtime/core/debug-event-bus.mjs +35 -0
- package/dist/v2/runtime/core/debug-event-bus.mjs.map +1 -0
- package/dist/v2/runtime/core/fetch-handler.cjs +8 -0
- package/dist/v2/runtime/core/fetch-handler.cjs.map +1 -1
- package/dist/v2/runtime/core/fetch-handler.d.cts.map +1 -1
- package/dist/v2/runtime/core/fetch-handler.d.mts.map +1 -1
- package/dist/v2/runtime/core/fetch-handler.mjs +8 -0
- package/dist/v2/runtime/core/fetch-handler.mjs.map +1 -1
- package/dist/v2/runtime/core/fetch-router.cjs +1 -0
- package/dist/v2/runtime/core/fetch-router.cjs.map +1 -1
- package/dist/v2/runtime/core/fetch-router.mjs +1 -0
- package/dist/v2/runtime/core/fetch-router.mjs.map +1 -1
- package/dist/v2/runtime/core/hooks.cjs.map +1 -1
- package/dist/v2/runtime/core/hooks.d.cts +2 -0
- package/dist/v2/runtime/core/hooks.d.cts.map +1 -1
- package/dist/v2/runtime/core/hooks.d.mts +2 -0
- package/dist/v2/runtime/core/hooks.d.mts.map +1 -1
- package/dist/v2/runtime/core/hooks.mjs.map +1 -1
- package/dist/v2/runtime/core/runtime.cjs +5 -0
- package/dist/v2/runtime/core/runtime.cjs.map +1 -1
- package/dist/v2/runtime/core/runtime.d.cts +5 -0
- package/dist/v2/runtime/core/runtime.d.cts.map +1 -1
- package/dist/v2/runtime/core/runtime.d.mts +5 -1
- package/dist/v2/runtime/core/runtime.d.mts.map +1 -1
- package/dist/v2/runtime/core/runtime.mjs +5 -0
- package/dist/v2/runtime/core/runtime.mjs.map +1 -1
- package/dist/v2/runtime/endpoints/express.cjs +5 -5
- package/dist/v2/runtime/endpoints/express.cjs.map +1 -1
- package/dist/v2/runtime/endpoints/express.mjs +5 -5
- package/dist/v2/runtime/endpoints/express.mjs.map +1 -1
- package/dist/v2/runtime/handlers/handle-connect.cjs +3 -2
- package/dist/v2/runtime/handlers/handle-connect.cjs.map +1 -1
- package/dist/v2/runtime/handlers/handle-connect.mjs +3 -2
- package/dist/v2/runtime/handlers/handle-connect.mjs.map +1 -1
- package/dist/v2/runtime/handlers/handle-debug-events.cjs +33 -0
- package/dist/v2/runtime/handlers/handle-debug-events.cjs.map +1 -0
- package/dist/v2/runtime/handlers/handle-debug-events.mjs +32 -0
- package/dist/v2/runtime/handlers/handle-debug-events.mjs.map +1 -0
- package/dist/v2/runtime/handlers/handle-run.cjs +1 -0
- package/dist/v2/runtime/handlers/handle-run.cjs.map +1 -1
- package/dist/v2/runtime/handlers/handle-run.mjs +1 -0
- package/dist/v2/runtime/handlers/handle-run.mjs.map +1 -1
- package/dist/v2/runtime/handlers/intelligence/connect.cjs +24 -4
- package/dist/v2/runtime/handlers/intelligence/connect.cjs.map +1 -1
- package/dist/v2/runtime/handlers/intelligence/connect.mjs +25 -5
- package/dist/v2/runtime/handlers/intelligence/connect.mjs.map +1 -1
- package/dist/v2/runtime/handlers/intelligence/run.cjs +111 -26
- package/dist/v2/runtime/handlers/intelligence/run.cjs.map +1 -1
- package/dist/v2/runtime/handlers/intelligence/run.mjs +111 -26
- package/dist/v2/runtime/handlers/intelligence/run.mjs.map +1 -1
- package/dist/v2/runtime/handlers/shared/intelligence-utils.cjs +7 -3
- package/dist/v2/runtime/handlers/shared/intelligence-utils.cjs.map +1 -1
- package/dist/v2/runtime/handlers/shared/intelligence-utils.mjs +7 -3
- package/dist/v2/runtime/handlers/shared/intelligence-utils.mjs.map +1 -1
- package/dist/v2/runtime/handlers/shared/resolve-intelligence-user.cjs +5 -1
- package/dist/v2/runtime/handlers/shared/resolve-intelligence-user.cjs.map +1 -1
- package/dist/v2/runtime/handlers/shared/resolve-intelligence-user.mjs +5 -1
- package/dist/v2/runtime/handlers/shared/resolve-intelligence-user.mjs.map +1 -1
- package/dist/v2/runtime/handlers/shared/sse-response.cjs +21 -1
- package/dist/v2/runtime/handlers/shared/sse-response.cjs.map +1 -1
- package/dist/v2/runtime/handlers/shared/sse-response.mjs +21 -1
- package/dist/v2/runtime/handlers/shared/sse-response.mjs.map +1 -1
- package/dist/v2/runtime/handlers/sse/connect.cjs +3 -1
- package/dist/v2/runtime/handlers/sse/connect.cjs.map +1 -1
- package/dist/v2/runtime/handlers/sse/connect.mjs +3 -1
- package/dist/v2/runtime/handlers/sse/connect.mjs.map +1 -1
- package/dist/v2/runtime/handlers/sse/run.cjs +3 -1
- package/dist/v2/runtime/handlers/sse/run.cjs.map +1 -1
- package/dist/v2/runtime/handlers/sse/run.mjs +3 -1
- package/dist/v2/runtime/handlers/sse/run.mjs.map +1 -1
- package/dist/v2/runtime/index.d.cts +1 -1
- package/dist/v2/runtime/index.d.mts +1 -2
- package/dist/v2/runtime/index.d.mts.map +1 -1
- package/dist/v2/runtime/intelligence-platform/client.cjs +6 -8
- package/dist/v2/runtime/intelligence-platform/client.cjs.map +1 -1
- package/dist/v2/runtime/intelligence-platform/client.d.cts +16 -21
- package/dist/v2/runtime/intelligence-platform/client.d.cts.map +1 -1
- package/dist/v2/runtime/intelligence-platform/client.d.mts +16 -21
- package/dist/v2/runtime/intelligence-platform/client.d.mts.map +1 -1
- package/dist/v2/runtime/intelligence-platform/client.mjs +6 -8
- package/dist/v2/runtime/intelligence-platform/client.mjs.map +1 -1
- package/dist/v2/runtime/runner/agent-runner.cjs.map +1 -1
- package/dist/v2/runtime/runner/agent-runner.d.cts +0 -1
- package/dist/v2/runtime/runner/agent-runner.d.cts.map +1 -1
- package/dist/v2/runtime/runner/agent-runner.d.mts +0 -1
- package/dist/v2/runtime/runner/agent-runner.d.mts.map +1 -1
- package/dist/v2/runtime/runner/agent-runner.mjs.map +1 -1
- package/dist/v2/runtime/runner/index.d.cts +1 -1
- package/dist/v2/runtime/runner/index.d.mts +1 -1
- package/dist/v2/runtime/runner/intelligence.cjs +47 -10
- package/dist/v2/runtime/runner/intelligence.cjs.map +1 -1
- package/dist/v2/runtime/runner/intelligence.d.cts +8 -1
- package/dist/v2/runtime/runner/intelligence.d.cts.map +1 -1
- package/dist/v2/runtime/runner/intelligence.d.mts +8 -1
- package/dist/v2/runtime/runner/intelligence.d.mts.map +1 -1
- package/dist/v2/runtime/runner/intelligence.mjs +47 -10
- package/dist/v2/runtime/runner/intelligence.mjs.map +1 -1
- package/dist/v2/runtime/telemetry/instance-created.cjs +33 -0
- package/dist/v2/runtime/telemetry/instance-created.cjs.map +1 -0
- package/dist/v2/runtime/telemetry/instance-created.mjs +33 -0
- package/dist/v2/runtime/telemetry/instance-created.mjs.map +1 -0
- package/dist/v2/runtime/telemetry/telemetry-client.cjs +1 -38
- package/dist/v2/runtime/telemetry/telemetry-client.cjs.map +1 -1
- package/dist/v2/runtime/telemetry/telemetry-client.mjs +1 -37
- package/dist/v2/runtime/telemetry/telemetry-client.mjs.map +1 -1
- package/package.json +8 -8
- package/src/agent/__tests__/agent-test-helpers.ts +31 -1
- package/src/agent/__tests__/converter-tanstack.test.ts +280 -0
- package/src/agent/converters/tanstack.ts +167 -10
- package/src/agents/langgraph/__tests__/event-source.test.ts +256 -0
- package/src/graphql/resolvers/__tests__/resolve-message-id.test.ts +25 -0
- package/src/graphql/resolvers/copilot.resolver.ts +2 -1
- package/src/graphql/resolvers/resolve-message-id.ts +14 -0
- package/src/lib/runtime/__tests__/handle-service-adapter.test.ts +108 -0
- package/src/lib/runtime/__tests__/retry-utils.test.ts +137 -0
- package/src/lib/runtime/agent-integrations/langgraph/__tests__/dispatch-event-filtering.test.ts +190 -0
- package/src/lib/runtime/agent-integrations/langgraph/agent.ts +8 -1
- package/src/lib/runtime/copilot-runtime.ts +20 -4
- package/src/lib/runtime/retry-utils.ts +41 -1
- package/src/v2/runtime/__tests__/express-fetch-bridge.test.ts +1 -1
- package/src/v2/runtime/__tests__/express-single-telemetry.integration.test.ts +65 -0
- package/src/v2/runtime/__tests__/express-telemetry.integration.test.ts +101 -0
- package/src/v2/runtime/__tests__/fetch-router.test.ts +22 -0
- package/src/v2/runtime/__tests__/handle-connect.test.ts +183 -23
- package/src/v2/runtime/__tests__/handle-run.test.ts +411 -33
- package/src/v2/runtime/__tests__/handle-threads.test.ts +66 -4
- package/src/v2/runtime/__tests__/hono-single-telemetry.integration.test.ts +46 -0
- package/src/v2/runtime/__tests__/hono-telemetry.integration.test.ts +99 -0
- package/src/v2/runtime/__tests__/integration/node-servers.integration.test.ts +19 -0
- package/src/v2/runtime/__tests__/integration/suites/debug-events.suite.ts +253 -0
- package/src/v2/runtime/__tests__/intelligence-run-telemetry.test.ts +194 -0
- package/src/v2/runtime/__tests__/runtime.test.ts +3 -1
- package/src/v2/runtime/__tests__/sse-response-telemetry.test.ts +108 -0
- package/src/v2/runtime/__tests__/telemetry.test.ts +0 -61
- package/src/v2/runtime/core/__tests__/debug-event-bus.test.ts +156 -0
- package/src/v2/runtime/core/debug-event-bus.ts +45 -0
- package/src/v2/runtime/core/fetch-handler.ts +7 -0
- package/src/v2/runtime/core/fetch-router.ts +11 -0
- package/src/v2/runtime/core/hooks.ts +2 -1
- package/src/v2/runtime/core/runtime.ts +12 -0
- package/src/v2/runtime/endpoints/express.ts +9 -3
- package/src/v2/runtime/handlers/__tests__/handle-debug-events.test.ts +176 -0
- package/src/v2/runtime/handlers/handle-connect.ts +2 -1
- package/src/v2/runtime/handlers/handle-debug-events.ts +52 -0
- package/src/v2/runtime/handlers/handle-run.ts +1 -0
- package/src/v2/runtime/handlers/intelligence/connect.ts +48 -11
- package/src/v2/runtime/handlers/intelligence/run.ts +162 -21
- package/src/v2/runtime/handlers/shared/intelligence-utils.ts +21 -1
- package/src/v2/runtime/handlers/shared/resolve-intelligence-user.ts +4 -1
- package/src/v2/runtime/handlers/shared/sse-response.ts +46 -0
- package/src/v2/runtime/handlers/sse/__tests__/sse-connect-agent-id.test.ts +71 -0
- package/src/v2/runtime/handlers/sse/connect.ts +6 -0
- package/src/v2/runtime/handlers/sse/run.ts +4 -0
- package/src/v2/runtime/intelligence-platform/__tests__/client.test.ts +33 -37
- package/src/v2/runtime/intelligence-platform/client.ts +37 -40
- package/src/v2/runtime/runner/__tests__/intelligence-runner.test.ts +66 -8
- package/src/v2/runtime/runner/agent-runner.ts +0 -1
- package/src/v2/runtime/runner/intelligence.ts +74 -15
- package/src/v2/runtime/telemetry/__tests__/instance-created.test.ts +96 -0
- package/src/v2/runtime/telemetry/instance-created.ts +44 -0
- package/src/v2/runtime/telemetry/telemetry-client.ts +1 -57
- package/dist/v2/runtime/intelligence-platform/index.d.mts +0 -2
- package/dist/v2/runtime/telemetry/utils.cjs +0 -15
- package/dist/v2/runtime/telemetry/utils.cjs.map +0 -1
- package/dist/v2/runtime/telemetry/utils.mjs +0 -14
- package/dist/v2/runtime/telemetry/utils.mjs.map +0 -1
- package/src/v2/runtime/telemetry/utils.ts +0 -15
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"copilot.resolver.cjs","names":["MessageRole","GraphQLError","ReplaySubject","Repeater","RuntimeEventTypes","LangGraphEventTypes","LangGraphInterruptEvent","RuntimeMetaEventName","CopilotKitLangGraphInterruptEvent","TextMessage","SuccessMessageStatus","ActionExecutionMessage","UnknownErrorResponse","SuccessResponseStatus","GuardrailsValidationFailureResponse","Subject","FailedMessageStatus","MessageStreamInterruptedResponse","ResultMessage","AgentStateMessage","CopilotKitError","CopilotKitLowLevelError","AgentsResponse","CopilotResponse","GraphQLJSONObject"],"sources":["../../../src/graphql/resolvers/copilot.resolver.ts"],"sourcesContent":["import { Arg, Ctx, Mutation, Query, Resolver } from \"type-graphql\";\nimport {\n ReplaySubject,\n Subject,\n Subscription,\n filter,\n finalize,\n firstValueFrom,\n shareReplay,\n skipWhile,\n take,\n takeWhile,\n tap,\n} from \"rxjs\";\nimport { GenerateCopilotResponseInput } from \"../inputs/generate-copilot-response.input\";\nimport { CopilotResponse } from \"../types/copilot-response.type\";\nimport {\n CopilotKitLangGraphInterruptEvent,\n LangGraphInterruptEvent,\n} from \"../types/meta-events.type\";\nimport { ActionInputAvailability, MessageRole } from \"../types/enums\";\nimport { Repeater } from \"graphql-yoga\";\nimport type {\n CopilotRequestContextProperties,\n GraphQLContext,\n} from \"../../lib/integrations\";\nimport {\n RuntimeEvent,\n RuntimeEventTypes,\n RuntimeMetaEventName,\n} from \"../../service-adapters/events\";\nimport {\n FailedMessageStatus,\n MessageStatusCode,\n MessageStatusUnion,\n SuccessMessageStatus,\n} from \"../types/message-status.type\";\nimport {\n ResponseStatusUnion,\n SuccessResponseStatus,\n} from \"../types/response-status.type\";\nimport { GraphQLJSONObject } from \"graphql-scalars\";\nimport { plainToInstance } from \"class-transformer\";\nimport { GuardrailsResult } from \"../types/guardrails-result.type\";\nimport { GraphQLError } from \"graphql\";\nimport {\n GuardrailsValidationFailureResponse,\n MessageStreamInterruptedResponse,\n UnknownErrorResponse,\n} from \"../../utils\";\nimport {\n ActionExecutionMessage,\n AgentStateMessage,\n Message,\n MessageType,\n ResultMessage,\n TextMessage,\n} from \"../types/converted\";\nimport telemetry from \"../../lib/telemetry-client\";\nimport { randomId } from \"@copilotkit/shared\";\nimport { AgentsResponse } from \"../types/agents-response.type\";\nimport { LangGraphEventTypes } from \"../../agents/langgraph/events\";\nimport {\n CopilotKitError,\n CopilotKitLowLevelError,\n isStructuredCopilotKitError,\n} from \"@copilotkit/shared\";\nimport { CopilotRuntime } from \"../../lib\";\n\nconst invokeGuardrails = async ({\n baseUrl,\n copilotCloudPublicApiKey,\n data,\n onResult,\n onError,\n}: {\n baseUrl: string;\n copilotCloudPublicApiKey: string;\n data: GenerateCopilotResponseInput;\n onResult: (result: GuardrailsResult) => void;\n onError: (err: Error) => void;\n}) => {\n if (\n data.messages.length &&\n data.messages[data.messages.length - 1].textMessage?.role ===\n MessageRole.user\n ) {\n const messages = data.messages\n .filter(\n (m) =>\n m.textMessage !== undefined &&\n (m.textMessage.role === MessageRole.user ||\n m.textMessage.role === MessageRole.assistant),\n )\n .map((m) => ({\n role: m.textMessage!.role,\n content: m.textMessage.content,\n }));\n\n const lastMessage = messages[messages.length - 1];\n const restOfMessages = messages.slice(0, -1);\n\n const body = {\n input: lastMessage.content,\n validTopics: data.cloud.guardrails.inputValidationRules.allowList,\n invalidTopics: data.cloud.guardrails.inputValidationRules.denyList,\n messages: restOfMessages,\n };\n\n const guardrailsResult = await fetch(`${baseUrl}/guardrails/validate`, {\n method: \"POST\",\n headers: {\n \"Content-Type\": \"application/json\",\n \"X-CopilotCloud-Public-API-Key\": copilotCloudPublicApiKey,\n },\n body: JSON.stringify(body),\n });\n\n if (guardrailsResult.ok) {\n const resultJson: GuardrailsResult = await guardrailsResult.json();\n onResult(resultJson);\n } else {\n onError(await guardrailsResult.json());\n }\n }\n};\n\n@Resolver(() => CopilotResponse)\nexport class CopilotResolver {\n @Query(() => String)\n async hello() {\n return \"Hello World\";\n }\n\n @Query(() => AgentsResponse)\n async availableAgents(@Ctx() ctx: GraphQLContext) {\n let logger = ctx.logger.child({\n component: \"CopilotResolver.availableAgents\",\n });\n\n logger.debug(\"Processing\");\n const agentsWithEndpoints = [];\n\n logger.debug(\"Event source created, creating response\");\n\n return {\n agents: agentsWithEndpoints.map(\n ({ endpoint, ...agentWithoutEndpoint }) => agentWithoutEndpoint,\n ),\n };\n }\n\n @Mutation(() => CopilotResponse)\n async generateCopilotResponse(\n @Ctx() ctx: GraphQLContext,\n @Arg(\"data\") data: GenerateCopilotResponseInput,\n @Arg(\"properties\", () => GraphQLJSONObject, { nullable: true })\n properties?: CopilotRequestContextProperties,\n ) {\n telemetry.capture(\"oss.runtime.copilot_request_created\", {\n \"cloud.guardrails.enabled\": data.cloud?.guardrails !== undefined,\n requestType: data.metadata.requestType,\n \"cloud.api_key_provided\": !!ctx.request.headers.get(\n \"x-copilotcloud-public-api-key\",\n ),\n ...(ctx.request.headers.get(\"x-copilotcloud-public-api-key\")\n ? {\n \"cloud.public_api_key\": ctx.request.headers.get(\n \"x-copilotcloud-public-api-key\",\n ),\n }\n : {}),\n ...(ctx._copilotkit.baseUrl\n ? {\n \"cloud.base_url\": ctx._copilotkit.baseUrl,\n }\n : {\n \"cloud.base_url\": \"https://api.cloud.copilotkit.ai\",\n }),\n });\n\n let logger = ctx.logger.child({\n component: \"CopilotResolver.generateCopilotResponse\",\n });\n logger.debug({ data }, \"Generating Copilot response\");\n\n if (properties) {\n logger.debug(\"Properties provided, merging with context properties\");\n ctx.properties = { ...ctx.properties, ...properties };\n }\n\n const copilotRuntime = ctx._copilotkit.runtime as unknown as CopilotRuntime;\n const serviceAdapter = ctx._copilotkit.serviceAdapter;\n\n let copilotCloudPublicApiKey: string | null = null;\n let copilotCloudBaseUrl: string;\n\n // Extract publicApiKey from headers for both cloud and non-cloud requests\n // This enables onTrace functionality regardless of cloud configuration\n const publicApiKeyFromHeaders = ctx.request.headers.get(\n \"x-copilotcloud-public-api-key\",\n );\n if (publicApiKeyFromHeaders) {\n copilotCloudPublicApiKey = publicApiKeyFromHeaders;\n }\n\n if (data.cloud) {\n logger = logger.child({ cloud: true });\n logger.debug(\n \"Cloud configuration provided, checking for public API key in headers\",\n );\n\n if (!copilotCloudPublicApiKey) {\n logger.error(\"Public API key not found in headers\");\n\n throw new GraphQLError(\n \"X-CopilotCloud-Public-API-Key header is required\",\n );\n }\n\n if (process.env.COPILOT_CLOUD_BASE_URL) {\n copilotCloudBaseUrl = process.env.COPILOT_CLOUD_BASE_URL;\n } else if (ctx._copilotkit.cloud?.baseUrl) {\n copilotCloudBaseUrl = ctx._copilotkit.cloud?.baseUrl;\n } else {\n copilotCloudBaseUrl = \"https://api.cloud.copilotkit.ai\";\n }\n\n logger = logger.child({ copilotCloudBaseUrl });\n }\n\n logger.debug(\"Setting up subjects\");\n const responseStatus$ = new ReplaySubject<typeof ResponseStatusUnion>();\n const interruptStreaming$ = new ReplaySubject<{\n reason: string;\n messageId?: string;\n }>();\n const guardrailsResult$ = new ReplaySubject<GuardrailsResult>();\n\n let outputMessages: Message[] = [];\n let resolveOutputMessagesPromise: (messages: Message[]) => void;\n let rejectOutputMessagesPromise: (err: Error) => void;\n\n const outputMessagesPromise = new Promise<Message[]>((resolve, reject) => {\n resolveOutputMessagesPromise = resolve;\n rejectOutputMessagesPromise = reject;\n });\n\n if (copilotCloudPublicApiKey) {\n ctx.properties[\"copilotCloudPublicApiKey\"] = copilotCloudPublicApiKey;\n }\n\n logger.debug(\"Processing\");\n let runtimeResponse;\n\n const {\n eventSource,\n threadId = randomId(),\n runId,\n serverSideActions,\n actionInputsWithoutAgents,\n extensions,\n } = runtimeResponse;\n\n logger.debug(\"Event source created, creating response\");\n // run and process the event stream\n const eventStream = eventSource\n .processRuntimeEvents({\n serverSideActions,\n guardrailsResult$: data.cloud?.guardrails ? guardrailsResult$ : null,\n actionInputsWithoutAgents: actionInputsWithoutAgents.filter(\n // TODO-AGENTS: do not exclude ALL server side actions\n (action) =>\n !serverSideActions.find(\n (serverSideAction) => serverSideAction.name == action.name,\n ),\n ),\n threadId,\n })\n .pipe(\n // shareReplay() ensures that later subscribers will see the whole stream instead of\n // just the events that were emitted after the subscriber was added.\n shareReplay(),\n finalize(() => {\n logger.debug(\"Event stream finalized\");\n }),\n );\n\n const response = {\n threadId,\n runId,\n status: firstValueFrom(responseStatus$),\n extensions,\n metaEvents: new Repeater(async (push, stop) => {\n let eventStreamSubscription: Subscription;\n\n eventStreamSubscription = eventStream.subscribe({\n next: async (event) => {\n if (event.type != RuntimeEventTypes.MetaEvent) {\n return;\n }\n switch (event.name) {\n // @ts-ignore\n case LangGraphEventTypes.OnInterrupt:\n push(\n plainToInstance(LangGraphInterruptEvent, {\n // @ts-ignore\n type: event.type,\n // @ts-ignore\n name: RuntimeMetaEventName.LangGraphInterruptEvent,\n // @ts-ignore\n value: event.value,\n }),\n );\n break;\n case RuntimeMetaEventName.LangGraphInterruptEvent:\n push(\n plainToInstance(LangGraphInterruptEvent, {\n type: event.type,\n name: event.name,\n value: event.value,\n }),\n );\n break;\n case RuntimeMetaEventName.CopilotKitLangGraphInterruptEvent:\n push(\n plainToInstance(CopilotKitLangGraphInterruptEvent, {\n type: event.type,\n name: event.name,\n data: {\n value: event.data.value,\n messages: event.data.messages.map((message) => {\n if (\n message.type === \"TextMessage\" ||\n (\"content\" in message && \"role\" in message)\n ) {\n return plainToInstance(TextMessage, {\n id: message.id,\n createdAt: new Date(),\n content: [(message as TextMessage).content],\n role: (message as TextMessage).role,\n status: new SuccessMessageStatus(),\n });\n }\n if (\"arguments\" in message) {\n return plainToInstance(ActionExecutionMessage, {\n name: message.name,\n id: message.id,\n arguments: [JSON.stringify(message.arguments)],\n createdAt: new Date(),\n status: new SuccessMessageStatus(),\n });\n }\n throw new Error(\n \"Unknown message in metaEvents copilot resolver\",\n );\n }),\n },\n }),\n );\n break;\n }\n },\n error: (err) => {\n // For structured CopilotKit errors, set proper error response status\n if (\n err?.name?.includes(\"CopilotKit\") ||\n err?.extensions?.visibility\n ) {\n responseStatus$.next(\n new UnknownErrorResponse({\n description: err.message || \"Agent error occurred\",\n }),\n );\n } else {\n responseStatus$.next(\n new UnknownErrorResponse({\n description: `An unknown error has occurred in the event stream`,\n }),\n );\n }\n\n eventStreamSubscription?.unsubscribe();\n stop();\n },\n complete: async () => {\n logger.debug(\"Meta events stream completed\");\n responseStatus$.next(new SuccessResponseStatus());\n eventStreamSubscription?.unsubscribe();\n stop();\n },\n });\n }),\n messages: new Repeater(async (pushMessage, stopStreamingMessages) => {\n logger.debug(\"Messages repeater created\");\n\n if (data.cloud?.guardrails) {\n logger = logger.child({ guardrails: true });\n logger.debug(\"Guardrails is enabled, validating input\");\n\n invokeGuardrails({\n baseUrl: copilotCloudBaseUrl,\n copilotCloudPublicApiKey,\n data,\n onResult: (result) => {\n logger.debug(\n { status: result.status },\n \"Guardrails validation done\",\n );\n guardrailsResult$.next(result);\n\n // Guardrails validation failed\n if (result.status === \"denied\") {\n // send the reason to the client and interrupt streaming\n responseStatus$.next(\n new GuardrailsValidationFailureResponse({\n guardrailsReason: result.reason,\n }),\n );\n interruptStreaming$.next({\n reason: `Interrupted due to Guardrails validation failure. Reason: ${result.reason}`,\n });\n\n // resolve messages promise to the middleware\n outputMessages = [\n plainToInstance(TextMessage, {\n id: randomId(),\n createdAt: new Date(),\n content: result.reason,\n role: MessageRole.assistant,\n }),\n ];\n resolveOutputMessagesPromise(outputMessages);\n }\n },\n onError: (err) => {\n logger.error({ err }, \"Error in guardrails validation\");\n responseStatus$.next(\n new UnknownErrorResponse({\n description: `An unknown error has occurred in the guardrails validation`,\n }),\n );\n interruptStreaming$.next({\n reason: `Interrupted due to unknown error in guardrails validation`,\n });\n\n // reject the middleware promise\n rejectOutputMessagesPromise(err);\n },\n });\n }\n\n let eventStreamSubscription: Subscription;\n\n logger.debug(\"Event stream created, subscribing to event stream\");\n\n eventStreamSubscription = eventStream.subscribe({\n next: async (event) => {\n switch (event.type) {\n case RuntimeEventTypes.MetaEvent:\n break;\n ////////////////////////////////\n // TextMessageStart\n ////////////////////////////////\n case RuntimeEventTypes.TextMessageStart:\n // create a sub stream that contains the message content\n const textMessageContentStream = eventStream.pipe(\n // skip until this message start event\n skipWhile((e: RuntimeEvent) => e !== event),\n // take until the message end event\n takeWhile(\n (e: RuntimeEvent) =>\n !(\n e.type === RuntimeEventTypes.TextMessageEnd &&\n (e as any).messageId == event.messageId\n ),\n ),\n // filter out any other message events or message ids\n filter(\n (e: RuntimeEvent) =>\n e.type == RuntimeEventTypes.TextMessageContent &&\n (e as any).messageId == event.messageId,\n ),\n );\n\n // signal when we are done streaming\n const streamingTextStatus = new Subject<\n typeof MessageStatusUnion\n >();\n\n const messageId = event.messageId;\n // push the new message\n pushMessage({\n id: messageId,\n parentMessageId: event.parentMessageId,\n status: firstValueFrom(streamingTextStatus),\n createdAt: new Date(),\n role: MessageRole.assistant,\n content: new Repeater(\n async (pushTextChunk, stopStreamingText) => {\n logger.debug(\"Text message content repeater created\");\n\n const textChunks: string[] = [];\n let textSubscription: Subscription;\n\n interruptStreaming$\n .pipe(\n shareReplay(),\n take(1),\n tap(({ reason, messageId }) => {\n logger.debug(\n { reason, messageId },\n \"Text streaming interrupted\",\n );\n\n streamingTextStatus.next(\n plainToInstance(FailedMessageStatus, { reason }),\n );\n\n responseStatus$.next(\n new MessageStreamInterruptedResponse({\n messageId,\n }),\n );\n stopStreamingText();\n textSubscription?.unsubscribe();\n }),\n )\n .subscribe();\n\n logger.debug(\n \"Subscribing to text message content stream\",\n );\n\n textSubscription = textMessageContentStream.subscribe({\n next: async (e: RuntimeEvent) => {\n if (e.type == RuntimeEventTypes.TextMessageContent) {\n await pushTextChunk(e.content);\n textChunks.push(e.content);\n }\n },\n error: (err) => {\n logger.error(\n { err },\n \"Error in text message content stream\",\n );\n interruptStreaming$.next({\n reason: \"Error streaming message content\",\n messageId,\n });\n stopStreamingText();\n textSubscription?.unsubscribe();\n },\n complete: () => {\n logger.debug(\"Text message content stream completed\");\n streamingTextStatus.next(new SuccessMessageStatus());\n stopStreamingText();\n textSubscription?.unsubscribe();\n\n outputMessages.push(\n plainToInstance(TextMessage, {\n id: messageId,\n createdAt: new Date(),\n content: textChunks.join(\"\"),\n role: MessageRole.assistant,\n }),\n );\n },\n });\n },\n ),\n });\n break;\n ////////////////////////////////\n // ActionExecutionStart\n ////////////////////////////////\n case RuntimeEventTypes.ActionExecutionStart:\n logger.debug(\"Action execution start event received\");\n const actionExecutionArgumentStream = eventStream.pipe(\n skipWhile((e: RuntimeEvent) => e !== event),\n // take until the action execution end event\n takeWhile(\n (e: RuntimeEvent) =>\n !(\n e.type === RuntimeEventTypes.ActionExecutionEnd &&\n (e as any).actionExecutionId == event.actionExecutionId\n ),\n ),\n // filter out any other action execution events or action execution ids\n filter(\n (e: RuntimeEvent) =>\n e.type == RuntimeEventTypes.ActionExecutionArgs &&\n (e as any).actionExecutionId == event.actionExecutionId,\n ),\n );\n const streamingArgumentsStatus = new Subject<\n typeof MessageStatusUnion\n >();\n pushMessage({\n id: event.actionExecutionId,\n parentMessageId: event.parentMessageId,\n status: firstValueFrom(streamingArgumentsStatus),\n createdAt: new Date(),\n name: event.actionName,\n arguments: new Repeater(\n async (pushArgumentsChunk, stopStreamingArguments) => {\n logger.debug(\"Action execution argument stream created\");\n\n const argumentChunks: string[] = [];\n let actionExecutionArgumentSubscription: Subscription;\n\n actionExecutionArgumentSubscription =\n actionExecutionArgumentStream.subscribe({\n next: async (e: RuntimeEvent) => {\n if (\n e.type == RuntimeEventTypes.ActionExecutionArgs\n ) {\n await pushArgumentsChunk(e.args);\n argumentChunks.push(e.args);\n }\n },\n error: (err) => {\n logger.error(\n { err },\n \"Error in action execution argument stream\",\n );\n streamingArgumentsStatus.next(\n plainToInstance(FailedMessageStatus, {\n reason:\n \"An unknown error has occurred in the action execution argument stream\",\n }),\n );\n stopStreamingArguments();\n actionExecutionArgumentSubscription?.unsubscribe();\n },\n complete: () => {\n logger.debug(\n \"Action execution argument stream completed\",\n );\n streamingArgumentsStatus.next(\n new SuccessMessageStatus(),\n );\n stopStreamingArguments();\n actionExecutionArgumentSubscription?.unsubscribe();\n\n outputMessages.push(\n plainToInstance(ActionExecutionMessage, {\n id: event.actionExecutionId,\n createdAt: new Date(),\n name: event.actionName,\n arguments: argumentChunks.join(\"\"),\n }),\n );\n },\n });\n },\n ),\n });\n break;\n ////////////////////////////////\n // ActionExecutionResult\n ////////////////////////////////\n case RuntimeEventTypes.ActionExecutionResult:\n logger.debug(\n { result: event.result },\n \"Action execution result event received\",\n );\n pushMessage({\n id: \"result-\" + event.actionExecutionId,\n status: new SuccessMessageStatus(),\n createdAt: new Date(),\n actionExecutionId: event.actionExecutionId,\n actionName: event.actionName,\n result: event.result,\n });\n\n outputMessages.push(\n plainToInstance(ResultMessage, {\n id: \"result-\" + event.actionExecutionId,\n createdAt: new Date(),\n actionExecutionId: event.actionExecutionId,\n actionName: event.actionName,\n result: event.result,\n }),\n );\n break;\n ////////////////////////////////\n // AgentStateMessage\n ////////////////////////////////\n case RuntimeEventTypes.AgentStateMessage:\n logger.debug({ event }, \"Agent message event received\");\n pushMessage({\n id: randomId(),\n status: new SuccessMessageStatus(),\n threadId: event.threadId,\n agentName: event.agentName,\n nodeName: event.nodeName,\n runId: event.runId,\n active: event.active,\n state: event.state,\n running: event.running,\n role: MessageRole.assistant,\n createdAt: new Date(),\n });\n outputMessages.push(\n plainToInstance(AgentStateMessage, {\n id: randomId(),\n threadId: event.threadId,\n agentName: event.agentName,\n nodeName: event.nodeName,\n runId: event.runId,\n active: event.active,\n state: event.state,\n running: event.running,\n role: MessageRole.assistant,\n createdAt: new Date(),\n }),\n );\n break;\n }\n },\n error: (err) => {\n // For structured CopilotKit errors, set proper error response status\n if (\n err instanceof CopilotKitError ||\n err instanceof CopilotKitLowLevelError ||\n (err instanceof Error &&\n err.name &&\n err.name.includes(\"CopilotKit\")) ||\n err?.extensions?.visibility\n ) {\n responseStatus$.next(\n new UnknownErrorResponse({\n description: err.message || \"Agent error occurred\",\n // Include original error information for frontend to extract\n originalError: {\n code: err.code || err.extensions?.code,\n statusCode: err.statusCode || err.extensions?.statusCode,\n severity: err.severity || err.extensions?.severity,\n visibility: err.visibility || err.extensions?.visibility,\n originalErrorType:\n err.originalErrorType ||\n err.extensions?.originalErrorType,\n extensions: err.extensions,\n },\n }),\n );\n eventStreamSubscription?.unsubscribe();\n rejectOutputMessagesPromise(err);\n stopStreamingMessages();\n return;\n }\n\n responseStatus$.next(\n new UnknownErrorResponse({\n description: `An unknown error has occurred in the event stream`,\n }),\n );\n eventStreamSubscription?.unsubscribe();\n stopStreamingMessages();\n\n rejectOutputMessagesPromise(err);\n },\n complete: async () => {\n logger.debug(\"Event stream completed\");\n if (data.cloud?.guardrails) {\n logger.debug(\n \"Guardrails is enabled, waiting for guardrails result\",\n );\n await firstValueFrom(guardrailsResult$);\n }\n responseStatus$.next(new SuccessResponseStatus());\n eventStreamSubscription?.unsubscribe();\n stopStreamingMessages();\n\n resolveOutputMessagesPromise(outputMessages);\n },\n });\n }),\n };\n\n return response;\n }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;AAqEA,MAAM,mBAAmB,OAAO,EAC9B,SACA,0BACA,MACA,UACA,cAOI;AACJ,KACE,KAAK,SAAS,UACd,KAAK,SAAS,KAAK,SAAS,SAAS,GAAG,aAAa,SACnDA,0BAAY,MACd;EACA,MAAM,WAAW,KAAK,SACnB,QACE,MACC,EAAE,gBAAgB,WACjB,EAAE,YAAY,SAASA,0BAAY,QAClC,EAAE,YAAY,SAASA,0BAAY,WACxC,CACA,KAAK,OAAO;GACX,MAAM,EAAE,YAAa;GACrB,SAAS,EAAE,YAAY;GACxB,EAAE;EAEL,MAAM,cAAc,SAAS,SAAS,SAAS;EAC/C,MAAM,iBAAiB,SAAS,MAAM,GAAG,GAAG;EAE5C,MAAM,OAAO;GACX,OAAO,YAAY;GACnB,aAAa,KAAK,MAAM,WAAW,qBAAqB;GACxD,eAAe,KAAK,MAAM,WAAW,qBAAqB;GAC1D,UAAU;GACX;EAED,MAAM,mBAAmB,MAAM,MAAM,GAAG,QAAQ,uBAAuB;GACrE,QAAQ;GACR,SAAS;IACP,gBAAgB;IAChB,iCAAiC;IAClC;GACD,MAAM,KAAK,UAAU,KAAK;GAC3B,CAAC;AAEF,MAAI,iBAAiB,GAEnB,UADqC,MAAM,iBAAiB,MAAM,CAC9C;MAEpB,SAAQ,MAAM,iBAAiB,MAAM,CAAC;;;AAMrC,4BAAM,gBAAgB;CAC3B,MACM,QAAQ;AACZ,SAAO;;CAGT,MACM,gBAAgB,AAAO,KAAqB;EAChD,IAAI,SAAS,IAAI,OAAO,MAAM,EAC5B,WAAW,mCACZ,CAAC;AAEF,SAAO,MAAM,aAAa;EAC1B,MAAM,sBAAsB,EAAE;AAE9B,SAAO,MAAM,0CAA0C;AAEvD,SAAO,EACL,QAAQ,oBAAoB,KACzB,EAAE,UAAU,GAAG,2BAA2B,qBAC5C,EACF;;CAGH,MACM,wBACJ,AAAO,KACP,AAAa,MACb,AACA,YACA;AACA,mCAAU,QAAQ,uCAAuC;GACvD,4BAA4B,KAAK,OAAO,eAAe;GACvD,aAAa,KAAK,SAAS;GAC3B,0BAA0B,CAAC,CAAC,IAAI,QAAQ,QAAQ,IAC9C,gCACD;GACD,GAAI,IAAI,QAAQ,QAAQ,IAAI,gCAAgC,GACxD,EACE,wBAAwB,IAAI,QAAQ,QAAQ,IAC1C,gCACD,EACF,GACD,EAAE;GACN,GAAI,IAAI,YAAY,UAChB,EACE,kBAAkB,IAAI,YAAY,SACnC,GACD,EACE,kBAAkB,mCACnB;GACN,CAAC;EAEF,IAAI,SAAS,IAAI,OAAO,MAAM,EAC5B,WAAW,2CACZ,CAAC;AACF,SAAO,MAAM,EAAE,MAAM,EAAE,8BAA8B;AAErD,MAAI,YAAY;AACd,UAAO,MAAM,uDAAuD;AACpE,OAAI,aAAa;IAAE,GAAG,IAAI;IAAY,GAAG;IAAY;;AAGhC,MAAI,YAAY;AAChB,MAAI,YAAY;EAEvC,IAAI,2BAA0C;EAC9C,IAAI;EAIJ,MAAM,0BAA0B,IAAI,QAAQ,QAAQ,IAClD,gCACD;AACD,MAAI,wBACF,4BAA2B;AAG7B,MAAI,KAAK,OAAO;AACd,YAAS,OAAO,MAAM,EAAE,OAAO,MAAM,CAAC;AACtC,UAAO,MACL,uEACD;AAED,OAAI,CAAC,0BAA0B;AAC7B,WAAO,MAAM,sCAAsC;AAEnD,UAAM,IAAIC,qBACR,mDACD;;AAGH,OAAI,QAAQ,IAAI,uBACd,uBAAsB,QAAQ,IAAI;YACzB,IAAI,YAAY,OAAO,QAChC,uBAAsB,IAAI,YAAY,OAAO;OAE7C,uBAAsB;AAGxB,YAAS,OAAO,MAAM,EAAE,qBAAqB,CAAC;;AAGhD,SAAO,MAAM,sBAAsB;EACnC,MAAM,kBAAkB,IAAIC,oBAA2C;EACvE,MAAM,sBAAsB,IAAIA,oBAG5B;EACJ,MAAM,oBAAoB,IAAIA,oBAAiC;EAE/D,IAAI,iBAA4B,EAAE;EAClC,IAAI;EACJ,IAAI;AAE0B,MAAI,SAAoB,SAAS,WAAW;AACxE,kCAA+B;AAC/B,iCAA8B;IAC9B;AAEF,MAAI,yBACF,KAAI,WAAW,8BAA8B;AAG/C,SAAO,MAAM,aAAa;EAC1B,IAAI;EAEJ,MAAM,EACJ,aACA,6CAAqB,EACrB,OACA,mBACA,2BACA,eACE;AAEJ,SAAO,MAAM,0CAA0C;EAEvD,MAAM,cAAc,YACjB,qBAAqB;GACpB;GACA,mBAAmB,KAAK,OAAO,aAAa,oBAAoB;GAChE,2BAA2B,0BAA0B,QAElD,WACC,CAAC,kBAAkB,MAChB,qBAAqB,iBAAiB,QAAQ,OAAO,KACvD,CACJ;GACD;GACD,CAAC,CACD,4BAGc,2BACE;AACb,UAAO,MAAM,yBAAyB;IACtC,CACH;AA+eH,SA7eiB;GACf;GACA;GACA,iCAAuB,gBAAgB;GACvC;GACA,YAAY,IAAIC,sBAAS,OAAO,MAAM,SAAS;IAC7C,IAAI;AAEJ,8BAA0B,YAAY,UAAU;KAC9C,MAAM,OAAO,UAAU;AACrB,UAAI,MAAM,QAAQC,iCAAkB,UAClC;AAEF,cAAQ,MAAM,MAAd;OAEE,KAAKC,qCAAoB;AACvB,oDACkBC,kDAAyB;SAEvC,MAAM,MAAM;SAEZ,MAAMC,oCAAqB;SAE3B,OAAO,MAAM;SACd,CAAC,CACH;AACD;OACF,KAAKA,oCAAqB;AACxB,oDACkBD,kDAAyB;SACvC,MAAM,MAAM;SACZ,MAAM,MAAM;SACZ,OAAO,MAAM;SACd,CAAC,CACH;AACD;OACF,KAAKC,oCAAqB;AACxB,oDACkBC,4DAAmC;SACjD,MAAM,MAAM;SACZ,MAAM,MAAM;SACZ,MAAM;UACJ,OAAO,MAAM,KAAK;UAClB,UAAU,MAAM,KAAK,SAAS,KAAK,YAAY;AAC7C,eACE,QAAQ,SAAS,iBAChB,aAAa,WAAW,UAAU,QAEnC,+CAAuBC,2BAAa;YAClC,IAAI,QAAQ;YACZ,2BAAW,IAAI,MAAM;YACrB,SAAS,CAAE,QAAwB,QAAQ;YAC3C,MAAO,QAAwB;YAC/B,QAAQ,IAAIC,kDAAsB;YACnC,CAAC;AAEJ,eAAI,eAAe,QACjB,+CAAuBC,sCAAwB;YAC7C,MAAM,QAAQ;YACd,IAAI,QAAQ;YACZ,WAAW,CAAC,KAAK,UAAU,QAAQ,UAAU,CAAC;YAC9C,2BAAW,IAAI,MAAM;YACrB,QAAQ,IAAID,kDAAsB;YACnC,CAAC;AAEJ,iBAAM,IAAI,MACR,iDACD;YACD;UACH;SACF,CAAC,CACH;AACD;;;KAGN,QAAQ,QAAQ;AAEd,UACE,KAAK,MAAM,SAAS,aAAa,IACjC,KAAK,YAAY,WAEjB,iBAAgB,KACd,IAAIE,4DAAqB,EACvB,aAAa,IAAI,WAAW,wBAC7B,CAAC,CACH;UAED,iBAAgB,KACd,IAAIA,4DAAqB,EACvB,aAAa,qDACd,CAAC,CACH;AAGH,+BAAyB,aAAa;AACtC,YAAM;;KAER,UAAU,YAAY;AACpB,aAAO,MAAM,+BAA+B;AAC5C,sBAAgB,KAAK,IAAIC,oDAAuB,CAAC;AACjD,+BAAyB,aAAa;AACtC,YAAM;;KAET,CAAC;KACF;GACF,UAAU,IAAIV,sBAAS,OAAO,aAAa,0BAA0B;AACnE,WAAO,MAAM,4BAA4B;AAEzC,QAAI,KAAK,OAAO,YAAY;AAC1B,cAAS,OAAO,MAAM,EAAE,YAAY,MAAM,CAAC;AAC3C,YAAO,MAAM,0CAA0C;AAEvD,sBAAiB;MACf,SAAS;MACT;MACA;MACA,WAAW,WAAW;AACpB,cAAO,MACL,EAAE,QAAQ,OAAO,QAAQ,EACzB,6BACD;AACD,yBAAkB,KAAK,OAAO;AAG9B,WAAI,OAAO,WAAW,UAAU;AAE9B,wBAAgB,KACd,IAAIW,2EAAoC,EACtC,kBAAkB,OAAO,QAC1B,CAAC,CACH;AACD,4BAAoB,KAAK,EACvB,QAAQ,6DAA6D,OAAO,UAC7E,CAAC;AAGF,yBAAiB,wCACCL,2BAAa;SAC3B,sCAAc;SACd,2BAAW,IAAI,MAAM;SACrB,SAAS,OAAO;SAChB,MAAMT,0BAAY;SACnB,CAAC,CACH;AACD,qCAA6B,eAAe;;;MAGhD,UAAU,QAAQ;AAChB,cAAO,MAAM,EAAE,KAAK,EAAE,iCAAiC;AACvD,uBAAgB,KACd,IAAIY,4DAAqB,EACvB,aAAa,8DACd,CAAC,CACH;AACD,2BAAoB,KAAK,EACvB,QAAQ,6DACT,CAAC;AAGF,mCAA4B,IAAI;;MAEnC,CAAC;;IAGJ,IAAI;AAEJ,WAAO,MAAM,oDAAoD;AAEjE,8BAA0B,YAAY,UAAU;KAC9C,MAAM,OAAO,UAAU;AACrB,cAAQ,MAAM,MAAd;OACE,KAAKR,iCAAkB,UACrB;OAIF,KAAKA,iCAAkB;QAErB,MAAM,2BAA2B,YAAY,0BAEhC,MAAoB,MAAM,MAAM,uBAGxC,MACC,EACE,EAAE,SAASA,iCAAkB,kBAC5B,EAAU,aAAa,MAAM,WAEnC,oBAGE,MACC,EAAE,QAAQA,iCAAkB,sBAC3B,EAAU,aAAa,MAAM,UACjC,CACF;QAGD,MAAM,sBAAsB,IAAIW,cAE7B;QAEH,MAAM,YAAY,MAAM;AAExB,oBAAY;SACV,IAAI;SACJ,iBAAiB,MAAM;SACvB,iCAAuB,oBAAoB;SAC3C,2BAAW,IAAI,MAAM;SACrB,MAAMf,0BAAY;SAClB,SAAS,IAAIG,sBACX,OAAO,eAAe,sBAAsB;AAC1C,iBAAO,MAAM,wCAAwC;UAErD,MAAM,aAAuB,EAAE;UAC/B,IAAI;AAEJ,8BACG,4BACc,iBACR,EAAE,iBACF,EAAE,QAAQ,gBAAgB;AAC7B,kBAAO,MACL;YAAE;YAAQ;YAAW,EACrB,6BACD;AAED,+BAAoB,4CACFa,iDAAqB,EAAE,QAAQ,CAAC,CACjD;AAED,2BAAgB,KACd,IAAIC,wEAAiC,EACnC,WACD,CAAC,CACH;AACD,8BAAmB;AACnB,6BAAkB,aAAa;YAC/B,CACH,CACA,WAAW;AAEd,iBAAO,MACL,6CACD;AAED,6BAAmB,yBAAyB,UAAU;WACpD,MAAM,OAAO,MAAoB;AAC/B,gBAAI,EAAE,QAAQb,iCAAkB,oBAAoB;AAClD,mBAAM,cAAc,EAAE,QAAQ;AAC9B,wBAAW,KAAK,EAAE,QAAQ;;;WAG9B,QAAQ,QAAQ;AACd,mBAAO,MACL,EAAE,KAAK,EACP,uCACD;AACD,gCAAoB,KAAK;aACvB,QAAQ;aACR;aACD,CAAC;AACF,+BAAmB;AACnB,8BAAkB,aAAa;;WAEjC,gBAAgB;AACd,mBAAO,MAAM,wCAAwC;AACrD,gCAAoB,KAAK,IAAIM,kDAAsB,CAAC;AACpD,+BAAmB;AACnB,8BAAkB,aAAa;AAE/B,2BAAe,4CACGD,2BAAa;aAC3B,IAAI;aACJ,2BAAW,IAAI,MAAM;aACrB,SAAS,WAAW,KAAK,GAAG;aAC5B,MAAMT,0BAAY;aACnB,CAAC,CACH;;WAEJ,CAAC;WAEL;SACF,CAAC;AACF;OAIF,KAAKI,iCAAkB;AACrB,eAAO,MAAM,wCAAwC;QACrD,MAAM,gCAAgC,YAAY,0BACrC,MAAoB,MAAM,MAAM,uBAGxC,MACC,EACE,EAAE,SAASA,iCAAkB,sBAC5B,EAAU,qBAAqB,MAAM,mBAE3C,oBAGE,MACC,EAAE,QAAQA,iCAAkB,uBAC3B,EAAU,qBAAqB,MAAM,kBACzC,CACF;QACD,MAAM,2BAA2B,IAAIW,cAElC;AACH,oBAAY;SACV,IAAI,MAAM;SACV,iBAAiB,MAAM;SACvB,iCAAuB,yBAAyB;SAChD,2BAAW,IAAI,MAAM;SACrB,MAAM,MAAM;SACZ,WAAW,IAAIZ,sBACb,OAAO,oBAAoB,2BAA2B;AACpD,iBAAO,MAAM,2CAA2C;UAExD,MAAM,iBAA2B,EAAE;UACnC,IAAI;AAEJ,gDACE,8BAA8B,UAAU;WACtC,MAAM,OAAO,MAAoB;AAC/B,gBACE,EAAE,QAAQC,iCAAkB,qBAC5B;AACA,mBAAM,mBAAmB,EAAE,KAAK;AAChC,4BAAe,KAAK,EAAE,KAAK;;;WAG/B,QAAQ,QAAQ;AACd,mBAAO,MACL,EAAE,KAAK,EACP,4CACD;AACD,qCAAyB,4CACPY,iDAAqB,EACnC,QACE,yEACH,CAAC,CACH;AACD,oCAAwB;AACxB,iDAAqC,aAAa;;WAEpD,gBAAgB;AACd,mBAAO,MACL,6CACD;AACD,qCAAyB,KACvB,IAAIN,kDAAsB,CAC3B;AACD,oCAAwB;AACxB,iDAAqC,aAAa;AAElD,2BAAe,4CACGC,sCAAwB;aACtC,IAAI,MAAM;aACV,2BAAW,IAAI,MAAM;aACrB,MAAM,MAAM;aACZ,WAAW,eAAe,KAAK,GAAG;aACnC,CAAC,CACH;;WAEJ,CAAC;WAEP;SACF,CAAC;AACF;OAIF,KAAKP,iCAAkB;AACrB,eAAO,MACL,EAAE,QAAQ,MAAM,QAAQ,EACxB,yCACD;AACD,oBAAY;SACV,IAAI,YAAY,MAAM;SACtB,QAAQ,IAAIM,kDAAsB;SAClC,2BAAW,IAAI,MAAM;SACrB,mBAAmB,MAAM;SACzB,YAAY,MAAM;SAClB,QAAQ,MAAM;SACf,CAAC;AAEF,uBAAe,4CACGQ,6BAAe;SAC7B,IAAI,YAAY,MAAM;SACtB,2BAAW,IAAI,MAAM;SACrB,mBAAmB,MAAM;SACzB,YAAY,MAAM;SAClB,QAAQ,MAAM;SACf,CAAC,CACH;AACD;OAIF,KAAKd,iCAAkB;AACrB,eAAO,MAAM,EAAE,OAAO,EAAE,+BAA+B;AACvD,oBAAY;SACV,sCAAc;SACd,QAAQ,IAAIM,kDAAsB;SAClC,UAAU,MAAM;SAChB,WAAW,MAAM;SACjB,UAAU,MAAM;SAChB,OAAO,MAAM;SACb,QAAQ,MAAM;SACd,OAAO,MAAM;SACb,SAAS,MAAM;SACf,MAAMV,0BAAY;SAClB,2BAAW,IAAI,MAAM;SACtB,CAAC;AACF,uBAAe,4CACGmB,iCAAmB;SACjC,sCAAc;SACd,UAAU,MAAM;SAChB,WAAW,MAAM;SACjB,UAAU,MAAM;SAChB,OAAO,MAAM;SACb,QAAQ,MAAM;SACd,OAAO,MAAM;SACb,SAAS,MAAM;SACf,MAAMnB,0BAAY;SAClB,2BAAW,IAAI,MAAM;SACtB,CAAC,CACH;AACD;;;KAGN,QAAQ,QAAQ;AAEd,UACE,eAAeoB,sCACf,eAAeC,8CACd,eAAe,SACd,IAAI,QACJ,IAAI,KAAK,SAAS,aAAa,IACjC,KAAK,YAAY,YACjB;AACA,uBAAgB,KACd,IAAIT,4DAAqB;QACvB,aAAa,IAAI,WAAW;QAE5B,eAAe;SACb,MAAM,IAAI,QAAQ,IAAI,YAAY;SAClC,YAAY,IAAI,cAAc,IAAI,YAAY;SAC9C,UAAU,IAAI,YAAY,IAAI,YAAY;SAC1C,YAAY,IAAI,cAAc,IAAI,YAAY;SAC9C,mBACE,IAAI,qBACJ,IAAI,YAAY;SAClB,YAAY,IAAI;SACjB;QACF,CAAC,CACH;AACD,gCAAyB,aAAa;AACtC,mCAA4B,IAAI;AAChC,8BAAuB;AACvB;;AAGF,sBAAgB,KACd,IAAIA,4DAAqB,EACvB,aAAa,qDACd,CAAC,CACH;AACD,+BAAyB,aAAa;AACtC,6BAAuB;AAEvB,kCAA4B,IAAI;;KAElC,UAAU,YAAY;AACpB,aAAO,MAAM,yBAAyB;AACtC,UAAI,KAAK,OAAO,YAAY;AAC1B,cAAO,MACL,uDACD;AACD,sCAAqB,kBAAkB;;AAEzC,sBAAgB,KAAK,IAAIC,oDAAuB,CAAC;AACjD,+BAAyB,aAAa;AACtC,6BAAuB;AAEvB,mCAA6B,eAAe;;KAE/C,CAAC;KACF;GACH;;;;+BA1oBU,OAAO;;;;;;+BAKPS,4CAAe;iEACA;;;;;;kCAiBZC,8CAAgB;iEAExB;gEACD,OAAO;gEACP,oBAAoBC,mCAAmB,EAAE,UAAU,MAAM,CAAC;;;;;;;;;gFA7BnDD,8CAAgB"}
|
|
1
|
+
{"version":3,"file":"copilot.resolver.cjs","names":["MessageRole","GraphQLError","ReplaySubject","Repeater","RuntimeEventTypes","LangGraphEventTypes","LangGraphInterruptEvent","RuntimeMetaEventName","CopilotKitLangGraphInterruptEvent","TextMessage","SuccessMessageStatus","ActionExecutionMessage","UnknownErrorResponse","SuccessResponseStatus","GuardrailsValidationFailureResponse","Subject","resolveMessageId","FailedMessageStatus","MessageStreamInterruptedResponse","ResultMessage","AgentStateMessage","CopilotKitError","CopilotKitLowLevelError","AgentsResponse","CopilotResponse","GraphQLJSONObject"],"sources":["../../../src/graphql/resolvers/copilot.resolver.ts"],"sourcesContent":["import { Arg, Ctx, Mutation, Query, Resolver } from \"type-graphql\";\nimport {\n ReplaySubject,\n Subject,\n Subscription,\n filter,\n finalize,\n firstValueFrom,\n shareReplay,\n skipWhile,\n take,\n takeWhile,\n tap,\n} from \"rxjs\";\nimport { GenerateCopilotResponseInput } from \"../inputs/generate-copilot-response.input\";\nimport { CopilotResponse } from \"../types/copilot-response.type\";\nimport {\n CopilotKitLangGraphInterruptEvent,\n LangGraphInterruptEvent,\n} from \"../types/meta-events.type\";\nimport { ActionInputAvailability, MessageRole } from \"../types/enums\";\nimport { Repeater } from \"graphql-yoga\";\nimport type {\n CopilotRequestContextProperties,\n GraphQLContext,\n} from \"../../lib/integrations\";\nimport {\n RuntimeEvent,\n RuntimeEventTypes,\n RuntimeMetaEventName,\n} from \"../../service-adapters/events\";\nimport {\n FailedMessageStatus,\n MessageStatusCode,\n MessageStatusUnion,\n SuccessMessageStatus,\n} from \"../types/message-status.type\";\nimport {\n ResponseStatusUnion,\n SuccessResponseStatus,\n} from \"../types/response-status.type\";\nimport { GraphQLJSONObject } from \"graphql-scalars\";\nimport { plainToInstance } from \"class-transformer\";\nimport { GuardrailsResult } from \"../types/guardrails-result.type\";\nimport { GraphQLError } from \"graphql\";\nimport {\n GuardrailsValidationFailureResponse,\n MessageStreamInterruptedResponse,\n UnknownErrorResponse,\n} from \"../../utils\";\nimport {\n ActionExecutionMessage,\n AgentStateMessage,\n Message,\n MessageType,\n ResultMessage,\n TextMessage,\n} from \"../types/converted\";\nimport telemetry from \"../../lib/telemetry-client\";\nimport { randomId } from \"@copilotkit/shared\";\nimport { resolveMessageId } from \"./resolve-message-id\";\nimport { AgentsResponse } from \"../types/agents-response.type\";\nimport { LangGraphEventTypes } from \"../../agents/langgraph/events\";\nimport {\n CopilotKitError,\n CopilotKitLowLevelError,\n isStructuredCopilotKitError,\n} from \"@copilotkit/shared\";\nimport { CopilotRuntime } from \"../../lib\";\n\nconst invokeGuardrails = async ({\n baseUrl,\n copilotCloudPublicApiKey,\n data,\n onResult,\n onError,\n}: {\n baseUrl: string;\n copilotCloudPublicApiKey: string;\n data: GenerateCopilotResponseInput;\n onResult: (result: GuardrailsResult) => void;\n onError: (err: Error) => void;\n}) => {\n if (\n data.messages.length &&\n data.messages[data.messages.length - 1].textMessage?.role ===\n MessageRole.user\n ) {\n const messages = data.messages\n .filter(\n (m) =>\n m.textMessage !== undefined &&\n (m.textMessage.role === MessageRole.user ||\n m.textMessage.role === MessageRole.assistant),\n )\n .map((m) => ({\n role: m.textMessage!.role,\n content: m.textMessage.content,\n }));\n\n const lastMessage = messages[messages.length - 1];\n const restOfMessages = messages.slice(0, -1);\n\n const body = {\n input: lastMessage.content,\n validTopics: data.cloud.guardrails.inputValidationRules.allowList,\n invalidTopics: data.cloud.guardrails.inputValidationRules.denyList,\n messages: restOfMessages,\n };\n\n const guardrailsResult = await fetch(`${baseUrl}/guardrails/validate`, {\n method: \"POST\",\n headers: {\n \"Content-Type\": \"application/json\",\n \"X-CopilotCloud-Public-API-Key\": copilotCloudPublicApiKey,\n },\n body: JSON.stringify(body),\n });\n\n if (guardrailsResult.ok) {\n const resultJson: GuardrailsResult = await guardrailsResult.json();\n onResult(resultJson);\n } else {\n onError(await guardrailsResult.json());\n }\n }\n};\n\n@Resolver(() => CopilotResponse)\nexport class CopilotResolver {\n @Query(() => String)\n async hello() {\n return \"Hello World\";\n }\n\n @Query(() => AgentsResponse)\n async availableAgents(@Ctx() ctx: GraphQLContext) {\n let logger = ctx.logger.child({\n component: \"CopilotResolver.availableAgents\",\n });\n\n logger.debug(\"Processing\");\n const agentsWithEndpoints = [];\n\n logger.debug(\"Event source created, creating response\");\n\n return {\n agents: agentsWithEndpoints.map(\n ({ endpoint, ...agentWithoutEndpoint }) => agentWithoutEndpoint,\n ),\n };\n }\n\n @Mutation(() => CopilotResponse)\n async generateCopilotResponse(\n @Ctx() ctx: GraphQLContext,\n @Arg(\"data\") data: GenerateCopilotResponseInput,\n @Arg(\"properties\", () => GraphQLJSONObject, { nullable: true })\n properties?: CopilotRequestContextProperties,\n ) {\n telemetry.capture(\"oss.runtime.copilot_request_created\", {\n \"cloud.guardrails.enabled\": data.cloud?.guardrails !== undefined,\n requestType: data.metadata.requestType,\n \"cloud.api_key_provided\": !!ctx.request.headers.get(\n \"x-copilotcloud-public-api-key\",\n ),\n ...(ctx.request.headers.get(\"x-copilotcloud-public-api-key\")\n ? {\n \"cloud.public_api_key\": ctx.request.headers.get(\n \"x-copilotcloud-public-api-key\",\n ),\n }\n : {}),\n ...(ctx._copilotkit.baseUrl\n ? {\n \"cloud.base_url\": ctx._copilotkit.baseUrl,\n }\n : {\n \"cloud.base_url\": \"https://api.cloud.copilotkit.ai\",\n }),\n });\n\n let logger = ctx.logger.child({\n component: \"CopilotResolver.generateCopilotResponse\",\n });\n logger.debug({ data }, \"Generating Copilot response\");\n\n if (properties) {\n logger.debug(\"Properties provided, merging with context properties\");\n ctx.properties = { ...ctx.properties, ...properties };\n }\n\n const copilotRuntime = ctx._copilotkit.runtime as unknown as CopilotRuntime;\n const serviceAdapter = ctx._copilotkit.serviceAdapter;\n\n let copilotCloudPublicApiKey: string | null = null;\n let copilotCloudBaseUrl: string;\n\n // Extract publicApiKey from headers for both cloud and non-cloud requests\n // This enables onTrace functionality regardless of cloud configuration\n const publicApiKeyFromHeaders = ctx.request.headers.get(\n \"x-copilotcloud-public-api-key\",\n );\n if (publicApiKeyFromHeaders) {\n copilotCloudPublicApiKey = publicApiKeyFromHeaders;\n }\n\n if (data.cloud) {\n logger = logger.child({ cloud: true });\n logger.debug(\n \"Cloud configuration provided, checking for public API key in headers\",\n );\n\n if (!copilotCloudPublicApiKey) {\n logger.error(\"Public API key not found in headers\");\n\n throw new GraphQLError(\n \"X-CopilotCloud-Public-API-Key header is required\",\n );\n }\n\n if (process.env.COPILOT_CLOUD_BASE_URL) {\n copilotCloudBaseUrl = process.env.COPILOT_CLOUD_BASE_URL;\n } else if (ctx._copilotkit.cloud?.baseUrl) {\n copilotCloudBaseUrl = ctx._copilotkit.cloud?.baseUrl;\n } else {\n copilotCloudBaseUrl = \"https://api.cloud.copilotkit.ai\";\n }\n\n logger = logger.child({ copilotCloudBaseUrl });\n }\n\n logger.debug(\"Setting up subjects\");\n const responseStatus$ = new ReplaySubject<typeof ResponseStatusUnion>();\n const interruptStreaming$ = new ReplaySubject<{\n reason: string;\n messageId?: string;\n }>();\n const guardrailsResult$ = new ReplaySubject<GuardrailsResult>();\n\n let outputMessages: Message[] = [];\n let resolveOutputMessagesPromise: (messages: Message[]) => void;\n let rejectOutputMessagesPromise: (err: Error) => void;\n\n const outputMessagesPromise = new Promise<Message[]>((resolve, reject) => {\n resolveOutputMessagesPromise = resolve;\n rejectOutputMessagesPromise = reject;\n });\n\n if (copilotCloudPublicApiKey) {\n ctx.properties[\"copilotCloudPublicApiKey\"] = copilotCloudPublicApiKey;\n }\n\n logger.debug(\"Processing\");\n let runtimeResponse;\n\n const {\n eventSource,\n threadId = randomId(),\n runId,\n serverSideActions,\n actionInputsWithoutAgents,\n extensions,\n } = runtimeResponse;\n\n logger.debug(\"Event source created, creating response\");\n // run and process the event stream\n const eventStream = eventSource\n .processRuntimeEvents({\n serverSideActions,\n guardrailsResult$: data.cloud?.guardrails ? guardrailsResult$ : null,\n actionInputsWithoutAgents: actionInputsWithoutAgents.filter(\n // TODO-AGENTS: do not exclude ALL server side actions\n (action) =>\n !serverSideActions.find(\n (serverSideAction) => serverSideAction.name == action.name,\n ),\n ),\n threadId,\n })\n .pipe(\n // shareReplay() ensures that later subscribers will see the whole stream instead of\n // just the events that were emitted after the subscriber was added.\n shareReplay(),\n finalize(() => {\n logger.debug(\"Event stream finalized\");\n }),\n );\n\n const response = {\n threadId,\n runId,\n status: firstValueFrom(responseStatus$),\n extensions,\n metaEvents: new Repeater(async (push, stop) => {\n let eventStreamSubscription: Subscription;\n\n eventStreamSubscription = eventStream.subscribe({\n next: async (event) => {\n if (event.type != RuntimeEventTypes.MetaEvent) {\n return;\n }\n switch (event.name) {\n // @ts-ignore\n case LangGraphEventTypes.OnInterrupt:\n push(\n plainToInstance(LangGraphInterruptEvent, {\n // @ts-ignore\n type: event.type,\n // @ts-ignore\n name: RuntimeMetaEventName.LangGraphInterruptEvent,\n // @ts-ignore\n value: event.value,\n }),\n );\n break;\n case RuntimeMetaEventName.LangGraphInterruptEvent:\n push(\n plainToInstance(LangGraphInterruptEvent, {\n type: event.type,\n name: event.name,\n value: event.value,\n }),\n );\n break;\n case RuntimeMetaEventName.CopilotKitLangGraphInterruptEvent:\n push(\n plainToInstance(CopilotKitLangGraphInterruptEvent, {\n type: event.type,\n name: event.name,\n data: {\n value: event.data.value,\n messages: event.data.messages.map((message) => {\n if (\n message.type === \"TextMessage\" ||\n (\"content\" in message && \"role\" in message)\n ) {\n return plainToInstance(TextMessage, {\n id: message.id,\n createdAt: new Date(),\n content: [(message as TextMessage).content],\n role: (message as TextMessage).role,\n status: new SuccessMessageStatus(),\n });\n }\n if (\"arguments\" in message) {\n return plainToInstance(ActionExecutionMessage, {\n name: message.name,\n id: message.id,\n arguments: [JSON.stringify(message.arguments)],\n createdAt: new Date(),\n status: new SuccessMessageStatus(),\n });\n }\n throw new Error(\n \"Unknown message in metaEvents copilot resolver\",\n );\n }),\n },\n }),\n );\n break;\n }\n },\n error: (err) => {\n // For structured CopilotKit errors, set proper error response status\n if (\n err?.name?.includes(\"CopilotKit\") ||\n err?.extensions?.visibility\n ) {\n responseStatus$.next(\n new UnknownErrorResponse({\n description: err.message || \"Agent error occurred\",\n }),\n );\n } else {\n responseStatus$.next(\n new UnknownErrorResponse({\n description: `An unknown error has occurred in the event stream`,\n }),\n );\n }\n\n eventStreamSubscription?.unsubscribe();\n stop();\n },\n complete: async () => {\n logger.debug(\"Meta events stream completed\");\n responseStatus$.next(new SuccessResponseStatus());\n eventStreamSubscription?.unsubscribe();\n stop();\n },\n });\n }),\n messages: new Repeater(async (pushMessage, stopStreamingMessages) => {\n logger.debug(\"Messages repeater created\");\n\n if (data.cloud?.guardrails) {\n logger = logger.child({ guardrails: true });\n logger.debug(\"Guardrails is enabled, validating input\");\n\n invokeGuardrails({\n baseUrl: copilotCloudBaseUrl,\n copilotCloudPublicApiKey,\n data,\n onResult: (result) => {\n logger.debug(\n { status: result.status },\n \"Guardrails validation done\",\n );\n guardrailsResult$.next(result);\n\n // Guardrails validation failed\n if (result.status === \"denied\") {\n // send the reason to the client and interrupt streaming\n responseStatus$.next(\n new GuardrailsValidationFailureResponse({\n guardrailsReason: result.reason,\n }),\n );\n interruptStreaming$.next({\n reason: `Interrupted due to Guardrails validation failure. Reason: ${result.reason}`,\n });\n\n // resolve messages promise to the middleware\n outputMessages = [\n plainToInstance(TextMessage, {\n id: randomId(),\n createdAt: new Date(),\n content: result.reason,\n role: MessageRole.assistant,\n }),\n ];\n resolveOutputMessagesPromise(outputMessages);\n }\n },\n onError: (err) => {\n logger.error({ err }, \"Error in guardrails validation\");\n responseStatus$.next(\n new UnknownErrorResponse({\n description: `An unknown error has occurred in the guardrails validation`,\n }),\n );\n interruptStreaming$.next({\n reason: `Interrupted due to unknown error in guardrails validation`,\n });\n\n // reject the middleware promise\n rejectOutputMessagesPromise(err);\n },\n });\n }\n\n let eventStreamSubscription: Subscription;\n\n logger.debug(\"Event stream created, subscribing to event stream\");\n\n eventStreamSubscription = eventStream.subscribe({\n next: async (event) => {\n switch (event.type) {\n case RuntimeEventTypes.MetaEvent:\n break;\n ////////////////////////////////\n // TextMessageStart\n ////////////////////////////////\n case RuntimeEventTypes.TextMessageStart:\n // create a sub stream that contains the message content\n const textMessageContentStream = eventStream.pipe(\n // skip until this message start event\n skipWhile((e: RuntimeEvent) => e !== event),\n // take until the message end event\n takeWhile(\n (e: RuntimeEvent) =>\n !(\n e.type === RuntimeEventTypes.TextMessageEnd &&\n (e as any).messageId == event.messageId\n ),\n ),\n // filter out any other message events or message ids\n filter(\n (e: RuntimeEvent) =>\n e.type == RuntimeEventTypes.TextMessageContent &&\n (e as any).messageId == event.messageId,\n ),\n );\n\n // signal when we are done streaming\n const streamingTextStatus = new Subject<\n typeof MessageStatusUnion\n >();\n\n const messageId = resolveMessageId(event.messageId);\n // push the new message\n pushMessage({\n id: messageId,\n parentMessageId: event.parentMessageId,\n status: firstValueFrom(streamingTextStatus),\n createdAt: new Date(),\n role: MessageRole.assistant,\n content: new Repeater(\n async (pushTextChunk, stopStreamingText) => {\n logger.debug(\"Text message content repeater created\");\n\n const textChunks: string[] = [];\n let textSubscription: Subscription;\n\n interruptStreaming$\n .pipe(\n shareReplay(),\n take(1),\n tap(({ reason, messageId }) => {\n logger.debug(\n { reason, messageId },\n \"Text streaming interrupted\",\n );\n\n streamingTextStatus.next(\n plainToInstance(FailedMessageStatus, { reason }),\n );\n\n responseStatus$.next(\n new MessageStreamInterruptedResponse({\n messageId,\n }),\n );\n stopStreamingText();\n textSubscription?.unsubscribe();\n }),\n )\n .subscribe();\n\n logger.debug(\n \"Subscribing to text message content stream\",\n );\n\n textSubscription = textMessageContentStream.subscribe({\n next: async (e: RuntimeEvent) => {\n if (e.type == RuntimeEventTypes.TextMessageContent) {\n await pushTextChunk(e.content);\n textChunks.push(e.content);\n }\n },\n error: (err) => {\n logger.error(\n { err },\n \"Error in text message content stream\",\n );\n interruptStreaming$.next({\n reason: \"Error streaming message content\",\n messageId,\n });\n stopStreamingText();\n textSubscription?.unsubscribe();\n },\n complete: () => {\n logger.debug(\"Text message content stream completed\");\n streamingTextStatus.next(new SuccessMessageStatus());\n stopStreamingText();\n textSubscription?.unsubscribe();\n\n outputMessages.push(\n plainToInstance(TextMessage, {\n id: messageId,\n createdAt: new Date(),\n content: textChunks.join(\"\"),\n role: MessageRole.assistant,\n }),\n );\n },\n });\n },\n ),\n });\n break;\n ////////////////////////////////\n // ActionExecutionStart\n ////////////////////////////////\n case RuntimeEventTypes.ActionExecutionStart:\n logger.debug(\"Action execution start event received\");\n const actionExecutionArgumentStream = eventStream.pipe(\n skipWhile((e: RuntimeEvent) => e !== event),\n // take until the action execution end event\n takeWhile(\n (e: RuntimeEvent) =>\n !(\n e.type === RuntimeEventTypes.ActionExecutionEnd &&\n (e as any).actionExecutionId == event.actionExecutionId\n ),\n ),\n // filter out any other action execution events or action execution ids\n filter(\n (e: RuntimeEvent) =>\n e.type == RuntimeEventTypes.ActionExecutionArgs &&\n (e as any).actionExecutionId == event.actionExecutionId,\n ),\n );\n const streamingArgumentsStatus = new Subject<\n typeof MessageStatusUnion\n >();\n pushMessage({\n id: event.actionExecutionId,\n parentMessageId: event.parentMessageId,\n status: firstValueFrom(streamingArgumentsStatus),\n createdAt: new Date(),\n name: event.actionName,\n arguments: new Repeater(\n async (pushArgumentsChunk, stopStreamingArguments) => {\n logger.debug(\"Action execution argument stream created\");\n\n const argumentChunks: string[] = [];\n let actionExecutionArgumentSubscription: Subscription;\n\n actionExecutionArgumentSubscription =\n actionExecutionArgumentStream.subscribe({\n next: async (e: RuntimeEvent) => {\n if (\n e.type == RuntimeEventTypes.ActionExecutionArgs\n ) {\n await pushArgumentsChunk(e.args);\n argumentChunks.push(e.args);\n }\n },\n error: (err) => {\n logger.error(\n { err },\n \"Error in action execution argument stream\",\n );\n streamingArgumentsStatus.next(\n plainToInstance(FailedMessageStatus, {\n reason:\n \"An unknown error has occurred in the action execution argument stream\",\n }),\n );\n stopStreamingArguments();\n actionExecutionArgumentSubscription?.unsubscribe();\n },\n complete: () => {\n logger.debug(\n \"Action execution argument stream completed\",\n );\n streamingArgumentsStatus.next(\n new SuccessMessageStatus(),\n );\n stopStreamingArguments();\n actionExecutionArgumentSubscription?.unsubscribe();\n\n outputMessages.push(\n plainToInstance(ActionExecutionMessage, {\n id: event.actionExecutionId,\n createdAt: new Date(),\n name: event.actionName,\n arguments: argumentChunks.join(\"\"),\n }),\n );\n },\n });\n },\n ),\n });\n break;\n ////////////////////////////////\n // ActionExecutionResult\n ////////////////////////////////\n case RuntimeEventTypes.ActionExecutionResult:\n logger.debug(\n { result: event.result },\n \"Action execution result event received\",\n );\n pushMessage({\n id: \"result-\" + event.actionExecutionId,\n status: new SuccessMessageStatus(),\n createdAt: new Date(),\n actionExecutionId: event.actionExecutionId,\n actionName: event.actionName,\n result: event.result,\n });\n\n outputMessages.push(\n plainToInstance(ResultMessage, {\n id: \"result-\" + event.actionExecutionId,\n createdAt: new Date(),\n actionExecutionId: event.actionExecutionId,\n actionName: event.actionName,\n result: event.result,\n }),\n );\n break;\n ////////////////////////////////\n // AgentStateMessage\n ////////////////////////////////\n case RuntimeEventTypes.AgentStateMessage:\n logger.debug({ event }, \"Agent message event received\");\n pushMessage({\n id: randomId(),\n status: new SuccessMessageStatus(),\n threadId: event.threadId,\n agentName: event.agentName,\n nodeName: event.nodeName,\n runId: event.runId,\n active: event.active,\n state: event.state,\n running: event.running,\n role: MessageRole.assistant,\n createdAt: new Date(),\n });\n outputMessages.push(\n plainToInstance(AgentStateMessage, {\n id: randomId(),\n threadId: event.threadId,\n agentName: event.agentName,\n nodeName: event.nodeName,\n runId: event.runId,\n active: event.active,\n state: event.state,\n running: event.running,\n role: MessageRole.assistant,\n createdAt: new Date(),\n }),\n );\n break;\n }\n },\n error: (err) => {\n // For structured CopilotKit errors, set proper error response status\n if (\n err instanceof CopilotKitError ||\n err instanceof CopilotKitLowLevelError ||\n (err instanceof Error &&\n err.name &&\n err.name.includes(\"CopilotKit\")) ||\n err?.extensions?.visibility\n ) {\n responseStatus$.next(\n new UnknownErrorResponse({\n description: err.message || \"Agent error occurred\",\n // Include original error information for frontend to extract\n originalError: {\n code: err.code || err.extensions?.code,\n statusCode: err.statusCode || err.extensions?.statusCode,\n severity: err.severity || err.extensions?.severity,\n visibility: err.visibility || err.extensions?.visibility,\n originalErrorType:\n err.originalErrorType ||\n err.extensions?.originalErrorType,\n extensions: err.extensions,\n },\n }),\n );\n eventStreamSubscription?.unsubscribe();\n rejectOutputMessagesPromise(err);\n stopStreamingMessages();\n return;\n }\n\n responseStatus$.next(\n new UnknownErrorResponse({\n description: `An unknown error has occurred in the event stream`,\n }),\n );\n eventStreamSubscription?.unsubscribe();\n stopStreamingMessages();\n\n rejectOutputMessagesPromise(err);\n },\n complete: async () => {\n logger.debug(\"Event stream completed\");\n if (data.cloud?.guardrails) {\n logger.debug(\n \"Guardrails is enabled, waiting for guardrails result\",\n );\n await firstValueFrom(guardrailsResult$);\n }\n responseStatus$.next(new SuccessResponseStatus());\n eventStreamSubscription?.unsubscribe();\n stopStreamingMessages();\n\n resolveOutputMessagesPromise(outputMessages);\n },\n });\n }),\n };\n\n return response;\n }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAsEA,MAAM,mBAAmB,OAAO,EAC9B,SACA,0BACA,MACA,UACA,cAOI;AACJ,KACE,KAAK,SAAS,UACd,KAAK,SAAS,KAAK,SAAS,SAAS,GAAG,aAAa,SACnDA,0BAAY,MACd;EACA,MAAM,WAAW,KAAK,SACnB,QACE,MACC,EAAE,gBAAgB,WACjB,EAAE,YAAY,SAASA,0BAAY,QAClC,EAAE,YAAY,SAASA,0BAAY,WACxC,CACA,KAAK,OAAO;GACX,MAAM,EAAE,YAAa;GACrB,SAAS,EAAE,YAAY;GACxB,EAAE;EAEL,MAAM,cAAc,SAAS,SAAS,SAAS;EAC/C,MAAM,iBAAiB,SAAS,MAAM,GAAG,GAAG;EAE5C,MAAM,OAAO;GACX,OAAO,YAAY;GACnB,aAAa,KAAK,MAAM,WAAW,qBAAqB;GACxD,eAAe,KAAK,MAAM,WAAW,qBAAqB;GAC1D,UAAU;GACX;EAED,MAAM,mBAAmB,MAAM,MAAM,GAAG,QAAQ,uBAAuB;GACrE,QAAQ;GACR,SAAS;IACP,gBAAgB;IAChB,iCAAiC;IAClC;GACD,MAAM,KAAK,UAAU,KAAK;GAC3B,CAAC;AAEF,MAAI,iBAAiB,GAEnB,UADqC,MAAM,iBAAiB,MAAM,CAC9C;MAEpB,SAAQ,MAAM,iBAAiB,MAAM,CAAC;;;AAMrC,4BAAM,gBAAgB;CAC3B,MACM,QAAQ;AACZ,SAAO;;CAGT,MACM,gBAAgB,AAAO,KAAqB;EAChD,IAAI,SAAS,IAAI,OAAO,MAAM,EAC5B,WAAW,mCACZ,CAAC;AAEF,SAAO,MAAM,aAAa;EAC1B,MAAM,sBAAsB,EAAE;AAE9B,SAAO,MAAM,0CAA0C;AAEvD,SAAO,EACL,QAAQ,oBAAoB,KACzB,EAAE,UAAU,GAAG,2BAA2B,qBAC5C,EACF;;CAGH,MACM,wBACJ,AAAO,KACP,AAAa,MACb,AACA,YACA;AACA,mCAAU,QAAQ,uCAAuC;GACvD,4BAA4B,KAAK,OAAO,eAAe;GACvD,aAAa,KAAK,SAAS;GAC3B,0BAA0B,CAAC,CAAC,IAAI,QAAQ,QAAQ,IAC9C,gCACD;GACD,GAAI,IAAI,QAAQ,QAAQ,IAAI,gCAAgC,GACxD,EACE,wBAAwB,IAAI,QAAQ,QAAQ,IAC1C,gCACD,EACF,GACD,EAAE;GACN,GAAI,IAAI,YAAY,UAChB,EACE,kBAAkB,IAAI,YAAY,SACnC,GACD,EACE,kBAAkB,mCACnB;GACN,CAAC;EAEF,IAAI,SAAS,IAAI,OAAO,MAAM,EAC5B,WAAW,2CACZ,CAAC;AACF,SAAO,MAAM,EAAE,MAAM,EAAE,8BAA8B;AAErD,MAAI,YAAY;AACd,UAAO,MAAM,uDAAuD;AACpE,OAAI,aAAa;IAAE,GAAG,IAAI;IAAY,GAAG;IAAY;;AAGhC,MAAI,YAAY;AAChB,MAAI,YAAY;EAEvC,IAAI,2BAA0C;EAC9C,IAAI;EAIJ,MAAM,0BAA0B,IAAI,QAAQ,QAAQ,IAClD,gCACD;AACD,MAAI,wBACF,4BAA2B;AAG7B,MAAI,KAAK,OAAO;AACd,YAAS,OAAO,MAAM,EAAE,OAAO,MAAM,CAAC;AACtC,UAAO,MACL,uEACD;AAED,OAAI,CAAC,0BAA0B;AAC7B,WAAO,MAAM,sCAAsC;AAEnD,UAAM,IAAIC,qBACR,mDACD;;AAGH,OAAI,QAAQ,IAAI,uBACd,uBAAsB,QAAQ,IAAI;YACzB,IAAI,YAAY,OAAO,QAChC,uBAAsB,IAAI,YAAY,OAAO;OAE7C,uBAAsB;AAGxB,YAAS,OAAO,MAAM,EAAE,qBAAqB,CAAC;;AAGhD,SAAO,MAAM,sBAAsB;EACnC,MAAM,kBAAkB,IAAIC,oBAA2C;EACvE,MAAM,sBAAsB,IAAIA,oBAG5B;EACJ,MAAM,oBAAoB,IAAIA,oBAAiC;EAE/D,IAAI,iBAA4B,EAAE;EAClC,IAAI;EACJ,IAAI;AAE0B,MAAI,SAAoB,SAAS,WAAW;AACxE,kCAA+B;AAC/B,iCAA8B;IAC9B;AAEF,MAAI,yBACF,KAAI,WAAW,8BAA8B;AAG/C,SAAO,MAAM,aAAa;EAC1B,IAAI;EAEJ,MAAM,EACJ,aACA,6CAAqB,EACrB,OACA,mBACA,2BACA,eACE;AAEJ,SAAO,MAAM,0CAA0C;EAEvD,MAAM,cAAc,YACjB,qBAAqB;GACpB;GACA,mBAAmB,KAAK,OAAO,aAAa,oBAAoB;GAChE,2BAA2B,0BAA0B,QAElD,WACC,CAAC,kBAAkB,MAChB,qBAAqB,iBAAiB,QAAQ,OAAO,KACvD,CACJ;GACD;GACD,CAAC,CACD,4BAGc,2BACE;AACb,UAAO,MAAM,yBAAyB;IACtC,CACH;AA+eH,SA7eiB;GACf;GACA;GACA,iCAAuB,gBAAgB;GACvC;GACA,YAAY,IAAIC,sBAAS,OAAO,MAAM,SAAS;IAC7C,IAAI;AAEJ,8BAA0B,YAAY,UAAU;KAC9C,MAAM,OAAO,UAAU;AACrB,UAAI,MAAM,QAAQC,iCAAkB,UAClC;AAEF,cAAQ,MAAM,MAAd;OAEE,KAAKC,qCAAoB;AACvB,oDACkBC,kDAAyB;SAEvC,MAAM,MAAM;SAEZ,MAAMC,oCAAqB;SAE3B,OAAO,MAAM;SACd,CAAC,CACH;AACD;OACF,KAAKA,oCAAqB;AACxB,oDACkBD,kDAAyB;SACvC,MAAM,MAAM;SACZ,MAAM,MAAM;SACZ,OAAO,MAAM;SACd,CAAC,CACH;AACD;OACF,KAAKC,oCAAqB;AACxB,oDACkBC,4DAAmC;SACjD,MAAM,MAAM;SACZ,MAAM,MAAM;SACZ,MAAM;UACJ,OAAO,MAAM,KAAK;UAClB,UAAU,MAAM,KAAK,SAAS,KAAK,YAAY;AAC7C,eACE,QAAQ,SAAS,iBAChB,aAAa,WAAW,UAAU,QAEnC,+CAAuBC,2BAAa;YAClC,IAAI,QAAQ;YACZ,2BAAW,IAAI,MAAM;YACrB,SAAS,CAAE,QAAwB,QAAQ;YAC3C,MAAO,QAAwB;YAC/B,QAAQ,IAAIC,kDAAsB;YACnC,CAAC;AAEJ,eAAI,eAAe,QACjB,+CAAuBC,sCAAwB;YAC7C,MAAM,QAAQ;YACd,IAAI,QAAQ;YACZ,WAAW,CAAC,KAAK,UAAU,QAAQ,UAAU,CAAC;YAC9C,2BAAW,IAAI,MAAM;YACrB,QAAQ,IAAID,kDAAsB;YACnC,CAAC;AAEJ,iBAAM,IAAI,MACR,iDACD;YACD;UACH;SACF,CAAC,CACH;AACD;;;KAGN,QAAQ,QAAQ;AAEd,UACE,KAAK,MAAM,SAAS,aAAa,IACjC,KAAK,YAAY,WAEjB,iBAAgB,KACd,IAAIE,4DAAqB,EACvB,aAAa,IAAI,WAAW,wBAC7B,CAAC,CACH;UAED,iBAAgB,KACd,IAAIA,4DAAqB,EACvB,aAAa,qDACd,CAAC,CACH;AAGH,+BAAyB,aAAa;AACtC,YAAM;;KAER,UAAU,YAAY;AACpB,aAAO,MAAM,+BAA+B;AAC5C,sBAAgB,KAAK,IAAIC,oDAAuB,CAAC;AACjD,+BAAyB,aAAa;AACtC,YAAM;;KAET,CAAC;KACF;GACF,UAAU,IAAIV,sBAAS,OAAO,aAAa,0BAA0B;AACnE,WAAO,MAAM,4BAA4B;AAEzC,QAAI,KAAK,OAAO,YAAY;AAC1B,cAAS,OAAO,MAAM,EAAE,YAAY,MAAM,CAAC;AAC3C,YAAO,MAAM,0CAA0C;AAEvD,sBAAiB;MACf,SAAS;MACT;MACA;MACA,WAAW,WAAW;AACpB,cAAO,MACL,EAAE,QAAQ,OAAO,QAAQ,EACzB,6BACD;AACD,yBAAkB,KAAK,OAAO;AAG9B,WAAI,OAAO,WAAW,UAAU;AAE9B,wBAAgB,KACd,IAAIW,2EAAoC,EACtC,kBAAkB,OAAO,QAC1B,CAAC,CACH;AACD,4BAAoB,KAAK,EACvB,QAAQ,6DAA6D,OAAO,UAC7E,CAAC;AAGF,yBAAiB,wCACCL,2BAAa;SAC3B,sCAAc;SACd,2BAAW,IAAI,MAAM;SACrB,SAAS,OAAO;SAChB,MAAMT,0BAAY;SACnB,CAAC,CACH;AACD,qCAA6B,eAAe;;;MAGhD,UAAU,QAAQ;AAChB,cAAO,MAAM,EAAE,KAAK,EAAE,iCAAiC;AACvD,uBAAgB,KACd,IAAIY,4DAAqB,EACvB,aAAa,8DACd,CAAC,CACH;AACD,2BAAoB,KAAK,EACvB,QAAQ,6DACT,CAAC;AAGF,mCAA4B,IAAI;;MAEnC,CAAC;;IAGJ,IAAI;AAEJ,WAAO,MAAM,oDAAoD;AAEjE,8BAA0B,YAAY,UAAU;KAC9C,MAAM,OAAO,UAAU;AACrB,cAAQ,MAAM,MAAd;OACE,KAAKR,iCAAkB,UACrB;OAIF,KAAKA,iCAAkB;QAErB,MAAM,2BAA2B,YAAY,0BAEhC,MAAoB,MAAM,MAAM,uBAGxC,MACC,EACE,EAAE,SAASA,iCAAkB,kBAC5B,EAAU,aAAa,MAAM,WAEnC,oBAGE,MACC,EAAE,QAAQA,iCAAkB,sBAC3B,EAAU,aAAa,MAAM,UACjC,CACF;QAGD,MAAM,sBAAsB,IAAIW,cAE7B;QAEH,MAAM,YAAYC,4CAAiB,MAAM,UAAU;AAEnD,oBAAY;SACV,IAAI;SACJ,iBAAiB,MAAM;SACvB,iCAAuB,oBAAoB;SAC3C,2BAAW,IAAI,MAAM;SACrB,MAAMhB,0BAAY;SAClB,SAAS,IAAIG,sBACX,OAAO,eAAe,sBAAsB;AAC1C,iBAAO,MAAM,wCAAwC;UAErD,MAAM,aAAuB,EAAE;UAC/B,IAAI;AAEJ,8BACG,4BACc,iBACR,EAAE,iBACF,EAAE,QAAQ,gBAAgB;AAC7B,kBAAO,MACL;YAAE;YAAQ;YAAW,EACrB,6BACD;AAED,+BAAoB,4CACFc,iDAAqB,EAAE,QAAQ,CAAC,CACjD;AAED,2BAAgB,KACd,IAAIC,wEAAiC,EACnC,WACD,CAAC,CACH;AACD,8BAAmB;AACnB,6BAAkB,aAAa;YAC/B,CACH,CACA,WAAW;AAEd,iBAAO,MACL,6CACD;AAED,6BAAmB,yBAAyB,UAAU;WACpD,MAAM,OAAO,MAAoB;AAC/B,gBAAI,EAAE,QAAQd,iCAAkB,oBAAoB;AAClD,mBAAM,cAAc,EAAE,QAAQ;AAC9B,wBAAW,KAAK,EAAE,QAAQ;;;WAG9B,QAAQ,QAAQ;AACd,mBAAO,MACL,EAAE,KAAK,EACP,uCACD;AACD,gCAAoB,KAAK;aACvB,QAAQ;aACR;aACD,CAAC;AACF,+BAAmB;AACnB,8BAAkB,aAAa;;WAEjC,gBAAgB;AACd,mBAAO,MAAM,wCAAwC;AACrD,gCAAoB,KAAK,IAAIM,kDAAsB,CAAC;AACpD,+BAAmB;AACnB,8BAAkB,aAAa;AAE/B,2BAAe,4CACGD,2BAAa;aAC3B,IAAI;aACJ,2BAAW,IAAI,MAAM;aACrB,SAAS,WAAW,KAAK,GAAG;aAC5B,MAAMT,0BAAY;aACnB,CAAC,CACH;;WAEJ,CAAC;WAEL;SACF,CAAC;AACF;OAIF,KAAKI,iCAAkB;AACrB,eAAO,MAAM,wCAAwC;QACrD,MAAM,gCAAgC,YAAY,0BACrC,MAAoB,MAAM,MAAM,uBAGxC,MACC,EACE,EAAE,SAASA,iCAAkB,sBAC5B,EAAU,qBAAqB,MAAM,mBAE3C,oBAGE,MACC,EAAE,QAAQA,iCAAkB,uBAC3B,EAAU,qBAAqB,MAAM,kBACzC,CACF;QACD,MAAM,2BAA2B,IAAIW,cAElC;AACH,oBAAY;SACV,IAAI,MAAM;SACV,iBAAiB,MAAM;SACvB,iCAAuB,yBAAyB;SAChD,2BAAW,IAAI,MAAM;SACrB,MAAM,MAAM;SACZ,WAAW,IAAIZ,sBACb,OAAO,oBAAoB,2BAA2B;AACpD,iBAAO,MAAM,2CAA2C;UAExD,MAAM,iBAA2B,EAAE;UACnC,IAAI;AAEJ,gDACE,8BAA8B,UAAU;WACtC,MAAM,OAAO,MAAoB;AAC/B,gBACE,EAAE,QAAQC,iCAAkB,qBAC5B;AACA,mBAAM,mBAAmB,EAAE,KAAK;AAChC,4BAAe,KAAK,EAAE,KAAK;;;WAG/B,QAAQ,QAAQ;AACd,mBAAO,MACL,EAAE,KAAK,EACP,4CACD;AACD,qCAAyB,4CACPa,iDAAqB,EACnC,QACE,yEACH,CAAC,CACH;AACD,oCAAwB;AACxB,iDAAqC,aAAa;;WAEpD,gBAAgB;AACd,mBAAO,MACL,6CACD;AACD,qCAAyB,KACvB,IAAIP,kDAAsB,CAC3B;AACD,oCAAwB;AACxB,iDAAqC,aAAa;AAElD,2BAAe,4CACGC,sCAAwB;aACtC,IAAI,MAAM;aACV,2BAAW,IAAI,MAAM;aACrB,MAAM,MAAM;aACZ,WAAW,eAAe,KAAK,GAAG;aACnC,CAAC,CACH;;WAEJ,CAAC;WAEP;SACF,CAAC;AACF;OAIF,KAAKP,iCAAkB;AACrB,eAAO,MACL,EAAE,QAAQ,MAAM,QAAQ,EACxB,yCACD;AACD,oBAAY;SACV,IAAI,YAAY,MAAM;SACtB,QAAQ,IAAIM,kDAAsB;SAClC,2BAAW,IAAI,MAAM;SACrB,mBAAmB,MAAM;SACzB,YAAY,MAAM;SAClB,QAAQ,MAAM;SACf,CAAC;AAEF,uBAAe,4CACGS,6BAAe;SAC7B,IAAI,YAAY,MAAM;SACtB,2BAAW,IAAI,MAAM;SACrB,mBAAmB,MAAM;SACzB,YAAY,MAAM;SAClB,QAAQ,MAAM;SACf,CAAC,CACH;AACD;OAIF,KAAKf,iCAAkB;AACrB,eAAO,MAAM,EAAE,OAAO,EAAE,+BAA+B;AACvD,oBAAY;SACV,sCAAc;SACd,QAAQ,IAAIM,kDAAsB;SAClC,UAAU,MAAM;SAChB,WAAW,MAAM;SACjB,UAAU,MAAM;SAChB,OAAO,MAAM;SACb,QAAQ,MAAM;SACd,OAAO,MAAM;SACb,SAAS,MAAM;SACf,MAAMV,0BAAY;SAClB,2BAAW,IAAI,MAAM;SACtB,CAAC;AACF,uBAAe,4CACGoB,iCAAmB;SACjC,sCAAc;SACd,UAAU,MAAM;SAChB,WAAW,MAAM;SACjB,UAAU,MAAM;SAChB,OAAO,MAAM;SACb,QAAQ,MAAM;SACd,OAAO,MAAM;SACb,SAAS,MAAM;SACf,MAAMpB,0BAAY;SAClB,2BAAW,IAAI,MAAM;SACtB,CAAC,CACH;AACD;;;KAGN,QAAQ,QAAQ;AAEd,UACE,eAAeqB,sCACf,eAAeC,8CACd,eAAe,SACd,IAAI,QACJ,IAAI,KAAK,SAAS,aAAa,IACjC,KAAK,YAAY,YACjB;AACA,uBAAgB,KACd,IAAIV,4DAAqB;QACvB,aAAa,IAAI,WAAW;QAE5B,eAAe;SACb,MAAM,IAAI,QAAQ,IAAI,YAAY;SAClC,YAAY,IAAI,cAAc,IAAI,YAAY;SAC9C,UAAU,IAAI,YAAY,IAAI,YAAY;SAC1C,YAAY,IAAI,cAAc,IAAI,YAAY;SAC9C,mBACE,IAAI,qBACJ,IAAI,YAAY;SAClB,YAAY,IAAI;SACjB;QACF,CAAC,CACH;AACD,gCAAyB,aAAa;AACtC,mCAA4B,IAAI;AAChC,8BAAuB;AACvB;;AAGF,sBAAgB,KACd,IAAIA,4DAAqB,EACvB,aAAa,qDACd,CAAC,CACH;AACD,+BAAyB,aAAa;AACtC,6BAAuB;AAEvB,kCAA4B,IAAI;;KAElC,UAAU,YAAY;AACpB,aAAO,MAAM,yBAAyB;AACtC,UAAI,KAAK,OAAO,YAAY;AAC1B,cAAO,MACL,uDACD;AACD,sCAAqB,kBAAkB;;AAEzC,sBAAgB,KAAK,IAAIC,oDAAuB,CAAC;AACjD,+BAAyB,aAAa;AACtC,6BAAuB;AAEvB,mCAA6B,eAAe;;KAE/C,CAAC;KACF;GACH;;;;+BA1oBU,OAAO;;;;;;+BAKPU,4CAAe;iEACA;;;;;;kCAiBZC,8CAAgB;iEAExB;gEACD,OAAO;gEACP,oBAAoBC,mCAAmB,EAAE,UAAU,MAAM,CAAC;;;;;;;;;gFA7BnDD,8CAAgB"}
|
|
@@ -11,6 +11,7 @@ import { ActionExecutionMessage, AgentStateMessage, ResultMessage, TextMessage }
|
|
|
11
11
|
import { RuntimeEventTypes, RuntimeMetaEventName } from "../../service-adapters/events.mjs";
|
|
12
12
|
import { GuardrailsValidationFailureResponse, MessageStreamInterruptedResponse, UnknownErrorResponse } from "../../utils/failed-response-status-reasons.mjs";
|
|
13
13
|
import telemetryClient from "../../lib/telemetry-client.mjs";
|
|
14
|
+
import { resolveMessageId } from "./resolve-message-id.mjs";
|
|
14
15
|
import { AgentsResponse } from "../types/agents-response.type.mjs";
|
|
15
16
|
import { LangGraphEventTypes } from "../../agents/langgraph/events.mjs";
|
|
16
17
|
import { __decorateParam } from "../../_virtual/_@oxc-project_runtime@0.112.0/helpers/decorateParam.mjs";
|
|
@@ -228,7 +229,7 @@ let CopilotResolver = class CopilotResolver {
|
|
|
228
229
|
case RuntimeEventTypes.TextMessageStart:
|
|
229
230
|
const textMessageContentStream = eventStream.pipe(skipWhile((e) => e !== event), takeWhile((e) => !(e.type === RuntimeEventTypes.TextMessageEnd && e.messageId == event.messageId)), filter((e) => e.type == RuntimeEventTypes.TextMessageContent && e.messageId == event.messageId));
|
|
230
231
|
const streamingTextStatus = new Subject();
|
|
231
|
-
const messageId = event.messageId;
|
|
232
|
+
const messageId = resolveMessageId(event.messageId);
|
|
232
233
|
pushMessage({
|
|
233
234
|
id: messageId,
|
|
234
235
|
parentMessageId: event.parentMessageId,
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"copilot.resolver.mjs","names":[],"sources":["../../../src/graphql/resolvers/copilot.resolver.ts"],"sourcesContent":["import { Arg, Ctx, Mutation, Query, Resolver } from \"type-graphql\";\nimport {\n ReplaySubject,\n Subject,\n Subscription,\n filter,\n finalize,\n firstValueFrom,\n shareReplay,\n skipWhile,\n take,\n takeWhile,\n tap,\n} from \"rxjs\";\nimport { GenerateCopilotResponseInput } from \"../inputs/generate-copilot-response.input\";\nimport { CopilotResponse } from \"../types/copilot-response.type\";\nimport {\n CopilotKitLangGraphInterruptEvent,\n LangGraphInterruptEvent,\n} from \"../types/meta-events.type\";\nimport { ActionInputAvailability, MessageRole } from \"../types/enums\";\nimport { Repeater } from \"graphql-yoga\";\nimport type {\n CopilotRequestContextProperties,\n GraphQLContext,\n} from \"../../lib/integrations\";\nimport {\n RuntimeEvent,\n RuntimeEventTypes,\n RuntimeMetaEventName,\n} from \"../../service-adapters/events\";\nimport {\n FailedMessageStatus,\n MessageStatusCode,\n MessageStatusUnion,\n SuccessMessageStatus,\n} from \"../types/message-status.type\";\nimport {\n ResponseStatusUnion,\n SuccessResponseStatus,\n} from \"../types/response-status.type\";\nimport { GraphQLJSONObject } from \"graphql-scalars\";\nimport { plainToInstance } from \"class-transformer\";\nimport { GuardrailsResult } from \"../types/guardrails-result.type\";\nimport { GraphQLError } from \"graphql\";\nimport {\n GuardrailsValidationFailureResponse,\n MessageStreamInterruptedResponse,\n UnknownErrorResponse,\n} from \"../../utils\";\nimport {\n ActionExecutionMessage,\n AgentStateMessage,\n Message,\n MessageType,\n ResultMessage,\n TextMessage,\n} from \"../types/converted\";\nimport telemetry from \"../../lib/telemetry-client\";\nimport { randomId } from \"@copilotkit/shared\";\nimport { AgentsResponse } from \"../types/agents-response.type\";\nimport { LangGraphEventTypes } from \"../../agents/langgraph/events\";\nimport {\n CopilotKitError,\n CopilotKitLowLevelError,\n isStructuredCopilotKitError,\n} from \"@copilotkit/shared\";\nimport { CopilotRuntime } from \"../../lib\";\n\nconst invokeGuardrails = async ({\n baseUrl,\n copilotCloudPublicApiKey,\n data,\n onResult,\n onError,\n}: {\n baseUrl: string;\n copilotCloudPublicApiKey: string;\n data: GenerateCopilotResponseInput;\n onResult: (result: GuardrailsResult) => void;\n onError: (err: Error) => void;\n}) => {\n if (\n data.messages.length &&\n data.messages[data.messages.length - 1].textMessage?.role ===\n MessageRole.user\n ) {\n const messages = data.messages\n .filter(\n (m) =>\n m.textMessage !== undefined &&\n (m.textMessage.role === MessageRole.user ||\n m.textMessage.role === MessageRole.assistant),\n )\n .map((m) => ({\n role: m.textMessage!.role,\n content: m.textMessage.content,\n }));\n\n const lastMessage = messages[messages.length - 1];\n const restOfMessages = messages.slice(0, -1);\n\n const body = {\n input: lastMessage.content,\n validTopics: data.cloud.guardrails.inputValidationRules.allowList,\n invalidTopics: data.cloud.guardrails.inputValidationRules.denyList,\n messages: restOfMessages,\n };\n\n const guardrailsResult = await fetch(`${baseUrl}/guardrails/validate`, {\n method: \"POST\",\n headers: {\n \"Content-Type\": \"application/json\",\n \"X-CopilotCloud-Public-API-Key\": copilotCloudPublicApiKey,\n },\n body: JSON.stringify(body),\n });\n\n if (guardrailsResult.ok) {\n const resultJson: GuardrailsResult = await guardrailsResult.json();\n onResult(resultJson);\n } else {\n onError(await guardrailsResult.json());\n }\n }\n};\n\n@Resolver(() => CopilotResponse)\nexport class CopilotResolver {\n @Query(() => String)\n async hello() {\n return \"Hello World\";\n }\n\n @Query(() => AgentsResponse)\n async availableAgents(@Ctx() ctx: GraphQLContext) {\n let logger = ctx.logger.child({\n component: \"CopilotResolver.availableAgents\",\n });\n\n logger.debug(\"Processing\");\n const agentsWithEndpoints = [];\n\n logger.debug(\"Event source created, creating response\");\n\n return {\n agents: agentsWithEndpoints.map(\n ({ endpoint, ...agentWithoutEndpoint }) => agentWithoutEndpoint,\n ),\n };\n }\n\n @Mutation(() => CopilotResponse)\n async generateCopilotResponse(\n @Ctx() ctx: GraphQLContext,\n @Arg(\"data\") data: GenerateCopilotResponseInput,\n @Arg(\"properties\", () => GraphQLJSONObject, { nullable: true })\n properties?: CopilotRequestContextProperties,\n ) {\n telemetry.capture(\"oss.runtime.copilot_request_created\", {\n \"cloud.guardrails.enabled\": data.cloud?.guardrails !== undefined,\n requestType: data.metadata.requestType,\n \"cloud.api_key_provided\": !!ctx.request.headers.get(\n \"x-copilotcloud-public-api-key\",\n ),\n ...(ctx.request.headers.get(\"x-copilotcloud-public-api-key\")\n ? {\n \"cloud.public_api_key\": ctx.request.headers.get(\n \"x-copilotcloud-public-api-key\",\n ),\n }\n : {}),\n ...(ctx._copilotkit.baseUrl\n ? {\n \"cloud.base_url\": ctx._copilotkit.baseUrl,\n }\n : {\n \"cloud.base_url\": \"https://api.cloud.copilotkit.ai\",\n }),\n });\n\n let logger = ctx.logger.child({\n component: \"CopilotResolver.generateCopilotResponse\",\n });\n logger.debug({ data }, \"Generating Copilot response\");\n\n if (properties) {\n logger.debug(\"Properties provided, merging with context properties\");\n ctx.properties = { ...ctx.properties, ...properties };\n }\n\n const copilotRuntime = ctx._copilotkit.runtime as unknown as CopilotRuntime;\n const serviceAdapter = ctx._copilotkit.serviceAdapter;\n\n let copilotCloudPublicApiKey: string | null = null;\n let copilotCloudBaseUrl: string;\n\n // Extract publicApiKey from headers for both cloud and non-cloud requests\n // This enables onTrace functionality regardless of cloud configuration\n const publicApiKeyFromHeaders = ctx.request.headers.get(\n \"x-copilotcloud-public-api-key\",\n );\n if (publicApiKeyFromHeaders) {\n copilotCloudPublicApiKey = publicApiKeyFromHeaders;\n }\n\n if (data.cloud) {\n logger = logger.child({ cloud: true });\n logger.debug(\n \"Cloud configuration provided, checking for public API key in headers\",\n );\n\n if (!copilotCloudPublicApiKey) {\n logger.error(\"Public API key not found in headers\");\n\n throw new GraphQLError(\n \"X-CopilotCloud-Public-API-Key header is required\",\n );\n }\n\n if (process.env.COPILOT_CLOUD_BASE_URL) {\n copilotCloudBaseUrl = process.env.COPILOT_CLOUD_BASE_URL;\n } else if (ctx._copilotkit.cloud?.baseUrl) {\n copilotCloudBaseUrl = ctx._copilotkit.cloud?.baseUrl;\n } else {\n copilotCloudBaseUrl = \"https://api.cloud.copilotkit.ai\";\n }\n\n logger = logger.child({ copilotCloudBaseUrl });\n }\n\n logger.debug(\"Setting up subjects\");\n const responseStatus$ = new ReplaySubject<typeof ResponseStatusUnion>();\n const interruptStreaming$ = new ReplaySubject<{\n reason: string;\n messageId?: string;\n }>();\n const guardrailsResult$ = new ReplaySubject<GuardrailsResult>();\n\n let outputMessages: Message[] = [];\n let resolveOutputMessagesPromise: (messages: Message[]) => void;\n let rejectOutputMessagesPromise: (err: Error) => void;\n\n const outputMessagesPromise = new Promise<Message[]>((resolve, reject) => {\n resolveOutputMessagesPromise = resolve;\n rejectOutputMessagesPromise = reject;\n });\n\n if (copilotCloudPublicApiKey) {\n ctx.properties[\"copilotCloudPublicApiKey\"] = copilotCloudPublicApiKey;\n }\n\n logger.debug(\"Processing\");\n let runtimeResponse;\n\n const {\n eventSource,\n threadId = randomId(),\n runId,\n serverSideActions,\n actionInputsWithoutAgents,\n extensions,\n } = runtimeResponse;\n\n logger.debug(\"Event source created, creating response\");\n // run and process the event stream\n const eventStream = eventSource\n .processRuntimeEvents({\n serverSideActions,\n guardrailsResult$: data.cloud?.guardrails ? guardrailsResult$ : null,\n actionInputsWithoutAgents: actionInputsWithoutAgents.filter(\n // TODO-AGENTS: do not exclude ALL server side actions\n (action) =>\n !serverSideActions.find(\n (serverSideAction) => serverSideAction.name == action.name,\n ),\n ),\n threadId,\n })\n .pipe(\n // shareReplay() ensures that later subscribers will see the whole stream instead of\n // just the events that were emitted after the subscriber was added.\n shareReplay(),\n finalize(() => {\n logger.debug(\"Event stream finalized\");\n }),\n );\n\n const response = {\n threadId,\n runId,\n status: firstValueFrom(responseStatus$),\n extensions,\n metaEvents: new Repeater(async (push, stop) => {\n let eventStreamSubscription: Subscription;\n\n eventStreamSubscription = eventStream.subscribe({\n next: async (event) => {\n if (event.type != RuntimeEventTypes.MetaEvent) {\n return;\n }\n switch (event.name) {\n // @ts-ignore\n case LangGraphEventTypes.OnInterrupt:\n push(\n plainToInstance(LangGraphInterruptEvent, {\n // @ts-ignore\n type: event.type,\n // @ts-ignore\n name: RuntimeMetaEventName.LangGraphInterruptEvent,\n // @ts-ignore\n value: event.value,\n }),\n );\n break;\n case RuntimeMetaEventName.LangGraphInterruptEvent:\n push(\n plainToInstance(LangGraphInterruptEvent, {\n type: event.type,\n name: event.name,\n value: event.value,\n }),\n );\n break;\n case RuntimeMetaEventName.CopilotKitLangGraphInterruptEvent:\n push(\n plainToInstance(CopilotKitLangGraphInterruptEvent, {\n type: event.type,\n name: event.name,\n data: {\n value: event.data.value,\n messages: event.data.messages.map((message) => {\n if (\n message.type === \"TextMessage\" ||\n (\"content\" in message && \"role\" in message)\n ) {\n return plainToInstance(TextMessage, {\n id: message.id,\n createdAt: new Date(),\n content: [(message as TextMessage).content],\n role: (message as TextMessage).role,\n status: new SuccessMessageStatus(),\n });\n }\n if (\"arguments\" in message) {\n return plainToInstance(ActionExecutionMessage, {\n name: message.name,\n id: message.id,\n arguments: [JSON.stringify(message.arguments)],\n createdAt: new Date(),\n status: new SuccessMessageStatus(),\n });\n }\n throw new Error(\n \"Unknown message in metaEvents copilot resolver\",\n );\n }),\n },\n }),\n );\n break;\n }\n },\n error: (err) => {\n // For structured CopilotKit errors, set proper error response status\n if (\n err?.name?.includes(\"CopilotKit\") ||\n err?.extensions?.visibility\n ) {\n responseStatus$.next(\n new UnknownErrorResponse({\n description: err.message || \"Agent error occurred\",\n }),\n );\n } else {\n responseStatus$.next(\n new UnknownErrorResponse({\n description: `An unknown error has occurred in the event stream`,\n }),\n );\n }\n\n eventStreamSubscription?.unsubscribe();\n stop();\n },\n complete: async () => {\n logger.debug(\"Meta events stream completed\");\n responseStatus$.next(new SuccessResponseStatus());\n eventStreamSubscription?.unsubscribe();\n stop();\n },\n });\n }),\n messages: new Repeater(async (pushMessage, stopStreamingMessages) => {\n logger.debug(\"Messages repeater created\");\n\n if (data.cloud?.guardrails) {\n logger = logger.child({ guardrails: true });\n logger.debug(\"Guardrails is enabled, validating input\");\n\n invokeGuardrails({\n baseUrl: copilotCloudBaseUrl,\n copilotCloudPublicApiKey,\n data,\n onResult: (result) => {\n logger.debug(\n { status: result.status },\n \"Guardrails validation done\",\n );\n guardrailsResult$.next(result);\n\n // Guardrails validation failed\n if (result.status === \"denied\") {\n // send the reason to the client and interrupt streaming\n responseStatus$.next(\n new GuardrailsValidationFailureResponse({\n guardrailsReason: result.reason,\n }),\n );\n interruptStreaming$.next({\n reason: `Interrupted due to Guardrails validation failure. Reason: ${result.reason}`,\n });\n\n // resolve messages promise to the middleware\n outputMessages = [\n plainToInstance(TextMessage, {\n id: randomId(),\n createdAt: new Date(),\n content: result.reason,\n role: MessageRole.assistant,\n }),\n ];\n resolveOutputMessagesPromise(outputMessages);\n }\n },\n onError: (err) => {\n logger.error({ err }, \"Error in guardrails validation\");\n responseStatus$.next(\n new UnknownErrorResponse({\n description: `An unknown error has occurred in the guardrails validation`,\n }),\n );\n interruptStreaming$.next({\n reason: `Interrupted due to unknown error in guardrails validation`,\n });\n\n // reject the middleware promise\n rejectOutputMessagesPromise(err);\n },\n });\n }\n\n let eventStreamSubscription: Subscription;\n\n logger.debug(\"Event stream created, subscribing to event stream\");\n\n eventStreamSubscription = eventStream.subscribe({\n next: async (event) => {\n switch (event.type) {\n case RuntimeEventTypes.MetaEvent:\n break;\n ////////////////////////////////\n // TextMessageStart\n ////////////////////////////////\n case RuntimeEventTypes.TextMessageStart:\n // create a sub stream that contains the message content\n const textMessageContentStream = eventStream.pipe(\n // skip until this message start event\n skipWhile((e: RuntimeEvent) => e !== event),\n // take until the message end event\n takeWhile(\n (e: RuntimeEvent) =>\n !(\n e.type === RuntimeEventTypes.TextMessageEnd &&\n (e as any).messageId == event.messageId\n ),\n ),\n // filter out any other message events or message ids\n filter(\n (e: RuntimeEvent) =>\n e.type == RuntimeEventTypes.TextMessageContent &&\n (e as any).messageId == event.messageId,\n ),\n );\n\n // signal when we are done streaming\n const streamingTextStatus = new Subject<\n typeof MessageStatusUnion\n >();\n\n const messageId = event.messageId;\n // push the new message\n pushMessage({\n id: messageId,\n parentMessageId: event.parentMessageId,\n status: firstValueFrom(streamingTextStatus),\n createdAt: new Date(),\n role: MessageRole.assistant,\n content: new Repeater(\n async (pushTextChunk, stopStreamingText) => {\n logger.debug(\"Text message content repeater created\");\n\n const textChunks: string[] = [];\n let textSubscription: Subscription;\n\n interruptStreaming$\n .pipe(\n shareReplay(),\n take(1),\n tap(({ reason, messageId }) => {\n logger.debug(\n { reason, messageId },\n \"Text streaming interrupted\",\n );\n\n streamingTextStatus.next(\n plainToInstance(FailedMessageStatus, { reason }),\n );\n\n responseStatus$.next(\n new MessageStreamInterruptedResponse({\n messageId,\n }),\n );\n stopStreamingText();\n textSubscription?.unsubscribe();\n }),\n )\n .subscribe();\n\n logger.debug(\n \"Subscribing to text message content stream\",\n );\n\n textSubscription = textMessageContentStream.subscribe({\n next: async (e: RuntimeEvent) => {\n if (e.type == RuntimeEventTypes.TextMessageContent) {\n await pushTextChunk(e.content);\n textChunks.push(e.content);\n }\n },\n error: (err) => {\n logger.error(\n { err },\n \"Error in text message content stream\",\n );\n interruptStreaming$.next({\n reason: \"Error streaming message content\",\n messageId,\n });\n stopStreamingText();\n textSubscription?.unsubscribe();\n },\n complete: () => {\n logger.debug(\"Text message content stream completed\");\n streamingTextStatus.next(new SuccessMessageStatus());\n stopStreamingText();\n textSubscription?.unsubscribe();\n\n outputMessages.push(\n plainToInstance(TextMessage, {\n id: messageId,\n createdAt: new Date(),\n content: textChunks.join(\"\"),\n role: MessageRole.assistant,\n }),\n );\n },\n });\n },\n ),\n });\n break;\n ////////////////////////////////\n // ActionExecutionStart\n ////////////////////////////////\n case RuntimeEventTypes.ActionExecutionStart:\n logger.debug(\"Action execution start event received\");\n const actionExecutionArgumentStream = eventStream.pipe(\n skipWhile((e: RuntimeEvent) => e !== event),\n // take until the action execution end event\n takeWhile(\n (e: RuntimeEvent) =>\n !(\n e.type === RuntimeEventTypes.ActionExecutionEnd &&\n (e as any).actionExecutionId == event.actionExecutionId\n ),\n ),\n // filter out any other action execution events or action execution ids\n filter(\n (e: RuntimeEvent) =>\n e.type == RuntimeEventTypes.ActionExecutionArgs &&\n (e as any).actionExecutionId == event.actionExecutionId,\n ),\n );\n const streamingArgumentsStatus = new Subject<\n typeof MessageStatusUnion\n >();\n pushMessage({\n id: event.actionExecutionId,\n parentMessageId: event.parentMessageId,\n status: firstValueFrom(streamingArgumentsStatus),\n createdAt: new Date(),\n name: event.actionName,\n arguments: new Repeater(\n async (pushArgumentsChunk, stopStreamingArguments) => {\n logger.debug(\"Action execution argument stream created\");\n\n const argumentChunks: string[] = [];\n let actionExecutionArgumentSubscription: Subscription;\n\n actionExecutionArgumentSubscription =\n actionExecutionArgumentStream.subscribe({\n next: async (e: RuntimeEvent) => {\n if (\n e.type == RuntimeEventTypes.ActionExecutionArgs\n ) {\n await pushArgumentsChunk(e.args);\n argumentChunks.push(e.args);\n }\n },\n error: (err) => {\n logger.error(\n { err },\n \"Error in action execution argument stream\",\n );\n streamingArgumentsStatus.next(\n plainToInstance(FailedMessageStatus, {\n reason:\n \"An unknown error has occurred in the action execution argument stream\",\n }),\n );\n stopStreamingArguments();\n actionExecutionArgumentSubscription?.unsubscribe();\n },\n complete: () => {\n logger.debug(\n \"Action execution argument stream completed\",\n );\n streamingArgumentsStatus.next(\n new SuccessMessageStatus(),\n );\n stopStreamingArguments();\n actionExecutionArgumentSubscription?.unsubscribe();\n\n outputMessages.push(\n plainToInstance(ActionExecutionMessage, {\n id: event.actionExecutionId,\n createdAt: new Date(),\n name: event.actionName,\n arguments: argumentChunks.join(\"\"),\n }),\n );\n },\n });\n },\n ),\n });\n break;\n ////////////////////////////////\n // ActionExecutionResult\n ////////////////////////////////\n case RuntimeEventTypes.ActionExecutionResult:\n logger.debug(\n { result: event.result },\n \"Action execution result event received\",\n );\n pushMessage({\n id: \"result-\" + event.actionExecutionId,\n status: new SuccessMessageStatus(),\n createdAt: new Date(),\n actionExecutionId: event.actionExecutionId,\n actionName: event.actionName,\n result: event.result,\n });\n\n outputMessages.push(\n plainToInstance(ResultMessage, {\n id: \"result-\" + event.actionExecutionId,\n createdAt: new Date(),\n actionExecutionId: event.actionExecutionId,\n actionName: event.actionName,\n result: event.result,\n }),\n );\n break;\n ////////////////////////////////\n // AgentStateMessage\n ////////////////////////////////\n case RuntimeEventTypes.AgentStateMessage:\n logger.debug({ event }, \"Agent message event received\");\n pushMessage({\n id: randomId(),\n status: new SuccessMessageStatus(),\n threadId: event.threadId,\n agentName: event.agentName,\n nodeName: event.nodeName,\n runId: event.runId,\n active: event.active,\n state: event.state,\n running: event.running,\n role: MessageRole.assistant,\n createdAt: new Date(),\n });\n outputMessages.push(\n plainToInstance(AgentStateMessage, {\n id: randomId(),\n threadId: event.threadId,\n agentName: event.agentName,\n nodeName: event.nodeName,\n runId: event.runId,\n active: event.active,\n state: event.state,\n running: event.running,\n role: MessageRole.assistant,\n createdAt: new Date(),\n }),\n );\n break;\n }\n },\n error: (err) => {\n // For structured CopilotKit errors, set proper error response status\n if (\n err instanceof CopilotKitError ||\n err instanceof CopilotKitLowLevelError ||\n (err instanceof Error &&\n err.name &&\n err.name.includes(\"CopilotKit\")) ||\n err?.extensions?.visibility\n ) {\n responseStatus$.next(\n new UnknownErrorResponse({\n description: err.message || \"Agent error occurred\",\n // Include original error information for frontend to extract\n originalError: {\n code: err.code || err.extensions?.code,\n statusCode: err.statusCode || err.extensions?.statusCode,\n severity: err.severity || err.extensions?.severity,\n visibility: err.visibility || err.extensions?.visibility,\n originalErrorType:\n err.originalErrorType ||\n err.extensions?.originalErrorType,\n extensions: err.extensions,\n },\n }),\n );\n eventStreamSubscription?.unsubscribe();\n rejectOutputMessagesPromise(err);\n stopStreamingMessages();\n return;\n }\n\n responseStatus$.next(\n new UnknownErrorResponse({\n description: `An unknown error has occurred in the event stream`,\n }),\n );\n eventStreamSubscription?.unsubscribe();\n stopStreamingMessages();\n\n rejectOutputMessagesPromise(err);\n },\n complete: async () => {\n logger.debug(\"Event stream completed\");\n if (data.cloud?.guardrails) {\n logger.debug(\n \"Guardrails is enabled, waiting for guardrails result\",\n );\n await firstValueFrom(guardrailsResult$);\n }\n responseStatus$.next(new SuccessResponseStatus());\n eventStreamSubscription?.unsubscribe();\n stopStreamingMessages();\n\n resolveOutputMessagesPromise(outputMessages);\n },\n });\n }),\n };\n\n return response;\n }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAqEA,MAAM,mBAAmB,OAAO,EAC9B,SACA,0BACA,MACA,UACA,cAOI;AACJ,KACE,KAAK,SAAS,UACd,KAAK,SAAS,KAAK,SAAS,SAAS,GAAG,aAAa,SACnD,YAAY,MACd;EACA,MAAM,WAAW,KAAK,SACnB,QACE,MACC,EAAE,gBAAgB,WACjB,EAAE,YAAY,SAAS,YAAY,QAClC,EAAE,YAAY,SAAS,YAAY,WACxC,CACA,KAAK,OAAO;GACX,MAAM,EAAE,YAAa;GACrB,SAAS,EAAE,YAAY;GACxB,EAAE;EAEL,MAAM,cAAc,SAAS,SAAS,SAAS;EAC/C,MAAM,iBAAiB,SAAS,MAAM,GAAG,GAAG;EAE5C,MAAM,OAAO;GACX,OAAO,YAAY;GACnB,aAAa,KAAK,MAAM,WAAW,qBAAqB;GACxD,eAAe,KAAK,MAAM,WAAW,qBAAqB;GAC1D,UAAU;GACX;EAED,MAAM,mBAAmB,MAAM,MAAM,GAAG,QAAQ,uBAAuB;GACrE,QAAQ;GACR,SAAS;IACP,gBAAgB;IAChB,iCAAiC;IAClC;GACD,MAAM,KAAK,UAAU,KAAK;GAC3B,CAAC;AAEF,MAAI,iBAAiB,GAEnB,UADqC,MAAM,iBAAiB,MAAM,CAC9C;MAEpB,SAAQ,MAAM,iBAAiB,MAAM,CAAC;;;AAMrC,4BAAM,gBAAgB;CAC3B,MACM,QAAQ;AACZ,SAAO;;CAGT,MACM,gBAAgB,AAAO,KAAqB;EAChD,IAAI,SAAS,IAAI,OAAO,MAAM,EAC5B,WAAW,mCACZ,CAAC;AAEF,SAAO,MAAM,aAAa;EAC1B,MAAM,sBAAsB,EAAE;AAE9B,SAAO,MAAM,0CAA0C;AAEvD,SAAO,EACL,QAAQ,oBAAoB,KACzB,EAAE,UAAU,GAAG,2BAA2B,qBAC5C,EACF;;CAGH,MACM,wBACJ,AAAO,KACP,AAAa,MACb,AACA,YACA;AACA,kBAAU,QAAQ,uCAAuC;GACvD,4BAA4B,KAAK,OAAO,eAAe;GACvD,aAAa,KAAK,SAAS;GAC3B,0BAA0B,CAAC,CAAC,IAAI,QAAQ,QAAQ,IAC9C,gCACD;GACD,GAAI,IAAI,QAAQ,QAAQ,IAAI,gCAAgC,GACxD,EACE,wBAAwB,IAAI,QAAQ,QAAQ,IAC1C,gCACD,EACF,GACD,EAAE;GACN,GAAI,IAAI,YAAY,UAChB,EACE,kBAAkB,IAAI,YAAY,SACnC,GACD,EACE,kBAAkB,mCACnB;GACN,CAAC;EAEF,IAAI,SAAS,IAAI,OAAO,MAAM,EAC5B,WAAW,2CACZ,CAAC;AACF,SAAO,MAAM,EAAE,MAAM,EAAE,8BAA8B;AAErD,MAAI,YAAY;AACd,UAAO,MAAM,uDAAuD;AACpE,OAAI,aAAa;IAAE,GAAG,IAAI;IAAY,GAAG;IAAY;;AAGhC,MAAI,YAAY;AAChB,MAAI,YAAY;EAEvC,IAAI,2BAA0C;EAC9C,IAAI;EAIJ,MAAM,0BAA0B,IAAI,QAAQ,QAAQ,IAClD,gCACD;AACD,MAAI,wBACF,4BAA2B;AAG7B,MAAI,KAAK,OAAO;AACd,YAAS,OAAO,MAAM,EAAE,OAAO,MAAM,CAAC;AACtC,UAAO,MACL,uEACD;AAED,OAAI,CAAC,0BAA0B;AAC7B,WAAO,MAAM,sCAAsC;AAEnD,UAAM,IAAI,aACR,mDACD;;AAGH,OAAI,QAAQ,IAAI,uBACd,uBAAsB,QAAQ,IAAI;YACzB,IAAI,YAAY,OAAO,QAChC,uBAAsB,IAAI,YAAY,OAAO;OAE7C,uBAAsB;AAGxB,YAAS,OAAO,MAAM,EAAE,qBAAqB,CAAC;;AAGhD,SAAO,MAAM,sBAAsB;EACnC,MAAM,kBAAkB,IAAI,eAA2C;EACvE,MAAM,sBAAsB,IAAI,eAG5B;EACJ,MAAM,oBAAoB,IAAI,eAAiC;EAE/D,IAAI,iBAA4B,EAAE;EAClC,IAAI;EACJ,IAAI;AAE0B,MAAI,SAAoB,SAAS,WAAW;AACxE,kCAA+B;AAC/B,iCAA8B;IAC9B;AAEF,MAAI,yBACF,KAAI,WAAW,8BAA8B;AAG/C,SAAO,MAAM,aAAa;EAC1B,IAAI;EAEJ,MAAM,EACJ,aACA,WAAW,UAAU,EACrB,OACA,mBACA,2BACA,eACE;AAEJ,SAAO,MAAM,0CAA0C;EAEvD,MAAM,cAAc,YACjB,qBAAqB;GACpB;GACA,mBAAmB,KAAK,OAAO,aAAa,oBAAoB;GAChE,2BAA2B,0BAA0B,QAElD,WACC,CAAC,kBAAkB,MAChB,qBAAqB,iBAAiB,QAAQ,OAAO,KACvD,CACJ;GACD;GACD,CAAC,CACD,KAGC,aAAa,EACb,eAAe;AACb,UAAO,MAAM,yBAAyB;IACtC,CACH;AA+eH,SA7eiB;GACf;GACA;GACA,QAAQ,eAAe,gBAAgB;GACvC;GACA,YAAY,IAAI,SAAS,OAAO,MAAM,SAAS;IAC7C,IAAI;AAEJ,8BAA0B,YAAY,UAAU;KAC9C,MAAM,OAAO,UAAU;AACrB,UAAI,MAAM,QAAQ,kBAAkB,UAClC;AAEF,cAAQ,MAAM,MAAd;OAEE,KAAK,oBAAoB;AACvB,aACE,gBAAgB,yBAAyB;SAEvC,MAAM,MAAM;SAEZ,MAAM,qBAAqB;SAE3B,OAAO,MAAM;SACd,CAAC,CACH;AACD;OACF,KAAK,qBAAqB;AACxB,aACE,gBAAgB,yBAAyB;SACvC,MAAM,MAAM;SACZ,MAAM,MAAM;SACZ,OAAO,MAAM;SACd,CAAC,CACH;AACD;OACF,KAAK,qBAAqB;AACxB,aACE,gBAAgB,mCAAmC;SACjD,MAAM,MAAM;SACZ,MAAM,MAAM;SACZ,MAAM;UACJ,OAAO,MAAM,KAAK;UAClB,UAAU,MAAM,KAAK,SAAS,KAAK,YAAY;AAC7C,eACE,QAAQ,SAAS,iBAChB,aAAa,WAAW,UAAU,QAEnC,QAAO,gBAAgB,aAAa;YAClC,IAAI,QAAQ;YACZ,2BAAW,IAAI,MAAM;YACrB,SAAS,CAAE,QAAwB,QAAQ;YAC3C,MAAO,QAAwB;YAC/B,QAAQ,IAAI,sBAAsB;YACnC,CAAC;AAEJ,eAAI,eAAe,QACjB,QAAO,gBAAgB,wBAAwB;YAC7C,MAAM,QAAQ;YACd,IAAI,QAAQ;YACZ,WAAW,CAAC,KAAK,UAAU,QAAQ,UAAU,CAAC;YAC9C,2BAAW,IAAI,MAAM;YACrB,QAAQ,IAAI,sBAAsB;YACnC,CAAC;AAEJ,iBAAM,IAAI,MACR,iDACD;YACD;UACH;SACF,CAAC,CACH;AACD;;;KAGN,QAAQ,QAAQ;AAEd,UACE,KAAK,MAAM,SAAS,aAAa,IACjC,KAAK,YAAY,WAEjB,iBAAgB,KACd,IAAI,qBAAqB,EACvB,aAAa,IAAI,WAAW,wBAC7B,CAAC,CACH;UAED,iBAAgB,KACd,IAAI,qBAAqB,EACvB,aAAa,qDACd,CAAC,CACH;AAGH,+BAAyB,aAAa;AACtC,YAAM;;KAER,UAAU,YAAY;AACpB,aAAO,MAAM,+BAA+B;AAC5C,sBAAgB,KAAK,IAAI,uBAAuB,CAAC;AACjD,+BAAyB,aAAa;AACtC,YAAM;;KAET,CAAC;KACF;GACF,UAAU,IAAI,SAAS,OAAO,aAAa,0BAA0B;AACnE,WAAO,MAAM,4BAA4B;AAEzC,QAAI,KAAK,OAAO,YAAY;AAC1B,cAAS,OAAO,MAAM,EAAE,YAAY,MAAM,CAAC;AAC3C,YAAO,MAAM,0CAA0C;AAEvD,sBAAiB;MACf,SAAS;MACT;MACA;MACA,WAAW,WAAW;AACpB,cAAO,MACL,EAAE,QAAQ,OAAO,QAAQ,EACzB,6BACD;AACD,yBAAkB,KAAK,OAAO;AAG9B,WAAI,OAAO,WAAW,UAAU;AAE9B,wBAAgB,KACd,IAAI,oCAAoC,EACtC,kBAAkB,OAAO,QAC1B,CAAC,CACH;AACD,4BAAoB,KAAK,EACvB,QAAQ,6DAA6D,OAAO,UAC7E,CAAC;AAGF,yBAAiB,CACf,gBAAgB,aAAa;SAC3B,IAAI,UAAU;SACd,2BAAW,IAAI,MAAM;SACrB,SAAS,OAAO;SAChB,MAAM,YAAY;SACnB,CAAC,CACH;AACD,qCAA6B,eAAe;;;MAGhD,UAAU,QAAQ;AAChB,cAAO,MAAM,EAAE,KAAK,EAAE,iCAAiC;AACvD,uBAAgB,KACd,IAAI,qBAAqB,EACvB,aAAa,8DACd,CAAC,CACH;AACD,2BAAoB,KAAK,EACvB,QAAQ,6DACT,CAAC;AAGF,mCAA4B,IAAI;;MAEnC,CAAC;;IAGJ,IAAI;AAEJ,WAAO,MAAM,oDAAoD;AAEjE,8BAA0B,YAAY,UAAU;KAC9C,MAAM,OAAO,UAAU;AACrB,cAAQ,MAAM,MAAd;OACE,KAAK,kBAAkB,UACrB;OAIF,KAAK,kBAAkB;QAErB,MAAM,2BAA2B,YAAY,KAE3C,WAAW,MAAoB,MAAM,MAAM,EAE3C,WACG,MACC,EACE,EAAE,SAAS,kBAAkB,kBAC5B,EAAU,aAAa,MAAM,WAEnC,EAED,QACG,MACC,EAAE,QAAQ,kBAAkB,sBAC3B,EAAU,aAAa,MAAM,UACjC,CACF;QAGD,MAAM,sBAAsB,IAAI,SAE7B;QAEH,MAAM,YAAY,MAAM;AAExB,oBAAY;SACV,IAAI;SACJ,iBAAiB,MAAM;SACvB,QAAQ,eAAe,oBAAoB;SAC3C,2BAAW,IAAI,MAAM;SACrB,MAAM,YAAY;SAClB,SAAS,IAAI,SACX,OAAO,eAAe,sBAAsB;AAC1C,iBAAO,MAAM,wCAAwC;UAErD,MAAM,aAAuB,EAAE;UAC/B,IAAI;AAEJ,8BACG,KACC,aAAa,EACb,KAAK,EAAE,EACP,KAAK,EAAE,QAAQ,gBAAgB;AAC7B,kBAAO,MACL;YAAE;YAAQ;YAAW,EACrB,6BACD;AAED,+BAAoB,KAClB,gBAAgB,qBAAqB,EAAE,QAAQ,CAAC,CACjD;AAED,2BAAgB,KACd,IAAI,iCAAiC,EACnC,WACD,CAAC,CACH;AACD,8BAAmB;AACnB,6BAAkB,aAAa;YAC/B,CACH,CACA,WAAW;AAEd,iBAAO,MACL,6CACD;AAED,6BAAmB,yBAAyB,UAAU;WACpD,MAAM,OAAO,MAAoB;AAC/B,gBAAI,EAAE,QAAQ,kBAAkB,oBAAoB;AAClD,mBAAM,cAAc,EAAE,QAAQ;AAC9B,wBAAW,KAAK,EAAE,QAAQ;;;WAG9B,QAAQ,QAAQ;AACd,mBAAO,MACL,EAAE,KAAK,EACP,uCACD;AACD,gCAAoB,KAAK;aACvB,QAAQ;aACR;aACD,CAAC;AACF,+BAAmB;AACnB,8BAAkB,aAAa;;WAEjC,gBAAgB;AACd,mBAAO,MAAM,wCAAwC;AACrD,gCAAoB,KAAK,IAAI,sBAAsB,CAAC;AACpD,+BAAmB;AACnB,8BAAkB,aAAa;AAE/B,2BAAe,KACb,gBAAgB,aAAa;aAC3B,IAAI;aACJ,2BAAW,IAAI,MAAM;aACrB,SAAS,WAAW,KAAK,GAAG;aAC5B,MAAM,YAAY;aACnB,CAAC,CACH;;WAEJ,CAAC;WAEL;SACF,CAAC;AACF;OAIF,KAAK,kBAAkB;AACrB,eAAO,MAAM,wCAAwC;QACrD,MAAM,gCAAgC,YAAY,KAChD,WAAW,MAAoB,MAAM,MAAM,EAE3C,WACG,MACC,EACE,EAAE,SAAS,kBAAkB,sBAC5B,EAAU,qBAAqB,MAAM,mBAE3C,EAED,QACG,MACC,EAAE,QAAQ,kBAAkB,uBAC3B,EAAU,qBAAqB,MAAM,kBACzC,CACF;QACD,MAAM,2BAA2B,IAAI,SAElC;AACH,oBAAY;SACV,IAAI,MAAM;SACV,iBAAiB,MAAM;SACvB,QAAQ,eAAe,yBAAyB;SAChD,2BAAW,IAAI,MAAM;SACrB,MAAM,MAAM;SACZ,WAAW,IAAI,SACb,OAAO,oBAAoB,2BAA2B;AACpD,iBAAO,MAAM,2CAA2C;UAExD,MAAM,iBAA2B,EAAE;UACnC,IAAI;AAEJ,gDACE,8BAA8B,UAAU;WACtC,MAAM,OAAO,MAAoB;AAC/B,gBACE,EAAE,QAAQ,kBAAkB,qBAC5B;AACA,mBAAM,mBAAmB,EAAE,KAAK;AAChC,4BAAe,KAAK,EAAE,KAAK;;;WAG/B,QAAQ,QAAQ;AACd,mBAAO,MACL,EAAE,KAAK,EACP,4CACD;AACD,qCAAyB,KACvB,gBAAgB,qBAAqB,EACnC,QACE,yEACH,CAAC,CACH;AACD,oCAAwB;AACxB,iDAAqC,aAAa;;WAEpD,gBAAgB;AACd,mBAAO,MACL,6CACD;AACD,qCAAyB,KACvB,IAAI,sBAAsB,CAC3B;AACD,oCAAwB;AACxB,iDAAqC,aAAa;AAElD,2BAAe,KACb,gBAAgB,wBAAwB;aACtC,IAAI,MAAM;aACV,2BAAW,IAAI,MAAM;aACrB,MAAM,MAAM;aACZ,WAAW,eAAe,KAAK,GAAG;aACnC,CAAC,CACH;;WAEJ,CAAC;WAEP;SACF,CAAC;AACF;OAIF,KAAK,kBAAkB;AACrB,eAAO,MACL,EAAE,QAAQ,MAAM,QAAQ,EACxB,yCACD;AACD,oBAAY;SACV,IAAI,YAAY,MAAM;SACtB,QAAQ,IAAI,sBAAsB;SAClC,2BAAW,IAAI,MAAM;SACrB,mBAAmB,MAAM;SACzB,YAAY,MAAM;SAClB,QAAQ,MAAM;SACf,CAAC;AAEF,uBAAe,KACb,gBAAgB,eAAe;SAC7B,IAAI,YAAY,MAAM;SACtB,2BAAW,IAAI,MAAM;SACrB,mBAAmB,MAAM;SACzB,YAAY,MAAM;SAClB,QAAQ,MAAM;SACf,CAAC,CACH;AACD;OAIF,KAAK,kBAAkB;AACrB,eAAO,MAAM,EAAE,OAAO,EAAE,+BAA+B;AACvD,oBAAY;SACV,IAAI,UAAU;SACd,QAAQ,IAAI,sBAAsB;SAClC,UAAU,MAAM;SAChB,WAAW,MAAM;SACjB,UAAU,MAAM;SAChB,OAAO,MAAM;SACb,QAAQ,MAAM;SACd,OAAO,MAAM;SACb,SAAS,MAAM;SACf,MAAM,YAAY;SAClB,2BAAW,IAAI,MAAM;SACtB,CAAC;AACF,uBAAe,KACb,gBAAgB,mBAAmB;SACjC,IAAI,UAAU;SACd,UAAU,MAAM;SAChB,WAAW,MAAM;SACjB,UAAU,MAAM;SAChB,OAAO,MAAM;SACb,QAAQ,MAAM;SACd,OAAO,MAAM;SACb,SAAS,MAAM;SACf,MAAM,YAAY;SAClB,2BAAW,IAAI,MAAM;SACtB,CAAC,CACH;AACD;;;KAGN,QAAQ,QAAQ;AAEd,UACE,eAAe,mBACf,eAAe,2BACd,eAAe,SACd,IAAI,QACJ,IAAI,KAAK,SAAS,aAAa,IACjC,KAAK,YAAY,YACjB;AACA,uBAAgB,KACd,IAAI,qBAAqB;QACvB,aAAa,IAAI,WAAW;QAE5B,eAAe;SACb,MAAM,IAAI,QAAQ,IAAI,YAAY;SAClC,YAAY,IAAI,cAAc,IAAI,YAAY;SAC9C,UAAU,IAAI,YAAY,IAAI,YAAY;SAC1C,YAAY,IAAI,cAAc,IAAI,YAAY;SAC9C,mBACE,IAAI,qBACJ,IAAI,YAAY;SAClB,YAAY,IAAI;SACjB;QACF,CAAC,CACH;AACD,gCAAyB,aAAa;AACtC,mCAA4B,IAAI;AAChC,8BAAuB;AACvB;;AAGF,sBAAgB,KACd,IAAI,qBAAqB,EACvB,aAAa,qDACd,CAAC,CACH;AACD,+BAAyB,aAAa;AACtC,6BAAuB;AAEvB,kCAA4B,IAAI;;KAElC,UAAU,YAAY;AACpB,aAAO,MAAM,yBAAyB;AACtC,UAAI,KAAK,OAAO,YAAY;AAC1B,cAAO,MACL,uDACD;AACD,aAAM,eAAe,kBAAkB;;AAEzC,sBAAgB,KAAK,IAAI,uBAAuB,CAAC;AACjD,+BAAyB,aAAa;AACtC,6BAAuB;AAEvB,mCAA6B,eAAe;;KAE/C,CAAC;KACF;GACH;;;;CA1oBF,YAAY,OAAO;;;;;;CAKnB,YAAY,eAAe;oBACL,KAAK;;;;;;CAiB3B,eAAe,gBAAgB;oBAE7B,KAAK;oBACL,IAAI,OAAO;oBACX,IAAI,oBAAoB,mBAAmB,EAAE,UAAU,MAAM,CAAC;;;;;;;;;8BA7BlE,eAAe,gBAAgB"}
|
|
1
|
+
{"version":3,"file":"copilot.resolver.mjs","names":[],"sources":["../../../src/graphql/resolvers/copilot.resolver.ts"],"sourcesContent":["import { Arg, Ctx, Mutation, Query, Resolver } from \"type-graphql\";\nimport {\n ReplaySubject,\n Subject,\n Subscription,\n filter,\n finalize,\n firstValueFrom,\n shareReplay,\n skipWhile,\n take,\n takeWhile,\n tap,\n} from \"rxjs\";\nimport { GenerateCopilotResponseInput } from \"../inputs/generate-copilot-response.input\";\nimport { CopilotResponse } from \"../types/copilot-response.type\";\nimport {\n CopilotKitLangGraphInterruptEvent,\n LangGraphInterruptEvent,\n} from \"../types/meta-events.type\";\nimport { ActionInputAvailability, MessageRole } from \"../types/enums\";\nimport { Repeater } from \"graphql-yoga\";\nimport type {\n CopilotRequestContextProperties,\n GraphQLContext,\n} from \"../../lib/integrations\";\nimport {\n RuntimeEvent,\n RuntimeEventTypes,\n RuntimeMetaEventName,\n} from \"../../service-adapters/events\";\nimport {\n FailedMessageStatus,\n MessageStatusCode,\n MessageStatusUnion,\n SuccessMessageStatus,\n} from \"../types/message-status.type\";\nimport {\n ResponseStatusUnion,\n SuccessResponseStatus,\n} from \"../types/response-status.type\";\nimport { GraphQLJSONObject } from \"graphql-scalars\";\nimport { plainToInstance } from \"class-transformer\";\nimport { GuardrailsResult } from \"../types/guardrails-result.type\";\nimport { GraphQLError } from \"graphql\";\nimport {\n GuardrailsValidationFailureResponse,\n MessageStreamInterruptedResponse,\n UnknownErrorResponse,\n} from \"../../utils\";\nimport {\n ActionExecutionMessage,\n AgentStateMessage,\n Message,\n MessageType,\n ResultMessage,\n TextMessage,\n} from \"../types/converted\";\nimport telemetry from \"../../lib/telemetry-client\";\nimport { randomId } from \"@copilotkit/shared\";\nimport { resolveMessageId } from \"./resolve-message-id\";\nimport { AgentsResponse } from \"../types/agents-response.type\";\nimport { LangGraphEventTypes } from \"../../agents/langgraph/events\";\nimport {\n CopilotKitError,\n CopilotKitLowLevelError,\n isStructuredCopilotKitError,\n} from \"@copilotkit/shared\";\nimport { CopilotRuntime } from \"../../lib\";\n\nconst invokeGuardrails = async ({\n baseUrl,\n copilotCloudPublicApiKey,\n data,\n onResult,\n onError,\n}: {\n baseUrl: string;\n copilotCloudPublicApiKey: string;\n data: GenerateCopilotResponseInput;\n onResult: (result: GuardrailsResult) => void;\n onError: (err: Error) => void;\n}) => {\n if (\n data.messages.length &&\n data.messages[data.messages.length - 1].textMessage?.role ===\n MessageRole.user\n ) {\n const messages = data.messages\n .filter(\n (m) =>\n m.textMessage !== undefined &&\n (m.textMessage.role === MessageRole.user ||\n m.textMessage.role === MessageRole.assistant),\n )\n .map((m) => ({\n role: m.textMessage!.role,\n content: m.textMessage.content,\n }));\n\n const lastMessage = messages[messages.length - 1];\n const restOfMessages = messages.slice(0, -1);\n\n const body = {\n input: lastMessage.content,\n validTopics: data.cloud.guardrails.inputValidationRules.allowList,\n invalidTopics: data.cloud.guardrails.inputValidationRules.denyList,\n messages: restOfMessages,\n };\n\n const guardrailsResult = await fetch(`${baseUrl}/guardrails/validate`, {\n method: \"POST\",\n headers: {\n \"Content-Type\": \"application/json\",\n \"X-CopilotCloud-Public-API-Key\": copilotCloudPublicApiKey,\n },\n body: JSON.stringify(body),\n });\n\n if (guardrailsResult.ok) {\n const resultJson: GuardrailsResult = await guardrailsResult.json();\n onResult(resultJson);\n } else {\n onError(await guardrailsResult.json());\n }\n }\n};\n\n@Resolver(() => CopilotResponse)\nexport class CopilotResolver {\n @Query(() => String)\n async hello() {\n return \"Hello World\";\n }\n\n @Query(() => AgentsResponse)\n async availableAgents(@Ctx() ctx: GraphQLContext) {\n let logger = ctx.logger.child({\n component: \"CopilotResolver.availableAgents\",\n });\n\n logger.debug(\"Processing\");\n const agentsWithEndpoints = [];\n\n logger.debug(\"Event source created, creating response\");\n\n return {\n agents: agentsWithEndpoints.map(\n ({ endpoint, ...agentWithoutEndpoint }) => agentWithoutEndpoint,\n ),\n };\n }\n\n @Mutation(() => CopilotResponse)\n async generateCopilotResponse(\n @Ctx() ctx: GraphQLContext,\n @Arg(\"data\") data: GenerateCopilotResponseInput,\n @Arg(\"properties\", () => GraphQLJSONObject, { nullable: true })\n properties?: CopilotRequestContextProperties,\n ) {\n telemetry.capture(\"oss.runtime.copilot_request_created\", {\n \"cloud.guardrails.enabled\": data.cloud?.guardrails !== undefined,\n requestType: data.metadata.requestType,\n \"cloud.api_key_provided\": !!ctx.request.headers.get(\n \"x-copilotcloud-public-api-key\",\n ),\n ...(ctx.request.headers.get(\"x-copilotcloud-public-api-key\")\n ? {\n \"cloud.public_api_key\": ctx.request.headers.get(\n \"x-copilotcloud-public-api-key\",\n ),\n }\n : {}),\n ...(ctx._copilotkit.baseUrl\n ? {\n \"cloud.base_url\": ctx._copilotkit.baseUrl,\n }\n : {\n \"cloud.base_url\": \"https://api.cloud.copilotkit.ai\",\n }),\n });\n\n let logger = ctx.logger.child({\n component: \"CopilotResolver.generateCopilotResponse\",\n });\n logger.debug({ data }, \"Generating Copilot response\");\n\n if (properties) {\n logger.debug(\"Properties provided, merging with context properties\");\n ctx.properties = { ...ctx.properties, ...properties };\n }\n\n const copilotRuntime = ctx._copilotkit.runtime as unknown as CopilotRuntime;\n const serviceAdapter = ctx._copilotkit.serviceAdapter;\n\n let copilotCloudPublicApiKey: string | null = null;\n let copilotCloudBaseUrl: string;\n\n // Extract publicApiKey from headers for both cloud and non-cloud requests\n // This enables onTrace functionality regardless of cloud configuration\n const publicApiKeyFromHeaders = ctx.request.headers.get(\n \"x-copilotcloud-public-api-key\",\n );\n if (publicApiKeyFromHeaders) {\n copilotCloudPublicApiKey = publicApiKeyFromHeaders;\n }\n\n if (data.cloud) {\n logger = logger.child({ cloud: true });\n logger.debug(\n \"Cloud configuration provided, checking for public API key in headers\",\n );\n\n if (!copilotCloudPublicApiKey) {\n logger.error(\"Public API key not found in headers\");\n\n throw new GraphQLError(\n \"X-CopilotCloud-Public-API-Key header is required\",\n );\n }\n\n if (process.env.COPILOT_CLOUD_BASE_URL) {\n copilotCloudBaseUrl = process.env.COPILOT_CLOUD_BASE_URL;\n } else if (ctx._copilotkit.cloud?.baseUrl) {\n copilotCloudBaseUrl = ctx._copilotkit.cloud?.baseUrl;\n } else {\n copilotCloudBaseUrl = \"https://api.cloud.copilotkit.ai\";\n }\n\n logger = logger.child({ copilotCloudBaseUrl });\n }\n\n logger.debug(\"Setting up subjects\");\n const responseStatus$ = new ReplaySubject<typeof ResponseStatusUnion>();\n const interruptStreaming$ = new ReplaySubject<{\n reason: string;\n messageId?: string;\n }>();\n const guardrailsResult$ = new ReplaySubject<GuardrailsResult>();\n\n let outputMessages: Message[] = [];\n let resolveOutputMessagesPromise: (messages: Message[]) => void;\n let rejectOutputMessagesPromise: (err: Error) => void;\n\n const outputMessagesPromise = new Promise<Message[]>((resolve, reject) => {\n resolveOutputMessagesPromise = resolve;\n rejectOutputMessagesPromise = reject;\n });\n\n if (copilotCloudPublicApiKey) {\n ctx.properties[\"copilotCloudPublicApiKey\"] = copilotCloudPublicApiKey;\n }\n\n logger.debug(\"Processing\");\n let runtimeResponse;\n\n const {\n eventSource,\n threadId = randomId(),\n runId,\n serverSideActions,\n actionInputsWithoutAgents,\n extensions,\n } = runtimeResponse;\n\n logger.debug(\"Event source created, creating response\");\n // run and process the event stream\n const eventStream = eventSource\n .processRuntimeEvents({\n serverSideActions,\n guardrailsResult$: data.cloud?.guardrails ? guardrailsResult$ : null,\n actionInputsWithoutAgents: actionInputsWithoutAgents.filter(\n // TODO-AGENTS: do not exclude ALL server side actions\n (action) =>\n !serverSideActions.find(\n (serverSideAction) => serverSideAction.name == action.name,\n ),\n ),\n threadId,\n })\n .pipe(\n // shareReplay() ensures that later subscribers will see the whole stream instead of\n // just the events that were emitted after the subscriber was added.\n shareReplay(),\n finalize(() => {\n logger.debug(\"Event stream finalized\");\n }),\n );\n\n const response = {\n threadId,\n runId,\n status: firstValueFrom(responseStatus$),\n extensions,\n metaEvents: new Repeater(async (push, stop) => {\n let eventStreamSubscription: Subscription;\n\n eventStreamSubscription = eventStream.subscribe({\n next: async (event) => {\n if (event.type != RuntimeEventTypes.MetaEvent) {\n return;\n }\n switch (event.name) {\n // @ts-ignore\n case LangGraphEventTypes.OnInterrupt:\n push(\n plainToInstance(LangGraphInterruptEvent, {\n // @ts-ignore\n type: event.type,\n // @ts-ignore\n name: RuntimeMetaEventName.LangGraphInterruptEvent,\n // @ts-ignore\n value: event.value,\n }),\n );\n break;\n case RuntimeMetaEventName.LangGraphInterruptEvent:\n push(\n plainToInstance(LangGraphInterruptEvent, {\n type: event.type,\n name: event.name,\n value: event.value,\n }),\n );\n break;\n case RuntimeMetaEventName.CopilotKitLangGraphInterruptEvent:\n push(\n plainToInstance(CopilotKitLangGraphInterruptEvent, {\n type: event.type,\n name: event.name,\n data: {\n value: event.data.value,\n messages: event.data.messages.map((message) => {\n if (\n message.type === \"TextMessage\" ||\n (\"content\" in message && \"role\" in message)\n ) {\n return plainToInstance(TextMessage, {\n id: message.id,\n createdAt: new Date(),\n content: [(message as TextMessage).content],\n role: (message as TextMessage).role,\n status: new SuccessMessageStatus(),\n });\n }\n if (\"arguments\" in message) {\n return plainToInstance(ActionExecutionMessage, {\n name: message.name,\n id: message.id,\n arguments: [JSON.stringify(message.arguments)],\n createdAt: new Date(),\n status: new SuccessMessageStatus(),\n });\n }\n throw new Error(\n \"Unknown message in metaEvents copilot resolver\",\n );\n }),\n },\n }),\n );\n break;\n }\n },\n error: (err) => {\n // For structured CopilotKit errors, set proper error response status\n if (\n err?.name?.includes(\"CopilotKit\") ||\n err?.extensions?.visibility\n ) {\n responseStatus$.next(\n new UnknownErrorResponse({\n description: err.message || \"Agent error occurred\",\n }),\n );\n } else {\n responseStatus$.next(\n new UnknownErrorResponse({\n description: `An unknown error has occurred in the event stream`,\n }),\n );\n }\n\n eventStreamSubscription?.unsubscribe();\n stop();\n },\n complete: async () => {\n logger.debug(\"Meta events stream completed\");\n responseStatus$.next(new SuccessResponseStatus());\n eventStreamSubscription?.unsubscribe();\n stop();\n },\n });\n }),\n messages: new Repeater(async (pushMessage, stopStreamingMessages) => {\n logger.debug(\"Messages repeater created\");\n\n if (data.cloud?.guardrails) {\n logger = logger.child({ guardrails: true });\n logger.debug(\"Guardrails is enabled, validating input\");\n\n invokeGuardrails({\n baseUrl: copilotCloudBaseUrl,\n copilotCloudPublicApiKey,\n data,\n onResult: (result) => {\n logger.debug(\n { status: result.status },\n \"Guardrails validation done\",\n );\n guardrailsResult$.next(result);\n\n // Guardrails validation failed\n if (result.status === \"denied\") {\n // send the reason to the client and interrupt streaming\n responseStatus$.next(\n new GuardrailsValidationFailureResponse({\n guardrailsReason: result.reason,\n }),\n );\n interruptStreaming$.next({\n reason: `Interrupted due to Guardrails validation failure. Reason: ${result.reason}`,\n });\n\n // resolve messages promise to the middleware\n outputMessages = [\n plainToInstance(TextMessage, {\n id: randomId(),\n createdAt: new Date(),\n content: result.reason,\n role: MessageRole.assistant,\n }),\n ];\n resolveOutputMessagesPromise(outputMessages);\n }\n },\n onError: (err) => {\n logger.error({ err }, \"Error in guardrails validation\");\n responseStatus$.next(\n new UnknownErrorResponse({\n description: `An unknown error has occurred in the guardrails validation`,\n }),\n );\n interruptStreaming$.next({\n reason: `Interrupted due to unknown error in guardrails validation`,\n });\n\n // reject the middleware promise\n rejectOutputMessagesPromise(err);\n },\n });\n }\n\n let eventStreamSubscription: Subscription;\n\n logger.debug(\"Event stream created, subscribing to event stream\");\n\n eventStreamSubscription = eventStream.subscribe({\n next: async (event) => {\n switch (event.type) {\n case RuntimeEventTypes.MetaEvent:\n break;\n ////////////////////////////////\n // TextMessageStart\n ////////////////////////////////\n case RuntimeEventTypes.TextMessageStart:\n // create a sub stream that contains the message content\n const textMessageContentStream = eventStream.pipe(\n // skip until this message start event\n skipWhile((e: RuntimeEvent) => e !== event),\n // take until the message end event\n takeWhile(\n (e: RuntimeEvent) =>\n !(\n e.type === RuntimeEventTypes.TextMessageEnd &&\n (e as any).messageId == event.messageId\n ),\n ),\n // filter out any other message events or message ids\n filter(\n (e: RuntimeEvent) =>\n e.type == RuntimeEventTypes.TextMessageContent &&\n (e as any).messageId == event.messageId,\n ),\n );\n\n // signal when we are done streaming\n const streamingTextStatus = new Subject<\n typeof MessageStatusUnion\n >();\n\n const messageId = resolveMessageId(event.messageId);\n // push the new message\n pushMessage({\n id: messageId,\n parentMessageId: event.parentMessageId,\n status: firstValueFrom(streamingTextStatus),\n createdAt: new Date(),\n role: MessageRole.assistant,\n content: new Repeater(\n async (pushTextChunk, stopStreamingText) => {\n logger.debug(\"Text message content repeater created\");\n\n const textChunks: string[] = [];\n let textSubscription: Subscription;\n\n interruptStreaming$\n .pipe(\n shareReplay(),\n take(1),\n tap(({ reason, messageId }) => {\n logger.debug(\n { reason, messageId },\n \"Text streaming interrupted\",\n );\n\n streamingTextStatus.next(\n plainToInstance(FailedMessageStatus, { reason }),\n );\n\n responseStatus$.next(\n new MessageStreamInterruptedResponse({\n messageId,\n }),\n );\n stopStreamingText();\n textSubscription?.unsubscribe();\n }),\n )\n .subscribe();\n\n logger.debug(\n \"Subscribing to text message content stream\",\n );\n\n textSubscription = textMessageContentStream.subscribe({\n next: async (e: RuntimeEvent) => {\n if (e.type == RuntimeEventTypes.TextMessageContent) {\n await pushTextChunk(e.content);\n textChunks.push(e.content);\n }\n },\n error: (err) => {\n logger.error(\n { err },\n \"Error in text message content stream\",\n );\n interruptStreaming$.next({\n reason: \"Error streaming message content\",\n messageId,\n });\n stopStreamingText();\n textSubscription?.unsubscribe();\n },\n complete: () => {\n logger.debug(\"Text message content stream completed\");\n streamingTextStatus.next(new SuccessMessageStatus());\n stopStreamingText();\n textSubscription?.unsubscribe();\n\n outputMessages.push(\n plainToInstance(TextMessage, {\n id: messageId,\n createdAt: new Date(),\n content: textChunks.join(\"\"),\n role: MessageRole.assistant,\n }),\n );\n },\n });\n },\n ),\n });\n break;\n ////////////////////////////////\n // ActionExecutionStart\n ////////////////////////////////\n case RuntimeEventTypes.ActionExecutionStart:\n logger.debug(\"Action execution start event received\");\n const actionExecutionArgumentStream = eventStream.pipe(\n skipWhile((e: RuntimeEvent) => e !== event),\n // take until the action execution end event\n takeWhile(\n (e: RuntimeEvent) =>\n !(\n e.type === RuntimeEventTypes.ActionExecutionEnd &&\n (e as any).actionExecutionId == event.actionExecutionId\n ),\n ),\n // filter out any other action execution events or action execution ids\n filter(\n (e: RuntimeEvent) =>\n e.type == RuntimeEventTypes.ActionExecutionArgs &&\n (e as any).actionExecutionId == event.actionExecutionId,\n ),\n );\n const streamingArgumentsStatus = new Subject<\n typeof MessageStatusUnion\n >();\n pushMessage({\n id: event.actionExecutionId,\n parentMessageId: event.parentMessageId,\n status: firstValueFrom(streamingArgumentsStatus),\n createdAt: new Date(),\n name: event.actionName,\n arguments: new Repeater(\n async (pushArgumentsChunk, stopStreamingArguments) => {\n logger.debug(\"Action execution argument stream created\");\n\n const argumentChunks: string[] = [];\n let actionExecutionArgumentSubscription: Subscription;\n\n actionExecutionArgumentSubscription =\n actionExecutionArgumentStream.subscribe({\n next: async (e: RuntimeEvent) => {\n if (\n e.type == RuntimeEventTypes.ActionExecutionArgs\n ) {\n await pushArgumentsChunk(e.args);\n argumentChunks.push(e.args);\n }\n },\n error: (err) => {\n logger.error(\n { err },\n \"Error in action execution argument stream\",\n );\n streamingArgumentsStatus.next(\n plainToInstance(FailedMessageStatus, {\n reason:\n \"An unknown error has occurred in the action execution argument stream\",\n }),\n );\n stopStreamingArguments();\n actionExecutionArgumentSubscription?.unsubscribe();\n },\n complete: () => {\n logger.debug(\n \"Action execution argument stream completed\",\n );\n streamingArgumentsStatus.next(\n new SuccessMessageStatus(),\n );\n stopStreamingArguments();\n actionExecutionArgumentSubscription?.unsubscribe();\n\n outputMessages.push(\n plainToInstance(ActionExecutionMessage, {\n id: event.actionExecutionId,\n createdAt: new Date(),\n name: event.actionName,\n arguments: argumentChunks.join(\"\"),\n }),\n );\n },\n });\n },\n ),\n });\n break;\n ////////////////////////////////\n // ActionExecutionResult\n ////////////////////////////////\n case RuntimeEventTypes.ActionExecutionResult:\n logger.debug(\n { result: event.result },\n \"Action execution result event received\",\n );\n pushMessage({\n id: \"result-\" + event.actionExecutionId,\n status: new SuccessMessageStatus(),\n createdAt: new Date(),\n actionExecutionId: event.actionExecutionId,\n actionName: event.actionName,\n result: event.result,\n });\n\n outputMessages.push(\n plainToInstance(ResultMessage, {\n id: \"result-\" + event.actionExecutionId,\n createdAt: new Date(),\n actionExecutionId: event.actionExecutionId,\n actionName: event.actionName,\n result: event.result,\n }),\n );\n break;\n ////////////////////////////////\n // AgentStateMessage\n ////////////////////////////////\n case RuntimeEventTypes.AgentStateMessage:\n logger.debug({ event }, \"Agent message event received\");\n pushMessage({\n id: randomId(),\n status: new SuccessMessageStatus(),\n threadId: event.threadId,\n agentName: event.agentName,\n nodeName: event.nodeName,\n runId: event.runId,\n active: event.active,\n state: event.state,\n running: event.running,\n role: MessageRole.assistant,\n createdAt: new Date(),\n });\n outputMessages.push(\n plainToInstance(AgentStateMessage, {\n id: randomId(),\n threadId: event.threadId,\n agentName: event.agentName,\n nodeName: event.nodeName,\n runId: event.runId,\n active: event.active,\n state: event.state,\n running: event.running,\n role: MessageRole.assistant,\n createdAt: new Date(),\n }),\n );\n break;\n }\n },\n error: (err) => {\n // For structured CopilotKit errors, set proper error response status\n if (\n err instanceof CopilotKitError ||\n err instanceof CopilotKitLowLevelError ||\n (err instanceof Error &&\n err.name &&\n err.name.includes(\"CopilotKit\")) ||\n err?.extensions?.visibility\n ) {\n responseStatus$.next(\n new UnknownErrorResponse({\n description: err.message || \"Agent error occurred\",\n // Include original error information for frontend to extract\n originalError: {\n code: err.code || err.extensions?.code,\n statusCode: err.statusCode || err.extensions?.statusCode,\n severity: err.severity || err.extensions?.severity,\n visibility: err.visibility || err.extensions?.visibility,\n originalErrorType:\n err.originalErrorType ||\n err.extensions?.originalErrorType,\n extensions: err.extensions,\n },\n }),\n );\n eventStreamSubscription?.unsubscribe();\n rejectOutputMessagesPromise(err);\n stopStreamingMessages();\n return;\n }\n\n responseStatus$.next(\n new UnknownErrorResponse({\n description: `An unknown error has occurred in the event stream`,\n }),\n );\n eventStreamSubscription?.unsubscribe();\n stopStreamingMessages();\n\n rejectOutputMessagesPromise(err);\n },\n complete: async () => {\n logger.debug(\"Event stream completed\");\n if (data.cloud?.guardrails) {\n logger.debug(\n \"Guardrails is enabled, waiting for guardrails result\",\n );\n await firstValueFrom(guardrailsResult$);\n }\n responseStatus$.next(new SuccessResponseStatus());\n eventStreamSubscription?.unsubscribe();\n stopStreamingMessages();\n\n resolveOutputMessagesPromise(outputMessages);\n },\n });\n }),\n };\n\n return response;\n }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;AAsEA,MAAM,mBAAmB,OAAO,EAC9B,SACA,0BACA,MACA,UACA,cAOI;AACJ,KACE,KAAK,SAAS,UACd,KAAK,SAAS,KAAK,SAAS,SAAS,GAAG,aAAa,SACnD,YAAY,MACd;EACA,MAAM,WAAW,KAAK,SACnB,QACE,MACC,EAAE,gBAAgB,WACjB,EAAE,YAAY,SAAS,YAAY,QAClC,EAAE,YAAY,SAAS,YAAY,WACxC,CACA,KAAK,OAAO;GACX,MAAM,EAAE,YAAa;GACrB,SAAS,EAAE,YAAY;GACxB,EAAE;EAEL,MAAM,cAAc,SAAS,SAAS,SAAS;EAC/C,MAAM,iBAAiB,SAAS,MAAM,GAAG,GAAG;EAE5C,MAAM,OAAO;GACX,OAAO,YAAY;GACnB,aAAa,KAAK,MAAM,WAAW,qBAAqB;GACxD,eAAe,KAAK,MAAM,WAAW,qBAAqB;GAC1D,UAAU;GACX;EAED,MAAM,mBAAmB,MAAM,MAAM,GAAG,QAAQ,uBAAuB;GACrE,QAAQ;GACR,SAAS;IACP,gBAAgB;IAChB,iCAAiC;IAClC;GACD,MAAM,KAAK,UAAU,KAAK;GAC3B,CAAC;AAEF,MAAI,iBAAiB,GAEnB,UADqC,MAAM,iBAAiB,MAAM,CAC9C;MAEpB,SAAQ,MAAM,iBAAiB,MAAM,CAAC;;;AAMrC,4BAAM,gBAAgB;CAC3B,MACM,QAAQ;AACZ,SAAO;;CAGT,MACM,gBAAgB,AAAO,KAAqB;EAChD,IAAI,SAAS,IAAI,OAAO,MAAM,EAC5B,WAAW,mCACZ,CAAC;AAEF,SAAO,MAAM,aAAa;EAC1B,MAAM,sBAAsB,EAAE;AAE9B,SAAO,MAAM,0CAA0C;AAEvD,SAAO,EACL,QAAQ,oBAAoB,KACzB,EAAE,UAAU,GAAG,2BAA2B,qBAC5C,EACF;;CAGH,MACM,wBACJ,AAAO,KACP,AAAa,MACb,AACA,YACA;AACA,kBAAU,QAAQ,uCAAuC;GACvD,4BAA4B,KAAK,OAAO,eAAe;GACvD,aAAa,KAAK,SAAS;GAC3B,0BAA0B,CAAC,CAAC,IAAI,QAAQ,QAAQ,IAC9C,gCACD;GACD,GAAI,IAAI,QAAQ,QAAQ,IAAI,gCAAgC,GACxD,EACE,wBAAwB,IAAI,QAAQ,QAAQ,IAC1C,gCACD,EACF,GACD,EAAE;GACN,GAAI,IAAI,YAAY,UAChB,EACE,kBAAkB,IAAI,YAAY,SACnC,GACD,EACE,kBAAkB,mCACnB;GACN,CAAC;EAEF,IAAI,SAAS,IAAI,OAAO,MAAM,EAC5B,WAAW,2CACZ,CAAC;AACF,SAAO,MAAM,EAAE,MAAM,EAAE,8BAA8B;AAErD,MAAI,YAAY;AACd,UAAO,MAAM,uDAAuD;AACpE,OAAI,aAAa;IAAE,GAAG,IAAI;IAAY,GAAG;IAAY;;AAGhC,MAAI,YAAY;AAChB,MAAI,YAAY;EAEvC,IAAI,2BAA0C;EAC9C,IAAI;EAIJ,MAAM,0BAA0B,IAAI,QAAQ,QAAQ,IAClD,gCACD;AACD,MAAI,wBACF,4BAA2B;AAG7B,MAAI,KAAK,OAAO;AACd,YAAS,OAAO,MAAM,EAAE,OAAO,MAAM,CAAC;AACtC,UAAO,MACL,uEACD;AAED,OAAI,CAAC,0BAA0B;AAC7B,WAAO,MAAM,sCAAsC;AAEnD,UAAM,IAAI,aACR,mDACD;;AAGH,OAAI,QAAQ,IAAI,uBACd,uBAAsB,QAAQ,IAAI;YACzB,IAAI,YAAY,OAAO,QAChC,uBAAsB,IAAI,YAAY,OAAO;OAE7C,uBAAsB;AAGxB,YAAS,OAAO,MAAM,EAAE,qBAAqB,CAAC;;AAGhD,SAAO,MAAM,sBAAsB;EACnC,MAAM,kBAAkB,IAAI,eAA2C;EACvE,MAAM,sBAAsB,IAAI,eAG5B;EACJ,MAAM,oBAAoB,IAAI,eAAiC;EAE/D,IAAI,iBAA4B,EAAE;EAClC,IAAI;EACJ,IAAI;AAE0B,MAAI,SAAoB,SAAS,WAAW;AACxE,kCAA+B;AAC/B,iCAA8B;IAC9B;AAEF,MAAI,yBACF,KAAI,WAAW,8BAA8B;AAG/C,SAAO,MAAM,aAAa;EAC1B,IAAI;EAEJ,MAAM,EACJ,aACA,WAAW,UAAU,EACrB,OACA,mBACA,2BACA,eACE;AAEJ,SAAO,MAAM,0CAA0C;EAEvD,MAAM,cAAc,YACjB,qBAAqB;GACpB;GACA,mBAAmB,KAAK,OAAO,aAAa,oBAAoB;GAChE,2BAA2B,0BAA0B,QAElD,WACC,CAAC,kBAAkB,MAChB,qBAAqB,iBAAiB,QAAQ,OAAO,KACvD,CACJ;GACD;GACD,CAAC,CACD,KAGC,aAAa,EACb,eAAe;AACb,UAAO,MAAM,yBAAyB;IACtC,CACH;AA+eH,SA7eiB;GACf;GACA;GACA,QAAQ,eAAe,gBAAgB;GACvC;GACA,YAAY,IAAI,SAAS,OAAO,MAAM,SAAS;IAC7C,IAAI;AAEJ,8BAA0B,YAAY,UAAU;KAC9C,MAAM,OAAO,UAAU;AACrB,UAAI,MAAM,QAAQ,kBAAkB,UAClC;AAEF,cAAQ,MAAM,MAAd;OAEE,KAAK,oBAAoB;AACvB,aACE,gBAAgB,yBAAyB;SAEvC,MAAM,MAAM;SAEZ,MAAM,qBAAqB;SAE3B,OAAO,MAAM;SACd,CAAC,CACH;AACD;OACF,KAAK,qBAAqB;AACxB,aACE,gBAAgB,yBAAyB;SACvC,MAAM,MAAM;SACZ,MAAM,MAAM;SACZ,OAAO,MAAM;SACd,CAAC,CACH;AACD;OACF,KAAK,qBAAqB;AACxB,aACE,gBAAgB,mCAAmC;SACjD,MAAM,MAAM;SACZ,MAAM,MAAM;SACZ,MAAM;UACJ,OAAO,MAAM,KAAK;UAClB,UAAU,MAAM,KAAK,SAAS,KAAK,YAAY;AAC7C,eACE,QAAQ,SAAS,iBAChB,aAAa,WAAW,UAAU,QAEnC,QAAO,gBAAgB,aAAa;YAClC,IAAI,QAAQ;YACZ,2BAAW,IAAI,MAAM;YACrB,SAAS,CAAE,QAAwB,QAAQ;YAC3C,MAAO,QAAwB;YAC/B,QAAQ,IAAI,sBAAsB;YACnC,CAAC;AAEJ,eAAI,eAAe,QACjB,QAAO,gBAAgB,wBAAwB;YAC7C,MAAM,QAAQ;YACd,IAAI,QAAQ;YACZ,WAAW,CAAC,KAAK,UAAU,QAAQ,UAAU,CAAC;YAC9C,2BAAW,IAAI,MAAM;YACrB,QAAQ,IAAI,sBAAsB;YACnC,CAAC;AAEJ,iBAAM,IAAI,MACR,iDACD;YACD;UACH;SACF,CAAC,CACH;AACD;;;KAGN,QAAQ,QAAQ;AAEd,UACE,KAAK,MAAM,SAAS,aAAa,IACjC,KAAK,YAAY,WAEjB,iBAAgB,KACd,IAAI,qBAAqB,EACvB,aAAa,IAAI,WAAW,wBAC7B,CAAC,CACH;UAED,iBAAgB,KACd,IAAI,qBAAqB,EACvB,aAAa,qDACd,CAAC,CACH;AAGH,+BAAyB,aAAa;AACtC,YAAM;;KAER,UAAU,YAAY;AACpB,aAAO,MAAM,+BAA+B;AAC5C,sBAAgB,KAAK,IAAI,uBAAuB,CAAC;AACjD,+BAAyB,aAAa;AACtC,YAAM;;KAET,CAAC;KACF;GACF,UAAU,IAAI,SAAS,OAAO,aAAa,0BAA0B;AACnE,WAAO,MAAM,4BAA4B;AAEzC,QAAI,KAAK,OAAO,YAAY;AAC1B,cAAS,OAAO,MAAM,EAAE,YAAY,MAAM,CAAC;AAC3C,YAAO,MAAM,0CAA0C;AAEvD,sBAAiB;MACf,SAAS;MACT;MACA;MACA,WAAW,WAAW;AACpB,cAAO,MACL,EAAE,QAAQ,OAAO,QAAQ,EACzB,6BACD;AACD,yBAAkB,KAAK,OAAO;AAG9B,WAAI,OAAO,WAAW,UAAU;AAE9B,wBAAgB,KACd,IAAI,oCAAoC,EACtC,kBAAkB,OAAO,QAC1B,CAAC,CACH;AACD,4BAAoB,KAAK,EACvB,QAAQ,6DAA6D,OAAO,UAC7E,CAAC;AAGF,yBAAiB,CACf,gBAAgB,aAAa;SAC3B,IAAI,UAAU;SACd,2BAAW,IAAI,MAAM;SACrB,SAAS,OAAO;SAChB,MAAM,YAAY;SACnB,CAAC,CACH;AACD,qCAA6B,eAAe;;;MAGhD,UAAU,QAAQ;AAChB,cAAO,MAAM,EAAE,KAAK,EAAE,iCAAiC;AACvD,uBAAgB,KACd,IAAI,qBAAqB,EACvB,aAAa,8DACd,CAAC,CACH;AACD,2BAAoB,KAAK,EACvB,QAAQ,6DACT,CAAC;AAGF,mCAA4B,IAAI;;MAEnC,CAAC;;IAGJ,IAAI;AAEJ,WAAO,MAAM,oDAAoD;AAEjE,8BAA0B,YAAY,UAAU;KAC9C,MAAM,OAAO,UAAU;AACrB,cAAQ,MAAM,MAAd;OACE,KAAK,kBAAkB,UACrB;OAIF,KAAK,kBAAkB;QAErB,MAAM,2BAA2B,YAAY,KAE3C,WAAW,MAAoB,MAAM,MAAM,EAE3C,WACG,MACC,EACE,EAAE,SAAS,kBAAkB,kBAC5B,EAAU,aAAa,MAAM,WAEnC,EAED,QACG,MACC,EAAE,QAAQ,kBAAkB,sBAC3B,EAAU,aAAa,MAAM,UACjC,CACF;QAGD,MAAM,sBAAsB,IAAI,SAE7B;QAEH,MAAM,YAAY,iBAAiB,MAAM,UAAU;AAEnD,oBAAY;SACV,IAAI;SACJ,iBAAiB,MAAM;SACvB,QAAQ,eAAe,oBAAoB;SAC3C,2BAAW,IAAI,MAAM;SACrB,MAAM,YAAY;SAClB,SAAS,IAAI,SACX,OAAO,eAAe,sBAAsB;AAC1C,iBAAO,MAAM,wCAAwC;UAErD,MAAM,aAAuB,EAAE;UAC/B,IAAI;AAEJ,8BACG,KACC,aAAa,EACb,KAAK,EAAE,EACP,KAAK,EAAE,QAAQ,gBAAgB;AAC7B,kBAAO,MACL;YAAE;YAAQ;YAAW,EACrB,6BACD;AAED,+BAAoB,KAClB,gBAAgB,qBAAqB,EAAE,QAAQ,CAAC,CACjD;AAED,2BAAgB,KACd,IAAI,iCAAiC,EACnC,WACD,CAAC,CACH;AACD,8BAAmB;AACnB,6BAAkB,aAAa;YAC/B,CACH,CACA,WAAW;AAEd,iBAAO,MACL,6CACD;AAED,6BAAmB,yBAAyB,UAAU;WACpD,MAAM,OAAO,MAAoB;AAC/B,gBAAI,EAAE,QAAQ,kBAAkB,oBAAoB;AAClD,mBAAM,cAAc,EAAE,QAAQ;AAC9B,wBAAW,KAAK,EAAE,QAAQ;;;WAG9B,QAAQ,QAAQ;AACd,mBAAO,MACL,EAAE,KAAK,EACP,uCACD;AACD,gCAAoB,KAAK;aACvB,QAAQ;aACR;aACD,CAAC;AACF,+BAAmB;AACnB,8BAAkB,aAAa;;WAEjC,gBAAgB;AACd,mBAAO,MAAM,wCAAwC;AACrD,gCAAoB,KAAK,IAAI,sBAAsB,CAAC;AACpD,+BAAmB;AACnB,8BAAkB,aAAa;AAE/B,2BAAe,KACb,gBAAgB,aAAa;aAC3B,IAAI;aACJ,2BAAW,IAAI,MAAM;aACrB,SAAS,WAAW,KAAK,GAAG;aAC5B,MAAM,YAAY;aACnB,CAAC,CACH;;WAEJ,CAAC;WAEL;SACF,CAAC;AACF;OAIF,KAAK,kBAAkB;AACrB,eAAO,MAAM,wCAAwC;QACrD,MAAM,gCAAgC,YAAY,KAChD,WAAW,MAAoB,MAAM,MAAM,EAE3C,WACG,MACC,EACE,EAAE,SAAS,kBAAkB,sBAC5B,EAAU,qBAAqB,MAAM,mBAE3C,EAED,QACG,MACC,EAAE,QAAQ,kBAAkB,uBAC3B,EAAU,qBAAqB,MAAM,kBACzC,CACF;QACD,MAAM,2BAA2B,IAAI,SAElC;AACH,oBAAY;SACV,IAAI,MAAM;SACV,iBAAiB,MAAM;SACvB,QAAQ,eAAe,yBAAyB;SAChD,2BAAW,IAAI,MAAM;SACrB,MAAM,MAAM;SACZ,WAAW,IAAI,SACb,OAAO,oBAAoB,2BAA2B;AACpD,iBAAO,MAAM,2CAA2C;UAExD,MAAM,iBAA2B,EAAE;UACnC,IAAI;AAEJ,gDACE,8BAA8B,UAAU;WACtC,MAAM,OAAO,MAAoB;AAC/B,gBACE,EAAE,QAAQ,kBAAkB,qBAC5B;AACA,mBAAM,mBAAmB,EAAE,KAAK;AAChC,4BAAe,KAAK,EAAE,KAAK;;;WAG/B,QAAQ,QAAQ;AACd,mBAAO,MACL,EAAE,KAAK,EACP,4CACD;AACD,qCAAyB,KACvB,gBAAgB,qBAAqB,EACnC,QACE,yEACH,CAAC,CACH;AACD,oCAAwB;AACxB,iDAAqC,aAAa;;WAEpD,gBAAgB;AACd,mBAAO,MACL,6CACD;AACD,qCAAyB,KACvB,IAAI,sBAAsB,CAC3B;AACD,oCAAwB;AACxB,iDAAqC,aAAa;AAElD,2BAAe,KACb,gBAAgB,wBAAwB;aACtC,IAAI,MAAM;aACV,2BAAW,IAAI,MAAM;aACrB,MAAM,MAAM;aACZ,WAAW,eAAe,KAAK,GAAG;aACnC,CAAC,CACH;;WAEJ,CAAC;WAEP;SACF,CAAC;AACF;OAIF,KAAK,kBAAkB;AACrB,eAAO,MACL,EAAE,QAAQ,MAAM,QAAQ,EACxB,yCACD;AACD,oBAAY;SACV,IAAI,YAAY,MAAM;SACtB,QAAQ,IAAI,sBAAsB;SAClC,2BAAW,IAAI,MAAM;SACrB,mBAAmB,MAAM;SACzB,YAAY,MAAM;SAClB,QAAQ,MAAM;SACf,CAAC;AAEF,uBAAe,KACb,gBAAgB,eAAe;SAC7B,IAAI,YAAY,MAAM;SACtB,2BAAW,IAAI,MAAM;SACrB,mBAAmB,MAAM;SACzB,YAAY,MAAM;SAClB,QAAQ,MAAM;SACf,CAAC,CACH;AACD;OAIF,KAAK,kBAAkB;AACrB,eAAO,MAAM,EAAE,OAAO,EAAE,+BAA+B;AACvD,oBAAY;SACV,IAAI,UAAU;SACd,QAAQ,IAAI,sBAAsB;SAClC,UAAU,MAAM;SAChB,WAAW,MAAM;SACjB,UAAU,MAAM;SAChB,OAAO,MAAM;SACb,QAAQ,MAAM;SACd,OAAO,MAAM;SACb,SAAS,MAAM;SACf,MAAM,YAAY;SAClB,2BAAW,IAAI,MAAM;SACtB,CAAC;AACF,uBAAe,KACb,gBAAgB,mBAAmB;SACjC,IAAI,UAAU;SACd,UAAU,MAAM;SAChB,WAAW,MAAM;SACjB,UAAU,MAAM;SAChB,OAAO,MAAM;SACb,QAAQ,MAAM;SACd,OAAO,MAAM;SACb,SAAS,MAAM;SACf,MAAM,YAAY;SAClB,2BAAW,IAAI,MAAM;SACtB,CAAC,CACH;AACD;;;KAGN,QAAQ,QAAQ;AAEd,UACE,eAAe,mBACf,eAAe,2BACd,eAAe,SACd,IAAI,QACJ,IAAI,KAAK,SAAS,aAAa,IACjC,KAAK,YAAY,YACjB;AACA,uBAAgB,KACd,IAAI,qBAAqB;QACvB,aAAa,IAAI,WAAW;QAE5B,eAAe;SACb,MAAM,IAAI,QAAQ,IAAI,YAAY;SAClC,YAAY,IAAI,cAAc,IAAI,YAAY;SAC9C,UAAU,IAAI,YAAY,IAAI,YAAY;SAC1C,YAAY,IAAI,cAAc,IAAI,YAAY;SAC9C,mBACE,IAAI,qBACJ,IAAI,YAAY;SAClB,YAAY,IAAI;SACjB;QACF,CAAC,CACH;AACD,gCAAyB,aAAa;AACtC,mCAA4B,IAAI;AAChC,8BAAuB;AACvB;;AAGF,sBAAgB,KACd,IAAI,qBAAqB,EACvB,aAAa,qDACd,CAAC,CACH;AACD,+BAAyB,aAAa;AACtC,6BAAuB;AAEvB,kCAA4B,IAAI;;KAElC,UAAU,YAAY;AACpB,aAAO,MAAM,yBAAyB;AACtC,UAAI,KAAK,OAAO,YAAY;AAC1B,cAAO,MACL,uDACD;AACD,aAAM,eAAe,kBAAkB;;AAEzC,sBAAgB,KAAK,IAAI,uBAAuB,CAAC;AACjD,+BAAyB,aAAa;AACtC,6BAAuB;AAEvB,mCAA6B,eAAe;;KAE/C,CAAC;KACF;GACH;;;;CA1oBF,YAAY,OAAO;;;;;;CAKnB,YAAY,eAAe;oBACL,KAAK;;;;;;CAiB3B,eAAe,gBAAgB;oBAE7B,KAAK;oBACL,IAAI,OAAO;oBACX,IAAI,oBAAoB,mBAAmB,EAAE,UAAU,MAAM,CAAC;;;;;;;;;8BA7BlE,eAAe,gBAAgB"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
require("reflect-metadata");
|
|
2
|
+
const require_runtime = require('../../_virtual/_rolldown/runtime.cjs');
|
|
3
|
+
let _copilotkit_shared = require("@copilotkit/shared");
|
|
4
|
+
|
|
5
|
+
//#region src/graphql/resolvers/resolve-message-id.ts
|
|
6
|
+
/**
|
|
7
|
+
* Resolve the id to use for a streamed TextMessageOutput.
|
|
8
|
+
*
|
|
9
|
+
* Upstream events occasionally arrive with `messageId` missing, null, or an
|
|
10
|
+
* empty string. We fall back to a freshly generated id in those cases so the
|
|
11
|
+
* resulting output never surfaces a null id to the GraphQL client (#2118).
|
|
12
|
+
*/
|
|
13
|
+
function resolveMessageId(eventMessageId) {
|
|
14
|
+
return eventMessageId || (0, _copilotkit_shared.randomId)();
|
|
15
|
+
}
|
|
16
|
+
|
|
17
|
+
//#endregion
|
|
18
|
+
exports.resolveMessageId = resolveMessageId;
|
|
19
|
+
//# sourceMappingURL=resolve-message-id.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"resolve-message-id.cjs","names":[],"sources":["../../../src/graphql/resolvers/resolve-message-id.ts"],"sourcesContent":["import { randomId } from \"@copilotkit/shared\";\n\n/**\n * Resolve the id to use for a streamed TextMessageOutput.\n *\n * Upstream events occasionally arrive with `messageId` missing, null, or an\n * empty string. We fall back to a freshly generated id in those cases so the\n * resulting output never surfaces a null id to the GraphQL client (#2118).\n */\nexport function resolveMessageId(\n eventMessageId: string | null | undefined,\n): string {\n return eventMessageId || randomId();\n}\n"],"mappings":";;;;;;;;;;;;AASA,SAAgB,iBACd,gBACQ;AACR,QAAO,oDAA4B"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
import "reflect-metadata";
|
|
2
|
+
import { randomId } from "@copilotkit/shared";
|
|
3
|
+
|
|
4
|
+
//#region src/graphql/resolvers/resolve-message-id.ts
|
|
5
|
+
/**
|
|
6
|
+
* Resolve the id to use for a streamed TextMessageOutput.
|
|
7
|
+
*
|
|
8
|
+
* Upstream events occasionally arrive with `messageId` missing, null, or an
|
|
9
|
+
* empty string. We fall back to a freshly generated id in those cases so the
|
|
10
|
+
* resulting output never surfaces a null id to the GraphQL client (#2118).
|
|
11
|
+
*/
|
|
12
|
+
function resolveMessageId(eventMessageId) {
|
|
13
|
+
return eventMessageId || randomId();
|
|
14
|
+
}
|
|
15
|
+
|
|
16
|
+
//#endregion
|
|
17
|
+
export { resolveMessageId };
|
|
18
|
+
//# sourceMappingURL=resolve-message-id.mjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"resolve-message-id.mjs","names":[],"sources":["../../../src/graphql/resolvers/resolve-message-id.ts"],"sourcesContent":["import { randomId } from \"@copilotkit/shared\";\n\n/**\n * Resolve the id to use for a streamed TextMessageOutput.\n *\n * Upstream events occasionally arrive with `messageId` missing, null, or an\n * empty string. We fall back to a freshly generated id in those cases so the\n * resulting output never surfaces a null id to the GraphQL client (#2118).\n */\nexport function resolveMessageId(\n eventMessageId: string | null | undefined,\n): string {\n return eventMessageId || randomId();\n}\n"],"mappings":";;;;;;;;;;;AASA,SAAgB,iBACd,gBACQ;AACR,QAAO,kBAAkB,UAAU"}
|
|
@@ -93,7 +93,14 @@ var LangGraphAgent = class extends _ag_ui_langgraph.LangGraphAgent {
|
|
|
93
93
|
return true;
|
|
94
94
|
}
|
|
95
95
|
run(input) {
|
|
96
|
-
|
|
96
|
+
const enrichedInput = {
|
|
97
|
+
...input,
|
|
98
|
+
forwardedProps: {
|
|
99
|
+
...input.forwardedProps,
|
|
100
|
+
streamSubgraphs: input.forwardedProps?.streamSubgraphs ?? true
|
|
101
|
+
}
|
|
102
|
+
};
|
|
103
|
+
return super.run(enrichedInput).pipe((0, rxjs.map)((processedEvent) => {
|
|
97
104
|
if (processedEvent.type === _ag_ui_client.EventType.RAW) {
|
|
98
105
|
const event = processedEvent.event ?? processedEvent.rawEvent;
|
|
99
106
|
const eventType = event.event;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"agent.cjs","names":["AGUILangGraphAgent","EventType","CustomEventNames","LangGraphEventTypes"],"sources":["../../../../../src/lib/runtime/agent-integrations/langgraph/agent.ts"],"sourcesContent":["import { map, Observable } from \"rxjs\";\nimport { LangGraphEventTypes } from \"../../../../agents/langgraph/events\";\nimport { BaseEvent, RawEvent } from \"@ag-ui/core\";\nimport {\n LangGraphAgent as AGUILangGraphAgent,\n LangGraphHttpAgent,\n type LangGraphAgentConfig,\n ProcessedEvents,\n SchemaKeys,\n type State,\n StateEnrichment,\n} from \"@ag-ui/langgraph\";\nimport { Message as LangGraphMessage } from \"@langchain/langgraph-sdk/dist/types.messages\";\nimport { ThreadState } from \"@langchain/langgraph-sdk\";\n\ninterface CopilotKitStateEnrichment {\n copilotkit: {\n actions: StateEnrichment[\"ag-ui\"][\"tools\"];\n context: StateEnrichment[\"ag-ui\"][\"context\"];\n };\n}\n\nimport { RunAgentInput, EventType, CustomEvent } from \"@ag-ui/client\";\n\n// Import and re-export from separate file to maintain API compatibility\nimport {\n CustomEventNames,\n TextMessageEvents,\n ToolCallEvents,\n PredictStateTool,\n} from \"./consts\";\nexport { CustomEventNames };\n\nexport class LangGraphAgent extends AGUILangGraphAgent {\n constructor(config: LangGraphAgentConfig) {\n super(config);\n }\n\n dispatchEvent(event: ProcessedEvents) {\n if (event.type === EventType.CUSTOM) {\n // const event = processedEvent as unknown as CustomEvent;\n const customEvent = event as unknown as CustomEvent;\n\n if (customEvent.name === CustomEventNames.CopilotKitManuallyEmitMessage) {\n this.subscriber.next({\n type: EventType.TEXT_MESSAGE_START,\n role: \"assistant\",\n messageId: customEvent.value.message_id,\n rawEvent: event,\n });\n this.subscriber.next({\n type: EventType.TEXT_MESSAGE_CONTENT,\n messageId: customEvent.value.message_id,\n delta: customEvent.value.message,\n rawEvent: event,\n });\n this.subscriber.next({\n type: EventType.TEXT_MESSAGE_END,\n messageId: customEvent.value.message_id,\n rawEvent: event,\n });\n return true;\n }\n\n if (\n customEvent.name === CustomEventNames.CopilotKitManuallyEmitToolCall\n ) {\n this.subscriber.next({\n type: EventType.TOOL_CALL_START,\n toolCallId: customEvent.value.id,\n toolCallName: customEvent.value.name,\n parentMessageId: customEvent.value.id,\n rawEvent: event,\n });\n this.subscriber.next({\n type: EventType.TOOL_CALL_ARGS,\n toolCallId: customEvent.value.id,\n delta: customEvent.value.args,\n rawEvent: event,\n });\n this.subscriber.next({\n type: EventType.TOOL_CALL_END,\n toolCallId: customEvent.value.id,\n rawEvent: event,\n });\n return true;\n }\n\n if (\n customEvent.name ===\n CustomEventNames.CopilotKitManuallyEmitIntermediateState\n ) {\n this.activeRun.manuallyEmittedState = customEvent.value;\n this.dispatchEvent({\n type: EventType.STATE_SNAPSHOT,\n snapshot: this.getStateSnapshot({\n values: this.activeRun.manuallyEmittedState,\n } as ThreadState<State>),\n rawEvent: event,\n });\n return true;\n }\n\n if (customEvent.name === CustomEventNames.CopilotKitExit) {\n this.subscriber.next({\n type: EventType.CUSTOM,\n name: \"Exit\",\n value: true,\n });\n return true;\n }\n }\n\n // Intercept all text message and tool call events and check if should disable\n const rawEvent = (event as ToolCallEvents | TextMessageEvents).rawEvent;\n if (!rawEvent) {\n this.subscriber.next(event);\n return true;\n }\n\n const isMessageEvent =\n event.type === EventType.TEXT_MESSAGE_START ||\n event.type === EventType.TEXT_MESSAGE_CONTENT ||\n event.type === EventType.TEXT_MESSAGE_END;\n const isToolEvent =\n event.type === EventType.TOOL_CALL_START ||\n event.type === EventType.TOOL_CALL_ARGS ||\n event.type === EventType.TOOL_CALL_END;\n if (\"copilotkit:emit-tool-calls\" in (rawEvent.metadata || {})) {\n if (\n rawEvent.metadata[\"copilotkit:emit-tool-calls\"] === false &&\n isToolEvent\n ) {\n return false;\n }\n }\n if (\"copilotkit:emit-messages\" in (rawEvent.metadata || {})) {\n if (\n rawEvent.metadata[\"copilotkit:emit-messages\"] === false &&\n isMessageEvent\n ) {\n // Clean up tracked message state to prevent stale records from\n // leaking into subsequent nodes that have emit-messages enabled.\n if (this.activeRun?.id) {\n this.messagesInProcess[this.activeRun.id] = null;\n }\n return false;\n }\n }\n\n this.subscriber.next(event);\n return true;\n }\n\n // @ts-ignore\n run(input: RunAgentInput): Observable<BaseEvent> {\n return super.run(input).pipe(\n map((processedEvent) => {\n // Turn raw event into emit state snapshot from tool call event\n if (processedEvent.type === EventType.RAW) {\n // Get the LangGraph event from the AGUI event.\n const event =\n (processedEvent as RawEvent).event ??\n (processedEvent as RawEvent).rawEvent;\n\n const eventType = event.event;\n const toolCallData = event.data?.chunk?.tool_call_chunks?.[0];\n const toolCallUsedToPredictState = event.metadata?.[\n \"copilotkit:emit-intermediate-state\"\n ]?.some(\n (predictStateTool: PredictStateTool) =>\n predictStateTool.tool === toolCallData?.name,\n );\n\n if (\n eventType === LangGraphEventTypes.OnChatModelStream &&\n toolCallUsedToPredictState\n ) {\n return {\n type: EventType.CUSTOM,\n name: \"PredictState\",\n value: event.metadata[\"copilotkit:emit-intermediate-state\"],\n };\n }\n }\n\n return processedEvent;\n }),\n );\n }\n\n langGraphDefaultMergeState(\n state: State,\n messages: LangGraphMessage[],\n input: RunAgentInput,\n ): State<StateEnrichment & CopilotKitStateEnrichment> {\n const aguiMergedState = super.langGraphDefaultMergeState(\n state,\n messages,\n input,\n );\n const { tools: returnedTools, \"ag-ui\": agui } = aguiMergedState;\n // tolerate undefined and de-duplicate by stable key (id | name | key)\n const rawCombinedTools = [\n ...((returnedTools as any[]) ?? []),\n ...((agui?.tools as any[]) ?? []),\n ];\n const combinedTools = Array.from(\n new Map(\n rawCombinedTools.map((t: any) => [\n t?.id ?? t?.name ?? t?.key ?? JSON.stringify(t),\n t,\n ]),\n ).values(),\n );\n\n return {\n ...aguiMergedState,\n copilotkit: {\n actions: combinedTools,\n context: agui?.context ?? [],\n },\n };\n }\n\n async getSchemaKeys(): Promise<SchemaKeys> {\n const CONSTANT_KEYS = [\"copilotkit\"];\n const schemaKeys = await super.getSchemaKeys();\n return {\n config: schemaKeys.config,\n input: schemaKeys.input ? [...schemaKeys.input, ...CONSTANT_KEYS] : null,\n output: schemaKeys.output\n ? [...schemaKeys.output, ...CONSTANT_KEYS]\n : null,\n context: schemaKeys.context\n ? [...schemaKeys.context, ...CONSTANT_KEYS]\n : null,\n };\n }\n}\n\nexport { LangGraphHttpAgent };\n"],"mappings":";;;;;;;;;AAiCA,IAAa,iBAAb,cAAoCA,gCAAmB;CACrD,YAAY,QAA8B;AACxC,QAAM,OAAO;;CAGf,cAAc,OAAwB;AACpC,MAAI,MAAM,SAASC,wBAAU,QAAQ;GAEnC,MAAM,cAAc;AAEpB,OAAI,YAAY,SAASC,gCAAiB,+BAA+B;AACvE,SAAK,WAAW,KAAK;KACnB,MAAMD,wBAAU;KAChB,MAAM;KACN,WAAW,YAAY,MAAM;KAC7B,UAAU;KACX,CAAC;AACF,SAAK,WAAW,KAAK;KACnB,MAAMA,wBAAU;KAChB,WAAW,YAAY,MAAM;KAC7B,OAAO,YAAY,MAAM;KACzB,UAAU;KACX,CAAC;AACF,SAAK,WAAW,KAAK;KACnB,MAAMA,wBAAU;KAChB,WAAW,YAAY,MAAM;KAC7B,UAAU;KACX,CAAC;AACF,WAAO;;AAGT,OACE,YAAY,SAASC,gCAAiB,gCACtC;AACA,SAAK,WAAW,KAAK;KACnB,MAAMD,wBAAU;KAChB,YAAY,YAAY,MAAM;KAC9B,cAAc,YAAY,MAAM;KAChC,iBAAiB,YAAY,MAAM;KACnC,UAAU;KACX,CAAC;AACF,SAAK,WAAW,KAAK;KACnB,MAAMA,wBAAU;KAChB,YAAY,YAAY,MAAM;KAC9B,OAAO,YAAY,MAAM;KACzB,UAAU;KACX,CAAC;AACF,SAAK,WAAW,KAAK;KACnB,MAAMA,wBAAU;KAChB,YAAY,YAAY,MAAM;KAC9B,UAAU;KACX,CAAC;AACF,WAAO;;AAGT,OACE,YAAY,SACZC,gCAAiB,yCACjB;AACA,SAAK,UAAU,uBAAuB,YAAY;AAClD,SAAK,cAAc;KACjB,MAAMD,wBAAU;KAChB,UAAU,KAAK,iBAAiB,EAC9B,QAAQ,KAAK,UAAU,sBACxB,CAAuB;KACxB,UAAU;KACX,CAAC;AACF,WAAO;;AAGT,OAAI,YAAY,SAASC,gCAAiB,gBAAgB;AACxD,SAAK,WAAW,KAAK;KACnB,MAAMD,wBAAU;KAChB,MAAM;KACN,OAAO;KACR,CAAC;AACF,WAAO;;;EAKX,MAAM,WAAY,MAA6C;AAC/D,MAAI,CAAC,UAAU;AACb,QAAK,WAAW,KAAK,MAAM;AAC3B,UAAO;;EAGT,MAAM,iBACJ,MAAM,SAASA,wBAAU,sBACzB,MAAM,SAASA,wBAAU,wBACzB,MAAM,SAASA,wBAAU;EAC3B,MAAM,cACJ,MAAM,SAASA,wBAAU,mBACzB,MAAM,SAASA,wBAAU,kBACzB,MAAM,SAASA,wBAAU;AAC3B,MAAI,iCAAiC,SAAS,YAAY,EAAE,GAC1D;OACE,SAAS,SAAS,kCAAkC,SACpD,YAEA,QAAO;;AAGX,MAAI,+BAA+B,SAAS,YAAY,EAAE,GACxD;OACE,SAAS,SAAS,gCAAgC,SAClD,gBACA;AAGA,QAAI,KAAK,WAAW,GAClB,MAAK,kBAAkB,KAAK,UAAU,MAAM;AAE9C,WAAO;;;AAIX,OAAK,WAAW,KAAK,MAAM;AAC3B,SAAO;;CAIT,IAAI,OAA6C;AAC/C,SAAO,MAAM,IAAI,MAAM,CAAC,oBACjB,mBAAmB;AAEtB,OAAI,eAAe,SAASA,wBAAU,KAAK;IAEzC,MAAM,QACH,eAA4B,SAC5B,eAA4B;IAE/B,MAAM,YAAY,MAAM;IACxB,MAAM,eAAe,MAAM,MAAM,OAAO,mBAAmB;IAC3D,MAAM,6BAA6B,MAAM,WACvC,uCACC,MACA,qBACC,iBAAiB,SAAS,cAAc,KAC3C;AAED,QACE,cAAcE,mCAAoB,qBAClC,2BAEA,QAAO;KACL,MAAMF,wBAAU;KAChB,MAAM;KACN,OAAO,MAAM,SAAS;KACvB;;AAIL,UAAO;IACP,CACH;;CAGH,2BACE,OACA,UACA,OACoD;EACpD,MAAM,kBAAkB,MAAM,2BAC5B,OACA,UACA,MACD;EACD,MAAM,EAAE,OAAO,eAAe,SAAS,SAAS;EAEhD,MAAM,mBAAmB,CACvB,GAAK,iBAA2B,EAAE,EAClC,GAAK,MAAM,SAAmB,EAAE,CACjC;EACD,MAAM,gBAAgB,MAAM,KAC1B,IAAI,IACF,iBAAiB,KAAK,MAAW,CAC/B,GAAG,MAAM,GAAG,QAAQ,GAAG,OAAO,KAAK,UAAU,EAAE,EAC/C,EACD,CAAC,CACH,CAAC,QAAQ,CACX;AAED,SAAO;GACL,GAAG;GACH,YAAY;IACV,SAAS;IACT,SAAS,MAAM,WAAW,EAAE;IAC7B;GACF;;CAGH,MAAM,gBAAqC;EACzC,MAAM,gBAAgB,CAAC,aAAa;EACpC,MAAM,aAAa,MAAM,MAAM,eAAe;AAC9C,SAAO;GACL,QAAQ,WAAW;GACnB,OAAO,WAAW,QAAQ,CAAC,GAAG,WAAW,OAAO,GAAG,cAAc,GAAG;GACpE,QAAQ,WAAW,SACf,CAAC,GAAG,WAAW,QAAQ,GAAG,cAAc,GACxC;GACJ,SAAS,WAAW,UAChB,CAAC,GAAG,WAAW,SAAS,GAAG,cAAc,GACzC;GACL"}
|
|
1
|
+
{"version":3,"file":"agent.cjs","names":["AGUILangGraphAgent","EventType","CustomEventNames","LangGraphEventTypes"],"sources":["../../../../../src/lib/runtime/agent-integrations/langgraph/agent.ts"],"sourcesContent":["import { map, Observable } from \"rxjs\";\nimport { LangGraphEventTypes } from \"../../../../agents/langgraph/events\";\nimport { BaseEvent, RawEvent } from \"@ag-ui/core\";\nimport {\n LangGraphAgent as AGUILangGraphAgent,\n LangGraphHttpAgent,\n type LangGraphAgentConfig,\n ProcessedEvents,\n SchemaKeys,\n type State,\n StateEnrichment,\n} from \"@ag-ui/langgraph\";\nimport { Message as LangGraphMessage } from \"@langchain/langgraph-sdk/dist/types.messages\";\nimport { ThreadState } from \"@langchain/langgraph-sdk\";\n\ninterface CopilotKitStateEnrichment {\n copilotkit: {\n actions: StateEnrichment[\"ag-ui\"][\"tools\"];\n context: StateEnrichment[\"ag-ui\"][\"context\"];\n };\n}\n\nimport { RunAgentInput, EventType, CustomEvent } from \"@ag-ui/client\";\n\n// Import and re-export from separate file to maintain API compatibility\nimport {\n CustomEventNames,\n TextMessageEvents,\n ToolCallEvents,\n PredictStateTool,\n} from \"./consts\";\nexport { CustomEventNames };\n\nexport class LangGraphAgent extends AGUILangGraphAgent {\n constructor(config: LangGraphAgentConfig) {\n super(config);\n }\n\n dispatchEvent(event: ProcessedEvents) {\n if (event.type === EventType.CUSTOM) {\n // const event = processedEvent as unknown as CustomEvent;\n const customEvent = event as unknown as CustomEvent;\n\n if (customEvent.name === CustomEventNames.CopilotKitManuallyEmitMessage) {\n this.subscriber.next({\n type: EventType.TEXT_MESSAGE_START,\n role: \"assistant\",\n messageId: customEvent.value.message_id,\n rawEvent: event,\n });\n this.subscriber.next({\n type: EventType.TEXT_MESSAGE_CONTENT,\n messageId: customEvent.value.message_id,\n delta: customEvent.value.message,\n rawEvent: event,\n });\n this.subscriber.next({\n type: EventType.TEXT_MESSAGE_END,\n messageId: customEvent.value.message_id,\n rawEvent: event,\n });\n return true;\n }\n\n if (\n customEvent.name === CustomEventNames.CopilotKitManuallyEmitToolCall\n ) {\n this.subscriber.next({\n type: EventType.TOOL_CALL_START,\n toolCallId: customEvent.value.id,\n toolCallName: customEvent.value.name,\n parentMessageId: customEvent.value.id,\n rawEvent: event,\n });\n this.subscriber.next({\n type: EventType.TOOL_CALL_ARGS,\n toolCallId: customEvent.value.id,\n delta: customEvent.value.args,\n rawEvent: event,\n });\n this.subscriber.next({\n type: EventType.TOOL_CALL_END,\n toolCallId: customEvent.value.id,\n rawEvent: event,\n });\n return true;\n }\n\n if (\n customEvent.name ===\n CustomEventNames.CopilotKitManuallyEmitIntermediateState\n ) {\n this.activeRun.manuallyEmittedState = customEvent.value;\n this.dispatchEvent({\n type: EventType.STATE_SNAPSHOT,\n snapshot: this.getStateSnapshot({\n values: this.activeRun.manuallyEmittedState,\n } as ThreadState<State>),\n rawEvent: event,\n });\n return true;\n }\n\n if (customEvent.name === CustomEventNames.CopilotKitExit) {\n this.subscriber.next({\n type: EventType.CUSTOM,\n name: \"Exit\",\n value: true,\n });\n return true;\n }\n }\n\n // Intercept all text message and tool call events and check if should disable\n const rawEvent = (event as ToolCallEvents | TextMessageEvents).rawEvent;\n if (!rawEvent) {\n this.subscriber.next(event);\n return true;\n }\n\n const isMessageEvent =\n event.type === EventType.TEXT_MESSAGE_START ||\n event.type === EventType.TEXT_MESSAGE_CONTENT ||\n event.type === EventType.TEXT_MESSAGE_END;\n const isToolEvent =\n event.type === EventType.TOOL_CALL_START ||\n event.type === EventType.TOOL_CALL_ARGS ||\n event.type === EventType.TOOL_CALL_END;\n if (\"copilotkit:emit-tool-calls\" in (rawEvent.metadata || {})) {\n if (\n rawEvent.metadata[\"copilotkit:emit-tool-calls\"] === false &&\n isToolEvent\n ) {\n return false;\n }\n }\n if (\"copilotkit:emit-messages\" in (rawEvent.metadata || {})) {\n if (\n rawEvent.metadata[\"copilotkit:emit-messages\"] === false &&\n isMessageEvent\n ) {\n // Clean up tracked message state to prevent stale records from\n // leaking into subsequent nodes that have emit-messages enabled.\n if (this.activeRun?.id) {\n this.messagesInProcess[this.activeRun.id] = null;\n }\n return false;\n }\n }\n\n this.subscriber.next(event);\n return true;\n }\n\n // @ts-ignore\n run(input: RunAgentInput): Observable<BaseEvent> {\n const enrichedInput = {\n ...input,\n forwardedProps: {\n ...input.forwardedProps,\n streamSubgraphs: input.forwardedProps?.streamSubgraphs ?? true,\n },\n };\n return super.run(enrichedInput).pipe(\n map((processedEvent) => {\n // Turn raw event into emit state snapshot from tool call event\n if (processedEvent.type === EventType.RAW) {\n // Get the LangGraph event from the AGUI event.\n const event =\n (processedEvent as RawEvent).event ??\n (processedEvent as RawEvent).rawEvent;\n\n const eventType = event.event;\n const toolCallData = event.data?.chunk?.tool_call_chunks?.[0];\n const toolCallUsedToPredictState = event.metadata?.[\n \"copilotkit:emit-intermediate-state\"\n ]?.some(\n (predictStateTool: PredictStateTool) =>\n predictStateTool.tool === toolCallData?.name,\n );\n\n if (\n eventType === LangGraphEventTypes.OnChatModelStream &&\n toolCallUsedToPredictState\n ) {\n return {\n type: EventType.CUSTOM,\n name: \"PredictState\",\n value: event.metadata[\"copilotkit:emit-intermediate-state\"],\n };\n }\n }\n\n return processedEvent;\n }),\n );\n }\n\n langGraphDefaultMergeState(\n state: State,\n messages: LangGraphMessage[],\n input: RunAgentInput,\n ): State<StateEnrichment & CopilotKitStateEnrichment> {\n const aguiMergedState = super.langGraphDefaultMergeState(\n state,\n messages,\n input,\n );\n const { tools: returnedTools, \"ag-ui\": agui } = aguiMergedState;\n // tolerate undefined and de-duplicate by stable key (id | name | key)\n const rawCombinedTools = [\n ...((returnedTools as any[]) ?? []),\n ...((agui?.tools as any[]) ?? []),\n ];\n const combinedTools = Array.from(\n new Map(\n rawCombinedTools.map((t: any) => [\n t?.id ?? t?.name ?? t?.key ?? JSON.stringify(t),\n t,\n ]),\n ).values(),\n );\n\n return {\n ...aguiMergedState,\n copilotkit: {\n actions: combinedTools,\n context: agui?.context ?? [],\n },\n };\n }\n\n async getSchemaKeys(): Promise<SchemaKeys> {\n const CONSTANT_KEYS = [\"copilotkit\"];\n const schemaKeys = await super.getSchemaKeys();\n return {\n config: schemaKeys.config,\n input: schemaKeys.input ? [...schemaKeys.input, ...CONSTANT_KEYS] : null,\n output: schemaKeys.output\n ? [...schemaKeys.output, ...CONSTANT_KEYS]\n : null,\n context: schemaKeys.context\n ? [...schemaKeys.context, ...CONSTANT_KEYS]\n : null,\n };\n }\n}\n\nexport { LangGraphHttpAgent };\n"],"mappings":";;;;;;;;;AAiCA,IAAa,iBAAb,cAAoCA,gCAAmB;CACrD,YAAY,QAA8B;AACxC,QAAM,OAAO;;CAGf,cAAc,OAAwB;AACpC,MAAI,MAAM,SAASC,wBAAU,QAAQ;GAEnC,MAAM,cAAc;AAEpB,OAAI,YAAY,SAASC,gCAAiB,+BAA+B;AACvE,SAAK,WAAW,KAAK;KACnB,MAAMD,wBAAU;KAChB,MAAM;KACN,WAAW,YAAY,MAAM;KAC7B,UAAU;KACX,CAAC;AACF,SAAK,WAAW,KAAK;KACnB,MAAMA,wBAAU;KAChB,WAAW,YAAY,MAAM;KAC7B,OAAO,YAAY,MAAM;KACzB,UAAU;KACX,CAAC;AACF,SAAK,WAAW,KAAK;KACnB,MAAMA,wBAAU;KAChB,WAAW,YAAY,MAAM;KAC7B,UAAU;KACX,CAAC;AACF,WAAO;;AAGT,OACE,YAAY,SAASC,gCAAiB,gCACtC;AACA,SAAK,WAAW,KAAK;KACnB,MAAMD,wBAAU;KAChB,YAAY,YAAY,MAAM;KAC9B,cAAc,YAAY,MAAM;KAChC,iBAAiB,YAAY,MAAM;KACnC,UAAU;KACX,CAAC;AACF,SAAK,WAAW,KAAK;KACnB,MAAMA,wBAAU;KAChB,YAAY,YAAY,MAAM;KAC9B,OAAO,YAAY,MAAM;KACzB,UAAU;KACX,CAAC;AACF,SAAK,WAAW,KAAK;KACnB,MAAMA,wBAAU;KAChB,YAAY,YAAY,MAAM;KAC9B,UAAU;KACX,CAAC;AACF,WAAO;;AAGT,OACE,YAAY,SACZC,gCAAiB,yCACjB;AACA,SAAK,UAAU,uBAAuB,YAAY;AAClD,SAAK,cAAc;KACjB,MAAMD,wBAAU;KAChB,UAAU,KAAK,iBAAiB,EAC9B,QAAQ,KAAK,UAAU,sBACxB,CAAuB;KACxB,UAAU;KACX,CAAC;AACF,WAAO;;AAGT,OAAI,YAAY,SAASC,gCAAiB,gBAAgB;AACxD,SAAK,WAAW,KAAK;KACnB,MAAMD,wBAAU;KAChB,MAAM;KACN,OAAO;KACR,CAAC;AACF,WAAO;;;EAKX,MAAM,WAAY,MAA6C;AAC/D,MAAI,CAAC,UAAU;AACb,QAAK,WAAW,KAAK,MAAM;AAC3B,UAAO;;EAGT,MAAM,iBACJ,MAAM,SAASA,wBAAU,sBACzB,MAAM,SAASA,wBAAU,wBACzB,MAAM,SAASA,wBAAU;EAC3B,MAAM,cACJ,MAAM,SAASA,wBAAU,mBACzB,MAAM,SAASA,wBAAU,kBACzB,MAAM,SAASA,wBAAU;AAC3B,MAAI,iCAAiC,SAAS,YAAY,EAAE,GAC1D;OACE,SAAS,SAAS,kCAAkC,SACpD,YAEA,QAAO;;AAGX,MAAI,+BAA+B,SAAS,YAAY,EAAE,GACxD;OACE,SAAS,SAAS,gCAAgC,SAClD,gBACA;AAGA,QAAI,KAAK,WAAW,GAClB,MAAK,kBAAkB,KAAK,UAAU,MAAM;AAE9C,WAAO;;;AAIX,OAAK,WAAW,KAAK,MAAM;AAC3B,SAAO;;CAIT,IAAI,OAA6C;EAC/C,MAAM,gBAAgB;GACpB,GAAG;GACH,gBAAgB;IACd,GAAG,MAAM;IACT,iBAAiB,MAAM,gBAAgB,mBAAmB;IAC3D;GACF;AACD,SAAO,MAAM,IAAI,cAAc,CAAC,oBACzB,mBAAmB;AAEtB,OAAI,eAAe,SAASA,wBAAU,KAAK;IAEzC,MAAM,QACH,eAA4B,SAC5B,eAA4B;IAE/B,MAAM,YAAY,MAAM;IACxB,MAAM,eAAe,MAAM,MAAM,OAAO,mBAAmB;IAC3D,MAAM,6BAA6B,MAAM,WACvC,uCACC,MACA,qBACC,iBAAiB,SAAS,cAAc,KAC3C;AAED,QACE,cAAcE,mCAAoB,qBAClC,2BAEA,QAAO;KACL,MAAMF,wBAAU;KAChB,MAAM;KACN,OAAO,MAAM,SAAS;KACvB;;AAIL,UAAO;IACP,CACH;;CAGH,2BACE,OACA,UACA,OACoD;EACpD,MAAM,kBAAkB,MAAM,2BAC5B,OACA,UACA,MACD;EACD,MAAM,EAAE,OAAO,eAAe,SAAS,SAAS;EAEhD,MAAM,mBAAmB,CACvB,GAAK,iBAA2B,EAAE,EAClC,GAAK,MAAM,SAAmB,EAAE,CACjC;EACD,MAAM,gBAAgB,MAAM,KAC1B,IAAI,IACF,iBAAiB,KAAK,MAAW,CAC/B,GAAG,MAAM,GAAG,QAAQ,GAAG,OAAO,KAAK,UAAU,EAAE,EAC/C,EACD,CAAC,CACH,CAAC,QAAQ,CACX;AAED,SAAO;GACL,GAAG;GACH,YAAY;IACV,SAAS;IACT,SAAS,MAAM,WAAW,EAAE;IAC7B;GACF;;CAGH,MAAM,gBAAqC;EACzC,MAAM,gBAAgB,CAAC,aAAa;EACpC,MAAM,aAAa,MAAM,MAAM,eAAe;AAC9C,SAAO;GACL,QAAQ,WAAW;GACnB,OAAO,WAAW,QAAQ,CAAC,GAAG,WAAW,OAAO,GAAG,cAAc,GAAG;GACpE,QAAQ,WAAW,SACf,CAAC,GAAG,WAAW,QAAQ,GAAG,cAAc,GACxC;GACJ,SAAS,WAAW,UAChB,CAAC,GAAG,WAAW,SAAS,GAAG,cAAc,GACzC;GACL"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"agent.d.cts","names":[],"sources":["../../../../../src/lib/runtime/agent-integrations/langgraph/agent.ts"],"mappings":";;;;;;;;;UAeU,yBAAA;EACR,UAAA;IACE,OAAA,EAAS,eAAA;IACT,OAAA,EAAS,eAAA;EAAA;AAAA;AAAA,cAeA,gBAAA,SAAuB,cAAA;cACtB,MAAA,EAAQ,oBAAA;EAIpB,aAAA,CAAc,KAAA,EAAO,eAAA;EAqHrB,GAAA,CAAI,KAAA,EAAO,aAAA,GAAgB,UAAA,CAAW,WAAA;
|
|
1
|
+
{"version":3,"file":"agent.d.cts","names":[],"sources":["../../../../../src/lib/runtime/agent-integrations/langgraph/agent.ts"],"mappings":";;;;;;;;;UAeU,yBAAA;EACR,UAAA;IACE,OAAA,EAAS,eAAA;IACT,OAAA,EAAS,eAAA;EAAA;AAAA;AAAA,cAeA,gBAAA,SAAuB,cAAA;cACtB,MAAA,EAAQ,oBAAA;EAIpB,aAAA,CAAc,KAAA,EAAO,eAAA;EAqHrB,GAAA,CAAI,KAAA,EAAO,aAAA,GAAgB,UAAA,CAAW,WAAA;EA2CtC,0BAAA,CACE,KAAA,EAAO,KAAA,EACP,QAAA,EAAU,SAAA,IACV,KAAA,EAAO,aAAA,GACN,KAAA,CAAM,eAAA,GAAkB,yBAAA;EA8BrB,aAAA,CAAA,GAAiB,OAAA,CAAQ,UAAA;AAAA"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"agent.d.mts","names":[],"sources":["../../../../../src/lib/runtime/agent-integrations/langgraph/agent.ts"],"mappings":";;;;;;;;;UAeU,yBAAA;EACR,UAAA;IACE,OAAA,EAAS,eAAA;IACT,OAAA,EAAS,eAAA;EAAA;AAAA;AAAA,cAeA,gBAAA,SAAuB,cAAA;cACtB,MAAA,EAAQ,oBAAA;EAIpB,aAAA,CAAc,KAAA,EAAO,eAAA;EAqHrB,GAAA,CAAI,KAAA,EAAO,aAAA,GAAgB,UAAA,CAAW,WAAA;
|
|
1
|
+
{"version":3,"file":"agent.d.mts","names":[],"sources":["../../../../../src/lib/runtime/agent-integrations/langgraph/agent.ts"],"mappings":";;;;;;;;;UAeU,yBAAA;EACR,UAAA;IACE,OAAA,EAAS,eAAA;IACT,OAAA,EAAS,eAAA;EAAA;AAAA;AAAA,cAeA,gBAAA,SAAuB,cAAA;cACtB,MAAA,EAAQ,oBAAA;EAIpB,aAAA,CAAc,KAAA,EAAO,eAAA;EAqHrB,GAAA,CAAI,KAAA,EAAO,aAAA,GAAgB,UAAA,CAAW,WAAA;EA2CtC,0BAAA,CACE,KAAA,EAAO,KAAA,EACP,QAAA,EAAU,SAAA,IACV,KAAA,EAAO,aAAA,GACN,KAAA,CAAM,eAAA,GAAkB,yBAAA;EA8BrB,aAAA,CAAA,GAAiB,OAAA,CAAQ,UAAA;AAAA"}
|
|
@@ -92,7 +92,14 @@ var LangGraphAgent$1 = class extends LangGraphAgent {
|
|
|
92
92
|
return true;
|
|
93
93
|
}
|
|
94
94
|
run(input) {
|
|
95
|
-
|
|
95
|
+
const enrichedInput = {
|
|
96
|
+
...input,
|
|
97
|
+
forwardedProps: {
|
|
98
|
+
...input.forwardedProps,
|
|
99
|
+
streamSubgraphs: input.forwardedProps?.streamSubgraphs ?? true
|
|
100
|
+
}
|
|
101
|
+
};
|
|
102
|
+
return super.run(enrichedInput).pipe(map((processedEvent) => {
|
|
96
103
|
if (processedEvent.type === EventType.RAW) {
|
|
97
104
|
const event = processedEvent.event ?? processedEvent.rawEvent;
|
|
98
105
|
const eventType = event.event;
|