@cloudbase/agent-server 1.0.1-alpha.8 → 1.0.1-alpha.9

package/CHANGELOG.md

@@ -1,5 +1,29 @@
 # @cloudbase/agent-server
 
+## 1.0.1-alpha.8
+
+### 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.8
+
+## 1.0.1-alpha.7
+
+### 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.7
+
 ## 1.0.1-alpha.6
 
 ### Patch Changes

package/dist/index.d.ts

@@ -1,3 +1,4 @@
+import * as _cloudbase_agent_observability_server from '@cloudbase/agent-observability/server';
 import { CreateCopilotRuntimeServerOptions } from '@copilotkit/runtime';
 import { CopilotRuntimeOptions } from '@copilotkit/runtime/v2';
 import expressLib, { Express } from 'express';
@@ -5,7 +6,7 @@ import * as _ag_ui_client from '@ag-ui/client';
 import { AbstractAgent, RunAgentInput } from '@ag-ui/client';
 import cors from 'cors';
 import { Logger, SendMessageInput } from '@cloudbase/agent-shared';
-export { LogFn, Logger, createConsoleLogger, noopLogger } from '@cloudbase/agent-shared';
+export { LogFn, Logger, createConsoleLogger, isErrorWithCode, noopLogger } from '@cloudbase/agent-shared';
 import { Repeater } from '@repeaterjs/repeater';
 import * as _whatwg_node_server from '@whatwg-node/server';
 import { OpenAI } from 'openai';
@@ -53,16 +54,47 @@ interface ICreateServer {
      * createExpressServer({ createAgent, logger: pino({ level: 'info' }) });
      */
     logger?: Logger;
+    /**
+     * Observability configuration for trace exporters.
+     *
+     * Requires @cloudbase/agent-observability package to be installed.
+     * If the package is not installed, this option is silently ignored.
+     *
+     * @example
+     * // Console exporter (from env AUTO_TRACES_STDOUT)
+     * createExpressServer({ createAgent, observability: { type: 'console' } });
+     *
+     * // OTLP exporter (Langfuse, Jaeger, etc.)
+     * createExpressServer({
+     *   createAgent,
+     *   observability: {
+     *     type: 'otlp',
+     *     url: 'https://cloud.langfuse.com/api/public/otlp/v1/traces',
+     *     headers: { 'Authorization': 'Basic xxx' }
+     *   }
+     * });
+     *
+     * // Multiple exporters
+     * createExpressServer({
+     *   createAgent,
+     *   observability: [
+     *     { type: 'console' },
+     *     { type: 'otlp', url: 'http://localhost:4318/v1/traces' }
+     *   ]
+     * });
+     */
+    observability?: _cloudbase_agent_observability_server.ObservabilityConfig | _cloudbase_agent_observability_server.ObservabilityConfig[];
 }
 interface IRun extends ICreateServer {
     port?: number | string;
 }
 interface ICreateExpressRoutes extends Omit<ICreateServer, "cors"> {
     express: Express;
+    observability?: ICreateServer['observability'];
 }
 declare function run(props: IRun): void;
 declare function createExpressServer(props: ICreateServer): Express;
