@aigne/transport 0.9.6 → 0.10.2

package/CHANGELOG.md

@@ -1,5 +1,53 @@
 # Changelog
 
+## [0.10.2](https://github.com/AIGNE-io/aigne-framework/compare/transport-v0.10.1...transport-v0.10.2) (2025-07-17)
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/openai bumped to 0.10.2
+  * devDependencies
+    * @aigne/agent-library bumped to 1.21.2
+    * @aigne/core bumped to 1.36.0
+    * @aigne/test-utils bumped to 0.5.10
+    * @aigne/default-memory bumped to 1.0.2
+
+## [0.10.1](https://github.com/AIGNE-io/aigne-framework/compare/transport-v0.10.0...transport-v0.10.1) (2025-07-17)
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/openai bumped to 0.10.1
+  * devDependencies
+    * @aigne/agent-library bumped to 1.21.1
+    * @aigne/core bumped to 1.35.0
+    * @aigne/test-utils bumped to 0.5.9
+    * @aigne/default-memory bumped to 1.0.1
+
+## [0.10.0](https://github.com/AIGNE-io/aigne-framework/compare/transport-v0.9.6...transport-v0.10.0) (2025-07-15)
+
+
+### Features
+
+* **memory:** support did space memory adapter ([#229](https://github.com/AIGNE-io/aigne-framework/issues/229)) ([6f69b64](https://github.com/AIGNE-io/aigne-framework/commit/6f69b64e98b963db9d6ab5357306b445385eaa68))
+* **model:** support aigne-hub model adapter ([#253](https://github.com/AIGNE-io/aigne-framework/issues/253)) ([4b33f8d](https://github.com/AIGNE-io/aigne-framework/commit/4b33f8d1a819f52357db81d502c56b55eaa0669f))
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/openai bumped to 0.10.0
+  * devDependencies
+    * @aigne/agent-library bumped to 1.21.0
+    * @aigne/core bumped to 1.34.0
+    * @aigne/test-utils bumped to 0.5.8
+    * @aigne/default-memory bumped to 1.0.0
+
 ## [0.9.6](https://github.com/AIGNE-io/aigne-framework/compare/transport-v0.9.5...transport-v0.9.6) (2025-07-14)
 
 

package/lib/cjs/http-client/base-client.d.ts

@@ -0,0 +1,92 @@
+import type { AgentResponse, AgentResponseStream, ChatModelOptions, InvokeOptions, Message } from "@aigne/core";
+import type { OpenAIChatModelOptions } from "@aigne/openai";
+/**
+ * Options for invoking an agent through the BaseClient.
+ * Extends the standard AgentInvokeOptions with client-specific options.
+ */
+export interface BaseClientInvokeOptions extends InvokeOptions {
+    /**
+     * Additional fetch API options to customize the HTTP request.
+     * These options will be merged with the default options used by the client.
+     */
+    fetchOptions?: Partial<RequestInit>;
+}
+export interface BaseClientOptions {
+    url: string;
+    accessKey?: string;
+    model?: string;
+    modelOptions?: ChatModelOptions;
+    clientOptions?: OpenAIChatModelOptions["clientOptions"];
+}
+/**
+ * Http client for interacting with a remote AIGNE server.
+ * BaseClient provides a client-side interface that matches the AIGNE API,
+ * allowing applications to invoke agents and receive responses from a remote AIGNE instance.
+ *
+ * @example
+ * Here's a simple example of how to use AIGNEClient:
+ * {@includeCode ../../test/http-client/http-client.test.ts#example-aigne-client-simple}
+ *
+ * @example
+ * Here's an example of how to use AIGNEClient with streaming response:
+ * {@includeCode ../../test/http-client/http-client.test.ts#example-aigne-client-streaming}
+ */
+export declare class BaseClient {
+    options: BaseClientOptions;
+    /**
+     * Creates a new AIGNEClient instance.
+     *
+     * @param options - Configuration options for connecting to the AIGNE server
+     */
+    constructor(options: BaseClientOptions);
+    /**
+     * Invokes an agent in non-streaming mode and returns the complete response.
+     *
+     * @param agent - Name of the agent to invoke
+     * @param input - Input message for the agent
+     * @param options - Options with streaming mode explicitly set to false or omitted
+     * @returns The complete agent response
+     *
+     * @example
+     * Here's a simple example of how to use AIGNEClient:
+     * {@includeCode ../../test/http-client/http-client.test.ts#example-aigne-client-simple}
+     */
+    __invoke<I extends Message, O extends Message>(agent?: string, input?: string | I, options?: BaseClientInvokeOptions & {
+        streaming?: false;
+    }): Promise<O>;
+    /**
+     * Invokes an agent with streaming mode enabled and returns a stream of response chunks.
+     *
+     * @param agent - Name of the agent to invoke
+     * @param input - Input message for the agent
+     * @param options - Options with streaming mode explicitly set to true
+     * @returns A stream of agent response chunks
+     *
+     * @example
+     * Here's an example of how to use AIGNEClient with streaming response:
+     * {@includeCode ../../test/http-client/http-client.test.ts#example-aigne-client-streaming}
+     */
+    __invoke<I extends Message, O extends Message>(agent?: string, input?: string | I, options?: BaseClientInvokeOptions & {
+        streaming: true;
+    }): Promise<AgentResponseStream<O>>;
+    /**
+     * Invokes an agent with the given input and options.
+     *
+     * @param agent - Name of the agent to invoke
+     * @param input - Input message for the agent
+     * @param options - Options for the invocation
+     * @returns Either a complete response or a response stream depending on the streaming option
+     */
+    __invoke<I extends Message, O extends Message>(agent?: string, input?: string | I, options?: BaseClientInvokeOptions): Promise<AgentResponse<O>>;
+    /**
+     * Enhanced fetch method that handles error responses from the AIGNE server.
+     * This method wraps the standard fetch API to provide better error handling and reporting.
+     *
+     * @param args - Standard fetch API arguments (url and options)
+     * @returns A Response object if the request was successful
+     * @throws Error with detailed information if the request failed
+     *
+     * @private
+     */
+    fetch(...args: Parameters<typeof globalThis.fetch>): Promise<Response>;
+}

package/lib/cjs/http-client/base-client.js

@@ -0,0 +1,101 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BaseClient = void 0;
+const event_stream_js_1 = require("@aigne/core/utils/event-stream.js");
+const type_utils_js_1 = require("@aigne/core/utils/type-utils.js");
+const constants_js_1 = require("../constants.js");
+/**
+ * Http client for interacting with a remote AIGNE server.
+ * BaseClient provides a client-side interface that matches the AIGNE API,
+ * allowing applications to invoke agents and receive responses from a remote AIGNE instance.
+ *
+ * @example
+ * Here's a simple example of how to use AIGNEClient:
+ * {@includeCode ../../test/http-client/http-client.test.ts#example-aigne-client-simple}
+ *
+ * @example
+ * Here's an example of how to use AIGNEClient with streaming response:
+ * {@includeCode ../../test/http-client/http-client.test.ts#example-aigne-client-streaming}
+ */
+class BaseClient {
+    options;
+    /**
+     * Creates a new AIGNEClient instance.
+     *
+     * @param options - Configuration options for connecting to the AIGNE server
+     */
+    constructor(options) {
+        this.options = options;
+    }
+    async __invoke(agent, input, options) {
+        const model = this.options.modelOptions?.model ?? this.options.model;
+        const headers = {
+            "Content-Type": "application/json",
+            ...options?.fetchOptions?.headers,
+        };
+        if (this.options?.accessKey) {
+            headers["Authorization"] = `Bearer ${this.options.accessKey}`;
+        }
+        const body = {
+            model,
+            input,
+            agent: agent ?? constants_js_1.ChatModelName,
+            options: options && {
+                ...(0, type_utils_js_1.omit)(options, "context"),
+                userContext: { ...options.userContext },
+                memories: [...(options.memories ?? [])],
+                modelOptions: this.options.modelOptions,
+                clientOptions: this.options.clientOptions,
+            },
+        };
+        const response = await this.fetch(this.options.url, {
+            ...options?.fetchOptions,
+            method: "POST",
+            headers: headers,
+            body: JSON.stringify(body),
+        });
+        // For non-streaming responses, simply parse the JSON response and return it
+        if (!options?.streaming) {
+            return await response.json();
+        }
+        // For streaming responses, set up the streaming pipeline
+        const stream = response.body;
+        if (!stream)
+            throw new Error("Response body is not a stream");
+        // Process the stream through a series of transforms:
+        // 1. Convert bytes to text
+        // 2. Parse SSE format into structured events
+        // 3. Convert events into a standardized agent response stream
+        return stream
+            .pipeThrough(new TextDecoderStream())
+            .pipeThrough(new event_stream_js_1.EventStreamParser())
+            .pipeThrough(new event_stream_js_1.AgentResponseStreamParser());
+    }
+    /**
+     * Enhanced fetch method that handles error responses from the AIGNE server.
+     * This method wraps the standard fetch API to provide better error handling and reporting.
+     *
+     * @param args - Standard fetch API arguments (url and options)
+     * @returns A Response object if the request was successful
+     * @throws Error with detailed information if the request failed
+     *
+     * @private
+     */
+    async fetch(...args) {
+        const result = await globalThis.fetch(...args);
+        if (!result.ok) {
+            let message;
+            try {
+                const text = await result.text();
+                const json = (0, type_utils_js_1.tryOrThrow)(() => JSON.parse(text));
+                message = json?.error?.message || text;
+            }
+            catch {
+                // ignore
+            }
+            throw new Error(`Failed to fetch url ${args[0]} with status ${result.status}: ${message}`);
+        }
+        return result;
+    }
+}
+exports.BaseClient = BaseClient;

package/lib/cjs/http-client/client.d.ts

@@ -1,27 +1,18 @@
 import { type Agent, type AgentResponse, type AgentResponseStream, type Context, type ContextEmitEventMap, type ContextEventMap, type ContextUsage, type InvokeOptions, type Message, type MessagePayload, type MessageQueueListener, type Unsubscribe, type UserAgent, type UserContext } from "@aigne/core";
 import type { Args, Listener } from "@aigne/core/utils/typed-event-emitter.js";
+import { BaseClient, type BaseClientInvokeOptions, type BaseClientOptions } from "./base-client.js";
 import { ClientAgent, type ClientAgentOptions } from "./client-agent.js";
 import { ClientChatModel } from "./client-chat-model.js";
 /**
  * Configuration options for the AIGNEHTTPClient.
  */
-export interface AIGNEHTTPClientOptions {
-    /**
-     * The URL of the AIGNE server to connect to.
-     * This should point to the base endpoint where the AIGNEServer is hosted.
-     */
-    url: string;
+export interface AIGNEHTTPClientOptions extends BaseClientOptions {
 }
 /**
  * Options for invoking an agent through the AIGNEHTTPClient.
  * Extends the standard AgentInvokeOptions with client-specific options.
  */
-export interface AIGNEHTTPClientInvokeOptions extends InvokeOptions {
-    /**
-     * Additional fetch API options to customize the HTTP request.
-     * These options will be merged with the default options used by the client.
-     */
-    fetchOptions?: Partial<RequestInit>;
+export interface AIGNEHTTPClientInvokeOptions extends BaseClientInvokeOptions {
 }
 /**
  * Http client for interacting with a remote AIGNE server.
@@ -36,7 +27,7 @@ export interface AIGNEHTTPClientInvokeOptions extends InvokeOptions {
  * Here's an example of how to use AIGNEClient with streaming response:
  * {@includeCode ../../test/http-client/http-client.test.ts#example-aigne-client-streaming}
  */
-export declare class AIGNEHTTPClient<U extends UserContext = UserContext> implements Context<U> {
+export declare class AIGNEHTTPClient<U extends UserContext = UserContext> extends BaseClient implements Context<U> {
     options: AIGNEHTTPClientOptions;
     /**
      * Creates a new AIGNEClient instance.
@@ -123,16 +114,5 @@ export declare class AIGNEHTTPClient<U extends UserContext = UserContext> implem
      * @returns Either a complete response or a response stream depending on the streaming option
      */
     _invoke<I extends Message, O extends Message>(agent: string, input: string | I, options?: AIGNEHTTPClientInvokeOptions): Promise<AgentResponse<O>>;
-    /**
-     * Enhanced fetch method that handles error responses from the AIGNE server.
-     * This method wraps the standard fetch API to provide better error handling and reporting.
-     *
-     * @param args - Standard fetch API arguments (url and options)
-     * @returns A Response object if the request was successful
-     * @throws Error with detailed information if the request failed
-     *
-     * @private
-     */
-    fetch(...args: Parameters<typeof globalThis.fetch>): Promise<Response>;
     getAgent<I extends Message = Message, O extends Message = Message>(options: ClientAgentOptions<I, O>): Promise<ClientAgent<I, O>>;
 }

package/lib/cjs/http-client/client.js

@@ -2,9 +2,9 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AIGNEHTTPClient = void 0;
 const core_1 = require("@aigne/core");
-const event_stream_js_1 = require("@aigne/core/utils/event-stream.js");
 const type_utils_js_1 = require("@aigne/core/utils/type-utils.js");
 const uuid_1 = require("uuid");
+const base_client_js_1 = require("./base-client.js");
 const client_agent_js_1 = require("./client-agent.js");
 const client_chat_model_js_1 = require("./client-chat-model.js");
 /**
@@ -20,7 +20,7 @@ const client_chat_model_js_1 = require("./client-chat-model.js");
  * Here's an example of how to use AIGNEClient with streaming response:
  * {@includeCode ../../test/http-client/http-client.test.ts#example-aigne-client-streaming}
  */
-class AIGNEHTTPClient {
+class AIGNEHTTPClient extends base_client_js_1.BaseClient {
     options;
     /**
      * Creates a new AIGNEClient instance.
@@ -28,6 +28,7 @@ class AIGNEHTTPClient {
      * @param options - Configuration options for connecting to the AIGNE server
      */
     constructor(options) {
+        super(options);
         this.options = options;
     }
     id = (0, uuid_1.v7)();
@@ -81,67 +82,12 @@ class AIGNEHTTPClient {
         console.error("Method not implemented.");
         return this;
     }
-    async _invoke(agent, input, options) {
-        // Send the agent invocation request to the AIGNE server
-        const response = await this.fetch(this.options.url, {
-            ...options?.fetchOptions,
-            method: "POST",
-            headers: {
-                "Content-Type": "application/json",
-                ...options?.fetchOptions?.headers,
-            },
-            body: JSON.stringify({
-                agent,
-                input,
-                options: options && {
-                    ...(0, type_utils_js_1.omit)(options, "context"),
-                    userContext: { ...this.userContext, ...options.userContext },
-                    memories: [...this.memories, ...(options.memories ?? [])],
-                },
-            }),
+    async _invoke(agent, input, options = {}) {
+        return this.__invoke(agent, input, {
+            ...(0, type_utils_js_1.omit)(options, "context"),
+            userContext: { ...this.userContext, ...options.userContext },
+            memories: [...this.memories, ...(options.memories ?? [])],
         });
-        // For non-streaming responses, simply parse the JSON response and return it
-        if (!options?.streaming) {
-            return await response.json();
-        }
-        // For streaming responses, set up the streaming pipeline
-        const stream = response.body;
-        if (!stream)
-            throw new Error("Response body is not a stream");
-        // Process the stream through a series of transforms:
-        // 1. Convert bytes to text
-        // 2. Parse SSE format into structured events
-        // 3. Convert events into a standardized agent response stream
-        return stream
-            .pipeThrough(new TextDecoderStream())
-            .pipeThrough(new event_stream_js_1.EventStreamParser())
-            .pipeThrough(new event_stream_js_1.AgentResponseStreamParser());
-    }
-    /**
-     * Enhanced fetch method that handles error responses from the AIGNE server.
-     * This method wraps the standard fetch API to provide better error handling and reporting.
-     *
-     * @param args - Standard fetch API arguments (url and options)
-     * @returns A Response object if the request was successful
-     * @throws Error with detailed information if the request failed
-     *
-     * @private
-     */
-    async fetch(...args) {
-        const result = await globalThis.fetch(...args);
-        if (!result.ok) {
-            let message;
-            try {
-                const text = await result.text();
-                const json = (0, type_utils_js_1.tryOrThrow)(() => JSON.parse(text));
-                message = json?.error?.message || text;
-            }
-            catch {
-                // ignore
-            }
-            throw new Error(`Failed to fetch url ${args[0]} with status ${result.status}: ${message}`);
-        }
-        return result;
     }
     async getAgent(options) {
         return new client_agent_js_1.ClientAgent(this, options);

package/lib/cjs/http-server/server.d.ts

@@ -1,5 +1,6 @@
 import { IncomingMessage, ServerResponse } from "node:http";
 import type { AIGNE, InvokeOptions, UserContext } from "@aigne/core";
+import type { PromiseOrValue } from "@aigne/core/utils/type-utils.js";
 import { z } from "zod";
 /**
  * Schema for validating agent invocation payloads.
@@ -26,55 +27,55 @@ export declare const invokePayloadSchema: z.ZodObject<{
             content: object;
         }[] | null | undefined>;
     }, "strip", z.ZodTypeAny, {
-        memories?: {
-            content: object;
-        }[] | undefined;
         streaming?: boolean | undefined;
         returnProgressChunks?: boolean | undefined;
         userContext?: Record<string, unknown> | undefined;
-    }, {
         memories?: {
             content: object;
-        }[] | null | undefined;
+        }[] | undefined;
+    }, {
         streaming?: boolean | null | undefined;
         returnProgressChunks?: boolean | null | undefined;
         userContext?: Record<string, unknown> | null | undefined;
-    }>>>, {
         memories?: {
             content: object;
-        }[] | undefined;
+        }[] | null | undefined;
+    }>>>, {
         streaming?: boolean | undefined;
         returnProgressChunks?: boolean | undefined;
         userContext?: Record<string, unknown> | undefined;
-    } | undefined, {
         memories?: {
             content: object;
-        }[] | null | undefined;
+        }[] | undefined;
+    } | undefined, {
         streaming?: boolean | null | undefined;
         returnProgressChunks?: boolean | null | undefined;
         userContext?: Record<string, unknown> | null | undefined;
+        memories?: {
+            content: object;
+        }[] | null | undefined;
     } | null | undefined>;
 }, "strip", z.ZodTypeAny, {
-    agent: string;
     input: Record<string, unknown>;
+    agent: string;
     options?: {
-        memories?: {
-            content: object;
-        }[] | undefined;
         streaming?: boolean | undefined;
         returnProgressChunks?: boolean | undefined;
         userContext?: Record<string, unknown> | undefined;
+        memories?: {
+            content: object;
+        }[] | undefined;
     } | undefined;
 }, {
-    agent: string;
     input: Record<string, unknown>;
+    agent: string;
     options?: {
-        memories?: {
-            content: object;
-        }[] | null | undefined;
         streaming?: boolean | null | undefined;
         returnProgressChunks?: boolean | null | undefined;
         userContext?: Record<string, unknown> | null | undefined;
+        memories?: {
+            content: object;
+        }[] | null | undefined;
     } | null | undefined;
 }>;
 /**
@@ -94,6 +95,7 @@ export interface AIGNEHTTPServerOptions {
     maximumBodySize?: string;
 }
 export interface AIGNEHTTPServerInvokeOptions<U extends UserContext = UserContext> extends Pick<InvokeOptions<U>, "returnProgressChunks" | "userContext" | "memories"> {
+    callback?: (result: Record<string, unknown>) => PromiseOrValue<void>;
 }
 /**
  * AIGNEHTTPServer provides HTTP API access to AIGNE capabilities.

package/lib/cjs/http-server/server.js

@@ -6,6 +6,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.AIGNEHTTPServer = exports.invokePayloadSchema = void 0;
 const node_http_1 = require("node:http");
 const event_stream_js_1 = require("@aigne/core/utils/event-stream.js");
+const stream_utils_js_1 = require("@aigne/core/utils/stream-utils.js");
 const type_utils_js_1 = require("@aigne/core/utils/type-utils.js");
 const content_type_1 = __importDefault(require("content-type"));
 const raw_body_1 = __importDefault(require("raw-body"));
@@ -82,6 +83,7 @@ class AIGNEHTTPServer {
         const result = await this._invoke(request, {
             userContext: opts?.userContext,
             memories: opts?.memories,
+            callback: opts?.callback,
         });
         if (response instanceof node_http_1.ServerResponse) {
             await this._writeResponse(result, response);
@@ -114,6 +116,7 @@ class AIGNEHTTPServer {
             };
             if (!streaming) {
                 const result = await aigne.invoke(agent, input, mergedOptions);
+                options.callback?.(result);
                 return new Response(JSON.stringify(result), {
                     headers: { "Content-Type": "application/json" },
                 });
@@ -123,7 +126,12 @@ class AIGNEHTTPServer {
                 returnActiveAgent: false,
                 streaming: true,
             });
-            return new Response(new event_stream_js_1.AgentResponseStreamSSE(stream), {
+            const newStream = (0, stream_utils_js_1.onAgentResponseStreamEnd)(stream, {
+                onResult: async (result) => {
+                    options.callback?.(result);
+                },
+            });
+            return new Response(new event_stream_js_1.AgentResponseStreamSSE(newStream), {
                 headers: {
                     "Content-Type": "text/event-stream",
                     "Cache-Control": "no-cache",

package/lib/cjs/package.json

@@ -1 +1,3 @@
-{"type": "commonjs"}
+{
+  "type": "commonjs"
+}

package/lib/dts/http-client/base-client.d.ts

@@ -0,0 +1,92 @@
+import type { AgentResponse, AgentResponseStream, ChatModelOptions, InvokeOptions, Message } from "@aigne/core";
+import type { OpenAIChatModelOptions } from "@aigne/openai";
+/**
+ * Options for invoking an agent through the BaseClient.
+ * Extends the standard AgentInvokeOptions with client-specific options.
+ */
+export interface BaseClientInvokeOptions extends InvokeOptions {
+    /**
+     * Additional fetch API options to customize the HTTP request.
+     * These options will be merged with the default options used by the client.
+     */
+    fetchOptions?: Partial<RequestInit>;
+}
+export interface BaseClientOptions {
+    url: string;
+    accessKey?: string;
+    model?: string;
+    modelOptions?: ChatModelOptions;
+    clientOptions?: OpenAIChatModelOptions["clientOptions"];
+}
+/**
+ * Http client for interacting with a remote AIGNE server.
+ * BaseClient provides a client-side interface that matches the AIGNE API,
+ * allowing applications to invoke agents and receive responses from a remote AIGNE instance.
+ *
+ * @example
+ * Here's a simple example of how to use AIGNEClient:
+ * {@includeCode ../../test/http-client/http-client.test.ts#example-aigne-client-simple}
+ *
+ * @example
+ * Here's an example of how to use AIGNEClient with streaming response:
+ * {@includeCode ../../test/http-client/http-client.test.ts#example-aigne-client-streaming}
+ */
+export declare class BaseClient {
+    options: BaseClientOptions;
+    /**
+     * Creates a new AIGNEClient instance.
+     *
+     * @param options - Configuration options for connecting to the AIGNE server
+     */
+    constructor(options: BaseClientOptions);
+    /**
+     * Invokes an agent in non-streaming mode and returns the complete response.
+     *
+     * @param agent - Name of the agent to invoke
+     * @param input - Input message for the agent
+     * @param options - Options with streaming mode explicitly set to false or omitted
+     * @returns The complete agent response
+     *
+     * @example
+     * Here's a simple example of how to use AIGNEClient:
+     * {@includeCode ../../test/http-client/http-client.test.ts#example-aigne-client-simple}
+     */
+    __invoke<I extends Message, O extends Message>(agent?: string, input?: string | I, options?: BaseClientInvokeOptions & {
+        streaming?: false;
+    }): Promise<O>;
+    /**
+     * Invokes an agent with streaming mode enabled and returns a stream of response chunks.
+     *
+     * @param agent - Name of the agent to invoke
+     * @param input - Input message for the agent
+     * @param options - Options with streaming mode explicitly set to true
+     * @returns A stream of agent response chunks
+     *
+     * @example
+     * Here's an example of how to use AIGNEClient with streaming response:
+     * {@includeCode ../../test/http-client/http-client.test.ts#example-aigne-client-streaming}
+     */
+    __invoke<I extends Message, O extends Message>(agent?: string, input?: string | I, options?: BaseClientInvokeOptions & {
+        streaming: true;
+    }): Promise<AgentResponseStream<O>>;
+    /**
+     * Invokes an agent with the given input and options.
+     *
+     * @param agent - Name of the agent to invoke
+     * @param input - Input message for the agent
+     * @param options - Options for the invocation
+     * @returns Either a complete response or a response stream depending on the streaming option
+     */
+    __invoke<I extends Message, O extends Message>(agent?: string, input?: string | I, options?: BaseClientInvokeOptions): Promise<AgentResponse<O>>;
+    /**
+     * Enhanced fetch method that handles error responses from the AIGNE server.
+     * This method wraps the standard fetch API to provide better error handling and reporting.
+     *
+     * @param args - Standard fetch API arguments (url and options)
+     * @returns A Response object if the request was successful
+     * @throws Error with detailed information if the request failed
+     *
+     * @private
+     */
+    fetch(...args: Parameters<typeof globalThis.fetch>): Promise<Response>;
+}

package/lib/dts/http-client/client.d.ts

@@ -1,27 +1,18 @@
 import { type Agent, type AgentResponse, type AgentResponseStream, type Context, type ContextEmitEventMap, type ContextEventMap, type ContextUsage, type InvokeOptions, type Message, type MessagePayload, type MessageQueueListener, type Unsubscribe, type UserAgent, type UserContext } from "@aigne/core";
 import type { Args, Listener } from "@aigne/core/utils/typed-event-emitter.js";
+import { BaseClient, type BaseClientInvokeOptions, type BaseClientOptions } from "./base-client.js";
 import { ClientAgent, type ClientAgentOptions } from "./client-agent.js";
 import { ClientChatModel } from "./client-chat-model.js";
 /**
  * Configuration options for the AIGNEHTTPClient.
  */
-export interface AIGNEHTTPClientOptions {
-    /**
-     * The URL of the AIGNE server to connect to.
-     * This should point to the base endpoint where the AIGNEServer is hosted.
-     */
-    url: string;
+export interface AIGNEHTTPClientOptions extends BaseClientOptions {
 }
 /**
  * Options for invoking an agent through the AIGNEHTTPClient.
  * Extends the standard AgentInvokeOptions with client-specific options.
  */
-export interface AIGNEHTTPClientInvokeOptions extends InvokeOptions {
-    /**
-     * Additional fetch API options to customize the HTTP request.
-     * These options will be merged with the default options used by the client.
-     */
-    fetchOptions?: Partial<RequestInit>;
+export interface AIGNEHTTPClientInvokeOptions extends BaseClientInvokeOptions {
 }
 /**
  * Http client for interacting with a remote AIGNE server.
@@ -36,7 +27,7 @@ export interface AIGNEHTTPClientInvokeOptions extends InvokeOptions {
  * Here's an example of how to use AIGNEClient with streaming response:
  * {@includeCode ../../test/http-client/http-client.test.ts#example-aigne-client-streaming}
  */
-export declare class AIGNEHTTPClient<U extends UserContext = UserContext> implements Context<U> {
+export declare class AIGNEHTTPClient<U extends UserContext = UserContext> extends BaseClient implements Context<U> {
     options: AIGNEHTTPClientOptions;
     /**
      * Creates a new AIGNEClient instance.
@@ -123,16 +114,5 @@ export declare class AIGNEHTTPClient<U extends UserContext = UserContext> implem
      * @returns Either a complete response or a response stream depending on the streaming option
      */
     _invoke<I extends Message, O extends Message>(agent: string, input: string | I, options?: AIGNEHTTPClientInvokeOptions): Promise<AgentResponse<O>>;
-    /**
-     * Enhanced fetch method that handles error responses from the AIGNE server.
-     * This method wraps the standard fetch API to provide better error handling and reporting.
-     *
-     * @param args - Standard fetch API arguments (url and options)
-     * @returns A Response object if the request was successful
-     * @throws Error with detailed information if the request failed
-     *
-     * @private
-     */
-    fetch(...args: Parameters<typeof globalThis.fetch>): Promise<Response>;
     getAgent<I extends Message = Message, O extends Message = Message>(options: ClientAgentOptions<I, O>): Promise<ClientAgent<I, O>>;
 }

package/lib/dts/http-server/server.d.ts

@@ -1,5 +1,6 @@
 import { IncomingMessage, ServerResponse } from "node:http";
 import type { AIGNE, InvokeOptions, UserContext } from "@aigne/core";
+import type { PromiseOrValue } from "@aigne/core/utils/type-utils.js";
 import { z } from "zod";
 /**
  * Schema for validating agent invocation payloads.
@@ -26,55 +27,55 @@ export declare const invokePayloadSchema: z.ZodObject<{
             content: object;
         }[] | null | undefined>;
     }, "strip", z.ZodTypeAny, {
-        memories?: {
-            content: object;
-        }[] | undefined;
         streaming?: boolean | undefined;
         returnProgressChunks?: boolean | undefined;
         userContext?: Record<string, unknown> | undefined;
-    }, {
         memories?: {
             content: object;
-        }[] | null | undefined;
+        }[] | undefined;
+    }, {
         streaming?: boolean | null | undefined;
         returnProgressChunks?: boolean | null | undefined;
         userContext?: Record<string, unknown> | null | undefined;
-    }>>>, {
         memories?: {
             content: object;
-        }[] | undefined;
+        }[] | null | undefined;
+    }>>>, {
         streaming?: boolean | undefined;
         returnProgressChunks?: boolean | undefined;
         userContext?: Record<string, unknown> | undefined;
-    } | undefined, {
         memories?: {
             content: object;
-        }[] | null | undefined;
+        }[] | undefined;
+    } | undefined, {
         streaming?: boolean | null | undefined;
         returnProgressChunks?: boolean | null | undefined;
         userContext?: Record<string, unknown> | null | undefined;
+        memories?: {
+            content: object;
+        }[] | null | undefined;
     } | null | undefined>;
 }, "strip", z.ZodTypeAny, {
-    agent: string;
     input: Record<string, unknown>;
+    agent: string;
     options?: {
-        memories?: {
-            content: object;
-        }[] | undefined;
         streaming?: boolean | undefined;
         returnProgressChunks?: boolean | undefined;
         userContext?: Record<string, unknown> | undefined;
+        memories?: {
+            content: object;
+        }[] | undefined;
     } | undefined;
 }, {
-    agent: string;
     input: Record<string, unknown>;
+    agent: string;
     options?: {
-        memories?: {
-            content: object;
-        }[] | null | undefined;
         streaming?: boolean | null | undefined;
         returnProgressChunks?: boolean | null | undefined;
         userContext?: Record<string, unknown> | null | undefined;
+        memories?: {
+            content: object;
+        }[] | null | undefined;
     } | null | undefined;
 }>;
 /**
@@ -94,6 +95,7 @@ export interface AIGNEHTTPServerOptions {
     maximumBodySize?: string;
 }
 export interface AIGNEHTTPServerInvokeOptions<U extends UserContext = UserContext> extends Pick<InvokeOptions<U>, "returnProgressChunks" | "userContext" | "memories"> {
+    callback?: (result: Record<string, unknown>) => PromiseOrValue<void>;
 }
 /**
  * AIGNEHTTPServer provides HTTP API access to AIGNE capabilities.

package/lib/esm/http-client/base-client.d.ts

@@ -0,0 +1,92 @@
+import type { AgentResponse, AgentResponseStream, ChatModelOptions, InvokeOptions, Message } from "@aigne/core";
+import type { OpenAIChatModelOptions } from "@aigne/openai";
+/**
+ * Options for invoking an agent through the BaseClient.
+ * Extends the standard AgentInvokeOptions with client-specific options.
+ */
+export interface BaseClientInvokeOptions extends InvokeOptions {
+    /**
+     * Additional fetch API options to customize the HTTP request.
+     * These options will be merged with the default options used by the client.
+     */
+    fetchOptions?: Partial<RequestInit>;
+}
+export interface BaseClientOptions {
+    url: string;
+    accessKey?: string;
+    model?: string;
+    modelOptions?: ChatModelOptions;
+    clientOptions?: OpenAIChatModelOptions["clientOptions"];
+}
+/**
+ * Http client for interacting with a remote AIGNE server.
+ * BaseClient provides a client-side interface that matches the AIGNE API,
+ * allowing applications to invoke agents and receive responses from a remote AIGNE instance.
+ *
+ * @example
+ * Here's a simple example of how to use AIGNEClient:
+ * {@includeCode ../../test/http-client/http-client.test.ts#example-aigne-client-simple}
+ *
+ * @example
+ * Here's an example of how to use AIGNEClient with streaming response:
+ * {@includeCode ../../test/http-client/http-client.test.ts#example-aigne-client-streaming}
+ */
+export declare class BaseClient {
+    options: BaseClientOptions;
+    /**
+     * Creates a new AIGNEClient instance.
+     *
+     * @param options - Configuration options for connecting to the AIGNE server
+     */
+    constructor(options: BaseClientOptions);
+    /**
+     * Invokes an agent in non-streaming mode and returns the complete response.
+     *
+     * @param agent - Name of the agent to invoke
+     * @param input - Input message for the agent
+     * @param options - Options with streaming mode explicitly set to false or omitted
+     * @returns The complete agent response
+     *
+     * @example
+     * Here's a simple example of how to use AIGNEClient:
+     * {@includeCode ../../test/http-client/http-client.test.ts#example-aigne-client-simple}
+     */
+    __invoke<I extends Message, O extends Message>(agent?: string, input?: string | I, options?: BaseClientInvokeOptions & {
+        streaming?: false;
+    }): Promise<O>;
+    /**
+     * Invokes an agent with streaming mode enabled and returns a stream of response chunks.
+     *
+     * @param agent - Name of the agent to invoke
+     * @param input - Input message for the agent
+     * @param options - Options with streaming mode explicitly set to true
+     * @returns A stream of agent response chunks
+     *
+     * @example
+     * Here's an example of how to use AIGNEClient with streaming response:
+     * {@includeCode ../../test/http-client/http-client.test.ts#example-aigne-client-streaming}
+     */
+    __invoke<I extends Message, O extends Message>(agent?: string, input?: string | I, options?: BaseClientInvokeOptions & {
+        streaming: true;
+    }): Promise<AgentResponseStream<O>>;
+    /**
+     * Invokes an agent with the given input and options.
+     *
+     * @param agent - Name of the agent to invoke
+     * @param input - Input message for the agent
+     * @param options - Options for the invocation
+     * @returns Either a complete response or a response stream depending on the streaming option
+     */
+    __invoke<I extends Message, O extends Message>(agent?: string, input?: string | I, options?: BaseClientInvokeOptions): Promise<AgentResponse<O>>;
+    /**
+     * Enhanced fetch method that handles error responses from the AIGNE server.
+     * This method wraps the standard fetch API to provide better error handling and reporting.
+     *
+     * @param args - Standard fetch API arguments (url and options)
+     * @returns A Response object if the request was successful
+     * @throws Error with detailed information if the request failed
+     *
+     * @private
+     */
+    fetch(...args: Parameters<typeof globalThis.fetch>): Promise<Response>;
+}

package/lib/esm/http-client/base-client.js

@@ -0,0 +1,97 @@
+import { AgentResponseStreamParser, EventStreamParser } from "@aigne/core/utils/event-stream.js";
+import { omit, tryOrThrow } from "@aigne/core/utils/type-utils.js";
+import { ChatModelName } from "../constants.js";
+/**
+ * Http client for interacting with a remote AIGNE server.
+ * BaseClient provides a client-side interface that matches the AIGNE API,
+ * allowing applications to invoke agents and receive responses from a remote AIGNE instance.
+ *
+ * @example
+ * Here's a simple example of how to use AIGNEClient:
+ * {@includeCode ../../test/http-client/http-client.test.ts#example-aigne-client-simple}
+ *
+ * @example
+ * Here's an example of how to use AIGNEClient with streaming response:
+ * {@includeCode ../../test/http-client/http-client.test.ts#example-aigne-client-streaming}
+ */
+export class BaseClient {
+    options;
+    /**
+     * Creates a new AIGNEClient instance.
+     *
+     * @param options - Configuration options for connecting to the AIGNE server
+     */
+    constructor(options) {
+        this.options = options;
+    }
+    async __invoke(agent, input, options) {
+        const model = this.options.modelOptions?.model ?? this.options.model;
+        const headers = {
+            "Content-Type": "application/json",
+            ...options?.fetchOptions?.headers,
+        };
+        if (this.options?.accessKey) {
+            headers["Authorization"] = `Bearer ${this.options.accessKey}`;
+        }
+        const body = {
+            model,
+            input,
+            agent: agent ?? ChatModelName,
+            options: options && {
+                ...omit(options, "context"),
+                userContext: { ...options.userContext },
+                memories: [...(options.memories ?? [])],
+                modelOptions: this.options.modelOptions,
+                clientOptions: this.options.clientOptions,
+            },
+        };
+        const response = await this.fetch(this.options.url, {
+            ...options?.fetchOptions,
+            method: "POST",
+            headers: headers,
+            body: JSON.stringify(body),
+        });
+        // For non-streaming responses, simply parse the JSON response and return it
+        if (!options?.streaming) {
+            return await response.json();
+        }
+        // For streaming responses, set up the streaming pipeline
+        const stream = response.body;
+        if (!stream)
+            throw new Error("Response body is not a stream");
+        // Process the stream through a series of transforms:
+        // 1. Convert bytes to text
+        // 2. Parse SSE format into structured events
+        // 3. Convert events into a standardized agent response stream
+        return stream
+            .pipeThrough(new TextDecoderStream())
+            .pipeThrough(new EventStreamParser())
+            .pipeThrough(new AgentResponseStreamParser());
+    }
+    /**
+     * Enhanced fetch method that handles error responses from the AIGNE server.
+     * This method wraps the standard fetch API to provide better error handling and reporting.
+     *
+     * @param args - Standard fetch API arguments (url and options)
+     * @returns A Response object if the request was successful
+     * @throws Error with detailed information if the request failed
+     *
+     * @private
+     */
+    async fetch(...args) {
+        const result = await globalThis.fetch(...args);
+        if (!result.ok) {
+            let message;
+            try {
+                const text = await result.text();
+                const json = tryOrThrow(() => JSON.parse(text));
+                message = json?.error?.message || text;
+            }
+            catch {
+                // ignore
+            }
+            throw new Error(`Failed to fetch url ${args[0]} with status ${result.status}: ${message}`);
+        }
+        return result;
+    }
+}

package/lib/esm/http-client/client.d.ts

@@ -1,27 +1,18 @@
 import { type Agent, type AgentResponse, type AgentResponseStream, type Context, type ContextEmitEventMap, type ContextEventMap, type ContextUsage, type InvokeOptions, type Message, type MessagePayload, type MessageQueueListener, type Unsubscribe, type UserAgent, type UserContext } from "@aigne/core";
 import type { Args, Listener } from "@aigne/core/utils/typed-event-emitter.js";
+import { BaseClient, type BaseClientInvokeOptions, type BaseClientOptions } from "./base-client.js";
 import { ClientAgent, type ClientAgentOptions } from "./client-agent.js";
 import { ClientChatModel } from "./client-chat-model.js";
 /**
  * Configuration options for the AIGNEHTTPClient.
  */
-export interface AIGNEHTTPClientOptions {
-    /**
-     * The URL of the AIGNE server to connect to.
-     * This should point to the base endpoint where the AIGNEServer is hosted.
-     */
-    url: string;
+export interface AIGNEHTTPClientOptions extends BaseClientOptions {
 }
 /**
  * Options for invoking an agent through the AIGNEHTTPClient.
  * Extends the standard AgentInvokeOptions with client-specific options.
  */
-export interface AIGNEHTTPClientInvokeOptions extends InvokeOptions {
-    /**
-     * Additional fetch API options to customize the HTTP request.
-     * These options will be merged with the default options used by the client.
-     */
-    fetchOptions?: Partial<RequestInit>;
+export interface AIGNEHTTPClientInvokeOptions extends BaseClientInvokeOptions {
 }
 /**
  * Http client for interacting with a remote AIGNE server.
@@ -36,7 +27,7 @@ export interface AIGNEHTTPClientInvokeOptions extends InvokeOptions {
  * Here's an example of how to use AIGNEClient with streaming response:
  * {@includeCode ../../test/http-client/http-client.test.ts#example-aigne-client-streaming}
  */
-export declare class AIGNEHTTPClient<U extends UserContext = UserContext> implements Context<U> {
+export declare class AIGNEHTTPClient<U extends UserContext = UserContext> extends BaseClient implements Context<U> {
     options: AIGNEHTTPClientOptions;
     /**
      * Creates a new AIGNEClient instance.
@@ -123,16 +114,5 @@ export declare class AIGNEHTTPClient<U extends UserContext = UserContext> implem
      * @returns Either a complete response or a response stream depending on the streaming option
      */
     _invoke<I extends Message, O extends Message>(agent: string, input: string | I, options?: AIGNEHTTPClientInvokeOptions): Promise<AgentResponse<O>>;
-    /**
-     * Enhanced fetch method that handles error responses from the AIGNE server.
-     * This method wraps the standard fetch API to provide better error handling and reporting.
-     *
-     * @param args - Standard fetch API arguments (url and options)
-     * @returns A Response object if the request was successful
-     * @throws Error with detailed information if the request failed
-     *
-     * @private
-     */
-    fetch(...args: Parameters<typeof globalThis.fetch>): Promise<Response>;
     getAgent<I extends Message = Message, O extends Message = Message>(options: ClientAgentOptions<I, O>): Promise<ClientAgent<I, O>>;
 }

package/lib/esm/http-client/client.js

@@ -1,7 +1,7 @@
 import { newEmptyContextUsage, } from "@aigne/core";
-import { AgentResponseStreamParser, EventStreamParser } from "@aigne/core/utils/event-stream.js";
-import { omit, tryOrThrow } from "@aigne/core/utils/type-utils.js";
+import { omit } from "@aigne/core/utils/type-utils.js";
 import { v7 } from "uuid";
+import { BaseClient } from "./base-client.js";
 import { ClientAgent } from "./client-agent.js";
 import { ClientChatModel } from "./client-chat-model.js";
 /**
@@ -17,7 +17,7 @@ import { ClientChatModel } from "./client-chat-model.js";
  * Here's an example of how to use AIGNEClient with streaming response:
  * {@includeCode ../../test/http-client/http-client.test.ts#example-aigne-client-streaming}
  */
-export class AIGNEHTTPClient {
+export class AIGNEHTTPClient extends BaseClient {
     options;
     /**
      * Creates a new AIGNEClient instance.
@@ -25,6 +25,7 @@ export class AIGNEHTTPClient {
      * @param options - Configuration options for connecting to the AIGNE server
      */
     constructor(options) {
+        super(options);
         this.options = options;
     }
     id = v7();
@@ -78,67 +79,12 @@ export class AIGNEHTTPClient {
         console.error("Method not implemented.");
         return this;
     }
-    async _invoke(agent, input, options) {
-        // Send the agent invocation request to the AIGNE server
-        const response = await this.fetch(this.options.url, {
-            ...options?.fetchOptions,
-            method: "POST",
-            headers: {
-                "Content-Type": "application/json",
-                ...options?.fetchOptions?.headers,
-            },
-            body: JSON.stringify({
-                agent,
-                input,
-                options: options && {
-                    ...omit(options, "context"),
-                    userContext: { ...this.userContext, ...options.userContext },
-                    memories: [...this.memories, ...(options.memories ?? [])],
-                },
-            }),
+    async _invoke(agent, input, options = {}) {
+        return this.__invoke(agent, input, {
+            ...omit(options, "context"),
+            userContext: { ...this.userContext, ...options.userContext },
+            memories: [...this.memories, ...(options.memories ?? [])],
         });
-        // For non-streaming responses, simply parse the JSON response and return it
-        if (!options?.streaming) {
-            return await response.json();
-        }
-        // For streaming responses, set up the streaming pipeline
-        const stream = response.body;
-        if (!stream)
-            throw new Error("Response body is not a stream");
-        // Process the stream through a series of transforms:
-        // 1. Convert bytes to text
-        // 2. Parse SSE format into structured events
-        // 3. Convert events into a standardized agent response stream
-        return stream
-            .pipeThrough(new TextDecoderStream())
-            .pipeThrough(new EventStreamParser())
-            .pipeThrough(new AgentResponseStreamParser());
-    }
-    /**
-     * Enhanced fetch method that handles error responses from the AIGNE server.
-     * This method wraps the standard fetch API to provide better error handling and reporting.
-     *
-     * @param args - Standard fetch API arguments (url and options)
-     * @returns A Response object if the request was successful
-     * @throws Error with detailed information if the request failed
-     *
-     * @private
-     */
-    async fetch(...args) {
-        const result = await globalThis.fetch(...args);
-        if (!result.ok) {
-            let message;
-            try {
-                const text = await result.text();
-                const json = tryOrThrow(() => JSON.parse(text));
-                message = json?.error?.message || text;
-            }
-            catch {
-                // ignore
-            }
-            throw new Error(`Failed to fetch url ${args[0]} with status ${result.status}: ${message}`);
-        }
-        return result;
     }
     async getAgent(options) {
         return new ClientAgent(this, options);

package/lib/esm/http-server/server.d.ts

@@ -1,5 +1,6 @@
 import { IncomingMessage, ServerResponse } from "node:http";
 import type { AIGNE, InvokeOptions, UserContext } from "@aigne/core";
+import type { PromiseOrValue } from "@aigne/core/utils/type-utils.js";
 import { z } from "zod";
 /**
  * Schema for validating agent invocation payloads.
@@ -26,55 +27,55 @@ export declare const invokePayloadSchema: z.ZodObject<{
             content: object;
         }[] | null | undefined>;
     }, "strip", z.ZodTypeAny, {
-        memories?: {
-            content: object;
-        }[] | undefined;
         streaming?: boolean | undefined;
         returnProgressChunks?: boolean | undefined;
         userContext?: Record<string, unknown> | undefined;
-    }, {
         memories?: {
             content: object;
-        }[] | null | undefined;
+        }[] | undefined;
+    }, {
         streaming?: boolean | null | undefined;
         returnProgressChunks?: boolean | null | undefined;
         userContext?: Record<string, unknown> | null | undefined;
-    }>>>, {
         memories?: {
             content: object;
-        }[] | undefined;
+        }[] | null | undefined;
+    }>>>, {
         streaming?: boolean | undefined;
         returnProgressChunks?: boolean | undefined;
         userContext?: Record<string, unknown> | undefined;
-    } | undefined, {
         memories?: {
             content: object;
-        }[] | null | undefined;
+        }[] | undefined;
+    } | undefined, {
         streaming?: boolean | null | undefined;
         returnProgressChunks?: boolean | null | undefined;
         userContext?: Record<string, unknown> | null | undefined;
+        memories?: {
+            content: object;
+        }[] | null | undefined;
     } | null | undefined>;
 }, "strip", z.ZodTypeAny, {
-    agent: string;
     input: Record<string, unknown>;
+    agent: string;
     options?: {
-        memories?: {
-            content: object;
-        }[] | undefined;
         streaming?: boolean | undefined;
         returnProgressChunks?: boolean | undefined;
         userContext?: Record<string, unknown> | undefined;
+        memories?: {
+            content: object;
+        }[] | undefined;
     } | undefined;
 }, {
-    agent: string;
     input: Record<string, unknown>;
+    agent: string;
     options?: {
-        memories?: {
-            content: object;
-        }[] | null | undefined;
         streaming?: boolean | null | undefined;
         returnProgressChunks?: boolean | null | undefined;
         userContext?: Record<string, unknown> | null | undefined;
+        memories?: {
+            content: object;
+        }[] | null | undefined;
     } | null | undefined;
 }>;
 /**
@@ -94,6 +95,7 @@ export interface AIGNEHTTPServerOptions {
     maximumBodySize?: string;
 }
 export interface AIGNEHTTPServerInvokeOptions<U extends UserContext = UserContext> extends Pick<InvokeOptions<U>, "returnProgressChunks" | "userContext" | "memories"> {
+    callback?: (result: Record<string, unknown>) => PromiseOrValue<void>;
 }
 /**
  * AIGNEHTTPServer provides HTTP API access to AIGNE capabilities.

package/lib/esm/http-server/server.js

@@ -1,5 +1,6 @@
 import { IncomingMessage, ServerResponse } from "node:http";
 import { AgentResponseStreamSSE } from "@aigne/core/utils/event-stream.js";
+import { onAgentResponseStreamEnd } from "@aigne/core/utils/stream-utils.js";
 import { checkArguments, isRecord, tryOrThrow } from "@aigne/core/utils/type-utils.js";
 import contentType from "content-type";
 import getRawBody from "raw-body";
@@ -76,6 +77,7 @@ export class AIGNEHTTPServer {
         const result = await this._invoke(request, {
             userContext: opts?.userContext,
             memories: opts?.memories,
+            callback: opts?.callback,
         });
         if (response instanceof ServerResponse) {
             await this._writeResponse(result, response);
@@ -108,6 +110,7 @@ export class AIGNEHTTPServer {
             };
             if (!streaming) {
                 const result = await aigne.invoke(agent, input, mergedOptions);
+                options.callback?.(result);
                 return new Response(JSON.stringify(result), {
                     headers: { "Content-Type": "application/json" },
                 });
@@ -117,7 +120,12 @@ export class AIGNEHTTPServer {
                 returnActiveAgent: false,
                 streaming: true,
             });
-            return new Response(new AgentResponseStreamSSE(stream), {
+            const newStream = onAgentResponseStreamEnd(stream, {
+                onResult: async (result) => {
+                    options.callback?.(result);
+                },
+            });
+            return new Response(new AgentResponseStreamSSE(newStream), {
                 headers: {
                     "Content-Type": "text/event-stream",
                     "Cache-Control": "no-cache",

package/lib/esm/package.json

@@ -1 +1,3 @@
-{"type": "module"}
+{
+  "type": "module"
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/transport",
-  "version": "0.9.6",
+  "version": "0.10.2",
   "description": "AIGNE Transport SDK providing HTTP client and server implementations for communication between AIGNE components",
   "publishConfig": {
     "access": "public"
@@ -39,7 +39,7 @@
     }
   },
   "dependencies": {
-    "@aigne/openai": "^0.9.2"
+    "@aigne/openai": "^0.10.2"
   },
   "devDependencies": {
     "@types/bun": "^1.2.18",
@@ -54,9 +54,10 @@
     "rimraf": "^6.0.1",
     "typescript": "^5.8.3",
     "uuid": "^11.1.0",
-    "@aigne/agent-library": "^1.20.5",
-    "@aigne/test-utils": "^0.5.7",
-    "@aigne/core": "^1.33.2"
+    "@aigne/agent-library": "^1.21.2",
+    "@aigne/core": "^1.36.0",
+    "@aigne/default-memory": "^1.0.2",
+    "@aigne/test-utils": "^0.5.10"
   },
   "scripts": {
     "lint": "tsc --noEmit",
@@ -64,6 +65,6 @@
     "clean": "rimraf lib test/coverage",
     "test": "bun test",
     "test:coverage": "bun test --coverage --coverage-reporter=lcov --coverage-reporter=text",
-    "postbuild": "echo '{\"type\": \"module\"}' > lib/esm/package.json && echo '{\"type\": \"commonjs\"}' > lib/cjs/package.json"
+    "postbuild": "node ../../scripts/post-build-lib.mjs"
   }
 }