fix inference code

MeetKai · Dec 23, 2024 · 376ebbc · 376ebbc
1 parent 74da07d
commit 376ebbc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/functionary/prompt_template/llama31_multi_tool_use_template.py b/functionary/prompt_template/llama31_multi_tool_use_template.py
@@ -127,7 +127,7 @@ def parse_assistant_response(
             for tool_call in tool_calls:
                 if tool_call["function"]["name"] == "multi_tool_use":
                     sub_tool_calls = []
-                    tool_use_list = json.loads(tool_call["function"]["arguments"])
+                    tool_use_list = json.loads(tool_call["function"]["arguments"])["tool_uses"]
                     for tool_use in tool_use_list:
                         sub_tool_calls.append(
                             {