@umsai/ums-code 0.1.4-v2 → 0.3.0-v2

package/dist/src/nonInteractive/io/StreamJsonInputReader.d.ts

@@ -0,0 +1,16 @@
+/**
+ * @license
+ * Copyright 2025 Qwen Team
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Readable } from 'node:stream';
+import type { CLIControlRequest, CLIControlResponse, CLIMessage, ControlCancelRequest } from '../types.js';
+export type StreamJsonInputMessage = CLIMessage | CLIControlRequest | CLIControlResponse | ControlCancelRequest;
+export declare class StreamJsonParseError extends Error {
+}
+export declare class StreamJsonInputReader {
+    private readonly input;
+    constructor(input?: Readable);
+    read(): AsyncGenerator<StreamJsonInputMessage>;
+    private parse;
+}

package/dist/src/nonInteractive/io/StreamJsonInputReader.js

@@ -0,0 +1,54 @@
+/**
+ * @license
+ * Copyright 2025 Qwen Team
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { createInterface } from 'node:readline/promises';
+import process from 'node:process';
+export class StreamJsonParseError extends Error {
+}
+export class StreamJsonInputReader {
+    input;
+    constructor(input = process.stdin) {
+        this.input = input;
+    }
+    async *read() {
+        const rl = createInterface({
+            input: this.input,
+            crlfDelay: Number.POSITIVE_INFINITY,
+            terminal: false,
+        });
+        try {
+            for await (const rawLine of rl) {
+                const line = rawLine.trim();
+                if (!line) {
+                    continue;
+                }
+                yield this.parse(line);
+            }
+        }
+        finally {
+            rl.close();
+        }
+    }
+    parse(line) {
+        try {
+            const parsed = JSON.parse(line);
+            if (!parsed || typeof parsed !== 'object') {
+                throw new StreamJsonParseError('Parsed value is not an object');
+            }
+            if (!('type' in parsed) || typeof parsed.type !== 'string') {
+                throw new StreamJsonParseError('Missing required "type" field');
+            }
+            return parsed;
+        }
+        catch (error) {
+            if (error instanceof StreamJsonParseError) {
+                throw error;
+            }
+            const reason = error instanceof Error ? error.message : String(error);
+            throw new StreamJsonParseError(`Failed to parse stream-json line: ${reason}`);
+        }
+    }
+}
+//# sourceMappingURL=StreamJsonInputReader.js.map

package/dist/src/nonInteractive/io/StreamJsonInputReader.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"StreamJsonInputReader.js","sourceRoot":"","sources":["../../../../src/nonInteractive/io/StreamJsonInputReader.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAEzD,OAAO,OAAO,MAAM,cAAc,CAAC;AAcnC,MAAM,OAAO,oBAAqB,SAAQ,KAAK;CAAG;AAElD,MAAM,OAAO,qBAAqB;IACf,KAAK,CAAW;IAEjC,YAAY,QAAkB,OAAO,CAAC,KAAK;QACzC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,CAAC,IAAI;QACT,MAAM,EAAE,GAAG,eAAe,CAAC;YACzB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,SAAS,EAAE,MAAM,CAAC,iBAAiB;YACnC,QAAQ,EAAE,KAAK;SAChB,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,IAAI,KAAK,EAAE,MAAM,OAAO,IAAI,EAAE,EAAE,CAAC;gBAC/B,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;gBAC5B,IAAI,CAAC,IAAI,EAAE,CAAC;oBACV,SAAS;gBACX,CAAC;gBAED,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACzB,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,EAAE,CAAC,KAAK,EAAE,CAAC;QACb,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,IAAY;QACxB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAA2B,CAAC;YAC1D,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC1C,MAAM,IAAI,oBAAoB,CAAC,+BAA+B,CAAC,CAAC;YAClE,CAAC;YACD,IAAI,CAAC,CAAC,MAAM,IAAI,MAAM,CAAC,IAAI,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC3D,MAAM,IAAI,oBAAoB,CAAC,+BAA+B,CAAC,CAAC;YAClE,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,oBAAoB,EAAE,CAAC;gBAC1C,MAAM,KAAK,CAAC;YACd,CAAC;YACD,MAAM,MAAM,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACtE,MAAM,IAAI,oBAAoB,CAC5B,qCAAqC,MAAM,EAAE,CAC9C,CAAC;QACJ,CAAC;IACH,CAAC;CACF"}

package/dist/src/nonInteractive/io/StreamJsonInputReader.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Qwen Team
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

package/dist/src/nonInteractive/io/StreamJsonInputReader.test.js

@@ -0,0 +1,178 @@
+/**
+ * @license
+ * Copyright 2025 Qwen Team
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { PassThrough } from 'node:stream';
+import { afterEach, describe, expect, it, vi } from 'vitest';
+import { StreamJsonInputReader, StreamJsonParseError, } from './StreamJsonInputReader.js';
+describe('StreamJsonInputReader', () => {
+    afterEach(() => {
+        vi.restoreAllMocks();
+    });
+    describe('read', () => {
+        /**
+         * Test parsing all supported message types in a single test
+         */
+        it('should parse valid messages of all types', async () => {
+            const input = new PassThrough();
+            const reader = new StreamJsonInputReader(input);
+            const messages = [
+                {
+                    type: 'user',
+                    session_id: 'test-session',
+                    message: {
+                        role: 'user',
+                        content: [{ type: 'text', text: 'hello world' }],
+                    },
+                    parent_tool_use_id: null,
+                },
+                {
+                    type: 'control_request',
+                    request_id: 'req-1',
+                    request: { subtype: 'initialize' },
+                },
+                {
+                    type: 'control_response',
+                    response: {
+                        subtype: 'success',
+                        request_id: 'req-1',
+                        response: { initialized: true },
+                    },
+                },
+                {
+                    type: 'control_cancel_request',
+                    request_id: 'req-1',
+                },
+            ];
+            for (const msg of messages) {
+                input.write(JSON.stringify(msg) + '\n');
+            }
+            input.end();
+            const parsed = [];
+            for await (const msg of reader.read()) {
+                parsed.push(msg);
+            }
+            expect(parsed).toHaveLength(messages.length);
+            expect(parsed).toEqual(messages);
+        });
+        it('should parse multiple messages', async () => {
+            const input = new PassThrough();
+            const reader = new StreamJsonInputReader(input);
+            const message1 = {
+                type: 'control_request',
+                request_id: 'req-1',
+                request: { subtype: 'initialize' },
+            };
+            const message2 = {
+                type: 'user',
+                session_id: 'test-session',
+                message: {
+                    role: 'user',
+                    content: [{ type: 'text', text: 'hello' }],
+                },
+                parent_tool_use_id: null,
+            };
+            input.write(JSON.stringify(message1) + '\n');
+            input.write(JSON.stringify(message2) + '\n');
+            input.end();
+            const messages = [];
+            for await (const msg of reader.read()) {
+                messages.push(msg);
+            }
+            expect(messages).toHaveLength(2);
+            expect(messages[0]).toEqual(message1);
+            expect(messages[1]).toEqual(message2);
+        });
+        it('should skip empty lines and trim whitespace', async () => {
+            const input = new PassThrough();
+            const reader = new StreamJsonInputReader(input);
+            const message = {
+                type: 'user',
+                session_id: 'test-session',
+                message: {
+                    role: 'user',
+                    content: [{ type: 'text', text: 'hello' }],
+                },
+                parent_tool_use_id: null,
+            };
+            input.write('\n');
+            input.write('  ' + JSON.stringify(message) + '  \n');
+            input.write('  \n');
+            input.write('\t\n');
+            input.end();
+            const messages = [];
+            for await (const msg of reader.read()) {
+                messages.push(msg);
+            }
+            expect(messages).toHaveLength(1);
+            expect(messages[0]).toEqual(message);
+        });
+        /**
+         * Consolidated error handling test cases
+         */
+        it.each([
+            {
+                name: 'invalid JSON',
+                input: '{"invalid": json}\n',
+                expectedError: 'Failed to parse stream-json line',
+            },
+            {
+                name: 'missing type field',
+                input: JSON.stringify({ session_id: 'test-session', message: 'hello' }) +
+                    '\n',
+                expectedError: 'Missing required "type" field',
+            },
+            {
+                name: 'non-object value (string)',
+                input: '"just a string"\n',
+                expectedError: 'Parsed value is not an object',
+            },
+            {
+                name: 'non-object value (null)',
+                input: 'null\n',
+                expectedError: 'Parsed value is not an object',
+            },
+            {
+                name: 'array value',
+                input: '[1, 2, 3]\n',
+                expectedError: 'Missing required "type" field',
+            },
+            {
+                name: 'type field not a string',
+                input: JSON.stringify({ type: 123, session_id: 'test-session' }) + '\n',
+                expectedError: 'Missing required "type" field',
+            },
+        ])('should throw StreamJsonParseError for $name', async ({ input: inputLine, expectedError }) => {
+            const input = new PassThrough();
+            const reader = new StreamJsonInputReader(input);
+            input.write(inputLine);
+            input.end();
+            const messages = [];
+            let error;
+            try {
+                for await (const msg of reader.read()) {
+                    messages.push(msg);
+                }
+            }
+            catch (e) {
+                error = e;
+            }
+            expect(messages).toHaveLength(0);
+            expect(error).toBeInstanceOf(StreamJsonParseError);
+            expect(error.message).toContain(expectedError);
+        });
+        it('should use process.stdin as default input', () => {
+            const reader = new StreamJsonInputReader();
+            // Access private field for testing constructor default parameter
+            expect(reader.input).toBe(process.stdin);
+        });
+        it('should use provided input stream', () => {
+            const customInput = new PassThrough();
+            const reader = new StreamJsonInputReader(customInput);
+            // Access private field for testing constructor parameter
+            expect(reader.input).toBe(customInput);
+        });
+    });
+});
+//# sourceMappingURL=StreamJsonInputReader.test.js.map

