From c6264a24ccd007d52da016dbc5f9797bdecb35ad Mon Sep 17 00:00:00 2001
From: Henry Heng <henryheng@flowiseai.com>
Date: Fri, 26 Jul 2024 00:22:42 +0100
Subject: [PATCH] Bugfix/Add streaming to chatfireworks (#2884)

add streaming to chatfireworks
---
 .../nodes/chatmodels/ChatFireworks/ChatFireworks.ts           | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/packages/components/nodes/chatmodels/ChatFireworks/ChatFireworks.ts b/packages/components/nodes/chatmodels/ChatFireworks/ChatFireworks.ts
index 91b19b22..471c0872 100644
--- a/packages/components/nodes/chatmodels/ChatFireworks/ChatFireworks.ts
+++ b/packages/components/nodes/chatmodels/ChatFireworks/ChatFireworks.ts
@@ -59,6 +59,7 @@ class ChatFireworks_ChatModels implements INode {
         const cache = nodeData.inputs?.cache as BaseCache
         const temperature = nodeData.inputs?.temperature as string
         const modelName = nodeData.inputs?.modelName as string
+        const streaming = nodeData.inputs?.streaming as boolean
 
         const credentialData = await getCredentialData(nodeData.credential ?? '', options)
         const fireworksApiKey = getCredentialParam('fireworksApiKey', credentialData, nodeData)
@@ -67,7 +68,8 @@ class ChatFireworks_ChatModels implements INode {
             fireworksApiKey,
             model: modelName,
             modelName,
-            temperature: temperature ? parseFloat(temperature) : undefined
+            temperature: temperature ? parseFloat(temperature) : undefined,
+            streaming: streaming ?? true
         }
         if (cache) obj.cache = cache