fix(ci): unblock lint and docs build

Author: Ali-Khodabakhsh

docs/configure-rails/guardrail-catalog/third-party.md

@@ -304,6 +304,7 @@ Pangea AI Guard <community/pangea>
 Patronus Evaluate API <community/patronus-evaluate-api>
 Patronus Lynx <community/patronus-lynx>
 PolicyAI <community/policyai>
+Polygraf <community/polygraf>
 Presidio <community/presidio>
 Private AI <community/privateai>
 Prompt Security <community/prompt-security>

nemoguardrails/embeddings/providers/google.py

@@ -107,6 +107,13 @@ def encode(self, documents: List[str]) -> List[List[float]]:
                 embed_kwargs["output_dimensionality"] = self.output_dimensionality
 
             results = self.client.models.embed_content(**embed_kwargs)
-            return [emb.values for emb in results.embeddings]
+            embeddings: List[List[float]] = []
+            for emb in results.embeddings:
+                # Some clients may return `None` values for embeddings on partial failures.
+                # We treat that as an error to keep the return type stable.
+                if emb.values is None:
+                    raise RuntimeError("Embedding response contained a null embedding vector.")
+                embeddings.append(list(emb.values))
+            return embeddings
         except Exception as e:
             raise RuntimeError(f"Failed to retrieve embeddings: {e}") from e

nemoguardrails/library/polygraf/request.py

@@ -23,9 +23,7 @@
 log = logging.getLogger(__name__)
 
 
-async def polygraf_request(
-    text: str, server_endpoint: str, api_key: Optional[str] = None
-) -> List[Dict[str, Any]]:
+async def polygraf_request(text: str, server_endpoint: str, api_key: Optional[str] = None) -> List[Dict[str, Any]]:
     """Send a PII detection request to the Polygraf API.
 
     Args:
@@ -55,13 +53,9 @@ async def polygraf_request(
         headers["Authorization"] = f"Bearer {api_key}"
 
     async with aiohttp.ClientSession() as session:
-        async with session.post(
-            server_endpoint, json=payload, headers=headers
-        ) as resp:
+        async with session.post(server_endpoint, json=payload, headers=headers) as resp:
             if resp.status != 200:
-                raise ValueError(
-                    f"Polygraf call failed with status code {resp.status}.\nDetails: {await resp.text()}"
-                )
+                raise ValueError(f"Polygraf call failed with status code {resp.status}.\nDetails: {await resp.text()}")
 
             try:
                 data = await resp.json()

pyproject.toml

@@ -174,7 +174,6 @@ include = [
   "nemoguardrails/tracing/**",
   "nemoguardrails/server/**",
   "tests/test_callbacks.py",
-  "nemoguardrails/benchmark/**",
   "nemoguardrails/guardrails/**"
 ]
 exclude = [