-declare function createExpressRoutes({ createAgent, basePath: _basePath, express, useAGUI: _useAGUI, aguiOptions, logger: _logger, }: ICreateExpressRoutes): expressLib.Express;
+declare function createExpressRoutes({ createAgent, basePath: _basePath, express, useAGUI: _useAGUI, aguiOptions, logger: _logger, observability, }: ICreateExpressRoutes): expressLib.Express;
 interface AGUIOptions {
     runtimeOptions?: Partial<CopilotRuntimeOptions>;
     endpointOptions?: Partial<CreateCopilotRuntimeServerOptions>;
@@ -188,4 +220,24 @@ declare function extractRequestId(headers: Headers | Record<string, string | str
  */
 declare function getOrGenerateRequestId(headers: Headers | Record<string, string | string[] | undefined>, prefix?: string): string;
 
-export { type AgentCreator, type AgentCreatorContext, index as agui, createExpressRoutes, createExpressServer, extractRequestId, generateRequestId, getOrGenerateRequestId, run };
+/**
+ * AG-Kit Error Handling
+ *
+ * Error handling follows these rules:
+ * 1. Pre-stream errors: Return JSON `{error: {code, message}, requestId}`
+ * 2. In-stream errors: Emit RunError event `{type: "RUN_ERROR", code, message}`
+ *
+ * Adapters can:
+ * - Directly emit RunError events into the stream
+ * - Throw errors with a `code` property (server extracts and converts to RunError)
+ * - Throw other errors (server uses fallback INTERNAL_ERROR code)
+ */
+declare const ErrorCode: {
+    /** Invalid request format or parameters (400) */
+    readonly INVALID_REQUEST: "INVALID_REQUEST";
+    /** Internal server error (500) */
+    readonly INTERNAL_ERROR: "INTERNAL_ERROR";
+};
+type ErrorCodeType = (typeof ErrorCode)[keyof typeof ErrorCode];
+
+export { type AgentCreator, type AgentCreatorContext, ErrorCode, type ErrorCodeType, index as agui, createExpressRoutes, createExpressServer, extractRequestId, generateRequestId, getOrGenerateRequestId, run };

package/dist/index.js

@@ -29,6 +29,7 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
+  ErrorCode: () => ErrorCode,
   agui: () => agui_exports,
   createConsoleLogger: () => import_agent_shared2.createConsoleLogger,
   createExpressRoutes: () => createExpressRoutes,
@@ -36,6 +37,7 @@ __export(index_exports, {
   extractRequestId: () => extractRequestId,
   generateRequestId: () => generateRequestId,
   getOrGenerateRequestId: () => getOrGenerateRequestId,
+  isErrorWithCode: () => import_agent_shared3.isErrorWithCode,
   noopLogger: () => import_agent_shared2.noopLogger,
   run: () => run
 });
@@ -266,13 +268,166 @@ function getOrGenerateRequestId(headers, prefix = "req") {
   return extractRequestId(headers) || generateRequestId(prefix);
 }
 
+// src/agui/sendMessageAGUI/schemas.ts
+var import_v4 = require("zod/v4");
+var FunctionCallSchema = import_v4.z.object({
+  name: import_v4.z.string(),
+  arguments: import_v4.z.string()
+});
+var ToolCallSchema = import_v4.z.object({
+  id: import_v4.z.string(),
+  type: import_v4.z.literal("function"),
+  function: FunctionCallSchema
+});
+var ToolSchema = import_v4.z.object({
+  name: import_v4.z.string(),
+  description: import_v4.z.string(),
+  parameters: import_v4.z.any()
+});
+var TextInputContentSchema = import_v4.z.object({
+  type: import_v4.z.literal("text"),
+  text: import_v4.z.string()
+});
+var BinaryInputContentObjectSchema = import_v4.z.object({
+  type: import_v4.z.literal("binary"),
+  mimeType: import_v4.z.string(),
+  id: import_v4.z.string().optional(),
+  url: import_v4.z.string().optional(),
+  data: import_v4.z.string().optional(),
+  filename: import_v4.z.string().optional()
+});
+var ensureBinaryPayload = (value, ctx) => {
+  if (!value.id && !value.url && !value.data) {
+    ctx.addIssue({
+      code: import_v4.z.ZodIssueCode.custom,
+      message: "BinaryInputContent requires at least one of id, url, or data.",
+      path: ["id"]
+    });
+  }
+};
+var BinaryInputContentSchema = BinaryInputContentObjectSchema.superRefine((value, ctx) => {
+  ensureBinaryPayload(value, ctx);
+});
+var InputContentBaseSchema = import_v4.z.discriminatedUnion("type", [
+  TextInputContentSchema,
+  BinaryInputContentObjectSchema
+]);
+var InputContentSchema = InputContentBaseSchema.superRefine(
+  (value, ctx) => {
+    if (value.type === "binary") {
+      ensureBinaryPayload(value, ctx);
+    }
+  }
+);
+var BaseMessageSchema = import_v4.z.object({
+  id: import_v4.z.string(),
+  role: import_v4.z.string(),
+  content: import_v4.z.string().optional(),
+  name: import_v4.z.string().optional()
+});
+var DeveloperMessageSchema = BaseMessageSchema.extend({
+  role: import_v4.z.literal("developer"),
+  content: import_v4.z.string()
+});
+var SystemMessageSchema = BaseMessageSchema.extend({
+  role: import_v4.z.literal("system"),
+  content: import_v4.z.string()
+});
+var AssistantMessageSchema = BaseMessageSchema.extend({
+  role: import_v4.z.literal("assistant"),
+  content: import_v4.z.string().optional(),
+  toolCalls: import_v4.z.array(ToolCallSchema).optional()
+});
+var UserMessageSchema = BaseMessageSchema.extend({
+  role: import_v4.z.literal("user"),
+  content: import_v4.z.union([import_v4.z.string(), import_v4.z.array(InputContentSchema)])
+});
+var ToolMessageSchema = import_v4.z.object({
+  id: import_v4.z.string(),
+  content: import_v4.z.string(),
+  role: import_v4.z.literal("tool"),
+  toolCallId: import_v4.z.string(),
+  error: import_v4.z.string().optional()
+});
+var ActivityMessageSchema = import_v4.z.object({
+  id: import_v4.z.string(),
+  role: import_v4.z.literal("activity"),
+  activityType: import_v4.z.string(),
+  content: import_v4.z.record(import_v4.z.string(), import_v4.z.any())
+});
+var MessageSchema = import_v4.z.discriminatedUnion("role", [
+  DeveloperMessageSchema,
+  SystemMessageSchema,
+  AssistantMessageSchema,
+  UserMessageSchema,
+  ToolMessageSchema,
+  ActivityMessageSchema
+]);
+var ContextSchema = import_v4.z.object({
+  description: import_v4.z.string(),
+  value: import_v4.z.string()
+});
+var ServerRunAgentInputSchema = import_v4.z.object({
+  threadId: import_v4.z.string().optional(),
+  // Modified: optional instead of required
+  runId: import_v4.z.string(),
+  parentRunId: import_v4.z.string().optional(),
+  state: import_v4.z.any(),
+  messages: import_v4.z.array(MessageSchema),
+  tools: import_v4.z.array(ToolSchema),
+  context: import_v4.z.array(ContextSchema),
+  forwardedProps: import_v4.z.any()
+});
+
+// src/errors/index.ts
+var import_agent_shared3 = require("@cloudbase/agent-shared");
+var ErrorCode = {
+  /** Invalid request format or parameters (400) */
+  INVALID_REQUEST: "INVALID_REQUEST",
+  /** Internal server error (500) */
+  INTERNAL_ERROR: "INTERNAL_ERROR"
+};
+
 // src/agui/sendMessageAGUI/server.ts
+var startObservation;
+var setupObservability;
+var observabilityLoadAttempted = false;
+async function loadObservability() {
+  if (!observabilityLoadAttempted) {
+    observabilityLoadAttempted = true;
+    try {
+      const obs = await import("@cloudbase/agent-observability");
+      const obsServer = await import("@cloudbase/agent-observability/server");
+      startObservation = obs.startObservation;
+      setupObservability = obsServer.setupObservability;
+      return true;
+    } catch (e) {
+      return false;
+    }
+  }
+  return !!startObservation;
+}
+async function ensureObservabilityReady() {
+  if (!setupObservability) return false;
+  try {
+    const timeoutPromise = new Promise((_, reject) => {
+      setTimeout(() => reject(new Error("Observability setup timeout")), 2e3);
+    });
+    await Promise.race([
+      setupObservability(),
+      timeoutPromise
+    ]);
+    return true;
+  } catch (e) {
+    return false;
+  }
+}
 function createServerAdapter2(createAgent, options) {
   var _a;
   const { logger: parentLogger = import_agent_shared2.noopLogger } = options ?? {};
   const adapterLogger = ((_a = parentLogger.child) == null ? void 0 : _a.call(parentLogger, { component: "sendMessageAGUI" })) ?? parentLogger;
   return (0, import_server3.createServerAdapter)(async (request) => {
-    var _a2, _b, _c, _d, _e, _f, _g, _h;
+    var _a2, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l;
     const requestId = getOrGenerateRequestId(request.headers, "agui");
     const logger = ((_a2 = adapterLogger.child) == null ? void 0 : _a2.call(adapterLogger, { requestId })) ?? adapterLogger;
     (_b = logger.info) == null ? void 0 : _b.call(logger, "Request received");
@@ -286,7 +441,7 @@ function createServerAdapter2(createAgent, options) {
         ...rawInput,
         runId: typeof rawInput.runId === "string" && rawInput.runId ? rawInput.runId : (0, import_uuid3.v4)()
       };
-      return import_client2.RunAgentInputSchema.parse(inputWithDefaults);
+      return ServerRunAgentInputSchema.parse(inputWithDefaults);
     });
     if ("error" in inputRes) {
       const { error } = inputRes;
@@ -297,7 +452,10 @@ function createServerAdapter2(createAgent, options) {
       );
       return new Response(
         JSON.stringify({
-          error: error instanceof Error ? error.message : String(error),
+          error: {
+            code: ErrorCode.INVALID_REQUEST,
+            message: error instanceof Error ? error.message : String(error)
+          },
           requestId
         }),
         {
@@ -333,9 +491,14 @@ function createServerAdapter2(createAgent, options) {
     if ("error" in createAgentRes) {
       const { error } = createAgentRes;
       (_h = logger.error) == null ? void 0 : _h.call(logger, { err: error }, "Agent creation failed");
+      const errorCode = (0, import_agent_shared3.isErrorWithCode)(error) ? error.code : ErrorCode.INTERNAL_ERROR;
+      const errorMessage = error instanceof Error ? error.message : String(error);
       return new Response(
         JSON.stringify({
-          error: error instanceof Error ? error.message : String(error),
+          error: {
+            code: errorCode,
+            message: errorMessage
+          },
           requestId
         }),
         {
@@ -344,8 +507,46 @@ function createServerAdapter2(createAgent, options) {
         }
       );
     }
-    const events = handler2(inputRes.result, createAgentRes.result.agent);
-    let heartbeat;
+    const hasObservability = await loadObservability();
+    let serverSpan = null;
+    let serverContextData = null;
+    if (hasObservability && startObservation) {
+      try {
+        const isReady = await ensureObservabilityReady();
+        if (isReady) {
+          serverSpan = startObservation(
+            "AG-UI.Server",
+            {
+              "http.method": request.method,
+              "http.url": request.url,
+              "http.host": request.headers.get("host") || "unknown",
+              "http.user_agent": request.headers.get("user-agent") || "unknown",
+              "agui.thread_id": inputRes.result.threadId,
+              "agui.run_id": inputRes.result.runId
+            },
+            { asType: "span" }
+          );
+          const spanContext = serverSpan.otelSpan.spanContext();
+          serverContextData = {
+            traceId: spanContext.traceId,
+            spanId: spanContext.spanId,
+            traceFlags: spanContext.traceFlags
+          };
+          inputRes.result.forwardedProps = {
+            ...inputRes.result.forwardedProps,
+            __agui_server_context: serverContextData
+          };
+          (_i = logger.debug) == null ? void 0 : _i.call(logger, "\u2713 Server span created:", {
+            traceId: serverContextData.traceId,
+            spanId: serverContextData.spanId
+          });
+        } else {
+          (_j = logger.debug) == null ? void 0 : _j.call(logger, "Observability not ready, skipping span creation");
+        }
+      } catch (e) {
+        (_k = logger.debug) == null ? void 0 : _k.call(logger, "Failed to create server span:", e);
+      }
+    }
     let cleanupCalled = false;
     const safeCleanup = () => {
       var _a3, _b2, _c2;
@@ -358,9 +559,36 @@ function createServerAdapter2(createAgent, options) {
         }
       }
     };
+    const eventsResult = safe(
+      () => handler2(
+        inputRes.result,
+        createAgentRes.result.agent
+      )
+    );
+    if ("error" in eventsResult) {
+      const { error } = eventsResult;
+      (_l = logger.error) == null ? void 0 : _l.call(logger, { err: error }, "Run agent failed");
+      const errorCode = (0, import_agent_shared3.isErrorWithCode)(error) ? error.code : ErrorCode.INTERNAL_ERROR;
+      const errorMessage = error instanceof Error ? error.message : String(error);
+      return new Response(
+        JSON.stringify({
+          error: {
+            code: errorCode,
+            message: errorMessage
+          },
+          requestId
+        }),
+        {
+          status: 500,
+          headers: { "Content-Type": "application/json" }
+        }
+      );
+    }
+    const { result: events } = eventsResult;
+    let heartbeat;
     const stream = new ReadableStream({
       async start(controller) {
-        var _a3, _b2, _c2, _d2;
+        var _a3, _b2, _c2, _d2, _e2;
         const encoder = new TextEncoder();
         heartbeat = setInterval(() => {
           controller.enqueue(encoder.encode(":ping\n\n"));
@@ -383,9 +611,12 @@ function createServerAdapter2(createAgent, options) {
           (_c2 = logger.info) == null ? void 0 : _c2.call(logger, { eventCount }, "Request completed");
         } catch (error) {
           (_d2 = logger.error) == null ? void 0 : _d2.call(logger, { err: error }, "Stream error");
+          const errorCode = (0, import_agent_shared3.isErrorWithCode)(error) ? error.code : ErrorCode.INTERNAL_ERROR;
+          const errorMessage = error instanceof Error ? error.message : String(error);
           const errorEvent = {
             type: import_client2.EventType.RUN_ERROR,
-            message: error instanceof Error ? error.message : String(error)
+            code: errorCode,
+            message: errorMessage
           };
           controller.enqueue(
             encoder.encode(`data: ${JSON.stringify(errorEvent)}
@@ -396,13 +627,21 @@ function createServerAdapter2(createAgent, options) {
           if (heartbeat) clearInterval(heartbeat);
           controller.close();
           safeCleanup();
+          if (serverSpan) {
+            serverSpan.end();
+            (_e2 = logger.debug) == null ? void 0 : _e2.call(logger, "\u2713 Server span ended");
+          }
         }
       },
       cancel() {
-        var _a3;
+        var _a3, _b2;
         (_a3 = logger.info) == null ? void 0 : _a3.call(logger, "Request cancelled by client");
         if (heartbeat) clearInterval(heartbeat);
         safeCleanup();
+        if (serverSpan) {
+          serverSpan.end();
+          (_b2 = logger.debug) == null ? void 0 : _b2.call(logger, "\u2713 Server span ended (cancelled)");
+        }
       }
     });
     const headers = new Headers({
@@ -422,6 +661,15 @@ async function safeAsync(fn) {
     return { error };
   }
 }
+function safe(fn) {
+  try {
+    return {
+      result: fn()
+    };
+  } catch (error) {
+    return { error };
+  }
+}
 
 // src/agui/healthz/index.ts
 var healthz_exports = {};
@@ -620,6 +868,13 @@ var import_cors = __toESM(require("cors"));
 var import_async_hooks = require("async_hooks");
 var import_server8 = require("@whatwg-node/server");
 var DefaultFetchAPI = __toESM(require("@whatwg-node/fetch"));
+async function setupObservabilityIfAvailable(configs) {
+  try {
+    const { setupObservability: setupObservability2 } = await import("@cloudbase/agent-observability/server");
+    await setupObservability2(configs);
+  } catch (error) {
+  }
+}
 var preparedAgentStorage = new import_async_hooks.AsyncLocalStorage();
 function agentCloneFn() {
   const preparedAgent = preparedAgentStorage.getStore();
@@ -654,7 +909,8 @@ function createExpressRoutes({
   express,
   useAGUI: _useAGUI,
   aguiOptions,
-  logger: _logger
+  logger: _logger,
+  observability
 }) {
   var _a, _b, _c;
   const useAGUI = _useAGUI ?? true;
@@ -662,6 +918,10 @@ function createExpressRoutes({
   const basePath = _basePath ?? (process.env.TENCENTCLOUD_RUNENV === "SCF" ? "/v1/aibot/bots/:agentId/" : "/");
   const serverLogger = ((_a = logger.child) == null ? void 0 : _a.call(logger, { component: "server" })) ?? logger;
   (_b = serverLogger.debug) == null ? void 0 : _b.call(serverLogger, { basePath, useAGUI }, "Initializing server routes");
+  if (observability) {
+    setupObservabilityIfAvailable(observability).catch(() => {
+    });
+  }
   const sendMessageServerAdapter = useAGUI ? sendMessageAGUI_exports.createServerAdapter(createAgent, { logger: serverLogger }) : sendMessage_exports.createServerAdapter(createAgent);
   if (useAGUI) {
     createAGUIRoute({
@@ -755,6 +1015,7 @@ function isCorsOptions(cors2) {
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  ErrorCode,
   agui,
   createConsoleLogger,
   createExpressRoutes,
@@ -762,6 +1023,7 @@ function isCorsOptions(cors2) {
   extractRequestId,
   generateRequestId,
   getOrGenerateRequestId,
+  isErrorWithCode,
   noopLogger,
   run
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cloudbase/agent-server",
-  "version": "1.0.1-alpha.8",
+  "version": "1.0.1-alpha.9",
   "main": "dist/index.js",
   "files": [
     "dist/",
@@ -20,13 +20,19 @@
     "express": "^5.1.0",
     "openai": "6.3.0",
     "uuid": "^10.0.0",
-    "zod": "^4.1.12",
-    "@cloudbase/agent-shared": "^1.0.1-alpha.8"
+    "@cloudbase/agent-shared": "^1.0.1-alpha.9"
+  },
+  "peerDependencies": {
+    "zod": "^3.25.0 || ^4.0.0"
+  },
+  "optionalDependencies": {
+    "@cloudbase/agent-observability": "1.0.1-alpha.9"
   },
   "devDependencies": {
     "@types/cors": "^2.8.19",
     "@types/express": "^5.0.3",
-    "tsup": "^8.5.0"
+    "tsup": "^8.5.0",
+    "zod": "^4.0.0"
   },
   "scripts": {
     "test": "echo \"Error: no test specified\" && exit 1",