blob42/langchain
1
0
Fork 0
forked from Archives/langchain
Code Issues Pull Requests Packages Projects Releases Wiki Activity

propogate kwargs fully (#6076)

Browse Source
This commit is contained in:
Harrison Chase 2023-06-12 22:37:55 -07:00 committed by GitHub
parent ec1a2adf9c
commit 6ac5d80286
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
langchain/chat_models/base.py
View File

@ -190,9 +190,10 @@ class BaseChatModel(BaseLanguageModel, ABC):
messages: List[BaseMessage],
stop: Optional[List[str]] = None,
callbacks: Callbacks = None,
**kwargs: Any,
) -> BaseMessage:
generation = self.generate(
[messages], stop=stop, callbacks=callbacks
[messages], stop=stop, callbacks=callbacks, **kwargs
).generations[0][0]
if isinstance(generation, ChatGeneration):
return generation.message
@ -227,7 +228,7 @@ class BaseChatModel(BaseLanguageModel, ABC):
_stop = None
else:
_stop = list(stop)
result = self([HumanMessage(content=text)], stop=_stop)
result = self([HumanMessage(content=text)], stop=_stop, **kwargs)
return result.content
def predict_messages(