From 0adbc9678fa897ca6a2a487eb2e5ef97106ccb3e Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Tue, 25 Mar 2025 13:38:53 -1000 Subject: [PATCH] fix: add mode attribute to test model metadata --- tests/basic/test_main.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/basic/test_main.py b/tests/basic/test_main.py index d956b606b..3d51d2e7d 100644 --- a/tests/basic/test_main.py +++ b/tests/basic/test_main.py @@ -1071,10 +1071,12 @@ class TestMain(TestCase): "unique-model-name": { "max_input_tokens": 8192, "litellm_provider": "test-provider", + "mode": "chat", # Added mode attribute }, "another-provider/another-unique-model": { "max_input_tokens": 4096, "litellm_provider": "another-provider", + "mode": "chat", # Added mode attribute }, } metadata_file.write_text(json.dumps(test_models))