Proper fix for model discovery per provider

2025-06-18 07:16:10 +04:00
parent 5199dd6ead
commit dad1e2d74e
15 changed files with 1250 additions and 65 deletions
--- a/tests/test_openai_compatible_token_usage.py
+++ b/tests/test_openai_compatible_token_usage.py
@@ -26,6 +26,12 @@ class TestOpenAICompatibleTokenUsage(unittest.TestCase):
            def validate_model_name(self, model_name):
                return True

+            def list_models(self, respect_restrictions=True):
+                return ["test-model"]
+
+            def list_all_known_models(self):
+                return ["test-model"]
+
        self.provider = TestProvider("test-key")

    def test_extract_usage_with_valid_tokens(self):