@botbotgo/agent-harness 0.0.475 → 0.0.476

package/dist/protocol/ag-ui/http.js

@@ -1,378 +1,3 @@
-import { createServer } from "node:http";
-import { createPersistentId } from "../../utils/id.js";
-import { createUpstreamTimelineReducer } from "../../projections/upstream-events.js";
-function normalizePath(value, fallback) {
-    const source = typeof value === "string" && value.trim().length > 0 ? value.trim() : fallback;
-    return source.startsWith("/") ? source : `/${source}`;
-}
-function readRequestBody(request) {
-    return new Promise((resolve, reject) => {
-        const chunks = [];
-        request.on("data", (chunk) => {
-            chunks.push(Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk));
-        });
-        request.on("end", () => resolve(Buffer.concat(chunks).toString("utf8")));
-        request.on("error", reject);
-    });
-}
-function writeSseEvent(response, payload) {
-    return new Promise((resolve, reject) => {
-        response.write(`data: ${JSON.stringify(payload)}\n\n`, (error) => {
-            if (error) {
-                reject(error);
-                return;
-            }
-            resolve();
-        });
-    });
-}
-function parseRequestInput(payload) {
-    if (!payload || typeof payload !== "object" || Array.isArray(payload)) {
-        throw new Error("AG-UI request input must be an object.");
-    }
-    const typed = payload;
-    if (typeof typed.input !== "string"
-        && !(Array.isArray(typed.input))) {
-        throw new Error("AG-UI request input requires `input`.");
-    }
-    return {
-        input: typed.input,
-        ...(typeof typed.agentId === "string" && typed.agentId.trim().length > 0 ? { agentId: typed.agentId.trim() } : {}),
-        ...(typeof typed.sessionId === "string" && typed.sessionId.trim().length > 0 ? { sessionId: typed.sessionId.trim() } : {}),
-        ...(typed.invocation && typeof typed.invocation === "object" && !Array.isArray(typed.invocation)
-            ? { invocation: typed.invocation }
-            : {}),
-    };
-}
-function createTimestamp() {
-    return new Date().toISOString();
-}
-function toRunStarted(event, input) {
-    return {
-        type: "RUN_STARTED",
-        timestamp: createTimestamp(),
-        sessionId: event.sessionId,
-        requestId: event.requestId,
-        input,
-    };
-}
-function asObject(value) {
-    return typeof value === "object" && value !== null ? value : null;
-}
-function readUpstreamEventName(event) {
-    return typeof asObject(event)?.event === "string" ? String(asObject(event)?.event) : "";
-}
-function readUpstreamToolName(event) {
-    return typeof asObject(event)?.name === "string" ? String(asObject(event)?.name) : "";
-}
-function readUpstreamRunType(event) {
-    return typeof asObject(event)?.run_type === "string" ? String(asObject(event)?.run_type) : "";
-}
-function isToolStartEvent(event) {
-    const eventName = readUpstreamEventName(event);
-    return eventName === "on_tool_start" || (eventName === "on_chain_start" && readUpstreamRunType(event) === "tool");
-}
-function isToolTerminalEvent(event) {
-    const eventName = readUpstreamEventName(event);
-    return eventName === "on_tool_end"
-        || eventName === "on_tool_error"
-        || ((eventName === "on_chain_end" || eventName === "on_chain_error") && readUpstreamRunType(event) === "tool");
-}
-function readToolArgs(event) {
-    const typed = asObject(event);
-    const data = asObject(typed?.data);
-    if (!data || !("input" in data)) {
-        return undefined;
-    }
-    return data.input;
-}
-function createToolCallState() {
-    const activeToolCallIds = new Map();
-    return {
-        start(toolName) {
-            const toolCallId = `tool-${createPersistentId()}`;
-            const activeIds = activeToolCallIds.get(toolName) ?? [];
-            activeIds.push(toolCallId);
-            activeToolCallIds.set(toolName, activeIds);
-            return toolCallId;
-        },
-        peek(toolName) {
-            const activeIds = activeToolCallIds.get(toolName);
-            return activeIds?.at(-1);
-        },
-        finish(toolName) {
-            const activeIds = activeToolCallIds.get(toolName);
-            if (!activeIds || activeIds.length === 0) {
-                return undefined;
-            }
-            const toolCallId = activeIds.pop();
-            if (!activeIds.length) {
-                activeToolCallIds.delete(toolName);
-            }
-            return toolCallId;
-        },
-    };
-}
-export async function serveAgUiOverHttp(runtime, options = {}) {
-    const hostname = options.hostname?.trim() || "127.0.0.1";
-    const port = typeof options.port === "number" && Number.isFinite(options.port) ? options.port : 0;
-    const runPath = normalizePath(options.runPath, "/run");
-    const httpServer = createServer(async (request, response) => {
-        const url = new URL(request.url ?? "/", `http://${hostname}`);
-        if (request.method !== "POST" || url.pathname !== runPath) {
-            response.statusCode = 404;
-            response.setHeader("content-type", "application/json; charset=utf-8");
-            response.end(JSON.stringify({ error: "Not Found", runPath }));
-            return;
-        }
-        response.statusCode = 200;
-        response.setHeader("content-type", "text/event-stream; charset=utf-8");
-        response.setHeader("cache-control", "no-cache, no-transform");
-        response.setHeader("connection", "keep-alive");
-        let requestId;
-        let sessionId;
-        let messageId;
-        let thinkingMessageId;
-        let textMessageStarted = false;
-        let thinkingMessageStarted = false;
-        const toolCalls = createToolCallState();
-        const upstreamReducer = createUpstreamTimelineReducer();
-        const ensureTextStart = async () => {
-            if (textMessageStarted) {
-                return;
-            }
-            messageId = messageId ?? `msg-${createPersistentId()}`;
-            textMessageStarted = true;
-            await writeSseEvent(response, {
-                type: "TEXT_MESSAGE_START",
-                timestamp: createTimestamp(),
-                messageId,
-                role: "assistant",
-            });
-        };
-        const ensureThinkingTextStart = async () => {
-            if (thinkingMessageStarted) {
-                return;
-            }
-            thinkingMessageId = thinkingMessageId ?? `thinking-${createPersistentId()}`;
-            thinkingMessageStarted = true;
-            await writeSseEvent(response, {
-                type: "THINKING_TEXT_MESSAGE_START",
-                timestamp: createTimestamp(),
-                messageId: thinkingMessageId,
-                role: "assistant",
-            });
-        };
-        const closeOpenMessages = async () => {
-            if (thinkingMessageStarted) {
-                await writeSseEvent(response, {
-                    type: "THINKING_TEXT_MESSAGE_END",
-                    timestamp: createTimestamp(),
-                    messageId: thinkingMessageId,
-                });
-                thinkingMessageStarted = false;
-            }
-            if (textMessageStarted) {
-                await writeSseEvent(response, {
-                    type: "TEXT_MESSAGE_END",
-                    timestamp: createTimestamp(),
-                    messageId: messageId,
-                });
-                textMessageStarted = false;
-            }
-        };
-        try {
-            const body = await readRequestBody(request);
-            const input = parseRequestInput(JSON.parse(body));
-            const result = await runtime.run({
-                agentId: input.agentId,
-                input: input.input,
-                ...(input.sessionId ? { sessionId: input.sessionId } : {}),
-                ...(input.invocation ? { invocation: input.invocation } : {}),
-                listeners: {
-                    onEvent: async (event) => {
-                        if (event.eventType === "request.created") {
-                            requestId = event.requestId;
-                            sessionId = event.sessionId;
-                            await writeSseEvent(response, toRunStarted(event, input.input));
-                            return;
-                        }
-                        if (event.eventType === "output.delta") {
-                            requestId = requestId ?? event.requestId;
-                            sessionId = sessionId ?? event.sessionId;
-                            const delta = typeof event.payload.content === "string" ? event.payload.content : "";
-                            if (!delta) {
-                                return;
-                            }
-                            await ensureTextStart();
-                            await writeSseEvent(response, {
-                                type: "TEXT_MESSAGE_CONTENT",
-                                timestamp: createTimestamp(),
-                                messageId: messageId,
-                                delta,
-                            });
-                        }
-                    },
-                    dataListener: async (event) => {
-                        if (event.type !== "progress.commentary") {
-                            return;
-                        }
-                        await writeSseEvent(response, {
-                            type: "STATUS_UPDATE",
-                            timestamp: createTimestamp(),
-                            message: event.text,
-                        });
-                    },
-                    onUpstreamEvent: async (event) => {
-                        const eventName = readUpstreamEventName(event);
-                        const toolName = readUpstreamToolName(event);
-                        if (isToolStartEvent(event) && toolName) {
-                            const toolCallId = toolCalls.start(toolName);
-                            await writeSseEvent(response, {
-                                type: "TOOL_CALL_START",
-                                timestamp: createTimestamp(),
-                                toolCallId,
-                                toolName,
-                            });
-                            const args = readToolArgs(event);
-                            if (args !== undefined) {
-                                await writeSseEvent(response, {
-                                    type: "TOOL_CALL_ARGS",
-                                    timestamp: createTimestamp(),
-                                    toolCallId,
-                                    toolName,
-                                    args,
-                                });
-                            }
-                        }
-                        const projections = upstreamReducer.consume(event);
-                        for (const projection of projections) {
-                            if (projection.type === "thinking") {
-                                if (!projection.text) {
-                                    continue;
-                                }
-                                await ensureThinkingTextStart();
-                                await writeSseEvent(response, {
-                                    type: "THINKING_TEXT_MESSAGE_CONTENT",
-                                    timestamp: createTimestamp(),
-                                    messageId: thinkingMessageId,
-                                    delta: projection.text,
-                                });
-                                continue;
-                            }
-                            if (projection.type === "step") {
-                                if (projection.status === "started") {
-                                    await writeSseEvent(response, {
-                                        type: "STEP_STARTED",
-                                        timestamp: createTimestamp(),
-                                        stepId: projection.key,
-                                        title: projection.step,
-                                        category: projection.category,
-                                    });
-                                    continue;
-                                }
-                                await writeSseEvent(response, {
-                                    type: "STEP_FINISHED",
-                                    timestamp: createTimestamp(),
-                                    stepId: projection.key,
-                                    title: projection.step,
-                                    category: projection.category,
-                                    status: projection.status,
-                                });
-                                continue;
-                            }
-                            const toolCallId = toolCalls.peek(projection.toolName) ?? toolCalls.start(projection.toolName);
-                            await writeSseEvent(response, {
-                                type: "TOOL_CALL_RESULT",
-                                timestamp: createTimestamp(),
-                                toolCallId,
-                                toolName: projection.toolName,
-                                result: projection.output,
-                                ...(projection.isError !== undefined ? { isError: projection.isError } : {}),
-                            });
-                        }
-                        if (isToolTerminalEvent(event) && toolName) {
-                            const toolCallId = toolCalls.finish(toolName) ?? `tool-${createPersistentId()}`;
-                            await writeSseEvent(response, {
-                                type: "TOOL_CALL_END",
-                                timestamp: createTimestamp(),
-                                toolCallId,
-                                toolName,
-                            });
-                        }
-                        if (eventName === "on_chat_model_end" && thinkingMessageStarted) {
-                            await writeSseEvent(response, {
-                                type: "THINKING_TEXT_MESSAGE_END",
-                                timestamp: createTimestamp(),
-                                messageId: thinkingMessageId,
-                            });
-                            thinkingMessageStarted = false;
-                        }
-                    },
-                },
-            });
-            requestId = requestId ?? result.requestId;
-            sessionId = sessionId ?? result.sessionId;
-            if (!textMessageStarted && result.output) {
-                await ensureTextStart();
-                await writeSseEvent(response, {
-                    type: "TEXT_MESSAGE_CONTENT",
-                    timestamp: createTimestamp(),
-                    messageId: messageId,
-                    delta: result.output,
-                });
-            }
-            await closeOpenMessages();
-            await writeSseEvent(response, {
-                type: "RUN_FINISHED",
-                timestamp: createTimestamp(),
-                sessionId: sessionId,
-                requestId: requestId,
-                state: result.state,
-                ...(result.approvalId ? { approvalId: result.approvalId } : {}),
-                ...(result.pendingActionId ? { pendingActionId: result.pendingActionId } : {}),
-            });
-        }
-        catch (error) {
-            await closeOpenMessages();
-            await writeSseEvent(response, {
-                type: "RUN_ERROR",
-                timestamp: createTimestamp(),
-                ...(sessionId ? { sessionId } : {}),
-                ...(requestId ? { requestId } : {}),
-                message: error instanceof Error ? error.message : "AG-UI request failed.",
-            });
-        }
-        finally {
-            response.end();
-        }
-    });
-    const completed = new Promise((resolve, reject) => {
-        httpServer.once("close", resolve);
-        httpServer.once("error", reject);
-    });
-    await new Promise((resolve, reject) => {
-        httpServer.listen(port, hostname, () => resolve());
-        httpServer.once("error", reject);
-    });
-    const address = httpServer.address();
-    const resolvedPort = typeof address === "object" && address ? address.port : port;
-    return {
-        hostname,
-        port: resolvedPort,
-        runPath,
-        runUrl: `http://${hostname}:${resolvedPort}${runPath}`,
-        completed,
-        close: async () => {
-            await new Promise((resolve, reject) => {
-                httpServer.close((error) => {
-                    if (error) {
-                        reject(error);
-                        return;
-                    }
-                    resolve();
-                });
-            });
-        },
-    };
-}
+import{createServer as D}from"node:http";import{createPersistentId as R}from"../../utils/id.js";import{createUpstreamTimelineReducer as $}from"../../projections/upstream-events.js";function j(t,e){const a=typeof t=="string"&&t.trim().length>0?t.trim():e;return a.startsWith("/")?a:`/${a}`}function k(t){return new Promise((e,a)=>{const c=[];t.on("data",u=>{c.push(Buffer.isBuffer(u)?u:Buffer.from(u))}),t.on("end",()=>e(Buffer.concat(c).toString("utf8"))),t.on("error",a)})}function o(t,e){return new Promise((a,c)=>{t.write(`data: ${JSON.stringify(e)}
+
+`,u=>{if(u){c(u);return}a()})})}function x(t){if(!t||typeof t!="object"||Array.isArray(t))throw new Error("AG-UI request input must be an object.");const e=t;if(typeof e.input!="string"&&!Array.isArray(e.input))throw new Error("AG-UI request input requires `input`.");return{input:e.input,...typeof e.agentId=="string"&&e.agentId.trim().length>0?{agentId:e.agentId.trim()}:{},...typeof e.sessionId=="string"&&e.sessionId.trim().length>0?{sessionId:e.sessionId.trim()}:{},...e.invocation&&typeof e.invocation=="object"&&!Array.isArray(e.invocation)?{invocation:e.invocation}:{}}}function s(){return new Date().toISOString()}function X(t,e){return{type:"RUN_STARTED",timestamp:s(),sessionId:t.sessionId,requestId:t.requestId,input:e}}function f(t){return typeof t=="object"&&t!==null?t:null}function U(t){return typeof f(t)?.event=="string"?String(f(t)?.event):""}function B(t){return typeof f(t)?.name=="string"?String(f(t)?.name):""}function M(t){return typeof f(t)?.run_type=="string"?String(f(t)?.run_type):""}function F(t){const e=U(t);return e==="on_tool_start"||e==="on_chain_start"&&M(t)==="tool"}function K(t){const e=U(t);return e==="on_tool_end"||e==="on_tool_error"||(e==="on_chain_end"||e==="on_chain_error")&&M(t)==="tool"}function J(t){const e=f(t),a=f(e?.data);if(!(!a||!("input"in a)))return a.input}function z(){const t=new Map;return{start(e){const a=`tool-${R()}`,c=t.get(e)??[];return c.push(a),t.set(e,c),a},peek(e){return t.get(e)?.at(-1)},finish(e){const a=t.get(e);if(!a||a.length===0)return;const c=a.pop();return a.length||t.delete(e),c}}}async function Y(t,e={}){const a=e.hostname?.trim()||"127.0.0.1",c=typeof e.port=="number"&&Number.isFinite(e.port)?e.port:0,u=j(e.runPath,"/run"),T=D(async(d,n)=>{const w=new URL(d.url??"/",`http://${a}`);if(d.method!=="POST"||w.pathname!==u){n.statusCode=404,n.setHeader("content-type","application/json; charset=utf-8"),n.end(JSON.stringify({error:"Not Found",runPath:u}));return}n.statusCode=200,n.setHeader("content-type","text/event-stream; charset=utf-8"),n.setHeader("cache-control","no-cache, no-transform"),n.setHeader("connection","keep-alive");let p,l,g,E,S=!1,_=!1;const N=z(),H=$(),L=async()=>{S||(g=g??`msg-${R()}`,S=!0,await o(n,{type:"TEXT_MESSAGE_START",timestamp:s(),messageId:g,role:"assistant"}))},b=async()=>{_||(E=E??`thinking-${R()}`,_=!0,await o(n,{type:"THINKING_TEXT_MESSAGE_START",timestamp:s(),messageId:E,role:"assistant"}))},P=async()=>{_&&(await o(n,{type:"THINKING_TEXT_MESSAGE_END",timestamp:s(),messageId:E}),_=!1),S&&(await o(n,{type:"TEXT_MESSAGE_END",timestamp:s(),messageId:g}),S=!1)};try{const h=await k(d),I=x(JSON.parse(h)),m=await t.run({agentId:I.agentId,input:I.input,...I.sessionId?{sessionId:I.sessionId}:{},...I.invocation?{invocation:I.invocation}:{},listeners:{onEvent:async r=>{if(r.eventType==="request.created"){p=r.requestId,l=r.sessionId,await o(n,X(r,I.input));return}if(r.eventType==="output.delta"){p=p??r.requestId,l=l??r.sessionId;const A=typeof r.payload.content=="string"?r.payload.content:"";if(!A)return;await L(),await o(n,{type:"TEXT_MESSAGE_CONTENT",timestamp:s(),messageId:g,delta:A})}},dataListener:async r=>{r.type==="progress.commentary"&&await o(n,{type:"STATUS_UPDATE",timestamp:s(),message:r.text})},onUpstreamEvent:async r=>{const A=U(r),y=B(r);if(F(r)&&y){const i=N.start(y);await o(n,{type:"TOOL_CALL_START",timestamp:s(),toolCallId:i,toolName:y});const O=J(r);O!==void 0&&await o(n,{type:"TOOL_CALL_ARGS",timestamp:s(),toolCallId:i,toolName:y,args:O})}const q=H.consume(r);for(const i of q){if(i.type==="thinking"){if(!i.text)continue;await b(),await o(n,{type:"THINKING_TEXT_MESSAGE_CONTENT",timestamp:s(),messageId:E,delta:i.text});continue}if(i.type==="step"){if(i.status==="started"){await o(n,{type:"STEP_STARTED",timestamp:s(),stepId:i.key,title:i.step,category:i.category});continue}await o(n,{type:"STEP_FINISHED",timestamp:s(),stepId:i.key,title:i.step,category:i.category,status:i.status});continue}const O=N.peek(i.toolName)??N.start(i.toolName);await o(n,{type:"TOOL_CALL_RESULT",timestamp:s(),toolCallId:O,toolName:i.toolName,result:i.output,...i.isError!==void 0?{isError:i.isError}:{}})}if(K(r)&&y){const i=N.finish(y)??`tool-${R()}`;await o(n,{type:"TOOL_CALL_END",timestamp:s(),toolCallId:i,toolName:y})}A==="on_chat_model_end"&&_&&(await o(n,{type:"THINKING_TEXT_MESSAGE_END",timestamp:s(),messageId:E}),_=!1)}}});p=p??m.requestId,l=l??m.sessionId,!S&&m.output&&(await L(),await o(n,{type:"TEXT_MESSAGE_CONTENT",timestamp:s(),messageId:g,delta:m.output})),await P(),await o(n,{type:"RUN_FINISHED",timestamp:s(),sessionId:l,requestId:p,state:m.state,...m.approvalId?{approvalId:m.approvalId}:{},...m.pendingActionId?{pendingActionId:m.pendingActionId}:{}})}catch(h){await P(),await o(n,{type:"RUN_ERROR",timestamp:s(),...l?{sessionId:l}:{},...p?{requestId:p}:{},message:h instanceof Error?h.message:"AG-UI request failed."})}finally{n.end()}}),v=new Promise((d,n)=>{T.once("close",d),T.once("error",n)});await new Promise((d,n)=>{T.listen(c,a,()=>d()),T.once("error",n)});const C=T.address(),G=typeof C=="object"&&C?C.port:c;return{hostname:a,port:G,runPath:u,runUrl:`http://${a}:${G}${u}`,completed:v,close:async()=>{await new Promise((d,n)=>{T.close(w=>{if(w){n(w);return}d()})})}}}export{Y as serveAgUiOverHttp};