Sfoglia il codice sorgente

fix: cohere tool call does not support single tool (#3373)

tags/0.6.2
takatost 1 anno fa
parent
commit
36b7dbb8d0
Nessun account collegato all'indirizzo email del committer

+ 3
- 2
api/core/model_runtime/model_providers/cohere/llm/llm.py Vedi File

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

Loading…
Annulla
Salva