dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-25 14:55:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

test: Implement TODOs in test_aider_extra_model_settings

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-11-19 12:06:15 -08:00
parent 34aff6c786
commit 92cf2cbd47
1 changed files with 11 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

13
tests/basic/test_models.py
View file

@ -117,11 +117,20 @@ class TestModels(unittest.TestCase):
# Test that defaults are applied when no exact match # Test that defaults are applied when no exact match
model = Model("claude-3-5-sonnet-20240620") model = Model("claude-3-5-sonnet-20240620")
# TODO: make sure Foo:bar and existing anthropic-beta headers are both here; check some_param; check max_tokens=8192 still there # Test that both the override and existing headers are present
model = Model("claude-3-5-sonnet-20240620")
self.assertEqual(model.extra_params["extra_headers"]["Foo"], "bar")
self.assertEqual(
model.extra_params["extra_headers"]["anthropic-beta"],
ANTHROPIC_BETA_HEADER,
)
self.assertEqual(model.extra_params["some_param"], "some value")
self.assertEqual(model.extra_params["max_tokens"], 8192)
# Test that exact match overrides defaults but not overrides # Test that exact match overrides defaults but not overrides
model = Model("gpt-4") model = Model("gpt-4")
# TODO: make sure Foo:bar header is there; check some_param self.assertEqual(model.extra_params["extra_headers"]["Foo"], "bar")
self.assertEqual(model.extra_params["some_param"], "some value")