clean up

samjulien · Nov 26, 2024 · aedaa51 · aedaa51
1 parent 481f6f0
commit aedaa51
Showing 1 changed file with 2 additions and 2 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
@@ -22,15 +22,15 @@ def _query_FC(self, inference_data: dict):
         if len(tools) > 0:
             api_response = self.client.chat.chat(
                 messages=message,
-                model=self.model_name.replace("-FC", ""),
+                model=self.model_name,
                 temperature=self.temperature,
                 tools=tools,
                 tool_choice="auto",
             )
         else:
             api_response = self.client.chat.chat(
                 messages=message,
-                model=self.model_name.replace("-FC", ""),
+                model=self.model_name,
                 temperature=self.temperature,
             )
         return api_response