@aigne/core 1.18.3 → 1.18.4

package/CHANGELOG.md

@@ -5,6 +5,13 @@
 
 * add user context support ([#131](https://github.com/AIGNE-io/aigne-framework/issues/131)) ([4dd9d20](https://github.com/AIGNE-io/aigne-framework/commit/4dd9d20953f6ac33933723db56efd9b44bafeb02))
 
+## [1.18.4](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.18.3...core-v1.18.4) (2025-06-05)
+
+
+### Bug Fixes
+
+* **core:** prioritize returning json chunks ([#151](https://github.com/AIGNE-io/aigne-framework/issues/151)) ([8bf49a1](https://github.com/AIGNE-io/aigne-framework/commit/8bf49a18c083b33d2e0b35e8d0f22f68d9d6effa))
+
 ## [1.18.3](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.18.2...core-v1.18.3) (2025-06-05)
 
 

package/lib/cjs/aigne/context.js

@@ -1,6 +1,10 @@
 "use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AIGNEContext = void 0;
+const fast_deep_equal_1 = __importDefault(require("fast-deep-equal"));
 const strict_event_emitter_1 = require("strict-event-emitter");
 const uuid_1 = require("uuid");
 const zod_1 = require("zod");
@@ -182,7 +186,6 @@ class AIGNEContextShared {
     }
     async *invokeAgent(agent, input, context, options) {
         let activeAgent = agent;
-        let output;
         for (;;) {
             const result = {};
             if (options?.sourceAgent && activeAgent !== options.sourceAgent) {
@@ -195,12 +198,14 @@ class AIGNEContextShared {
             }
             const stream = await activeAgent.invoke(input, { ...options, context, streaming: true });
             for await (const value of stream) {
-                if (value.delta.text) {
-                    yield { delta: { text: value.delta.text } };
-                }
                 if (value.delta.json) {
+                    value.delta.json = omitExistsProperties(result, value.delta.json);
                     Object.assign(result, value.delta.json);
                 }
+                delete value.delta.json?.[types_js_1.transferAgentOutputKey];
+                if ((0, agent_js_1.isEmptyChunk)(value))
+                    continue;
+                yield value;
             }
             if (!options?.disableTransfer) {
                 const transferToAgent = (0, types_js_1.isTransferAgentOutput)(result)
@@ -211,21 +216,22 @@ class AIGNEContextShared {
                     continue;
                 }
             }
-            output = result;
             break;
         }
-        if (!output)
-            throw new Error("Unexpected empty output");
         yield {
             delta: {
-                json: {
-                    ...output,
-                    __activeAgent__: activeAgent,
-                },
+                json: { __activeAgent__: activeAgent },
             },
         };
     }
 }
+function omitExistsProperties(result, { ...delta }) {
+    for (const [key, val] of Object.entries(delta)) {
+        if ((0, fast_deep_equal_1.default)(result[key], val))
+            delete delta[key];
+    }
+    return (0, type_utils_js_1.isEmpty)(delta) ? undefined : delta;
+}
 async function* withAbortSignal(signal, error, fn) {
     const iterator = fn();
     const timeoutPromise = (0, promise_js_1.promiseWithResolvers)();

package/lib/esm/aigne/context.js

@@ -1,13 +1,14 @@
+import equal from "fast-deep-equal";
 import { Emitter } from "strict-event-emitter";
 import { v7 } from "uuid";
 import { z } from "zod";
-import { Agent, } from "../agents/agent.js";
+import { Agent, isEmptyChunk, } from "../agents/agent.js";
 import { isTransferAgentOutput, transferAgentOutputKey, } from "../agents/types.js";
 import { UserAgent } from "../agents/user-agent.js";
 import { createMessage } from "../prompt/prompt-builder.js";
 import { promiseWithResolvers } from "../utils/promise.js";
 import { agentResponseStreamToObject, asyncGeneratorToReadableStream, onAgentResponseStreamEnd, } from "../utils/stream-utils.js";
-import { checkArguments, isNil, omitBy, } from "../utils/type-utils.js";
+import { checkArguments, isEmpty, isNil, omitBy, } from "../utils/type-utils.js";
 import { MessageQueue, toMessagePayload, } from "./message-queue.js";
 import { newEmptyContextUsage } from "./usage.js";
 /**
@@ -178,7 +179,6 @@ class AIGNEContextShared {
     }
     async *invokeAgent(agent, input, context, options) {
         let activeAgent = agent;
-        let output;
         for (;;) {
             const result = {};
             if (options?.sourceAgent && activeAgent !== options.sourceAgent) {
@@ -191,12 +191,14 @@ class AIGNEContextShared {
             }
             const stream = await activeAgent.invoke(input, { ...options, context, streaming: true });
             for await (const value of stream) {
-                if (value.delta.text) {
-                    yield { delta: { text: value.delta.text } };
-                }
                 if (value.delta.json) {
+                    value.delta.json = omitExistsProperties(result, value.delta.json);
                     Object.assign(result, value.delta.json);
                 }
+                delete value.delta.json?.[transferAgentOutputKey];
+                if (isEmptyChunk(value))
+                    continue;
+                yield value;
             }
             if (!options?.disableTransfer) {
                 const transferToAgent = isTransferAgentOutput(result)
@@ -207,21 +209,22 @@ class AIGNEContextShared {
                     continue;
                 }
             }
-            output = result;
             break;
         }
-        if (!output)
-            throw new Error("Unexpected empty output");
         yield {
             delta: {
-                json: {
-                    ...output,
-                    __activeAgent__: activeAgent,
-                },
+                json: { __activeAgent__: activeAgent },
             },
         };
     }
 }
+function omitExistsProperties(result, { ...delta }) {
+    for (const [key, val] of Object.entries(delta)) {
+        if (equal(result[key], val))
+            delete delta[key];
+    }
+    return isEmpty(delta) ? undefined : delta;
+}
 async function* withAbortSignal(signal, error, fn) {
     const iterator = fn();
     const timeoutPromise = promiseWithResolvers();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/core",
-  "version": "1.18.3",
+  "version": "1.18.4",
   "description": "AIGNE core library for building AI-powered applications",
   "publishConfig": {
     "access": "public"