listpage-next-ai 0.0.271 → 0.0.273

package/dist/cjs/index.d.ts

@@ -3,12 +3,15 @@ import { BaseMessage } from 'langchain';
 import { BaseMessageChunk } from 'langchain';
 import { ChatOpenAICallOptions } from '@langchain/openai';
 import { ChatOpenAICompletions } from '@langchain/openai';
+import { IterableReadableStream } from '@langchain/core/utils/stream';
 import { LanguageModelLike } from '@langchain/core/language_models/base';
 import { MessageStructure } from '@langchain/core/messages';
 import { MessageType } from '@langchain/core/messages';
 import { OpenAIClient } from '@langchain/openai';
 import { ReactAgent } from 'langchain';
+import { ReplaySubject } from 'rxjs';
 import { ResponseFormatUndefined } from 'langchain';
+import { StreamEvent } from '@langchain/core/tracers/log_stream';
 
 export declare interface AgentOptions {
     features: {
@@ -100,6 +103,28 @@ declare interface KnowledgeToolOptions {
     onRetrieveFullDocs: (index: string, title: string) => Promise<string>;
 }
 
+export declare function simplifyStreamEvents(events: IterableReadableStream<StreamEvent>): Promise<AsyncGenerator<{
+    type: string;
+    data: any;
+}, void, unknown>>;
+
+export declare function streamEvents2Subject<Event = any>(events: IterableReadableStream<Event>, transform?: (event: Event) => {
+    type: string;
+    data: any;
+}, callbacks?: {
+    before?: (subject: ReplaySubject<{
+        type: string;
+        data: any;
+    }>) => Promise<void>;
+    after?: (subject: ReplaySubject<{
+        type: string;
+        data: any;
+    }>) => Promise<void>;
+}): ReplaySubject<{
+    type: string;
+    data: any;
+}>;
+
 declare interface WordToolOptions {
     glossry: {
         name: string;

package/dist/cjs/index.js

@@ -25,6 +25,8 @@ var __webpack_exports__ = {};
 __webpack_require__.r(__webpack_exports__);
 __webpack_require__.d(__webpack_exports__, {
     createReactAgent: ()=>createReactAgent,
+    simplifyStreamEvents: ()=>simplifyStreamEvents,
+    streamEvents2Subject: ()=>streamEvents2Subject,
     ChatVolcengine: ()=>ChatVolcengine,
     AgentOptions: ()=>types_namespaceObject.AgentOptions
 });
@@ -499,13 +501,91 @@ class ChatVolcengine extends openai_namespaceObject.ChatOpenAICompletions {
         return langChainMessage;
     }
 }
+const external_rxjs_namespaceObject = require("rxjs");
+async function simplifyStreamEvents(events) {
+    return async function*() {
+        for await (const event of events){
+            const { event: type, data, name } = event;
+            if ('on_chat_model_stream' === type) {
+                const chunk = data?.chunk;
+                if (chunk) {
+                    const reasoning = chunk.additional_kwargs?.reasoning_content;
+                    if (reasoning) yield {
+                        type: 'thinking',
+                        data: reasoning
+                    };
+                    const content = chunk.content;
+                    if (content && 'string' == typeof content) yield {
+                        type: 'chunk',
+                        data: content
+                    };
+                }
+            } else if ('on_tool_start' === type) {
+                if ('write_todos' !== name) yield {
+                    type: 'tool_start',
+                    data: {
+                        name: name,
+                        input: data?.input?.input,
+                        start_timestamp: +Date.now()
+                    }
+                };
+            } else if ('on_tool_end' === type) switch(name){
+                case 'write_todos':
+                    yield {
+                        type: 'write_todos',
+                        data: data?.output?.update?.todos
+                    };
+                    break;
+                case 'generate_chart':
+                    yield {
+                        type: 'chunk',
+                        data: `\n<chart-view data-view="${data?.output?.content}"></chart-view>\n`
+                    };
+                    yield {
+                        type: 'tool_end',
+                        data: {
+                            name: name,
+                            output: data?.output?.content,
+                            end_timestamp: +Date.now()
+                        }
+                    };
+                    break;
+                default:
+                    yield {
+                        type: 'tool_end',
+                        data: {
+                            name: name,
+                            output: data?.output?.content,
+                            end_timestamp: +Date.now()
+                        }
+                    };
+            }
+        }
+    }();
+}
+function streamEvents2Subject(events, transform, callbacks) {
+    const subject = new external_rxjs_namespaceObject.ReplaySubject();
+    (async ()=>{
+        await callbacks?.before?.(subject);
+        for await (const event of events){
+            const transformedEvent = transform?.(event) || event;
+            subject.next(transformedEvent);
+        }
+        await callbacks?.after?.(subject);
+    })();
+    return subject;
+}
 exports.AgentOptions = __webpack_exports__.AgentOptions;
 exports.ChatVolcengine = __webpack_exports__.ChatVolcengine;
 exports.createReactAgent = __webpack_exports__.createReactAgent;
+exports.simplifyStreamEvents = __webpack_exports__.simplifyStreamEvents;
+exports.streamEvents2Subject = __webpack_exports__.streamEvents2Subject;
 for(var __webpack_i__ in __webpack_exports__)if (-1 === [
     "AgentOptions",
     "ChatVolcengine",
-    "createReactAgent"
+    "createReactAgent",
+    "simplifyStreamEvents",
+    "streamEvents2Subject"
 ].indexOf(__webpack_i__)) exports[__webpack_i__] = __webpack_exports__[__webpack_i__];
 Object.defineProperty(exports, '__esModule', {
     value: true

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "listpage-next-ai",
-  "version": "0.0.271",
+  "version": "0.0.273",
   "description": "A React component library for creating filter forms with Ant Design",
   "main": "./dist/cjs/index.js",
   "module": "./dist/esm/index.js",