Move check for response property

ShishirPatil · Nov 25, 2024 · d81532f · d81532f
1 parent 82474c4
commit d81532f
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/berkeley-function-call-leaderboard/bfcl/model_handler/proprietary_model/writer.py b/berkeley-function-call-leaderboard/bfcl/model_handler/proprietary_model/writer.py
@@ -50,10 +50,6 @@ def _query_FC(self, inference_data: dict):
         inference_data["inference_input_log"] = {"message": repr(message), "tools": tools}
 
         if len(tools) > 0:
-            for tool in tools:
-                if "response" in tool["function"]:
-                    del tool["function"]["response"]
-
             api_response = self.client.chat.chat(
                 messages=message,
                 model=self.model_name.replace("-FC", ""),

diff --git a/berkeley-function-call-leaderboard/bfcl/model_handler/utils.py b/berkeley-function-call-leaderboard/bfcl/model_handler/utils.py
@@ -221,6 +221,7 @@ def convert_to_tool(functions, mapping, model_style):
                 ModelStyle.Anthropic,
                 ModelStyle.Google,
                 ModelStyle.FIREWORK_AI,
+                ModelStyle.WRITER,
             ]:
                 item[
                     "description"