TestAzureOpenAIO3Mini

2025-12-20 23:18:19 +05:30 · 2025-12-20 23:18:19 +05:30 · de65f0e777
commit de65f0e777
parent 99da2d413d
1 changed files with 2 additions and 0 deletions
--- a/tests/llm_translation/test_azure_o_series.py
+++ b/tests/llm_translation/test_azure_o_series.py
@ -20,6 +20,8 @@ from base_llm_unit_tests import BaseLLMChatTest, BaseOSeriesModelsTest

 class TestAzureOpenAIO3Mini(BaseOSeriesModelsTest, BaseLLMChatTest):
    def get_base_completion_call_args(self):
+        # Clear the LLM client cache to prevent test pollution from cached clients
+        litellm.in_memory_llm_clients_cache.flush_cache()
        return {
            "model": "azure/o3-mini",
            "api_key": os.getenv("AZURE_API_KEY"),