@gea-ai/agent-sdk 0.1.261010-alpha.0 → 0.1.261014-alpha.0

package/dist/agent-ai.d.ts

@@ -6,4 +6,5 @@ export declare function createAgentAI(input: {
     modelId: string;
     signal(): AbortSignal;
     recordUsage(entry: AgentContextUsage): void;
+    startUsage?(callId: string, model: string): Promise<void>;
 }): AgentAI;

package/dist/agent-ai.js

@@ -29,9 +29,9 @@ export function createAgentAI(input) {
     };
     // The binding owns the model span. Forward the active SDK parent so generic
     // native calls and Agent-owned calls share the same trace without two model spans.
-    const observe = (options, result) => result.then((value) => {
+    const observe = (options, result, callId) => result.then((value) => {
         input.recordUsage({
-            callId: value.callId,
+            callId: callId ?? value.callId,
             model: value.model,
             label: options.label,
             usage: value.usage,
@@ -40,6 +40,7 @@ export function createAgentAI(input) {
         return value;
     }, (error) => {
         input.recordUsage({
+            callId,
             model: options.model,
             label: options.label,
             usage: null,
@@ -53,20 +54,26 @@ export function createAgentAI(input) {
     return {
         async generateText(options) {
             const parameters = prepare(options);
+            const callId = input.startUsage ? crypto.randomUUID() : undefined;
+            if (callId)
+                await input.startUsage(callId, parameters.model);
             return observe(parameters, (async () => {
                 if (!input.binding.generateText)
                     throw new Error("AI generation binding is unavailable. Update Worker Runtime.");
                 return input.binding.generateText(parameters);
-            })());
+            })(), callId);
         },
         async streamText(options) {
             const parameters = prepare(options);
+            const callId = input.startUsage ? crypto.randomUUID() : undefined;
+            if (callId)
+                await input.startUsage(callId, parameters.model);
             const opened = (async () => {
                 if (!input.binding.streamText)
                     throw new Error("AI generation binding is unavailable. Update Worker Runtime.");
                 return input.binding.streamText(parameters);
             })();
-            const result = observe(parameters, opened.then((call) => call.result));
+            const result = observe(parameters, opened.then((call) => call.result), callId);
             result.catch(() => { });
             const call = await opened;
             return { stream: call.stream, result };

package/dist/agent-call.d.ts

@@ -1,10 +1,37 @@
+import { z } from "zod";
 import { type AgentWorkerToolInitialization } from "@gea-ai/contract/agent-worker-tools";
 import type { AgentHostBinding } from "./agent-worker.js";
 import type { StudioArtifactScope } from "@gea-ai/contract/studio-artifacts";
+/** Fresh Sessions keep the declared-target rule. A resume only needs sessionId. */
+export declare function workerAgentToolInputSchema(input: {
+    agents: NonNullable<AgentWorkerToolInitialization["agents"]>;
+    canCopy?: boolean;
+    artifactScope?: StudioArtifactScope;
+}): z.ZodObject<{
+    sessionId: z.ZodOptional<z.ZodUUID>;
+    target: z.ZodOptional<z.ZodString>;
+    message: z.ZodUnion<readonly [z.ZodString, z.ZodObject<{
+        role: z.ZodLiteral<"user">;
+        parts: z.ZodArray<z.ZodObject<{
+            type: z.ZodLiteral<"text">;
+            text: z.ZodString;
+        }, z.core.$strict>>;
+    }, z.core.$strict>]> | z.ZodUnion<readonly [z.ZodString, z.ZodObject<{
+        role: z.ZodLiteral<"user">;
+        parts: z.ZodArray<z.ZodUnion<readonly [z.ZodObject<{
+            type: z.ZodLiteral<"text">;
+            text: z.ZodString;
+        }, z.core.$strict>, z.ZodObject<{
+            type: z.ZodLiteral<"file">;
+            file_id: z.ZodUUID;
+        }, z.core.$strict>]>>;
+    }, z.core.$strict>]>;
+}, z.core.$strip>;
 /** The host opens a fresh Worker invocation; only its visible answer returns to the caller. */
 export declare function createWorkerAgentTool(input: {
     agents: NonNullable<AgentWorkerToolInitialization["agents"]>;
     host: AgentHostBinding;
+    runId?: string;
     canCopy?: boolean;
     artifactScope?: StudioArtifactScope;
 }): ({
@@ -14,7 +41,7 @@ export declare function createWorkerAgentTool(input: {
     metadata?: import("@ai-sdk/provider").JSONObject;
     inputSchema: import("ai").FlexibleSchema<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -35,7 +62,7 @@ export declare function createWorkerAgentTool(input: {
     contextSchema?: import("ai").FlexibleSchema<import("@ai-sdk/provider-utils").Context> | undefined;
     needsApproval?: boolean | import("@ai-sdk/provider-utils").ToolNeedsApprovalFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -60,7 +87,7 @@ export declare function createWorkerAgentTool(input: {
     onInputAvailable?: ((options: {
         input: {
             sessionId?: string | undefined;
-            target: string | undefined;
+            target?: string | undefined;
             message: string | {
                 role: "user";
                 parts: ({
@@ -83,7 +110,7 @@ export declare function createWorkerAgentTool(input: {
         toolCallId: string;
         input: {
             sessionId?: string | undefined;
-            target: string | undefined;
+            target?: string | undefined;
             message: string | {
                 role: "user";
                 parts: ({
@@ -120,6 +147,16 @@ export declare function createWorkerAgentTool(input: {
         } | {
             error: string;
             message: string;
+        } | {
+            code: "UPSTREAM_HTTP_ERROR";
+            status: number;
+            message: string;
+        } | {
+            code: string;
+            reason: string;
+            status: number;
+            requestId?: string | undefined;
+            message: string;
         }>;
     }) => import("@ai-sdk/provider-utils").ToolResultOutput | PromiseLike<import("@ai-sdk/provider-utils").ToolResultOutput>) | undefined;
 } & {
@@ -142,10 +179,20 @@ export declare function createWorkerAgentTool(input: {
     } | {
         error: string;
         message: string;
+    } | {
+        code: "UPSTREAM_HTTP_ERROR";
+        status: number;
+        message: string;
+    } | {
+        code: string;
+        reason: string;
+        status: number;
+        requestId?: string | undefined;
+        message: string;
     }> | undefined;
     execute: import("ai").ToolExecuteFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -181,6 +228,16 @@ export declare function createWorkerAgentTool(input: {
     } | {
         error: string;
         message: string;
+    } | {
+        code: "UPSTREAM_HTTP_ERROR";
+        status: number;
+        message: string;
+    } | {
+        code: string;
+        reason: string;
+        status: number;
+        requestId?: string | undefined;
+        message: string;
     }, NoInfer<import("@ai-sdk/provider-utils").Context>>;
 } & {
     description?: string | ((options: {
@@ -191,7 +248,7 @@ export declare function createWorkerAgentTool(input: {
     inputExamples?: {
         input: NoInfer<{
             sessionId?: string | undefined;
-            target: string | undefined;
+            target?: string | undefined;
             message: string | {
                 role: "user";
                 parts: ({
@@ -219,7 +276,7 @@ export declare function createWorkerAgentTool(input: {
 } & {
     execute: import("ai").ToolExecuteFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -255,6 +312,16 @@ export declare function createWorkerAgentTool(input: {
     } | {
         error: string;
         message: string;
+    } | {
+        code: "UPSTREAM_HTTP_ERROR";
+        status: number;
+        message: string;
+    } | {
+        code: string;
+        reason: string;
+        status: number;
+        requestId?: string | undefined;
+        message: string;
     }, NoInfer<import("@ai-sdk/provider-utils").Context>>;
 }) | ({
     deferLoading?: boolean;
@@ -263,7 +330,7 @@ export declare function createWorkerAgentTool(input: {
     metadata?: import("@ai-sdk/provider").JSONObject;
     inputSchema: import("ai").FlexibleSchema<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -284,7 +351,7 @@ export declare function createWorkerAgentTool(input: {
     contextSchema?: import("ai").FlexibleSchema<import("@ai-sdk/provider-utils").Context> | undefined;
     needsApproval?: boolean | import("@ai-sdk/provider-utils").ToolNeedsApprovalFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -309,7 +376,7 @@ export declare function createWorkerAgentTool(input: {
     onInputAvailable?: ((options: {
         input: {
             sessionId?: string | undefined;
-            target: string | undefined;
+            target?: string | undefined;
             message: string | {
                 role: "user";
                 parts: ({
@@ -332,7 +399,7 @@ export declare function createWorkerAgentTool(input: {
         toolCallId: string;
         input: {
             sessionId?: string | undefined;
-            target: string | undefined;
+            target?: string | undefined;
             message: string | {
                 role: "user";
                 parts: ({
@@ -369,6 +436,16 @@ export declare function createWorkerAgentTool(input: {
         } | {
             error: string;
             message: string;
+        } | {
+            code: "UPSTREAM_HTTP_ERROR";
+            status: number;
+            message: string;
+        } | {
+            code: string;
+            reason: string;
+            status: number;
+            requestId?: string | undefined;
+            message: string;
         }>;
     }) => import("@ai-sdk/provider-utils").ToolResultOutput | PromiseLike<import("@ai-sdk/provider-utils").ToolResultOutput>) | undefined;
 } & {
@@ -391,10 +468,20 @@ export declare function createWorkerAgentTool(input: {
     } | {
         error: string;
         message: string;
+    } | {
+        code: "UPSTREAM_HTTP_ERROR";
+        status: number;
+        message: string;
+    } | {
+        code: string;
+        reason: string;
+        status: number;
+        requestId?: string | undefined;
+        message: string;
     }> | undefined;
     execute: import("ai").ToolExecuteFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -430,6 +517,16 @@ export declare function createWorkerAgentTool(input: {
     } | {
         error: string;
         message: string;
+    } | {
+        code: "UPSTREAM_HTTP_ERROR";
+        status: number;
+        message: string;
+    } | {
+        code: string;
+        reason: string;
+        status: number;
+        requestId?: string | undefined;
+        message: string;
     }, NoInfer<import("@ai-sdk/provider-utils").Context>>;
 } & {
     description?: string | ((options: {
@@ -440,7 +537,7 @@ export declare function createWorkerAgentTool(input: {
     inputExamples?: {
         input: NoInfer<{
             sessionId?: string | undefined;
-            target: string | undefined;
+            target?: string | undefined;
             message: string | {
                 role: "user";
                 parts: ({
@@ -468,7 +565,7 @@ export declare function createWorkerAgentTool(input: {
 } & {
     execute: import("ai").ToolExecuteFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -504,6 +601,16 @@ export declare function createWorkerAgentTool(input: {
     } | {
         error: string;
         message: string;
+    } | {
+        code: "UPSTREAM_HTTP_ERROR";
+        status: number;
+        message: string;
+    } | {
+        code: string;
+        reason: string;
+        status: number;
+        requestId?: string | undefined;
+        message: string;
     }, NoInfer<import("@ai-sdk/provider-utils").Context>>;
 }) | ({
     deferLoading?: boolean;
@@ -512,7 +619,7 @@ export declare function createWorkerAgentTool(input: {
     metadata?: import("@ai-sdk/provider").JSONObject;
     inputSchema: import("ai").FlexibleSchema<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -533,7 +640,7 @@ export declare function createWorkerAgentTool(input: {
     contextSchema?: import("ai").FlexibleSchema<import("@ai-sdk/provider-utils").Context> | undefined;
     needsApproval?: boolean | import("@ai-sdk/provider-utils").ToolNeedsApprovalFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -558,7 +665,7 @@ export declare function createWorkerAgentTool(input: {
     onInputAvailable?: ((options: {
         input: {
             sessionId?: string | undefined;
-            target: string | undefined;
+            target?: string | undefined;
             message: string | {
                 role: "user";
                 parts: ({
@@ -581,7 +688,7 @@ export declare function createWorkerAgentTool(input: {
         toolCallId: string;
         input: {
             sessionId?: string | undefined;
-            target: string | undefined;
+            target?: string | undefined;
             message: string | {
                 role: "user";
                 parts: ({
@@ -618,6 +725,16 @@ export declare function createWorkerAgentTool(input: {
         } | {
             error: string;
             message: string;
+        } | {
+            code: "UPSTREAM_HTTP_ERROR";
+            status: number;
+            message: string;
+        } | {
+            code: string;
+            reason: string;
+            status: number;
+            requestId?: string | undefined;
+            message: string;
         }>;
     }) => import("@ai-sdk/provider-utils").ToolResultOutput | PromiseLike<import("@ai-sdk/provider-utils").ToolResultOutput>) | undefined;
 } & {
@@ -640,10 +757,20 @@ export declare function createWorkerAgentTool(input: {
     } | {
         error: string;
         message: string;
+    } | {
+        code: "UPSTREAM_HTTP_ERROR";
+        status: number;
+        message: string;
+    } | {
+        code: string;
+        reason: string;
+        status: number;
+        requestId?: string | undefined;
+        message: string;
     }> | undefined;
     execute: import("ai").ToolExecuteFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -679,6 +806,16 @@ export declare function createWorkerAgentTool(input: {
     } | {
         error: string;
         message: string;
+    } | {
+        code: "UPSTREAM_HTTP_ERROR";
+        status: number;
+        message: string;
+    } | {
+        code: string;
+        reason: string;
+        status: number;
+        requestId?: string | undefined;
+        message: string;
     }, NoInfer<import("@ai-sdk/provider-utils").Context>>;
 } & {
     type: 'provider';
@@ -693,7 +830,7 @@ export declare function createWorkerAgentTool(input: {
 } & {
     execute: import("ai").ToolExecuteFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -729,6 +866,16 @@ export declare function createWorkerAgentTool(input: {
     } | {
         error: string;
         message: string;
+    } | {
+        code: "UPSTREAM_HTTP_ERROR";
+        status: number;
+        message: string;
+    } | {
+        code: string;
+        reason: string;
+        status: number;
+        requestId?: string | undefined;
+        message: string;
     }, NoInfer<import("@ai-sdk/provider-utils").Context>>;
 }) | ({
     deferLoading?: boolean;
@@ -737,7 +884,7 @@ export declare function createWorkerAgentTool(input: {
     metadata?: import("@ai-sdk/provider").JSONObject;
     inputSchema: import("ai").FlexibleSchema<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -758,7 +905,7 @@ export declare function createWorkerAgentTool(input: {
     contextSchema?: import("ai").FlexibleSchema<import("@ai-sdk/provider-utils").Context> | undefined;
     needsApproval?: boolean | import("@ai-sdk/provider-utils").ToolNeedsApprovalFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -783,7 +930,7 @@ export declare function createWorkerAgentTool(input: {
     onInputAvailable?: ((options: {
         input: {
             sessionId?: string | undefined;
-            target: string | undefined;
+            target?: string | undefined;
             message: string | {
                 role: "user";
                 parts: ({
@@ -806,7 +953,7 @@ export declare function createWorkerAgentTool(input: {
         toolCallId: string;
         input: {
             sessionId?: string | undefined;
-            target: string | undefined;
+            target?: string | undefined;
             message: string | {
                 role: "user";
                 parts: ({
@@ -843,6 +990,16 @@ export declare function createWorkerAgentTool(input: {
         } | {
             error: string;
             message: string;
+        } | {
+            code: "UPSTREAM_HTTP_ERROR";
+            status: number;
+            message: string;
+        } | {
+            code: string;
+            reason: string;
+            status: number;
+            requestId?: string | undefined;
+            message: string;
         }>;
     }) => import("@ai-sdk/provider-utils").ToolResultOutput | PromiseLike<import("@ai-sdk/provider-utils").ToolResultOutput>) | undefined;
 } & {
@@ -865,10 +1022,20 @@ export declare function createWorkerAgentTool(input: {
     } | {
         error: string;
         message: string;
+    } | {
+        code: "UPSTREAM_HTTP_ERROR";
+        status: number;
+        message: string;
+    } | {
+        code: string;
+        reason: string;
+        status: number;
+        requestId?: string | undefined;
+        message: string;
     }> | undefined;
     execute: import("ai").ToolExecuteFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -904,6 +1071,16 @@ export declare function createWorkerAgentTool(input: {
     } | {
         error: string;
         message: string;
+    } | {
+        code: "UPSTREAM_HTTP_ERROR";
+        status: number;
+        message: string;
+    } | {
+        code: string;
+        reason: string;
+        status: number;
+        requestId?: string | undefined;
+        message: string;
     }, NoInfer<import("@ai-sdk/provider-utils").Context>>;
 } & {
     type: 'provider';
@@ -918,7 +1095,7 @@ export declare function createWorkerAgentTool(input: {
 } & {
     execute: import("ai").ToolExecuteFunction<{
         sessionId?: string | undefined;
-        target: string | undefined;
+        target?: string | undefined;
         message: string | {
             role: "user";
             parts: ({
@@ -954,6 +1131,16 @@ export declare function createWorkerAgentTool(input: {
     } | {
         error: string;
         message: string;
+    } | {
+        code: "UPSTREAM_HTTP_ERROR";
+        status: number;
+        message: string;
+    } | {
+        code: string;
+        reason: string;
+        status: number;
+        requestId?: string | undefined;
+        message: string;
     }, NoInfer<import("@ai-sdk/provider-utils").Context>>;
 });
 export declare function createWorkerRunCancelTool({ host, }: {
@@ -1225,7 +1412,18 @@ export declare function createWorkerRunCancelTool({ host, }: {
         status: "abort_requested" | "not_running";
     }, NoInfer<import("@ai-sdk/provider-utils").Context>>;
 });
+export declare function readChildHttpRejection(response: Response): Promise<{
+    body: unknown;
+    bodyPrefix: string;
+    bodyState: string;
+    logReason: string;
+    code: string;
+    reason: string;
+    requestId: string | undefined;
+}>;
 export declare function readWorkerAgentResponse(response: Response, options?: {
+    runId?: string;
+    requestPath?: string;
     persistedSessionWaiting?: () => Promise<boolean>;
 }): Promise<{
     status: "completed" | "suspended";