pytest-llm-assert 0.2.0__tar.gz → 0.2.1__tar.gz

{pytest_llm_assert-0.2.0 → pytest_llm_assert-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-llm-assert
-Version: 0.2.0
+Version: 0.2.1
 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.0 → pytest_llm_assert-0.2.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "pytest-llm-assert"
-version = "0.2.0"
+version = "0.2.1"
 description = "Simple LLM-powered assertions for any pytest test"
 readme = "README.md"
 requires-python = ">=3.11"

{pytest_llm_assert-0.2.0 → pytest_llm_assert-0.2.1}/src/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 → pytest_llm_assert-0.2.1}/uv.lock

@@ -1561,7 +1561,7 @@ wheels = [
 
 [[package]]
 name = "pytest-llm-assert"
-version = "0.2.0"
+version = "0.2.1"
 source = { editable = "." }
 dependencies = [
     { name = "azure-identity" },