hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-09 10:56:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(api/model_runtime/azure/llm): Switch to tool_call. (#5541)

Browse Source
This commit is contained in:
-LAN-
2024-06-24 15:35:21 +08:00
committed by GitHub
parent 41ceb6a4eb
commit ba67206bb9
2 changed files with 137 additions and 142 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
api/tests/integration_tests/model_runtime/__mock/openai_chat.py
View File

@@ -73,17 +73,15 @@ class MockChatClass:
return FunctionCall(name=function_name, arguments=dumps(parameters))
@staticmethod
def generate_tool_calls(
tools: list[ChatCompletionToolParam] | NotGiven = NOT_GIVEN,
) -> Optional[list[ChatCompletionMessageToolCall]]:
def generate_tool_calls(tools = NOT_GIVEN) -> Optional[list[ChatCompletionMessageToolCall]]:
list_tool_calls = []
if not tools or len(tools) == 0:
return None
tool: ChatCompletionToolParam = tools[0]
tool = tools[0]
if tools['type'] != 'function':
if 'type' in tools and tools['type'] != 'function':
return None
function = tool['function']
function_call = MockChatClass.generate_function_call(functions=[function])