VibecoderMcSwaggins commited on
Commit
58ff8fb
·
1 Parent(s): 4e438f3

chore: update model names in .env.example and tests

Browse files

Revised model names in .env.example to reflect new defaults and updated test cases in test_judges_factory.py to match the changes. The OpenAI model is now set to "gpt-5.1" and the Anthropic model to "claude-sonnet-4-5-20250929".

.env.example CHANGED
@@ -7,9 +7,9 @@ LLM_PROVIDER=openai
7
  OPENAI_API_KEY=sk-your-key-here
8
  ANTHROPIC_API_KEY=sk-ant-your-key-here
9
 
10
- # Model names (optional - sensible defaults)
11
- ANTHROPIC_MODEL=claude-3-5-sonnet-20240620
12
- OPENAI_MODEL=gpt-4-turbo
13
 
14
  # ============== EMBEDDINGS ==============
15
 
 
7
  OPENAI_API_KEY=sk-your-key-here
8
  ANTHROPIC_API_KEY=sk-ant-your-key-here
9
 
10
+ # Model names (optional - sensible defaults set in config.py)
11
+ # ANTHROPIC_MODEL=claude-sonnet-4-5-20250929
12
+ # OPENAI_MODEL=gpt-5.1
13
 
14
  # ============== EMBEDDINGS ==============
15
 
tests/unit/agent_factory/test_judges_factory.py CHANGED
@@ -25,22 +25,22 @@ def test_get_model_openai(mock_settings):
25
  """Test that OpenAI model is returned when provider is openai."""
26
  mock_settings.llm_provider = "openai"
27
  mock_settings.openai_api_key = "sk-test"
28
- mock_settings.openai_model = "gpt-4o"
29
 
30
  model = get_model()
31
  assert isinstance(model, OpenAIModel)
32
- assert model.model_name == "gpt-4o"
33
 
34
 
35
  def test_get_model_anthropic(mock_settings):
36
  """Test that Anthropic model is returned when provider is anthropic."""
37
  mock_settings.llm_provider = "anthropic"
38
  mock_settings.anthropic_api_key = "sk-ant-test"
39
- mock_settings.anthropic_model = "claude-3-5-sonnet"
40
 
41
  model = get_model()
42
  assert isinstance(model, AnthropicModel)
43
- assert model.model_name == "claude-3-5-sonnet"
44
 
45
 
46
  def test_get_model_huggingface(mock_settings):
@@ -58,7 +58,7 @@ def test_get_model_default_fallback(mock_settings):
58
  """Test fallback to OpenAI if provider is unknown."""
59
  mock_settings.llm_provider = "unknown_provider"
60
  mock_settings.openai_api_key = "sk-test"
61
- mock_settings.openai_model = "gpt-4o"
62
 
63
  model = get_model()
64
  assert isinstance(model, OpenAIModel)
 
25
  """Test that OpenAI model is returned when provider is openai."""
26
  mock_settings.llm_provider = "openai"
27
  mock_settings.openai_api_key = "sk-test"
28
+ mock_settings.openai_model = "gpt-5.1"
29
 
30
  model = get_model()
31
  assert isinstance(model, OpenAIModel)
32
+ assert model.model_name == "gpt-5.1"
33
 
34
 
35
  def test_get_model_anthropic(mock_settings):
36
  """Test that Anthropic model is returned when provider is anthropic."""
37
  mock_settings.llm_provider = "anthropic"
38
  mock_settings.anthropic_api_key = "sk-ant-test"
39
+ mock_settings.anthropic_model = "claude-sonnet-4-5-20250929"
40
 
41
  model = get_model()
42
  assert isinstance(model, AnthropicModel)
43
+ assert model.model_name == "claude-sonnet-4-5-20250929"
44
 
45
 
46
  def test_get_model_huggingface(mock_settings):
 
58
  """Test fallback to OpenAI if provider is unknown."""
59
  mock_settings.llm_provider = "unknown_provider"
60
  mock_settings.openai_api_key = "sk-test"
61
+ mock_settings.openai_model = "gpt-5.1"
62
 
63
  model = get_model()
64
  assert isinstance(model, OpenAIModel)