From 038f057e591e2715c6abdbc1f9ddeaead7703790 Mon Sep 17 00:00:00 2001 From: AI Christianson Date: Sun, 2 Feb 2025 18:53:01 -0500 Subject: [PATCH] Fix tests. --- tests/ra_aid/test_llm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/ra_aid/test_llm.py b/tests/ra_aid/test_llm.py index 69a72d5..97465f6 100644 --- a/tests/ra_aid/test_llm.py +++ b/tests/ra_aid/test_llm.py @@ -54,7 +54,7 @@ def test_initialize_expert_defaults(clean_env, mock_openai, monkeypatch): monkeypatch.setenv("EXPERT_OPENAI_API_KEY", "test-key") _llm = initialize_expert_llm() - mock_openai.assert_called_once_with(api_key="test-key", model="o1", temperature=0) + mock_openai.assert_called_once_with(api_key="test-key", model="o1") def test_initialize_expert_openai_custom(clean_env, mock_openai, monkeypatch): @@ -348,7 +348,7 @@ def test_environment_variable_precedence(clean_env, mock_openai, monkeypatch): # Test LLM client creation with expert mode _llm = create_llm_client("openai", "o1", is_expert=True) - mock_openai.assert_called_with(api_key="expert-key", model="o1", temperature=0) + mock_openai.assert_called_with(api_key="expert-key", model="o1") # Test environment validation monkeypatch.setenv("EXPERT_OPENAI_API_KEY", "")