package/dist/src/nonInteractive/io/StreamJsonInputReader.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"StreamJsonInputReader.test.js","sourceRoot":"","sources":["../../../../src/nonInteractive/io/StreamJsonInputReader.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC7D,OAAO,EACL,qBAAqB,EACrB,oBAAoB,GAErB,MAAM,4BAA4B,CAAC;AAEpC,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE;QACpB;;WAEG;QACH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,KAAK,GAAG,IAAI,WAAW,EAAE,CAAC;YAChC,MAAM,MAAM,GAAG,IAAI,qBAAqB,CAAC,KAAK,CAAC,CAAC;YAEhD,MAAM,QAAQ,GAAG;gBACf;oBACE,IAAI,EAAE,MAAM;oBACZ,UAAU,EAAE,cAAc;oBAC1B,OAAO,EAAE;wBACP,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;qBACjD;oBACD,kBAAkB,EAAE,IAAI;iBACzB;gBACD;oBACE,IAAI,EAAE,iBAAiB;oBACvB,UAAU,EAAE,OAAO;oBACnB,OAAO,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE;iBACnC;gBACD;oBACE,IAAI,EAAE,kBAAkB;oBACxB,QAAQ,EAAE;wBACR,OAAO,EAAE,SAAS;wBAClB,UAAU,EAAE,OAAO;wBACnB,QAAQ,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE;qBAChC;iBACF;gBACD;oBACE,IAAI,EAAE,wBAAwB;oBAC9B,UAAU,EAAE,OAAO;iBACpB;aACF,CAAC;YAEF,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;gBAC3B,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;YAC1C,CAAC;YACD,KAAK,CAAC,GAAG,EAAE,CAAC;YAEZ,MAAM,MAAM,GAA6B,EAAE,CAAC;YAC5C,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;gBACtC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACnB,CAAC;YAED,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,KAAK,GAAG,IAAI,WAAW,EAAE,CAAC;YAChC,MAAM,MAAM,GAAG,IAAI,qBAAqB,CAAC,KAAK,CAAC,CAAC;YAEhD,MAAM,QAAQ,GAAG;gBACf,IAAI,EAAE,iBAAiB;gBACvB,UAAU,EAAE,OAAO;gBACnB,OAAO,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE;aACnC,CAAC;YAEF,MAAM,QAAQ,GAAG;gBACf,IAAI,EAAE,MAAM;gBACZ,UAAU,EAAE,cAAc;gBAC1B,OAAO,EAAE;oBACP,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;iBAC3C;gBACD,kBAAkB,EAAE,IAAI;aACzB,CAAC;YAEF,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC,CAAC;YAC7C,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC,CAAC;YAC7C,KAAK,CAAC,GAAG,EAAE,CAAC;YAEZ,MAAM,QAAQ,GAA6B,EAAE,CAAC;YAC9C,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;gBACtC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACrB,CAAC;YAED,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACjC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YACtC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,KAAK,GAAG,IAAI,WAAW,EAAE,CAAC;YAChC,MAAM,MAAM,GAAG,IAAI,qBAAqB,CAAC,KAAK,CAAC,CAAC;YAEhD,MAAM,OAAO,GAAG;gBACd,IAAI,EAAE,MAAM;gBACZ,UAAU,EAAE,cAAc;gBAC1B,OAAO,EAAE;oBACP,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;iBAC3C;gBACD,kBAAkB,EAAE,IAAI;aACzB,CAAC;YAEF,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAClB,KAAK,CAAC,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,MAAM,CAAC,CAAC;YACrD,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACpB,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACpB,KAAK,CAAC,GAAG,EAAE,CAAC;YAEZ,MAAM,QAAQ,GAA6B,EAAE,CAAC;YAC9C,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;gBACtC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACrB,CAAC;YAED,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACjC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH;;WAEG;QACH,EAAE,CAAC,IAAI,CAAC;YACN;gBACE,IAAI,EAAE,cAAc;gBACpB,KAAK,EAAE,qBAAqB;gBAC5B,aAAa,EAAE,kCAAkC;aAClD;YACD;gBACE,IAAI,EAAE,oBAAoB;gBAC1B,KAAK,EACH,IAAI,CAAC,SAAS,CAAC,EAAE,UAAU,EAAE,cAAc,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;oBAChE,IAAI;gBACN,aAAa,EAAE,+BAA+B;aAC/C;YACD;gBACE,IAAI,EAAE,2BAA2B;gBACjC,KAAK,EAAE,mBAAmB;gBAC1B,aAAa,EAAE,+BAA+B;aAC/C;YACD;gBACE,IAAI,EAAE,yBAAyB;gBAC/B,KAAK,EAAE,QAAQ;gBACf,aAAa,EAAE,+BAA+B;aAC/C;YACD;gBACE,IAAI,EAAE,aAAa;gBACnB,KAAK,EAAE,aAAa;gBACpB,aAAa,EAAE,+BAA+B;aAC/C;YACD;gBACE,IAAI,EAAE,yBAAyB;gBAC/B,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,UAAU,EAAE,cAAc,EAAE,CAAC,GAAG,IAAI;gBACvE,aAAa,EAAE,+BAA+B;aAC/C;SACF,CAAC,CACA,6CAA6C,EAC7C,KAAK,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,aAAa,EAAE,EAAE,EAAE;YAC5C,MAAM,KAAK,GAAG,IAAI,WAAW,EAAE,CAAC;YAChC,MAAM,MAAM,GAAG,IAAI,qBAAqB,CAAC,KAAK,CAAC,CAAC;YAEhD,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;YACvB,KAAK,CAAC,GAAG,EAAE,CAAC;YAEZ,MAAM,QAAQ,GAA6B,EAAE,CAAC;YAC9C,IAAI,KAAc,CAAC;YAEnB,IAAI,CAAC;gBACH,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;oBACtC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACrB,CAAC;YACH,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,KAAK,GAAG,CAAC,CAAC;YACZ,CAAC;YAED,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACjC,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC;YACnD,MAAM,CAAE,KAA8B,CAAC,OAAO,CAAC,CAAC,SAAS,CACvD,aAAa,CACd,CAAC;QACJ,CAAC,CACF,CAAC;QAEF,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,MAAM,GAAG,IAAI,qBAAqB,EAAE,CAAC;YAC3C,iEAAiE;YACjE,MAAM,CAAE,MAAqD,CAAC,KAAK,CAAC,CAAC,IAAI,CACvE,OAAO,CAAC,KAAK,CACd,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,WAAW,GAAG,IAAI,WAAW,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,IAAI,qBAAqB,CAAC,WAAW,CAAC,CAAC;YACtD,yDAAyD;YACzD,MAAM,CAAE,MAAmD,CAAC,KAAK,CAAC,CAAC,IAAI,CACrE,WAAW,CACZ,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/nonInteractive/io/StreamJsonOutputAdapter.d.ts

@@ -0,0 +1,69 @@
+/**
+ * @license
+ * Copyright 2025 Qwen Team
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Config } from '@umsai/ums-code-core';
+import type { CLIAssistantMessage, CLIMessage, ControlMessage, TextBlock, ThinkingBlock, ToolUseBlock } from '../types.js';
+import { BaseJsonOutputAdapter, type MessageState, type ResultOptions, type JsonOutputAdapterInterface } from './BaseJsonOutputAdapter.js';
+/**
+ * Stream JSON output adapter that emits messages immediately
+ * as they are completed during the streaming process.
+ * Supports both main agent and subagent messages through distinct APIs.
+ */
+export declare class StreamJsonOutputAdapter extends BaseJsonOutputAdapter implements JsonOutputAdapterInterface {
+    private readonly includePartialMessages;
+    constructor(config: Config, includePartialMessages: boolean);
+    /**
+     * Emits message immediately to stdout (stream mode).
+     */
+    protected emitMessageImpl(message: CLIMessage | ControlMessage): void;
+    /**
+     * Stream mode emits stream events when includePartialMessages is enabled.
+     */
+    protected shouldEmitStreamEvents(): boolean;
+    finalizeAssistantMessage(): CLIAssistantMessage;
+    emitResult(options: ResultOptions): void;
+    emitMessage(message: CLIMessage | ControlMessage): void;
+    send(message: CLIMessage | ControlMessage): void;
+    /**
+     * Overrides base class hook to emit stream event when text block is created.
+     */
+    protected onTextBlockCreated(state: MessageState, index: number, block: TextBlock, parentToolUseId: string | null): void;
+    /**
+     * Overrides base class hook to emit stream event when text is appended.
+     */
+    protected onTextAppended(state: MessageState, index: number, fragment: string, parentToolUseId: string | null): void;
+    /**
+     * Overrides base class hook to emit stream event when thinking block is created.
+     */
+    protected onThinkingBlockCreated(state: MessageState, index: number, block: ThinkingBlock, parentToolUseId: string | null): void;
+    /**
+     * Overrides base class hook to emit stream event when thinking is appended.
+     */
+    protected onThinkingAppended(state: MessageState, index: number, fragment: string, parentToolUseId: string | null): void;
+    /**
+     * Overrides base class hook to emit stream event when tool_use block is created.
+     */
+    protected onToolUseBlockCreated(state: MessageState, index: number, block: ToolUseBlock, parentToolUseId: string | null): void;
+    /**
+     * Overrides base class hook to emit stream event when tool_use input is set.
+     */
+    protected onToolUseInputSet(state: MessageState, index: number, input: unknown, parentToolUseId: string | null): void;
+    /**
+     * Overrides base class hook to emit stream event when block is closed.
+     */
+    protected onBlockClosed(state: MessageState, index: number, parentToolUseId: string | null): void;
+    /**
+     * Overrides base class hook to emit message_start event when message is started.
+     * Only emits for main agent, not for subagents.
+     */
+    protected onEnsureMessageStarted(state: MessageState, parentToolUseId: string | null): void;
+    /**
+     * Emits stream events when partial messages are enabled.
+     * This is a private method specific to StreamJsonOutputAdapter.
+     * @param event - Stream event to emit
+     * @param parentToolUseId - null for main agent, string for subagent
+     */
+    private emitStreamEventIfEnabled;
+}

package/dist/src/nonInteractive/io/StreamJsonOutputAdapter.js

@@ -0,0 +1,185 @@
+/**
+ * @license
+ * Copyright 2025 Qwen Team
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { randomUUID } from 'node:crypto';
+import { BaseJsonOutputAdapter, } from './BaseJsonOutputAdapter.js';
+/**
+ * Stream JSON output adapter that emits messages immediately
+ * as they are completed during the streaming process.
+ * Supports both main agent and subagent messages through distinct APIs.
+ */
+export class StreamJsonOutputAdapter extends BaseJsonOutputAdapter {
+    includePartialMessages;
+    constructor(config, includePartialMessages) {
+        super(config);
+        this.includePartialMessages = includePartialMessages;
+    }
+    /**
+     * Emits message immediately to stdout (stream mode).
+     */
+    emitMessageImpl(message) {
+        // Track assistant messages for result generation
+        if (typeof message === 'object' &&
+            message !== null &&
+            'type' in message &&
+            message.type === 'assistant') {
+            this.updateLastAssistantMessage(message);
+        }
+        // Emit messages immediately in stream mode
+        process.stdout.write(`${JSON.stringify(message)}\n`);
+    }
+    /**
+     * Stream mode emits stream events when includePartialMessages is enabled.
+     */
+    shouldEmitStreamEvents() {
+        return this.includePartialMessages;
+    }
+    finalizeAssistantMessage() {
+        const state = this.mainAgentMessageState;
+        if (state.finalized) {
+            return this.buildMessage(null);
+        }
+        state.finalized = true;
+        this.finalizePendingBlocks(state, null);
+        const orderedOpenBlocks = Array.from(state.openBlocks).sort((a, b) => a - b);
+        for (const index of orderedOpenBlocks) {
+            this.onBlockClosed(state, index, null);
+            this.closeBlock(state, index);
+        }
+        if (state.messageStarted && this.includePartialMessages) {
+            this.emitStreamEventIfEnabled({ type: 'message_stop' }, null);
+        }
+        const message = this.buildMessage(null);
+        this.updateLastAssistantMessage(message);
+        this.emitMessageImpl(message);
+        return message;
+    }
+    emitResult(options) {
+        const resultMessage = this.buildResultMessage(options, this.lastAssistantMessage);
+        this.emitMessageImpl(resultMessage);
+    }
+    emitMessage(message) {
+        // In stream mode, emit immediately
+        this.emitMessageImpl(message);
+    }
+    send(message) {
+        this.emitMessage(message);
+    }
+    /**
+     * Overrides base class hook to emit stream event when text block is created.
+     */
+    onTextBlockCreated(state, index, block, parentToolUseId) {
+        this.emitStreamEventIfEnabled({
+            type: 'content_block_start',
+            index,
+            content_block: block,
+        }, parentToolUseId);
+    }
+    /**
+     * Overrides base class hook to emit stream event when text is appended.
+     */
+    onTextAppended(state, index, fragment, parentToolUseId) {
+        this.emitStreamEventIfEnabled({
+            type: 'content_block_delta',
+            index,
+            delta: { type: 'text_delta', text: fragment },
+        }, parentToolUseId);
+    }
+    /**
+     * Overrides base class hook to emit stream event when thinking block is created.
+     */
+    onThinkingBlockCreated(state, index, block, parentToolUseId) {
+        this.emitStreamEventIfEnabled({
+            type: 'content_block_start',
+            index,
+            content_block: block,
+        }, parentToolUseId);
+    }
+    /**
+     * Overrides base class hook to emit stream event when thinking is appended.
+     */
+    onThinkingAppended(state, index, fragment, parentToolUseId) {
+        this.emitStreamEventIfEnabled({
+            type: 'content_block_delta',
+            index,
+            delta: { type: 'thinking_delta', thinking: fragment },
+        }, parentToolUseId);
+    }
+    /**
+     * Overrides base class hook to emit stream event when tool_use block is created.
+     */
+    onToolUseBlockCreated(state, index, block, parentToolUseId) {
+        this.emitStreamEventIfEnabled({
+            type: 'content_block_start',
+            index,
+            content_block: block,
+        }, parentToolUseId);
+    }
+    /**
+     * Overrides base class hook to emit stream event when tool_use input is set.
+     */
+    onToolUseInputSet(state, index, input, parentToolUseId) {
+        this.emitStreamEventIfEnabled({
+            type: 'content_block_delta',
+            index,
+            delta: {
+                type: 'input_json_delta',
+                partial_json: JSON.stringify(input),
+            },
+        }, parentToolUseId);
+    }
+    /**
+     * Overrides base class hook to emit stream event when block is closed.
+     */
+    onBlockClosed(state, index, parentToolUseId) {
+        if (this.includePartialMessages) {
+            this.emitStreamEventIfEnabled({
+                type: 'content_block_stop',
+                index,
+            }, parentToolUseId);
+        }
+    }
+    /**
+     * Overrides base class hook to emit message_start event when message is started.
+     * Only emits for main agent, not for subagents.
+     */
+    onEnsureMessageStarted(state, parentToolUseId) {
+        // Only emit message_start for main agent, not for subagents
+        if (parentToolUseId === null) {
+            this.emitStreamEventIfEnabled({
+                type: 'message_start',
+                message: {
+                    id: state.messageId,
+                    role: 'assistant',
+                    model: this.config.getModel(),
+                },
+            }, null);
+        }
+    }
+    /**
+     * Emits stream events when partial messages are enabled.
+     * This is a private method specific to StreamJsonOutputAdapter.
+     * @param event - Stream event to emit
+     * @param parentToolUseId - null for main agent, string for subagent
+     */
+    emitStreamEventIfEnabled(event, parentToolUseId) {
+        if (!this.includePartialMessages) {
+            return;
+        }
+        const state = this.getMessageState(parentToolUseId);
+        const enrichedEvent = state.messageStarted
+            ? { ...event, message_id: state.messageId }
+            : event;
+        const partial = {
+            type: 'stream_event',
+            uuid: randomUUID(),
+            session_id: this.getSessionId(),
+            parent_tool_use_id: parentToolUseId,
+            event: enrichedEvent,
+        };
+        this.emitMessageImpl(partial);
+    }
+}
+//# sourceMappingURL=StreamJsonOutputAdapter.js.map

package/dist/src/nonInteractive/io/StreamJsonOutputAdapter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"StreamJsonOutputAdapter.js","sourceRoot":"","sources":["../../../../src/nonInteractive/io/StreamJsonOutputAdapter.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAYzC,OAAO,EACL,qBAAqB,GAItB,MAAM,4BAA4B,CAAC;AAEpC;;;;GAIG;AACH,MAAM,OAAO,uBACX,SAAQ,qBAAqB;IAKV;IAFnB,YACE,MAAc,EACG,sBAA+B;QAEhD,KAAK,CAAC,MAAM,CAAC,CAAC;QAFG,2BAAsB,GAAtB,sBAAsB,CAAS;IAGlD,CAAC;IAED;;OAEG;IACO,eAAe,CAAC,OAAoC;QAC5D,iDAAiD;QACjD,IACE,OAAO,OAAO,KAAK,QAAQ;YAC3B,OAAO,KAAK,IAAI;YAChB,MAAM,IAAI,OAAO;YACjB,OAAO,CAAC,IAAI,KAAK,WAAW,EAC5B,CAAC;YACD,IAAI,CAAC,0BAA0B,CAAC,OAA8B,CAAC,CAAC;QAClE,CAAC;QAED,2CAA2C;QAC3C,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACvD,CAAC;IAED;;OAEG;IACO,sBAAsB;QAC9B,OAAO,IAAI,CAAC,sBAAsB,CAAC;IACrC,CAAC;IAED,wBAAwB;QACtB,MAAM,KAAK,GAAG,IAAI,CAAC,qBAAqB,CAAC;QACzC,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YACpB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QACjC,CAAC;QACD,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC;QAEvB,IAAI,CAAC,qBAAqB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QACxC,MAAM,iBAAiB,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CACzD,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAChB,CAAC;QACF,KAAK,MAAM,KAAK,IAAI,iBAAiB,EAAE,CAAC;YACtC,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;YACvC,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QAChC,CAAC;QAED,IAAI,KAAK,CAAC,cAAc,IAAI,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACxD,IAAI,CAAC,wBAAwB,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,EAAE,IAAI,CAAC,CAAC;QAChE,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QACxC,IAAI,CAAC,0BAA0B,CAAC,OAAO,CAAC,CAAC;QACzC,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;QAC9B,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,UAAU,CAAC,OAAsB;QAC/B,MAAM,aAAa,GAAG,IAAI,CAAC,kBAAkB,CAC3C,OAAO,EACP,IAAI,CAAC,oBAAoB,CAC1B,CAAC;QACF,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC;IACtC,CAAC;IAED,WAAW,CAAC,OAAoC;QAC9C,mCAAmC;QACnC,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;IAChC,CAAC;IAED,IAAI,CAAC,OAAoC;QACvC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IAC5B,CAAC;IAED;;OAEG;IACgB,kBAAkB,CACnC,KAAmB,EACnB,KAAa,EACb,KAAgB,EAChB,eAA8B;QAE9B,IAAI,CAAC,wBAAwB,CAC3B;YACE,IAAI,EAAE,qBAAqB;YAC3B,KAAK;YACL,aAAa,EAAE,KAAK;SACrB,EACD,eAAe,CAChB,CAAC;IACJ,CAAC;IAED;;OAEG;IACgB,cAAc,CAC/B,KAAmB,EACnB,KAAa,EACb,QAAgB,EAChB,eAA8B;QAE9B,IAAI,CAAC,wBAAwB,CAC3B;YACE,IAAI,EAAE,qBAAqB;YAC3B,KAAK;YACL,KAAK,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,QAAQ,EAAE;SAC9C,EACD,eAAe,CAChB,CAAC;IACJ,CAAC;IAED;;OAEG;IACgB,sBAAsB,CACvC,KAAmB,EACnB,KAAa,EACb,KAAoB,EACpB,eAA8B;QAE9B,IAAI,CAAC,wBAAwB,CAC3B;YACE,IAAI,EAAE,qBAAqB;YAC3B,KAAK;YACL,aAAa,EAAE,KAAK;SACrB,EACD,eAAe,CAChB,CAAC;IACJ,CAAC;IAED;;OAEG;IACgB,kBAAkB,CACnC,KAAmB,EACnB,KAAa,EACb,QAAgB,EAChB,eAA8B;QAE9B,IAAI,CAAC,wBAAwB,CAC3B;YACE,IAAI,EAAE,qBAAqB;YAC3B,KAAK;YACL,KAAK,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,QAAQ,EAAE,QAAQ,EAAE;SACtD,EACD,eAAe,CAChB,CAAC;IACJ,CAAC;IAED;;OAEG;IACgB,qBAAqB,CACtC,KAAmB,EACnB,KAAa,EACb,KAAmB,EACnB,eAA8B;QAE9B,IAAI,CAAC,wBAAwB,CAC3B;YACE,IAAI,EAAE,qBAAqB;YAC3B,KAAK;YACL,aAAa,EAAE,KAAK;SACrB,EACD,eAAe,CAChB,CAAC;IACJ,CAAC;IAED;;OAEG;IACgB,iBAAiB,CAClC,KAAmB,EACnB,KAAa,EACb,KAAc,EACd,eAA8B;QAE9B,IAAI,CAAC,wBAAwB,CAC3B;YACE,IAAI,EAAE,qBAAqB;YAC3B,KAAK;YACL,KAAK,EAAE;gBACL,IAAI,EAAE,kBAAkB;gBACxB,YAAY,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;aACpC;SACF,EACD,eAAe,CAChB,CAAC;IACJ,CAAC;IAED;;OAEG;IACgB,aAAa,CAC9B,KAAmB,EACnB,KAAa,EACb,eAA8B;QAE9B,IAAI,IAAI,CAAC,sBAAsB,EAAE,CAAC;YAChC,IAAI,CAAC,wBAAwB,CAC3B;gBACE,IAAI,EAAE,oBAAoB;gBAC1B,KAAK;aACN,EACD,eAAe,CAChB,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;;OAGG;IACgB,sBAAsB,CACvC,KAAmB,EACnB,eAA8B;QAE9B,4DAA4D;QAC5D,IAAI,eAAe,KAAK,IAAI,EAAE,CAAC;YAC7B,IAAI,CAAC,wBAAwB,CAC3B;gBACE,IAAI,EAAE,eAAe;gBACrB,OAAO,EAAE;oBACP,EAAE,EAAE,KAAK,CAAC,SAAU;oBACpB,IAAI,EAAE,WAAW;oBACjB,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE;iBAC9B;aACF,EACD,IAAI,CACL,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACK,wBAAwB,CAC9B,KAAkB,EAClB,eAA8B;QAE9B,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACjC,OAAO;QACT,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;QACpD,MAAM,aAAa,GAAG,KAAK,CAAC,cAAc;YACxC,CAAC,CAAE,EAAE,GAAG,KAAK,EAAE,UAAU,EAAE,KAAK,CAAC,SAAS,EAEtC;YACJ,CAAC,CAAC,KAAK,CAAC;QAEV,MAAM,OAAO,GAA+B;YAC1C,IAAI,EAAE,cAAc;YACpB,IAAI,EAAE,UAAU,EAAE;YAClB,UAAU,EAAE,IAAI,CAAC,YAAY,EAAE;YAC/B,kBAAkB,EAAE,eAAe;YACnC,KAAK,EAAE,aAAa;SACrB,CAAC;QACF,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;IAChC,CAAC;CACF"}

package/dist/src/nonInteractive/io/StreamJsonOutputAdapter.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Qwen Team
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};