zach
/
RA.Aid
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix tests.

This commit is contained in:
AI Christianson 2025-02-02 18:53:01 -05:00
parent 53ccc46392
commit 038f057e59
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/ra_aid/test_llm.py
View File

@ -54,7 +54,7 @@ def test_initialize_expert_defaults(clean_env, mock_openai, monkeypatch):
monkeypatch.setenv("EXPERT_OPENAI_API_KEY", "test-key") monkeypatch.setenv("EXPERT_OPENAI_API_KEY", "test-key")
_llm = initialize_expert_llm() _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): 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 # Test LLM client creation with expert mode
_llm = create_llm_client("openai", "o1", is_expert=True) _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 # Test environment validation
monkeypatch.setenv("EXPERT_OPENAI_API_KEY", "") monkeypatch.setenv("EXPERT_OPENAI_API_KEY", "")