| @@ -323,6 +323,9 @@ class CohereLargeLanguageModel(LargeLanguageModel): | |||
| model_parameters['stop_sequences'] = stop | |||
| if tools: | |||
| if len(tools) == 1: | |||
| raise ValueError("Cohere tool call requires at least two tools to be specified.") | |||
| model_parameters['tools'] = self._convert_tools(tools) | |||
| message, chat_histories, tool_results \ | |||
| @@ -476,8 +479,6 @@ class CohereLargeLanguageModel(LargeLanguageModel): | |||
| index += 1 | |||
| elif isinstance(chunk, StreamedChatResponse_ToolCallsGeneration): | |||
| chunk = cast(StreamedChatResponse_ToolCallsGeneration, chunk) | |||
| tool_calls = [] | |||
| if chunk.tool_calls: | |||
| for cohere_tool_call in chunk.tool_calls: | |||
| tool_call = AssistantPromptMessage.ToolCall( | |||