propogate kwargs fully (#6076)

searx_updates
Harrison Chase 11 months ago committed by GitHub
parent ec1a2adf9c
commit 6ac5d80286
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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(

Loading…
Cancel
Save