averello/aider-mirror
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2026-02-02 11:33:14 +01:00
Code Issues Packages Projects Releases Wiki Activity

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
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
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")