pytest-llm-assert 0.2.0__py3-none-any.whl → 0.2.2__py3-none-any.whl

pytest_llm_assert/core.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+import functools
 import json
 import os
 import re
@@ -19,6 +20,27 @@ if TYPE_CHECKING:
     from typing import Any
 
 
+@functools.cache
+def _get_azure_ad_token_provider() -> Callable[[], str] | None:
+    """Get Azure AD token provider for Entra ID authentication.
+
+    Uses LiteLLM's built-in helper which leverages DefaultAzureCredential.
+    Cached at module level to avoid recreating credentials on each call.
+    """
+    try:
+        from litellm.secret_managers.get_azure_ad_token_provider import (
+            get_azure_ad_token_provider,
+        )
+
+        return get_azure_ad_token_provider()
+    except ImportError:
+        # azure-identity not installed
+        return None
+    except Exception:
+        # Credential not available
+        return None
+
+
 @dataclass(slots=True)
 class LLMResponse:
     """Response details from the last LLM call.
@@ -93,7 +115,7 @@ class LLMAssert:
 
         # Auto-configure Azure Entra ID when no API key is provided
         if self._is_azure_model() and not self._has_azure_api_key():
-            self._azure_ad_token_provider = self._get_azure_ad_token_provider()
+            self._azure_ad_token_provider = _get_azure_ad_token_provider()
 
     def _is_azure_model(self) -> bool:
         """Check if the model is an Azure OpenAI model."""
@@ -103,29 +125,6 @@ class LLMAssert:
         """Check if an Azure API key is available."""
         return bool(self.api_key or os.environ.get("AZURE_API_KEY"))
 
-    @staticmethod
-    def _get_azure_ad_token_provider() -> Callable[[], str] | None:
-        """Get Azure AD token provider for Entra ID authentication.
-
-        Uses LiteLLM's built-in helper which leverages DefaultAzureCredential:
-        - Azure CLI credentials (az login)
-        - Managed Identity
-        - Environment variables (AZURE_CLIENT_ID, AZURE_CLIENT_SECRET, ...)
-        - Visual Studio Code credentials
-        """
-        try:
-            from litellm.secret_managers.get_azure_ad_token_provider import (
-                get_azure_ad_token_provider,
-            )
-
-            return get_azure_ad_token_provider()
-        except ImportError:
-            # azure-identity not installed
-            return None
-        except Exception:
-            # Credential not available
-            return None
-
     @property
     def system_prompt(self) -> str:
         """Get the system prompt used for LLM assertions."""

{pytest_llm_assert-0.2.0.dist-info → pytest_llm_assert-0.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-llm-assert
-Version: 0.2.0
+Version: 0.2.2
 Summary: Simple LLM-powered assertions for any pytest test
 Project-URL: Homepage, https://github.com/sbroenne/pytest-llm-assert
 Project-URL: Documentation, https://github.com/sbroenne/pytest-llm-assert#readme

pytest_llm_assert-0.2.2.dist-info/RECORD

@@ -0,0 +1,9 @@
+pytest_llm_assert/__init__.py,sha256=OcOVlsvqZBBxMzrQssLbaWVkc4qKSvdOMtLVibzDfFQ,233
+pytest_llm_assert/core.py,sha256=0JOttHcZrJF2rA3xwG8BaCBIiEWvQCxFvT5HsYROygg,7983
+pytest_llm_assert/plugin.py,sha256=g3sotHAeUXMuOsFQdaoIbn0CY24i-1CPv0EglrC5qtE,1327
+pytest_llm_assert/prompts/system_prompt.md,sha256=RhSaYrpOjVcVwuG_af_Q50kHFhqXGOCKzubSYBXFzTA,181
+pytest_llm_assert-0.2.2.dist-info/METADATA,sha256=PDLTGM1cYknujIreiKa06IuwprnBcrEb4FfNJdmi8oY,4723
+pytest_llm_assert-0.2.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+pytest_llm_assert-0.2.2.dist-info/entry_points.txt,sha256=YEYg83TT6znVYdvFvZHJEOJ8XsZbcrqV9pY8uM-ThQE,49
+pytest_llm_assert-0.2.2.dist-info/licenses/LICENSE,sha256=wHrdHpzRm4rdlyMdj-sQw7aou6kHPujW0VmRBEhInJ8,1072
+pytest_llm_assert-0.2.2.dist-info/RECORD,,

pytest_llm_assert-0.2.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-pytest_llm_assert/__init__.py,sha256=OcOVlsvqZBBxMzrQssLbaWVkc4qKSvdOMtLVibzDfFQ,233
-pytest_llm_assert/core.py,sha256=B3lah5pPZ3VMbtzZ3bBtdJ2o8r_mZb7_aw375_hBC2Q,8145
-pytest_llm_assert/plugin.py,sha256=g3sotHAeUXMuOsFQdaoIbn0CY24i-1CPv0EglrC5qtE,1327
-pytest_llm_assert/prompts/system_prompt.md,sha256=RhSaYrpOjVcVwuG_af_Q50kHFhqXGOCKzubSYBXFzTA,181
-pytest_llm_assert-0.2.0.dist-info/METADATA,sha256=SdFu02CtgVAK14bwjP10clZgxFLQF8yEtiIm5Ct6BJE,4723
-pytest_llm_assert-0.2.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-pytest_llm_assert-0.2.0.dist-info/entry_points.txt,sha256=YEYg83TT6znVYdvFvZHJEOJ8XsZbcrqV9pY8uM-ThQE,49
-pytest_llm_assert-0.2.0.dist-info/licenses/LICENSE,sha256=wHrdHpzRm4rdlyMdj-sQw7aou6kHPujW0VmRBEhInJ8,1072
-pytest_llm_assert-0.2.0.dist-info/RECORD,,