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 index 8695b4a61c..ca70b5eedb 100644 --- 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 index 55131d84e3..2a0e827985 100644 --- 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"