Преглед изворни кода

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

tags/0.6.2
takatost пре 1 година
родитељ
комит
36b7dbb8d0
No account linked to committer's email address
1 измењених фајлова са 3 додато и 2 уклоњено
  1. 3
    2
      api/core/model_runtime/model_providers/cohere/llm/llm.py

+ 3
- 2
api/core/model_runtime/model_providers/cohere/llm/llm.py Прегледај датотеку

@@ -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…
Откажи
Сачувај