Fix stale tests.

Author: amanjaiswal73892

tests/experiments/test_launch_exp.py

@@ -122,7 +122,7 @@ def test_4o_mini_on_miniwob_tiny_test():
 
 
 if __name__ == "__main__":
-    test_timeout_ray()
+    # test_timeout_ray()
     # test_4o_mini_on_miniwob_tiny_test()
     # test_launch_system_ray()
     # test_launch_system_sequntial()

tests/llm/test_response_api.py

@@ -36,9 +36,7 @@ def create_mock_openai_chat_completion(
             )
             tool_call_mock.id = tc["id"]
             tool_call_mock.type = tc["type"]
-            tool_call_mock.function = MagicMock(
-                spec=openai.types.chat.chat_completion_message_tool_call.Function
-            )
+            tool_call_mock.function = MagicMock()
             tool_call_mock.function.name = tc["function"]["name"]
             tool_call_mock.function.arguments = tc["function"]["arguments"]
             message.tool_calls.append(tool_call_mock)
@@ -800,4 +798,5 @@ def test_tool_call_to_python_code():
 
 
 if __name__ == "__main__":
-    test_tool_call_to_python_code()
+    # test_tool_call_to_python_code()
+    # test_openai_chat_completion_model_parse_and_cost()

tests/llm/test_tracking.py

@@ -27,11 +27,6 @@ def test_get_action_decorator():
 OPENROUTER_API_KEY_AVAILABLE = os.environ.get("OPENROUTER_API_KEY") is not None
 
 OPENROUTER_MODELS = (
-    "openai/o1-mini-2024-09-12",
-    "openai/o1-preview-2024-09-12",
-    "openai/gpt-4o-2024-08-06",
-    "openai/gpt-4o-2024-05-13",
-    "anthropic/claude-3.5-sonnet:beta",
     "anthropic/claude-3.5-sonnet",
     "meta-llama/llama-3.1-405b-instruct",
     "meta-llama/llama-3.1-70b-instruct",

tests/test_ui_assistant.py

@@ -4,6 +4,6 @@
 
 def test_make_exp_args():
     """Basic unit test to detect refactoring errors."""
-    exp_args = make_exp_args(AGENT_4o)
+    exp_args = make_exp_args(AGENT_4o, "https://www.google.com")
 
     assert exp_args.agent_args.flags.action.demo_mode == "default"