unique_web_search 2026.44.0.dev2__tar.gz → 2026.44.0.dev3__tar.gz

{unique_web_search-2026.44.0.dev2 → unique_web_search-2026.44.0.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-web-search
-Version: 2026.44.0.dev2
+Version: 2026.44.0.dev3
 Summary: 
 Author: Andreas Hauri, Gustav Hartz, Rami Azouz
 Author-email: Andreas Hauri <andreas@unique.ch>, Gustav Hartz <gustav.hartz.ext@unique.ch>, Rami Azouz <rami.ext@unique.ch>
@@ -25,8 +25,8 @@ Requires-Dist: rapidfuzz>=3.0.0,<4
 Requires-Dist: httpx>=0.28.1
 Requires-Dist: certifi>=2025.11.12
 Requires-Dist: jinja2>=3.1.6
-Requires-Dist: unique-search-proxy-core>=2026.44.0.dev4,<2026.44.0rc0
-Requires-Dist: unique-search-proxy-sdk>=2026.44.0.dev2,<2026.44.0rc0
+Requires-Dist: unique-search-proxy-core>=2026.44.0.dev5,<2026.44.0rc0
+Requires-Dist: unique-search-proxy-sdk>=2026.44.0.dev3,<2026.44.0rc0
 Requires-Dist: openai>=1.105.0,<3
 Requires-Python: >=3.12, <4.0
 Description-Content-Type: text/markdown

{unique_web_search-2026.44.0.dev2 → unique_web_search-2026.44.0.dev3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_web_search"
-version = "2026.44.0.dev2"
+version = "2026.44.0.dev3"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12,<4.0"
@@ -31,8 +31,8 @@ dependencies = [
     "httpx>=0.28.1",
     "certifi>=2025.11.12",
     "jinja2>=3.1.6",
-    "unique-search-proxy-core>=2026.44.0.dev4,<2026.44.0rc0",
-    "unique-search-proxy-sdk>=2026.44.0.dev2,<2026.44.0rc0",
+    "unique-search-proxy-core>=2026.44.0.dev5,<2026.44.0rc0",
+    "unique-search-proxy-sdk>=2026.44.0.dev3,<2026.44.0rc0",
     "openai>=1.105.0,<3",
 ]
 

{unique_web_search-2026.44.0.dev2 → unique_web_search-2026.44.0.dev3}/src/unique_web_search/services/search_engine/README.md

@@ -212,7 +212,14 @@ BING_AGENT_DEFAULT_MARKET=fr-FR
 
 **Provider:** Gemini + Google Search grounding on Vertex AI · Config extends `VertexAIAgentConfig`.
 
-Requires GCP ADC / workload identity. Fields include `vertexai_model_name`, `generation_instructions`, `enable_entreprise_search`, `enable_redirect_resolution`, `requires_scraping`.
+Requires GCP ADC / workload identity. Fields include `vertexai_model_name`,
+`generation_instructions`, `enable_entreprise_search`,
+`enable_redirect_resolution`, and `requires_scraping`.
+
+In proxy mode, Vertex grounding redirect URLs are resolved by Search Proxy,
+where every redirect hop passes URL-safety and shared egress controls. Legacy
+mode returns redirect URLs unchanged instead of resolving them directly from
+assistants-core.
 
 ```bash
 VERTEXAI_AGENT_DEFAULT_VERTEX_AI_MODEL_NAME=gemini-3-flash-preview

{unique_web_search-2026.44.0.dev2 → unique_web_search-2026.44.0.dev3}/src/unique_web_search/services/search_engine/utils/grounding/response_parsing.py

@@ -46,21 +46,19 @@ async def convert_response_to_search_results(
 
 
 class JsonConversionStrategy(ResponseParser):
-    """Extract a fenced JSON block from the response and validate it directly.
+    """Validate a ``GroundingSearchResults``-shaped JSON response directly.
 
-    Expects the response to contain a ``GroundingSearchResults``-shaped
-    JSON object wrapped in a ```json ... ``` code fence.
+    Accepts JSON wrapped in a ```json ... ``` code fence or a bare JSON object.
 
     Raises:
-        ValueError: If no JSON code fence is found in the response.
+        ValueError: If the response does not contain valid results JSON.
     """
 
     async def __call__(self, response: str) -> list[WebSearchResult]:
         json_match = _JSON_PATTERN.search(response)
-        if not json_match:
-            raise ValueError("No JSON found in the response")
+        payload = json_match.group(1) if json_match else response
         return GroundingSearchResults.model_validate_json(
-            json_match.group(1)
+            payload
         ).to_web_search_results()
 
 

{unique_web_search-2026.44.0.dev2 → unique_web_search-2026.44.0.dev3}/src/unique_web_search/services/search_engine/vertexai.py

@@ -1,9 +1,5 @@
-import asyncio
-import logging
 from typing import override
 
-from httpx import AsyncClient as HttpxAsyncClient
-from httpx import HTTPError
 from pydantic import Field
 from unique_search_proxy_core.agent_engines.base import AgentEngineType
 from unique_search_proxy_core.agent_engines.vertexai.schema import VertexAIAgentConfig
@@ -19,10 +15,7 @@ from unique_web_search.services.proxy.bridge import (
 )
 from unique_web_search.services.search_engine.base import SearchEngine, SearchEngineMode
 from unique_web_search.services.search_engine.registry import register_search_engine
-from unique_web_search.services.search_engine.schema import (
-    WebSearchResult,
-    WebSearchResults,
-)
+from unique_web_search.services.search_engine.schema import WebSearchResult
 from unique_web_search.services.search_engine.utils.grounding import (
     JsonConversionStrategy,
     LLMParserStrategy,
@@ -35,8 +28,6 @@ from unique_web_search.services.search_engine.utils.grounding.vertexai import (
     get_vertex_grounding_with_structured_output_config,
 )
 
-_LOGGER = logging.getLogger(__name__)
-
 
 class VertexAIConfig(VertexAIAgentConfig):
     language_model: LMI = get_LMI_default_field(
@@ -49,11 +40,6 @@ class VertexAIConfig(VertexAIAgentConfig):
         description="Whether the search engine requires scraping.",
     )
 
-    enable_redirect_resolution: bool = Field(
-        default=True,
-        description="Whether to enable redirect resolution.",
-    )
-
 
 @register_search_engine(
     name="vertexai",
@@ -87,16 +73,6 @@ class VertexAI(SearchEngine[VertexAIConfig]):
     def requires_scraping(self) -> bool:
         return self.config.requires_scraping
 
-    @override
-    async def _postprocess_results(
-        self,
-        results: list[WebSearchResult],
-    ) -> list[WebSearchResult]:
-        if self.config.enable_redirect_resolution:
-            resolved = await resolve_all(WebSearchResults(results=results))
-            return resolved.results
-        return results
-
     @override
     async def _legacy_search(
         self,
@@ -127,21 +103,4 @@ class VertexAI(SearchEngine[VertexAIConfig]):
             answer_with_citations, self.response_parsers
         )
 
-        return await self._postprocess_results(results)
-
-
-async def resolve_url(client: HttpxAsyncClient, web_search_result: WebSearchResult):
-    try:
-        resp = await client.head(web_search_result.url, follow_redirects=True)
-        web_search_result.url = str(resp.url)
-        return web_search_result
-    except HTTPError as e:
-        _LOGGER.error(f"Unable to redirect URL: {web_search_result.url}: {e}")
-        return web_search_result
-
-
-async def resolve_all(web_search_results: WebSearchResults):
-    async with HttpxAsyncClient(follow_redirects=True, timeout=10) as client:
-        tasks = [resolve_url(client, result) for result in web_search_results.results]
-        results = await asyncio.gather(*tasks)
-        return WebSearchResults(results=results)
+        return results