plugin-sensitive-filter-xr 0.1.7 → 0.1.19

package/dist/lib/sensitiveFilter.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"sensitiveFilter.d.ts","sourceRoot":"","sources":["../../src/lib/sensitiveFilter.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAa,oBAAoB,EAA8B,MAAM,kBAAkB,CAAA;AAGnG,OAAO,EACL,eAAe,EAGf,uBAAuB,EACvB,wBAAwB,EAEzB,MAAM,sBAAsB,CAAA;AAC7B,OAAO,EAML,qBAAqB,EAKtB,MAAM,YAAY,CAAA;AAixBnB,qBAEa,yBAA0B,YAAW,wBAAwB,CAAC,qBAAqB,CAAC;IAE/F,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAY;IAEvC,QAAQ,CAAC,IAAI,EAAE,oBAAoB,CA6NlC;IAEK,gBAAgB,CACpB,OAAO,EAAE,qBAAqB,EAC9B,OAAO,EAAE,uBAAuB,GAC/B,OAAO,CAAC,eAAe,CAAC;IAa3B,OAAO,CAAC,wBAAwB;IAgThC,OAAO,CAAC,uBAAuB;CA2fhC;AAED,YAAY,EAAE,qBAAqB,EAAE,CAAA"}
+{"version":3,"file":"sensitiveFilter.d.ts","sourceRoot":"","sources":["../../src/lib/sensitiveFilter.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAa,oBAAoB,EAA8B,MAAM,kBAAkB,CAAA;AAGnG,OAAO,EACL,eAAe,EAGf,uBAAuB,EACvB,wBAAwB,EAEzB,MAAM,sBAAsB,CAAA;AAC7B,OAAO,EAML,qBAAqB,EAKtB,MAAM,YAAY,CAAA;AAoxBnB,qBAEa,yBAA0B,YAAW,wBAAwB,CAAC,qBAAqB,CAAC;IAE/F,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAY;IAEvC,QAAQ,CAAC,IAAI,EAAE,oBAAoB,CA6NlC;IAEK,gBAAgB,CACpB,OAAO,EAAE,qBAAqB,EAC9B,OAAO,EAAE,uBAAuB,GAC/B,OAAO,CAAC,eAAe,CAAC;IAa3B,OAAO,CAAC,wBAAwB;IAgThC,OAAO,CAAC,uBAAuB;CA6fhC;AAED,YAAY,EAAE,qBAAqB,EAAE,CAAA"}

package/dist/lib/sensitiveFilter.js

@@ -234,16 +234,19 @@ class BufferedOutputProxyChatModel extends BaseChatModel {
         const internalOptions = buildInternalSourceOptions(options);
         const streamFn = this.innerModel?.stream;
         if (typeof streamFn === 'function') {
-            let mergedChunk = null;
-            for await (const rawChunk of streamFn.call(this.innerModel, messages, internalOptions)) {
-                const chunk = toAiMessageChunk(rawChunk);
-                if (!chunk) {
-                    continue;
+            const stream = await streamFn.call(this.innerModel, messages, internalOptions);
+            if (stream && typeof stream[Symbol.asyncIterator] === 'function') {
+                let mergedChunk = null;
+                for await (const rawChunk of stream) {
+                    const chunk = toAiMessageChunk(rawChunk);
+                    if (!chunk) {
+                        continue;
+                    }
+                    mergedChunk = mergedChunk ? mergedChunk.concat(chunk) : chunk;
+                }
+                if (mergedChunk) {
+                    return toAiMessage(mergedChunk);
                 }
-                mergedChunk = mergedChunk ? mergedChunk.concat(chunk) : chunk;
-            }
-            if (mergedChunk) {
-                return toAiMessage(mergedChunk);
             }
         }
         return toAiMessage(await this.innerModel.invoke(messages, internalOptions));
@@ -1097,7 +1100,9 @@ let SensitiveFilterMiddleware = class SensitiveFilterMiddleware {
             const llmConfig = getLlmConfig();
             if (!modelPromise) {
                 modelPromise = this.commandBus.execute(new CreateModelClientCommand(buildInternalModelConfig(llmConfig.model), {
-                    usageCallback: () => { },
+                    usageCallback: () => {
+                        //
+                    },
                 }));
             }
             return modelPromise;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "plugin-sensitive-filter-xr",
-  "version": "0.1.7",
+  "version": "0.1.19",
   "author": {
     "name": "XpertAI",
     "url": "https://xpertai.cn"