@cloudbase/agent-server 0.0.11 → 0.0.12

package/CHANGELOG.md

@@ -1,5 +1,17 @@
 # @cloudbase/agent-server
 
+## 1.0.1-alpha.6
+
+### Patch Changes
+
+- alpha release 0.1.2-alpha.1
+  - Update all public packages to version 0.1.2-alpha.1
+  - Trigger automated alpha release workflow
+  - Includes latest features and improvements
+
+- Updated dependencies
+  - @cloudbase/agent-shared@1.0.1-alpha.6
+
 ## 1.0.1-alpha.5
 
 ### Patch Changes

package/dist/index.js

@@ -231,15 +231,68 @@ function handler2(input, agent) {
 // src/agui/sendMessageAGUI/server.ts
 var import_server3 = require("@whatwg-node/server");
 var import_client2 = require("@ag-ui/client");
+var import_uuid2 = require("uuid");
 function createServerAdapter2(createAgent) {
   return (0, import_server3.createServerAdapter)(async (request) => {
-    const input = import_client2.RunAgentInputSchema.parse(await request.json());
-    const { agent: unknownAgent, cleanup } = await Promise.resolve(
-      createAgent({ request })
-    );
-    const agent = "toAGUIAgent" in unknownAgent ? unknownAgent.toAGUIAgent() : unknownAgent;
-    const events = handler2(input, agent);
+    const inputRes = await safeAsync(async () => {
+      const rawInput = await request.json();
+      const inputWithDefaults = {
+        tools: [],
+        context: [],
+        state: {},
+        forwardedProps: {},
+        ...rawInput,
+        runId: typeof rawInput.runId === "string" && rawInput.runId ? rawInput.runId : (0, import_uuid2.v4)()
+      };
+      return import_client2.RunAgentInputSchema.parse(inputWithDefaults);
+    });
+    if ("error" in inputRes) {
+      const { error } = inputRes;
+      console.error("[AGUI Server] Pre-stream error:", error);
+      return new Response(
+        JSON.stringify({
+          error: error instanceof Error ? error.message : String(error)
+        }),
+        {
+          status: 400,
+          headers: { "Content-Type": "application/json" }
+        }
+      );
+    }
+    const createAgentRes = await safeAsync(async () => {
+      const res = await Promise.resolve(createAgent({ request }));
+      return {
+        cleanup: res.cleanup,
+        agent: "toAGUIAgent" in res.agent ? res.agent.toAGUIAgent() : res.agent
+      };
+    });
+    if ("error" in createAgentRes) {
+      const { error } = createAgentRes;
+      console.error("[AGUI Server] Pre-stream error:", error);
+      return new Response(
+        JSON.stringify({
+          error: error instanceof Error ? error.message : String(error)
+        }),
+        {
+          status: 500,
+          headers: { "Content-Type": "application/json" }
+        }
+      );
+    }
+    const events = handler2(inputRes.result, createAgentRes.result.agent);
     let heartbeat;
+    let cleanupCalled = false;
+    const safeCleanup = () => {
+      var _a, _b;
+      if (!cleanupCalled) {
+        cleanupCalled = true;
+        try {
+          (_b = (_a = createAgentRes.result).cleanup) == null ? void 0 : _b.call(_a);
+        } catch (e) {
+          console.error(e);
+        }
+      }
+    };
     const stream = new ReadableStream({
       async start(controller) {
         const encoder = new TextEncoder();
@@ -253,14 +306,26 @@ function createServerAdapter2(createAgent) {
 `;
             controller.enqueue(encoder.encode(sseChunk));
           }
+        } catch (error) {
+          console.error("[AGUI Server] Stream error:", error);
+          const errorEvent = {
+            type: import_client2.EventType.RUN_ERROR,
+            message: error instanceof Error ? error.message : String(error)
+          };
+          controller.enqueue(
+            encoder.encode(`data: ${JSON.stringify(errorEvent)}
+
+`)
+          );
         } finally {
           if (heartbeat) clearInterval(heartbeat);
           controller.close();
-          cleanup == null ? void 0 : cleanup();
+          safeCleanup();
         }
       },
       cancel() {
         if (heartbeat) clearInterval(heartbeat);
+        safeCleanup();
       }
     });
     const headers = new Headers({
@@ -271,6 +336,15 @@ function createServerAdapter2(createAgent) {
     return new Response(stream, { status: 200, headers });
   });
 }
+async function safeAsync(fn) {
+  try {
+    return {
+      result: await fn()
+    };
+  } catch (error) {
+    return { error };
+  }
+}
 
 // src/agui/healthz/index.ts
 var healthz_exports = {};
@@ -288,7 +362,7 @@ __export(openai_exports, {
 });
 
 // src/agui/openai/handler.ts
-var import_uuid2 = require("uuid");
+var import_uuid3 = require("uuid");
 var import_repeater3 = require("@repeaterjs/repeater");
 function handler3(input, agent) {
   var _a;
@@ -300,12 +374,12 @@ function handler3(input, agent) {
         description: tool.function.description,
         parameters: tool.function.parameters
       })),
-      conversationId: (0, import_uuid2.v4)()
+      conversationId: (0, import_uuid3.v4)()
     },
     agent
   );
   return new import_repeater3.Repeater(async (push, stop) => {
-    const id = (0, import_uuid2.v4)();
+    const id = (0, import_uuid3.v4)();
     let tools = [];
     let lastWithToolCall = false;
     let maxIndex = 0;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cloudbase/agent-server",
-  "version": "0.0.11",
+  "version": "0.0.12",
   "main": "dist/index.js",
   "files": [
     "dist/",
@@ -21,7 +21,7 @@
     "openai": "6.3.0",
     "uuid": "^10.0.0",
     "zod": "^4.1.12",
-    "@cloudbase/agent-shared": "^0.0.11"
+    "@cloudbase/agent-shared": "^0.0.12"
   },
   "devDependencies": {
     "@types/cors": "^2.8.19",