unique-search-proxy-core 2026.36.0.dev2__tar.gz → 2026.36.0.dev3__tar.gz

{unique_search_proxy_core-2026.36.0.dev2 → unique_search_proxy_core-2026.36.0.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-search-proxy-core
-Version: 2026.36.0.dev2
+Version: 2026.36.0.dev3
 Summary: Shared Pydantic types for the Unique Search Proxy API
 Author: ThePhilAz
 Author-email: ThePhilAz <rami.azouz@philico.com>
@@ -8,7 +8,7 @@ Requires-Dist: httpx>=0.28.0,<0.29
 Requires-Dist: pydantic>=2.12.5,<3.0.0
 Requires-Dist: pydantic-settings>=2.10.1,<3
 Requires-Dist: pyhumps>=3.8.0,<4
-Requires-Dist: unique-toolkit[monitoring]>=2026.36.0.dev6,<2026.36.0rc0
+Requires-Dist: unique-toolkit[monitoring]>=2026.36.0.dev8,<2026.36.0rc0
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
 

{unique_search_proxy_core-2026.36.0.dev2 → unique_search_proxy_core-2026.36.0.dev3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique-search-proxy-core"
-version = "2026.36.0.dev2"
+version = "2026.36.0.dev3"
 description = "Shared Pydantic types for the Unique Search Proxy API"
 authors = [{ name = "ThePhilAz", email = "rami.azouz@philico.com" }]
 readme = "README.md"
@@ -10,7 +10,7 @@ dependencies = [
     "pydantic>=2.12.5,<3.0.0",
     "pydantic-settings>=2.10.1,<3",
     "pyhumps>=3.8.0,<4",
-    "unique-toolkit[monitoring]>=2026.36.0.dev6,<2026.36.0rc0",
+    "unique-toolkit[monitoring]>=2026.36.0.dev8,<2026.36.0rc0",
 
 ]
 
@@ -61,6 +61,10 @@ depcheck = "deptry ."
 [tool.pytest.ini_options]
 addopts = "--strict-markers --import-mode=importlib"
 asyncio_mode = "auto"
+markers = [
+    "ai: AI-authored or AI-generated tests",
+    "unit: unit tests",
+]
 
 [tool.uv.sources]
 unique-toolkit = { workspace = true }

unique_search_proxy_core-2026.36.0.dev3/unique_search_proxy_core/agent_engines/vertexai/schema.py

@@ -0,0 +1,73 @@
+from __future__ import annotations
+
+from typing import Annotated, ClassVar, Literal
+
+from pydantic import Field, field_validator
+from unique_toolkit._common.pydantic.rjsf_tags import RJSFMetaTag
+
+from unique_search_proxy_core.agent_engines.base import (
+    AgentEngineType,
+    BaseAgentEngineConfig,
+)
+from unique_search_proxy_core.agent_engines.vertexai.settings import (
+    resolve_enable_enterprise_search,
+    vertex_ai_env_settings,
+)
+
+
+def _get_force_activate_enterprise_search_description() -> str:
+    if vertex_ai_env_settings.force_activate_enterprise_search:
+        return "This parameter has been enforced by infra team."
+
+    return (
+        "When true, ground answers with Vertex AI Enterprise Web Search "
+        "(Agent Platform Search, SEC4-compliant)"
+    )
+
+
+def _enterprise_search_rjsf_tag() -> RJSFMetaTag:
+    return RJSFMetaTag.BooleanWidget.checkbox(
+        title="Enable Enterprise Search",
+        help=_get_force_activate_enterprise_search_description(),
+        disabled=vertex_ai_env_settings.force_activate_enterprise_search,
+    )
+
+
+EnableEnterpriseSearch = Annotated[bool, _enterprise_search_rjsf_tag()]
+
+
+class VertexAIAgentConfig(BaseAgentEngineConfig[Literal[AgentEngineType.VERTEXAI]]):
+    """Deployment + request defaults for Vertex AI grounding (Google GenAI)."""
+
+    _request_model_name: ClassVar[str] = "VertexAIAgentSearchRequest"
+
+    engine: Annotated[
+        Literal[AgentEngineType.VERTEXAI], RJSFMetaTag.SpecialWidget.hidden()
+    ] = Field(
+        default=AgentEngineType.VERTEXAI,
+        title="Agent engine",
+        description="Provider discriminator; must be `vertexai` for this config.",
+    )
+    vertexai_model_name: str = Field(
+        default=vertex_ai_env_settings.default_vertex_ai_model_name,
+        description="Gemini model name for grounded generation",
+    )
+    enable_enterprise_search: EnableEnterpriseSearch = Field(
+        default=vertex_ai_env_settings.force_activate_enterprise_search,
+        validate_default=True,
+    )
+
+    @field_validator("enable_enterprise_search", mode="before")
+    @classmethod
+    def validate_enable_enterprise_search(cls, v: object) -> bool:
+        """Pin the value on when infra forces enterprise search for ZDR tenants."""
+        return resolve_enable_enterprise_search(bool(v) if v is not None else False)
+
+
+VertexAIAgentSearchRequest = VertexAIAgentConfig.request_model()
+
+
+__all__ = [
+    "VertexAIAgentConfig",
+    "VertexAIAgentSearchRequest",
+]

unique_search_proxy_core-2026.36.0.dev3/unique_search_proxy_core/agent_engines/vertexai/settings.py

@@ -0,0 +1,52 @@
+import os
+import sys
+from pathlib import Path
+
+from pydantic import Field
+from pydantic_settings import BaseSettings, SettingsConfigDict
+
+_ENV_PREFIX = "VERTEXAI_AGENT_"
+
+
+class _VertexAIEnvSettings(BaseSettings):
+    default_vertex_ai_model_name: str = Field(
+        default="gemini-3-flash-preview",
+        description="Default Vertex AI model name for grounded generation",
+    )
+    force_activate_enterprise_search: bool = Field(
+        default=False,
+        description=(
+            "When true, pin Enable Enterprise Search on for ZDR tenants so the "
+            "toggle cannot be turned off in config or at request time."
+        ),
+    )
+
+
+def _get_settings() -> _VertexAIEnvSettings:
+    if "pytest" in sys.modules:
+        env_file = Path(os.getcwd()) / "tests/test.env"
+    else:
+        env_file = Path(os.getcwd()) / ".env"
+
+    class _Settings(_VertexAIEnvSettings):
+        model_config = SettingsConfigDict(
+            env_file=env_file,
+            env_prefix=_ENV_PREFIX,
+            extra="ignore",
+        )
+
+    return _Settings()
+
+
+vertex_ai_env_settings = _get_settings()
+
+
+def resolve_enable_enterprise_search(value: bool | None = False) -> bool:
+    """Return the effective enterprise-search flag after infra enforcement.
+
+    ZDR tenants set ``VERTEXAI_AGENT_FORCE_ACTIVATE_ENTERPRISE_SEARCH=true`` so
+    the cheaper standard grounding edition cannot be selected.
+    """
+    if vertex_ai_env_settings.force_activate_enterprise_search:
+        return True
+    return bool(value)

unique_search_proxy_core-2026.36.0.dev2/unique_search_proxy_core/agent_engines/vertexai/schema.py

@@ -1,45 +0,0 @@
-from __future__ import annotations
-
-from typing import Annotated, ClassVar, Literal
-
-from pydantic import Field
-from unique_toolkit._common.pydantic.rjsf_tags import RJSFMetaTag
-
-from unique_search_proxy_core.agent_engines.base import (
-    AgentEngineType,
-    BaseAgentEngineConfig,
-)
-
-
-class VertexAIAgentConfig(BaseAgentEngineConfig[Literal[AgentEngineType.VERTEXAI]]):
-    """Deployment + request defaults for Vertex AI grounding (Google GenAI)."""
-
-    _request_model_name: ClassVar[str] = "VertexAIAgentSearchRequest"
-
-    engine: Annotated[
-        Literal[AgentEngineType.VERTEXAI], RJSFMetaTag.SpecialWidget.hidden()
-    ] = Field(
-        default=AgentEngineType.VERTEXAI,
-        title="Agent engine",
-        description="Provider discriminator; must be `vertexai` for this config.",
-    )
-    vertexai_model_name: str = Field(
-        default="gemini-3-flash-preview",
-        description="Gemini model name for grounded generation",
-    )
-    enable_enterprise_search: bool = Field(
-        default=False,
-        description=(
-            "When true, ground answers with Vertex AI Enterprise Web Search "
-            "(Agent Platform Search, SEC4-compliant)"
-        ),
-    )
-
-
-VertexAIAgentSearchRequest = VertexAIAgentConfig.request_model()
-
-
-__all__ = [
-    "VertexAIAgentConfig",
-    "VertexAIAgentSearchRequest",
-]