@aigne/core 1.38.1 → 1.39.0

package/CHANGELOG.md

@@ -12,6 +12,25 @@
   * dependencies
     * @aigne/observability bumped to 0.1.0
 
+## [1.39.0](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.38.1...core-v1.39.0) (2025-07-28)
+
+
+### Features
+
+* **cli:** add inquirer/prompts integrations for cli ([#286](https://github.com/AIGNE-io/aigne-framework/issues/286)) ([33af756](https://github.com/AIGNE-io/aigne-framework/commit/33af7567fe2e7f9fb4b1633127e1d54fd65cb2a8))
+
+
+### Bug Fixes
+
+* **observability:** uniq index on insert and perf on trace query ([#268](https://github.com/AIGNE-io/aigne-framework/issues/268)) ([bd02d2e](https://github.com/AIGNE-io/aigne-framework/commit/bd02d2ef4dadc3df7e4806746fede2faa5cc50bb))
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/observability-api bumped to 0.8.2
+
 ## [1.38.1](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.38.0...core-v1.38.1) (2025-07-24)
 
 

package/lib/cjs/agents/agent.d.ts

@@ -1,4 +1,5 @@
 import { nodejs } from "@aigne/platform-helpers/nodejs/index.js";
+import type * as prompts from "@inquirer/prompts";
 import { ZodObject, type ZodType } from "zod";
 import type { AgentEvent, Context, UserContext } from "../aigne/context.js";
 import type { MessagePayload } from "../aigne/message-queue.js";
@@ -146,7 +147,13 @@ export interface AgentInvokeOptions<U extends UserContext = UserContext> {
     /**
      * Optional hooks for agent invocation
      */
-    hooks?: AgentHooks<any, any>;
+    hooks?: AgentHooks<any, any> | AgentHooks<any, any>[];
+    /**
+     * @inquirer/prompts used for collecting user input during agent execution
+     *
+     * This property only exists in the CLI context by command `aigne run`
+     */
+    prompts?: typeof prompts;
 }
 /**
  * Agent is the base class for all agents.
@@ -536,13 +543,16 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
      */
     onStart?: ((event: {
         context: Context;
+        agent: Agent;
         input: I;
     }) => PromiseOrValue<void | {
         input?: I;
+        options?: any;
     }>) | Agent<{
         input: I;
     }, {
         input?: I;
+        options?: any;
     }>;
     /**
      * Called when agent processing completes or fails
@@ -555,6 +565,7 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
      */
     onEnd?: ((event: XOr<{
         context: Context;
+        agent: Agent;
         input: I;
         output: O;
         error: Error;
@@ -569,6 +580,7 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
     }>;
     onSuccess?: ((event: {
         context: Context;
+        agent: Agent;
         input: I;
         output: O;
     }) => PromiseOrValue<void | {
@@ -581,6 +593,7 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
     }>;
     onError?: ((event: {
         context: Context;
+        agent: Agent;
         input: I;
         error: Error;
     }) => void) | Agent<{
@@ -597,6 +610,7 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
      */
     onSkillStart?: ((event: {
         context: Context;
+        agent: Agent;
         skill: Agent;
         input: Message;
     }) => PromiseOrValue<void>) | Agent<{
@@ -614,6 +628,7 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
      */
     onSkillEnd?: ((event: XOr<{
         context: Context;
+        agent: Agent;
         skill: Agent;
         input: Message;
         output: Message;
@@ -644,7 +659,7 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
         input: I;
     }>;
 }
-export type AgentHookInput<H extends keyof AgentHooks, Hook extends AgentHooks[H] = AgentHooks[H]> = Hook extends (input: infer I) => any ? Omit<I, "context"> : never;
+export type AgentHookInput<H extends keyof AgentHooks, Hook extends AgentHooks[H] = AgentHooks[H]> = Hook extends (input: infer I) => any ? Omit<I, "context" | "agent"> : never;
 export type AgentHookOutput<H extends keyof AgentHooks, Hook extends AgentHooks[H] = AgentHooks[H]> = Hook extends (...args: any[]) => any ? Exclude<Awaited<ReturnType<Hook>>, void> | undefined : never;
 /**
  * Response type for an agent, can be:

package/lib/cjs/agents/agent.js

@@ -348,6 +348,8 @@ class Agent {
             const s = await this.callHooks("onStart", { input }, opts);
             if (s?.input)
                 input = s.input;
+            if (s?.options)
+                Object.assign(opts, s.options);
             input = (0, type_utils_js_1.checkArguments)(`Agent ${this.name} input`, this.inputSchema, input);
             await this.preprocess(input, opts);
             this.checkContextStatus(opts);
@@ -386,10 +388,10 @@ class Agent {
             if (!h)
                 continue;
             if (typeof h === "function") {
-                Object.assign(result, await h({ ...input, context }));
+                Object.assign(result, await h({ ...input, context, agent: this }));
             }
             else {
-                Object.assign(result, await context.invoke(h, input));
+                Object.assign(result, await context.invoke(h, { ...input, agent: this }));
             }
         }
         return result;

package/lib/cjs/aigne/context.d.ts

@@ -156,6 +156,7 @@ export declare class AIGNEContext implements Context {
     get observer(): AIGNEObserver | undefined;
     get limits(): ContextLimits | undefined;
     get status(): "normal" | "timeout";
+    get spans(): Span[];
     get usage(): ContextUsage;
     get userContext(): Context["userContext"];
     set userContext(userContext: Context["userContext"]);
@@ -170,6 +171,7 @@ export declare class AIGNEContext implements Context {
     subscribe: Context["subscribe"];
     unsubscribe: Context["unsubscribe"];
     emit<K extends keyof ContextEmitEventMap>(eventName: K, ...args: Args<K, ContextEmitEventMap>): boolean;
+    private endAllSpans;
     private trace;
     on<K extends keyof ContextEventMap>(eventName: K, listener: Listener<K, ContextEventMap>): this;
     once<K extends keyof ContextEventMap>(eventName: K, listener: Listener<K, ContextEventMap>): this;
@@ -177,7 +179,7 @@ export declare class AIGNEContext implements Context {
 }
 declare class AIGNEContextShared {
     private readonly parent?;
-    span?: Span;
+    spans: Span[];
     constructor(parent?: (Pick<Context, "model" | "skills" | "limits" | "observer"> & {
         messageQueue?: MessageQueue;
         events?: Emitter<any>;
@@ -188,6 +190,7 @@ declare class AIGNEContextShared {
     get skills(): Agent<any, any>[] | undefined;
     get observer(): AIGNEObserver | undefined;
     get limits(): ContextLimits | undefined;
+    addSpan(span: Span): void;
     usage: ContextUsage;
     userContext: Context["userContext"];
     memories: Context["memories"];

package/lib/cjs/aigne/context.js

@@ -40,11 +40,14 @@ class AIGNEContext {
             }
         }
         else {
-            this.internal = new AIGNEContextShared(parent instanceof AIGNEContext ? parent.internal : parent);
             this.span = tracer?.startSpan("AIGNEContext");
+            this.internal = new AIGNEContextShared(parent instanceof AIGNEContext ? parent.internal : parent);
             // 修改了 rootId 是否会之前的有影响？，之前为 this.id
             this.rootId = this.span?.spanContext?.().traceId ?? (0, uuid_1.v7)();
         }
+        if (this.span) {
+            this.internal.addSpan(this.span);
+        }
         this.id = this.span?.spanContext()?.spanId ?? (0, uuid_1.v7)();
     }
     id;
@@ -70,6 +73,9 @@ class AIGNEContext {
     get status() {
         return this.internal.status;
     }
+    get spans() {
+        return this.internal.spans;
+    }
     get usage() {
         return this.internal.usage;
     }
@@ -111,12 +117,18 @@ class AIGNEContext {
         const newContext = options?.newContext === false ? this : this.newContext();
         return Promise.resolve(newContext.internal.invoke(agent, message, newContext, options)).then(async (response) => {
             if (!options?.streaming) {
-                let { __activeAgent__: activeAgent, ...output } = await (0, stream_utils_js_1.agentResponseStreamToObject)(response);
-                output = await this.onInvocationResult(output, options);
-                if (options?.returnActiveAgent) {
-                    return [output, activeAgent];
+                try {
+                    let { __activeAgent__: activeAgent, ...output } = await (0, stream_utils_js_1.agentResponseStreamToObject)(response);
+                    output = await this.onInvocationResult(output, options);
+                    if (options?.returnActiveAgent) {
+                        return [output, activeAgent];
+                    }
+                    return output;
+                }
+                catch (error) {
+                    this.endAllSpans(error);
+                    throw error;
                 }
-                return output;
             }
             const activeAgentPromise = (0, promise_js_1.promiseWithResolvers)();
             const stream = (0, stream_utils_js_1.onAgentResponseStreamEnd)((0, stream_utils_js_1.asyncGeneratorToReadableStream)(response), {
@@ -135,6 +147,10 @@ class AIGNEContext {
                     activeAgentPromise.resolve(output.__activeAgent__);
                     return await this.onInvocationResult(output, options);
                 },
+                onError: (error) => {
+                    this.endAllSpans(error);
+                    return error;
+                },
             });
             const finalStream = !options.returnProgressChunks
                 ? stream
@@ -189,6 +205,12 @@ class AIGNEContext {
         this.trace(eventName, args, b);
         return this.internal.events.emit(eventName, ...newArgs);
     }
+    async endAllSpans(error) {
+        this.spans.forEach((span) => {
+            span.setStatus({ code: api_1.SpanStatusCode.ERROR, message: error?.message });
+            span.end();
+        });
+    }
     async trace(eventName, args, b) {
         const span = this.span;
         if (!span)
@@ -220,11 +242,7 @@ class AIGNEContext {
                         logger_js_1.logger.error("parse memories error", _e.message);
                         span.setAttribute("memories", JSON.stringify([]));
                     }
-                    await this.observer?.traceExporter
-                        ?.upsertInitialSpan?.(span)
-                        .catch((err) => {
-                        logger_js_1.logger.error("upsertInitialSpan error", err?.message || err);
-                    });
+                    await this.observer?.flush(span);
                     break;
                 }
                 case "agentSucceed": {
@@ -268,7 +286,7 @@ class AIGNEContext {
 exports.AIGNEContext = AIGNEContext;
 class AIGNEContextShared {
     parent;
-    span;
+    spans = [];
     constructor(parent) {
         this.parent = parent;
         this.messageQueue = this.parent?.messageQueue ?? new message_queue_js_1.MessageQueue();
@@ -288,6 +306,9 @@ class AIGNEContextShared {
     get limits() {
         return this.parent?.limits;
     }
+    addSpan(span) {
+        this.spans.push(span);
+    }
     usage = (0, usage_js_1.newEmptyContextUsage)();
     userContext = {};
     memories = [];
@@ -317,7 +338,7 @@ class AIGNEContextShared {
             for (;;) {
                 const result = {};
                 if (options?.sourceAgent && activeAgent !== options.sourceAgent) {
-                    for (const { onHandoff } of [options.hooks ?? {}, ...options.sourceAgent.hooks]) {
+                    for (const { onHandoff } of (0, type_utils_js_1.flat)(options.hooks, options.sourceAgent.hooks)) {
                         if (!onHandoff)
                             continue;
                         await (typeof onHandoff === "function"

package/lib/dts/agents/agent.d.ts

@@ -1,4 +1,5 @@
 import { nodejs } from "@aigne/platform-helpers/nodejs/index.js";
+import type * as prompts from "@inquirer/prompts";
 import { ZodObject, type ZodType } from "zod";
 import type { AgentEvent, Context, UserContext } from "../aigne/context.js";
 import type { MessagePayload } from "../aigne/message-queue.js";
@@ -146,7 +147,13 @@ export interface AgentInvokeOptions<U extends UserContext = UserContext> {
     /**
      * Optional hooks for agent invocation
      */
-    hooks?: AgentHooks<any, any>;
+    hooks?: AgentHooks<any, any> | AgentHooks<any, any>[];
+    /**
+     * @inquirer/prompts used for collecting user input during agent execution
+     *
+     * This property only exists in the CLI context by command `aigne run`
+     */
+    prompts?: typeof prompts;
 }
 /**
  * Agent is the base class for all agents.
@@ -536,13 +543,16 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
      */
     onStart?: ((event: {
         context: Context;
+        agent: Agent;
         input: I;
     }) => PromiseOrValue<void | {
         input?: I;
+        options?: any;
     }>) | Agent<{
         input: I;
     }, {
         input?: I;
+        options?: any;
     }>;
     /**
      * Called when agent processing completes or fails
@@ -555,6 +565,7 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
      */
     onEnd?: ((event: XOr<{
         context: Context;
+        agent: Agent;
         input: I;
         output: O;
         error: Error;
@@ -569,6 +580,7 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
     }>;
     onSuccess?: ((event: {
         context: Context;
+        agent: Agent;
         input: I;
         output: O;
     }) => PromiseOrValue<void | {
@@ -581,6 +593,7 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
     }>;
     onError?: ((event: {
         context: Context;
+        agent: Agent;
         input: I;
         error: Error;
     }) => void) | Agent<{
@@ -597,6 +610,7 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
      */
     onSkillStart?: ((event: {
         context: Context;
+        agent: Agent;
         skill: Agent;
         input: Message;
     }) => PromiseOrValue<void>) | Agent<{
@@ -614,6 +628,7 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
      */
     onSkillEnd?: ((event: XOr<{
         context: Context;
+        agent: Agent;
         skill: Agent;
         input: Message;
         output: Message;
@@ -644,7 +659,7 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
         input: I;
     }>;
 }
-export type AgentHookInput<H extends keyof AgentHooks, Hook extends AgentHooks[H] = AgentHooks[H]> = Hook extends (input: infer I) => any ? Omit<I, "context"> : never;
+export type AgentHookInput<H extends keyof AgentHooks, Hook extends AgentHooks[H] = AgentHooks[H]> = Hook extends (input: infer I) => any ? Omit<I, "context" | "agent"> : never;
 export type AgentHookOutput<H extends keyof AgentHooks, Hook extends AgentHooks[H] = AgentHooks[H]> = Hook extends (...args: any[]) => any ? Exclude<Awaited<ReturnType<Hook>>, void> | undefined : never;
 /**
  * Response type for an agent, can be:

package/lib/dts/aigne/context.d.ts

@@ -156,6 +156,7 @@ export declare class AIGNEContext implements Context {
     get observer(): AIGNEObserver | undefined;
     get limits(): ContextLimits | undefined;
     get status(): "normal" | "timeout";
+    get spans(): Span[];
     get usage(): ContextUsage;
     get userContext(): Context["userContext"];
     set userContext(userContext: Context["userContext"]);
@@ -170,6 +171,7 @@ export declare class AIGNEContext implements Context {
     subscribe: Context["subscribe"];
     unsubscribe: Context["unsubscribe"];
     emit<K extends keyof ContextEmitEventMap>(eventName: K, ...args: Args<K, ContextEmitEventMap>): boolean;
+    private endAllSpans;
     private trace;
     on<K extends keyof ContextEventMap>(eventName: K, listener: Listener<K, ContextEventMap>): this;
     once<K extends keyof ContextEventMap>(eventName: K, listener: Listener<K, ContextEventMap>): this;
@@ -177,7 +179,7 @@ export declare class AIGNEContext implements Context {
 }
 declare class AIGNEContextShared {
     private readonly parent?;
-    span?: Span;
+    spans: Span[];
     constructor(parent?: (Pick<Context, "model" | "skills" | "limits" | "observer"> & {
         messageQueue?: MessageQueue;
         events?: Emitter<any>;
@@ -188,6 +190,7 @@ declare class AIGNEContextShared {
     get skills(): Agent<any, any>[] | undefined;
     get observer(): AIGNEObserver | undefined;
     get limits(): ContextLimits | undefined;
+    addSpan(span: Span): void;
     usage: ContextUsage;
     userContext: Context["userContext"];
     memories: Context["memories"];

package/lib/esm/agents/agent.d.ts

@@ -1,4 +1,5 @@
 import { nodejs } from "@aigne/platform-helpers/nodejs/index.js";
+import type * as prompts from "@inquirer/prompts";
 import { ZodObject, type ZodType } from "zod";
 import type { AgentEvent, Context, UserContext } from "../aigne/context.js";
 import type { MessagePayload } from "../aigne/message-queue.js";
@@ -146,7 +147,13 @@ export interface AgentInvokeOptions<U extends UserContext = UserContext> {
     /**
      * Optional hooks for agent invocation
      */
-    hooks?: AgentHooks<any, any>;
+    hooks?: AgentHooks<any, any> | AgentHooks<any, any>[];
+    /**
+     * @inquirer/prompts used for collecting user input during agent execution
+     *
+     * This property only exists in the CLI context by command `aigne run`
+     */
+    prompts?: typeof prompts;
 }
 /**
  * Agent is the base class for all agents.
@@ -536,13 +543,16 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
      */
     onStart?: ((event: {
         context: Context;
+        agent: Agent;
         input: I;
     }) => PromiseOrValue<void | {
         input?: I;
+        options?: any;
     }>) | Agent<{
         input: I;
     }, {
         input?: I;
+        options?: any;
     }>;
     /**
      * Called when agent processing completes or fails
@@ -555,6 +565,7 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
      */
     onEnd?: ((event: XOr<{
         context: Context;
+        agent: Agent;
         input: I;
         output: O;
         error: Error;
@@ -569,6 +580,7 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
     }>;
     onSuccess?: ((event: {
         context: Context;
+        agent: Agent;
         input: I;
         output: O;
     }) => PromiseOrValue<void | {
@@ -581,6 +593,7 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
     }>;
     onError?: ((event: {
         context: Context;
+        agent: Agent;
         input: I;
         error: Error;
     }) => void) | Agent<{
@@ -597,6 +610,7 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
      */
     onSkillStart?: ((event: {
         context: Context;
+        agent: Agent;
         skill: Agent;
         input: Message;
     }) => PromiseOrValue<void>) | Agent<{
@@ -614,6 +628,7 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
      */
     onSkillEnd?: ((event: XOr<{
         context: Context;
+        agent: Agent;
         skill: Agent;
         input: Message;
         output: Message;
@@ -644,7 +659,7 @@ export interface AgentHooks<I extends Message = Message, O extends Message = Mes
         input: I;
     }>;
 }
-export type AgentHookInput<H extends keyof AgentHooks, Hook extends AgentHooks[H] = AgentHooks[H]> = Hook extends (input: infer I) => any ? Omit<I, "context"> : never;
+export type AgentHookInput<H extends keyof AgentHooks, Hook extends AgentHooks[H] = AgentHooks[H]> = Hook extends (input: infer I) => any ? Omit<I, "context" | "agent"> : never;
 export type AgentHookOutput<H extends keyof AgentHooks, Hook extends AgentHooks[H] = AgentHooks[H]> = Hook extends (...args: any[]) => any ? Exclude<Awaited<ReturnType<Hook>>, void> | undefined : never;
 /**
  * Response type for an agent, can be:

package/lib/esm/agents/agent.js

@@ -303,6 +303,8 @@ export class Agent {
             const s = await this.callHooks("onStart", { input }, opts);
             if (s?.input)
                 input = s.input;
+            if (s?.options)
+                Object.assign(opts, s.options);
             input = checkArguments(`Agent ${this.name} input`, this.inputSchema, input);
             await this.preprocess(input, opts);
             this.checkContextStatus(opts);
@@ -341,10 +343,10 @@ export class Agent {
             if (!h)
                 continue;
             if (typeof h === "function") {
-                Object.assign(result, await h({ ...input, context }));
+                Object.assign(result, await h({ ...input, context, agent: this }));
             }
             else {
-                Object.assign(result, await context.invoke(h, input));
+                Object.assign(result, await context.invoke(h, { ...input, agent: this }));
             }
         }
         return result;

package/lib/esm/aigne/context.d.ts

@@ -156,6 +156,7 @@ export declare class AIGNEContext implements Context {
     get observer(): AIGNEObserver | undefined;
     get limits(): ContextLimits | undefined;
     get status(): "normal" | "timeout";
+    get spans(): Span[];
     get usage(): ContextUsage;
     get userContext(): Context["userContext"];
     set userContext(userContext: Context["userContext"]);
@@ -170,6 +171,7 @@ export declare class AIGNEContext implements Context {
     subscribe: Context["subscribe"];
     unsubscribe: Context["unsubscribe"];
     emit<K extends keyof ContextEmitEventMap>(eventName: K, ...args: Args<K, ContextEmitEventMap>): boolean;
+    private endAllSpans;
     private trace;
     on<K extends keyof ContextEventMap>(eventName: K, listener: Listener<K, ContextEventMap>): this;
     once<K extends keyof ContextEventMap>(eventName: K, listener: Listener<K, ContextEventMap>): this;
@@ -177,7 +179,7 @@ export declare class AIGNEContext implements Context {
 }
 declare class AIGNEContextShared {
     private readonly parent?;
-    span?: Span;
+    spans: Span[];
     constructor(parent?: (Pick<Context, "model" | "skills" | "limits" | "observer"> & {
         messageQueue?: MessageQueue;
         events?: Emitter<any>;
@@ -188,6 +190,7 @@ declare class AIGNEContextShared {
     get skills(): Agent<any, any>[] | undefined;
     get observer(): AIGNEObserver | undefined;
     get limits(): ContextLimits | undefined;
+    addSpan(span: Span): void;
     usage: ContextUsage;
     userContext: Context["userContext"];
     memories: Context["memories"];

package/lib/esm/aigne/context.js

@@ -10,7 +10,7 @@ import { AgentResponseProgressStream } from "../utils/event-stream.js";
 import { logger } from "../utils/logger.js";
 import { promiseWithResolvers } from "../utils/promise.js";
 import { agentResponseStreamToObject, asyncGeneratorToReadableStream, mergeReadableStreams, onAgentResponseStreamEnd, } from "../utils/stream-utils.js";
-import { checkArguments, isEmpty, isNil, omit, } from "../utils/type-utils.js";
+import { checkArguments, flat, isEmpty, isNil, omit, } from "../utils/type-utils.js";
 import { MessageQueue, toMessagePayload, } from "./message-queue.js";
 import { newEmptyContextUsage } from "./usage.js";
 /**
@@ -34,11 +34,14 @@ export class AIGNEContext {
             }
         }
         else {
-            this.internal = new AIGNEContextShared(parent instanceof AIGNEContext ? parent.internal : parent);
             this.span = tracer?.startSpan("AIGNEContext");
+            this.internal = new AIGNEContextShared(parent instanceof AIGNEContext ? parent.internal : parent);
             // 修改了 rootId 是否会之前的有影响？，之前为 this.id
             this.rootId = this.span?.spanContext?.().traceId ?? v7();
         }
+        if (this.span) {
+            this.internal.addSpan(this.span);
+        }
         this.id = this.span?.spanContext()?.spanId ?? v7();
     }
     id;
@@ -64,6 +67,9 @@ export class AIGNEContext {
     get status() {
         return this.internal.status;
     }
+    get spans() {
+        return this.internal.spans;
+    }
     get usage() {
         return this.internal.usage;
     }
@@ -105,12 +111,18 @@ export class AIGNEContext {
         const newContext = options?.newContext === false ? this : this.newContext();
         return Promise.resolve(newContext.internal.invoke(agent, message, newContext, options)).then(async (response) => {
             if (!options?.streaming) {
-                let { __activeAgent__: activeAgent, ...output } = await agentResponseStreamToObject(response);
-                output = await this.onInvocationResult(output, options);
-                if (options?.returnActiveAgent) {
-                    return [output, activeAgent];
+                try {
+                    let { __activeAgent__: activeAgent, ...output } = await agentResponseStreamToObject(response);
+                    output = await this.onInvocationResult(output, options);
+                    if (options?.returnActiveAgent) {
+                        return [output, activeAgent];
+                    }
+                    return output;
+                }
+                catch (error) {
+                    this.endAllSpans(error);
+                    throw error;
                 }
-                return output;
             }
             const activeAgentPromise = promiseWithResolvers();
             const stream = onAgentResponseStreamEnd(asyncGeneratorToReadableStream(response), {
@@ -129,6 +141,10 @@ export class AIGNEContext {
                     activeAgentPromise.resolve(output.__activeAgent__);
                     return await this.onInvocationResult(output, options);
                 },
+                onError: (error) => {
+                    this.endAllSpans(error);
+                    return error;
+                },
             });
             const finalStream = !options.returnProgressChunks
                 ? stream
@@ -183,6 +199,12 @@ export class AIGNEContext {
         this.trace(eventName, args, b);
         return this.internal.events.emit(eventName, ...newArgs);
     }
+    async endAllSpans(error) {
+        this.spans.forEach((span) => {
+            span.setStatus({ code: SpanStatusCode.ERROR, message: error?.message });
+            span.end();
+        });
+    }
     async trace(eventName, args, b) {
         const span = this.span;
         if (!span)
@@ -214,11 +236,7 @@ export class AIGNEContext {
                         logger.error("parse memories error", _e.message);
                         span.setAttribute("memories", JSON.stringify([]));
                     }
-                    await this.observer?.traceExporter
-                        ?.upsertInitialSpan?.(span)
-                        .catch((err) => {
-                        logger.error("upsertInitialSpan error", err?.message || err);
-                    });
+                    await this.observer?.flush(span);
                     break;
                 }
                 case "agentSucceed": {
@@ -261,7 +279,7 @@ export class AIGNEContext {
 }
 class AIGNEContextShared {
     parent;
-    span;
+    spans = [];
     constructor(parent) {
         this.parent = parent;
         this.messageQueue = this.parent?.messageQueue ?? new MessageQueue();
@@ -281,6 +299,9 @@ class AIGNEContextShared {
     get limits() {
         return this.parent?.limits;
     }
+    addSpan(span) {
+        this.spans.push(span);
+    }
     usage = newEmptyContextUsage();
     userContext = {};
     memories = [];
@@ -310,7 +331,7 @@ class AIGNEContextShared {
             for (;;) {
                 const result = {};
                 if (options?.sourceAgent && activeAgent !== options.sourceAgent) {
-                    for (const { onHandoff } of [options.hooks ?? {}, ...options.sourceAgent.hooks]) {
+                    for (const { onHandoff } of flat(options.hooks, options.sourceAgent.hooks)) {
                         if (!onHandoff)
                             continue;
                         await (typeof onHandoff === "function"

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/core",
-  "version": "1.38.1",
+  "version": "1.39.0",
   "description": "AIGNE core library for building AI-powered applications",
   "publishConfig": {
     "access": "public"
@@ -63,6 +63,7 @@
   },
   "dependencies": {
     "@aigne/json-schema-to-zod": "^1.3.3",
+    "@inquirer/prompts": "^7.6.0",
     "@modelcontextprotocol/sdk": "^1.15.0",
     "@opentelemetry/api": "^1.9.0",
     "@opentelemetry/sdk-trace-base": "^2.0.1",
@@ -85,7 +86,7 @@
     "yaml": "^2.8.0",
     "zod": "^3.25.67",
     "zod-to-json-schema": "^3.24.6",
-    "@aigne/observability-api": "^0.8.1",
+    "@aigne/observability-api": "^0.8.2",
     "@aigne/platform-helpers": "^0.4.0"
   },
   "devDependencies": {