Update agent_instance.py

This commit is contained in:
Hassan El Mghari 2024-09-23 07:46:08 -07:00 committed by GitHub
parent 8bf8c07eb3
commit e77d0e69ee
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -392,7 +392,7 @@ class ChatAgent(ShieldRunnerMixin):
input_messages, input_messages,
tools=self._get_tools(), tools=self._get_tools(),
tool_prompt_format=self.agent_config.tool_prompt_format, tool_prompt_format=self.agent_config.tool_prompt_format,
stream=True, stream=stream,
sampling_params=sampling_params, sampling_params=sampling_params,
): ):
event = chunk.event event = chunk.event