add geo_location in async

Author: KarolisKluonaitis

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "oxylabs-ai-studio"
-version = "0.2.11"
+version = "0.2.12"
 description = "Oxylabs studio python sdk"
 readme = "README.md"
 keywords = ["oxylabs", "ai", "studio"]

src/oxylabs_ai_studio/apps/ai_crawler.py

@@ -103,6 +103,7 @@ async def crawl_async(
         schema: dict[str, Any] | None = None,
         render_javascript: bool = False,
         return_sources_limit: int = 25,
+        geo_location: str | None = None,
     ) -> AiCrawlerJob:
         """Async version of crawl."""
         if output_format == "json" and schema is None:
@@ -116,6 +117,7 @@ async def crawl_async(
             "auxiliary_prompt": user_prompt,
             "render_html": render_javascript,
             "return_sources_limit": return_sources_limit,
+            "geo_location": geo_location,
         }
         async with self.async_client() as client:
             create_response = await client.post(url="/extract/run", json=body)

src/oxylabs_ai_studio/apps/ai_scraper.py

@@ -96,6 +96,7 @@ async def scrape_async(
         output_format: Literal["json", "markdown"] = "markdown",
         schema: dict[str, Any] | None = None,
         render_javascript: bool = False,
+        geo_location: str | None = None,
     ) -> AiScraperJob:
         """Async version of scrape."""
         if output_format == "json" and schema is None:
@@ -106,6 +107,7 @@ async def scrape_async(
             "output_format": output_format,
             "openapi_schema": schema,
             "render_html": render_javascript,
+            "geo_location": geo_location,
         }
         async with self.async_client() as client:
             create_response = await client.post(url="/scrape", json=body)

src/oxylabs_ai_studio/apps/ai_search.py

@@ -88,6 +88,7 @@ async def search_async(
         limit: int = 10,
         render_javascript: bool = False,
         return_content: bool = True,
+        geo_location: str | None = None,
     ) -> AiSearchJob:
         """Async version of search."""
         if not query:
@@ -98,6 +99,7 @@ async def search_async(
             "limit": limit,
             "render_html": render_javascript,
             "return_content": return_content,
+            "geo_location": geo_location,
         }
         async with self.async_client() as client:
             create_response = await client.post(url="/search/run", json=body)

src/oxylabs_ai_studio/apps/browser_agent.py

@@ -97,6 +97,7 @@ async def run_async(
         user_prompt: str = "",
         output_format: Literal["json", "markdown", "html", "screenshot"] = "markdown",
         schema: dict[str, Any] | None = None,
+        geo_location: str | None = None,
     ) -> BrowserAgentJob:
         """Async version of run."""
         if output_format == "json" and schema is None:
@@ -107,6 +108,7 @@ async def run_async(
             "output_format": output_format,
             "openapi_schema": schema,
             "auxiliary_prompt": user_prompt,
+            "geo_location": geo_location,
         }
         async with self.async_client() as client:
             create_response = await client.post(url="/browser-agent/run", json=body)