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

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

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

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

{pytest_llm_assert-0.1.1 → pytest_llm_assert-0.2.0}/src/pytest_llm_assert/core.py

@@ -2,7 +2,6 @@
 
 from __future__ import annotations
 
-import functools
 import json
 import os
 import re
@@ -20,27 +19,6 @@ 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.
@@ -115,7 +93,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 = _get_azure_ad_token_provider()
+            self._azure_ad_token_provider = self._get_azure_ad_token_provider()
 
     def _is_azure_model(self) -> bool:
         """Check if the model is an Azure OpenAI model."""
@@ -125,6 +103,29 @@ 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.1.1 → pytest_llm_assert-0.2.0}/uv.lock

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