disabled ouput parser streaming

2026-06-28 13:00:56 +03:00 · 2023-10-31 14:07:29 +00:00
parent 8857530f29
commit f57a08f59b
11 changed files with 32 additions and 28 deletions
@@ -0,0 +1,71 @@
+import { getBaseClasses, INode, INodeData, INodeParams } from '../../../src'
+import { BaseOutputParser } from 'langchain/schema/output_parser'
+import { CustomListOutputParser as LangchainCustomListOutputParser } from 'langchain/output_parsers'
+import { CATEGORY } from '../OutputParserHelpers'
+
+class CustomListOutputParser implements INode {
+    label: string
+    name: string
+    version: number
+    description: string
+    type: string
+    icon: string
+    category: string
+    baseClasses: string[]
+    inputs: INodeParams[]
+    credential: INodeParams
+
+    constructor() {
+        this.label = 'Custom List Output Parser'
+        this.name = 'customListOutputParser'
+        this.version = 1.0
+        this.type = 'CustomListOutputParser'
+        this.description = 'Parse the output of an LLM call as a list of values.'
+        this.icon = 'list.png'
+        this.category = CATEGORY
+        this.baseClasses = [this.type, ...getBaseClasses(BaseOutputParser)]
+        this.inputs = [
+            {
+                label: 'Length',
+                name: 'length',
+                type: 'number',
+                default: 5,
+                step: 1,
+                description: 'Number of values to return'
+            },
+            {
+                label: 'Separator',
+                name: 'separator',
+                type: 'string',
+                description: 'Separator between values',
+                default: ','
+            },
+            {
+                label: 'Autofix',
+                name: 'autofixParser',
+                type: 'boolean',
+                optional: true,
+                description: 'In the event that the first call fails, will make another call to the model to fix any errors.'
+            }
+        ]
+    }
+
+    async init(nodeData: INodeData): Promise<any> {
+        const separator = nodeData.inputs?.separator as string
+        const lengthStr = nodeData.inputs?.length as string
+        const autoFix = nodeData.inputs?.autofixParser as boolean
+        let length = 5
+        if (lengthStr) length = parseInt(lengthStr, 10)
+
+        const parser = new LangchainCustomListOutputParser({ length: length, separator: separator })
+        Object.defineProperty(parser, 'autoFix', {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: autoFix
+        })
+        return parser
+    }
+}
+
+module.exports = { nodeClass: CustomListOutputParser }