github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-09 14:35:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
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
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
libs/langchain/langchain/chat_models/fireworks.py
View File

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