langchain[patch]: Pass kwargs to chat fireworks (#14183)

Otherwise `.bind()` isn't really any good
This commit is contained in:
William FH 2023-12-03 15:12:02 -08:00 committed by GitHub
parent e961c57fd2
commit 246dc4f9cc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -135,6 +135,7 @@ class ChatFireworks(BaseChatModel):
"model": self.model,
"messages": message_dicts,
**self.model_kwargs,
**kwargs,
}
response = completion_with_retry(
self,
@ -157,6 +158,7 @@ class ChatFireworks(BaseChatModel):
"model": self.model,
"messages": message_dicts,
**self.model_kwargs,
**kwargs,
}
response = await acompletion_with_retry(
self, self.use_retry, run_manager=run_manager, stop=stop, **params
@ -200,6 +202,7 @@ class ChatFireworks(BaseChatModel):
"messages": message_dicts,
"stream": True,
**self.model_kwargs,
**kwargs,
}
for chunk in completion_with_retry(
self, self.use_retry, run_manager=run_manager, stop=stop, **params
@ -230,6 +233,7 @@ class ChatFireworks(BaseChatModel):
"messages": message_dicts,
"stream": True,
**self.model_kwargs,
**kwargs,
}
async for chunk in await acompletion_with_retry_streaming(
self, self.use_retry, run_manager=run_manager, stop=stop, **params