feat: add location_geo_code param to search endpoint and camelCase serialization

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: VinciGit00

scrapegraph-py/scrapegraph_py/async_client.py

@@ -355,6 +355,7 @@ async def search(
         query: str,
         num_results: int = 5,
         output_schema: Optional[Any] = None,
+        location_geo_code: Optional[str] = None,
         llm_config: Optional[LlmConfig] = None,
     ) -> Dict[str, Any]:
         """Search the web and extract structured results.
@@ -363,6 +364,7 @@ async def search(
             query: The search query
             num_results: Number of results (3-20, default 5)
             output_schema: JSON Schema dict or Pydantic BaseModel class for output structure
+            location_geo_code: Two-letter country code for geo-targeted results (e.g. 'us', 'gb')
             llm_config: LLM configuration options
         """
         logger.info(f"Searching: {query}")
@@ -382,6 +384,7 @@ async def search(
             query=query,
             num_results=num_results,
             output_schema=schema_dict,
+            location_geo_code=location_geo_code,
             llm_config=llm_config,
         )
         return await self._make_request(

scrapegraph-py/scrapegraph_py/client.py

@@ -402,6 +402,7 @@ def search(
         query: str,
         num_results: int = 5,
         output_schema: Optional[Any] = None,
+        location_geo_code: Optional[str] = None,
         llm_config: Optional[LlmConfig] = None,
     ) -> Dict[str, Any]:
         """Search the web and extract structured results.
@@ -410,6 +411,7 @@ def search(
             query: The search query
             num_results: Number of results (3-20, default 5)
             output_schema: JSON Schema dict or Pydantic BaseModel class for output structure
+            location_geo_code: Two-letter country code for geo-targeted results (e.g. 'us', 'gb')
             llm_config: LLM configuration options
         """
         logger.info(f"Searching: {query}")
@@ -429,6 +431,7 @@ def search(
             query=query,
             num_results=num_results,
             output_schema=schema_dict,
+            location_geo_code=location_geo_code,
             llm_config=llm_config,
         )
         return self._make_request(

scrapegraph-py/scrapegraph_py/models/search.py

@@ -6,22 +6,34 @@
 
 from typing import Any, Dict, Optional
 
-from pydantic import BaseModel, Field, conint, model_validator
+from pydantic import BaseModel, ConfigDict, Field, conint, model_validator
+from pydantic.alias_generators import to_camel
 
 from .shared import LlmConfig
 
 
 class SearchRequest(BaseModel):
     """Request model for POST /v2/search."""
 
+    model_config = ConfigDict(
+        alias_generator=to_camel,
+        populate_by_name=True,
+    )
+
     query: str = Field(..., description="The search query")
     num_results: conint(ge=3, le=20) = Field(
         default=5, description="Number of results to return (3-20)"
     )
     output_schema: Optional[Dict[str, Any]] = Field(
         default=None,
+        alias="schema",
         description="JSON Schema defining the structure of the extracted data",
     )
+    location_geo_code: Optional[str] = Field(
+        default=None,
+        max_length=10,
+        description="Two-letter country code for geo-targeted search results (e.g. 'us', 'gb', 'de')",
+    )
     llm_config: Optional[LlmConfig] = Field(
         default=None, description="LLM configuration options"
     )
@@ -34,4 +46,5 @@ def validate_query(self) -> "SearchRequest":
 
     def model_dump(self, *args: Any, **kwargs: Any) -> Dict[str, Any]:
         kwargs.setdefault("exclude_none", True)
+        kwargs.setdefault("by_alias", True)
         return super().model_dump(*args, **kwargs)

scrapegraph-py/tests/test_async_client.py

@@ -124,6 +124,20 @@ async def test_search(client):
         assert "results" in result
 
 
+@pytest.mark.asyncio
+async def test_search_with_location_geo_code(client):
+    with aioresponses() as mocked:
+        mocked.post(
+            f"{API_BASE_URL}/search",
+            payload={
+                "request_id": str(uuid4()),
+                "results": [{"url": "https://example.it"}],
+            },
+        )
+        result = await client.search("best restaurants", location_geo_code="it")
+        assert "results" in result
+
+
 # ------------------------------------------------------------------
 # Credits
 # ------------------------------------------------------------------

scrapegraph-py/tests/test_client.py

@@ -159,6 +159,17 @@ def test_search_with_num_results(client):
     assert "results" in result
 
 
+@responses.activate
+def test_search_with_location_geo_code(client):
+    responses.add(
+        responses.POST,
+        f"{API_BASE_URL}/search",
+        json={"request_id": str(uuid4()), "results": [{"url": "https://example.it"}]},
+    )
+    result = client.search("best restaurants", location_geo_code="it")
+    assert "results" in result
+
+
 # ------------------------------------------------------------------
 # Credits
 # ------------------------------------------------------------------

scrapegraph-py/tests/test_models.py

@@ -122,6 +122,22 @@ def test_search_request_num_results_bounds():
         SearchRequest(query="test", num_results=21)
 
 
+def test_search_request_with_location_geo_code():
+    req = SearchRequest(query="test", location_geo_code="us")
+    assert req.location_geo_code == "us"
+
+
+def test_search_request_location_geo_code_excluded_when_none():
+    req = SearchRequest(query="test")
+    data = req.model_dump()
+    assert "location_geo_code" not in data
+
+
+def test_search_request_location_geo_code_too_long():
+    with pytest.raises(ValueError):
+        SearchRequest(query="test", location_geo_code="a" * 11)
+
+
 # ------------------------------------------------------------------
 # Crawl
 # ------------------------------------------------------------------