@copilotkitnext/core 1.52.2-next.0 → 1.52.2-next.2

package/dist/index.cjs

@@ -4,6 +4,7 @@ let _copilotkitnext_shared = require("@copilotkitnext/shared");
 let rxjs = require("rxjs");
 let rxjs_operators = require("rxjs/operators");
 let zod_to_json_schema = require("zod-to-json-schema");
+let phoenix = require("phoenix");
 
 //#region src/agent.ts
 /**
@@ -1717,12 +1718,162 @@ function completePartialMarkdown(input) {
 	return result;
 }
 
+//#endregion
+//#region src/intelligence-agent.ts
+var IntelligenceAgent = class IntelligenceAgent extends _ag_ui_client.AbstractAgent {
+	config;
+	socket = null;
+	activeChannel = null;
+	threadId = null;
+	constructor(config) {
+		super();
+		this.config = config;
+	}
+	clone() {
+		return new IntelligenceAgent(this.config);
+	}
+	abortRun() {
+		if (!this.threadId) return;
+		if (typeof fetch === "undefined") {
+			this.cleanup();
+			return;
+		}
+		const { runtimeUrl, agentId, headers, credentials } = this.config;
+		const stopPath = `${runtimeUrl}/agent/${encodeURIComponent(agentId)}/stop/${encodeURIComponent(this.threadId)}`;
+		const origin = typeof window !== "undefined" && window.location ? window.location.origin : "http://localhost";
+		const stopUrl = new URL(stopPath, new URL(runtimeUrl, origin));
+		fetch(stopUrl.toString(), {
+			method: "POST",
+			headers: {
+				"Content-Type": "application/json",
+				...headers
+			},
+			...credentials ? { credentials } : {}
+		}).catch((error) => {
+			console.error("IntelligenceAgent: stop request failed", error);
+		});
+		this.cleanup();
+	}
+	/**
+	* Connect to a Phoenix channel scoped to the thread, trigger the run via
+	* REST, and relay server-pushed AG-UI events to the Observable subscriber.
+	*
+	* The server pushes each AG-UI event using its EventType string as the
+	* Phoenix event name (e.g. "TEXT_MESSAGE_CHUNK", "TOOL_CALL_START"), with
+	* the full BaseEvent as payload. RUN_FINISHED and RUN_ERROR are terminal
+	* events that complete or error the Observable.
+	*/
+	run(input) {
+		return new rxjs.Observable((observer) => {
+			this.threadId = input.threadId;
+			const socket = new phoenix.Socket(this.config.url, { params: this.config.socketParams ?? {} });
+			this.socket = socket;
+			socket.connect();
+			const channel = socket.channel(`agent:${input.threadId}`, { runId: input.runId });
+			this.activeChannel = channel;
+			channel.on(_copilotkitnext_shared.AG_UI_CHANNEL_EVENT, (payload) => {
+				observer.next(payload);
+				if (payload.type === _ag_ui_client.EventType.RUN_FINISHED) {
+					observer.complete();
+					this.cleanup();
+				} else if (payload.type === _ag_ui_client.EventType.RUN_ERROR) {
+					observer.error(new Error(payload.message ?? "Run error"));
+					this.cleanup();
+				}
+			});
+			channel.join().receive("ok", () => {
+				const { runtimeUrl, agentId, headers, credentials } = this.config;
+				const runPath = `${runtimeUrl}/agent/${encodeURIComponent(agentId)}/run`;
+				const origin = typeof window !== "undefined" && window.location ? window.location.origin : "http://localhost";
+				const runUrl = new URL(runPath, new URL(runtimeUrl, origin));
+				fetch(runUrl.toString(), {
+					method: "POST",
+					headers: {
+						"Content-Type": "application/json",
+						...headers
+					},
+					body: JSON.stringify({
+						threadId: input.threadId,
+						runId: input.runId,
+						messages: input.messages,
+						tools: input.tools,
+						context: input.context,
+						state: input.state,
+						forwardedProps: input.forwardedProps
+					}),
+					...credentials ? { credentials } : {}
+				}).catch((error) => {
+					observer.error(/* @__PURE__ */ new Error(`REST run request failed: ${error.message ?? error}`));
+					this.cleanup();
+				});
+			}).receive("error", (resp) => {
+				observer.error(/* @__PURE__ */ new Error(`Failed to join channel: ${JSON.stringify(resp)}`));
+				this.cleanup();
+			}).receive("timeout", () => {
+				observer.error(/* @__PURE__ */ new Error("Timed out joining channel"));
+				this.cleanup();
+			});
+			return () => {
+				this.cleanup();
+			};
+		});
+	}
+	/**
+	* Reconnect to an existing thread by joining the Phoenix channel in
+	* "connect" mode and requesting the server replay history.
+	*/
+	connect(input) {
+		return new rxjs.Observable((observer) => {
+			this.threadId = input.threadId;
+			const socket = new phoenix.Socket(this.config.url, { params: this.config.socketParams ?? {} });
+			this.socket = socket;
+			socket.connect();
+			const channel = socket.channel(`agent:${input.threadId}`, { mode: "connect" });
+			this.activeChannel = channel;
+			channel.on(_copilotkitnext_shared.AG_UI_CHANNEL_EVENT, (payload) => {
+				observer.next(payload);
+				if (payload.type === _ag_ui_client.EventType.RUN_FINISHED || payload.type === _ag_ui_client.EventType.RUN_ERROR) {
+					observer.complete();
+					this.cleanup();
+				}
+			});
+			channel.join().receive("ok", () => {
+				channel.push(_ag_ui_client.EventType.CUSTOM, {
+					type: _ag_ui_client.EventType.CUSTOM,
+					name: "connect",
+					value: { threadId: input.threadId }
+				});
+			}).receive("error", (resp) => {
+				observer.error(/* @__PURE__ */ new Error(`Failed to join channel: ${JSON.stringify(resp)}`));
+				this.cleanup();
+			}).receive("timeout", () => {
+				observer.error(/* @__PURE__ */ new Error("Timed out joining channel"));
+				this.cleanup();
+			});
+			return () => {
+				this.cleanup();
+			};
+		});
+	}
+	cleanup() {
+		if (this.activeChannel) {
+			this.activeChannel.leave();
+			this.activeChannel = null;
+		}
+		if (this.socket) {
+			this.socket.disconnect();
+			this.socket = null;
+		}
+	}
+};
+
 //#endregion
 exports.AgentRegistry = AgentRegistry;
 exports.ContextStore = ContextStore;
 exports.CopilotKitCore = CopilotKitCore;
 exports.CopilotKitCoreErrorCode = CopilotKitCoreErrorCode;
 exports.CopilotKitCoreRuntimeConnectionStatus = CopilotKitCoreRuntimeConnectionStatus;
+exports.IntelligenceAgent = IntelligenceAgent;
 exports.ProxiedCopilotRuntimeAgent = ProxiedCopilotRuntimeAgent;
 exports.RunHandler = RunHandler;
 exports.StateManager = StateManager;