update ai_map

Author: KarolisKluonaitis

examples/map.py examples/ai_map.pyexamples/map.py renamed to examples/ai_map.py

@@ -1,15 +1,14 @@
 from oxylabs_ai_studio.apps.ai_map import AiMap
 
 
-map = AiMap(api_key="<API_KEY>")
+ai_map = AiMap(api_key="<API_KEY>")
 
 payload = {
     "url": "https://career.oxylabs.io",
     "user_prompt": "job ad pages",
     "return_sources_limit": 10,
-    "max_depth": 1,
     "geo_location": None,
     "render_javascript": False,
 }
-result = map.map(**payload)
+result = ai_map.map(**payload)
 print(result.data)

pyproject.toml

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

readme.md

@@ -135,16 +135,15 @@ print(result.data)
 from oxylabs_ai_studio.apps.ai_map import AiMap
 
 
-map = AiMap(api_key="<API_KEY>")
+ai_map = AiMap(api_key="<API_KEY>")
 payload = {
     "url": "https://career.oxylabs.io",
     "user_prompt": "job ad pages",
     "return_sources_limit": 10,
-    "max_depth": 1,
     "geo_location": None,
     "render_javascript": False,
 }
-result = map.map(**payload)
+result = ai_map.map(**payload)
 print(result.data)
 ```
 **Parameters:**

src/oxylabs_ai_studio/apps/ai_map.py

@@ -8,16 +8,17 @@
 from oxylabs_ai_studio.client import OxyStudioAIClient
 from oxylabs_ai_studio.logger import get_logger
 
-POLL_MAX_ATTEMPTS = 10
+MAP_TIMEOUT_SECONDS = 60 * 5
 POLL_INTERVAL_SECONDS = 3
+POLL_MAX_ATTEMPTS = MAP_TIMEOUT_SECONDS // POLL_INTERVAL_SECONDS
 
 logger = get_logger(__name__)
 
 
 class AiMapJob(BaseModel):
     run_id: str
     message: str | None = None
-    data: dict[str, Any] | str | None
+    data: dict[str, Any] | list[str] | None
 
 
 class AiMap(OxyStudioAIClient):
@@ -31,15 +32,13 @@ def map(
         url: str,
         user_prompt: str,
         return_sources_limit: int = 25,
-        max_depth: int = 3,
         geo_location: str | None = None,
         render_javascript: bool = False,
     ) -> AiMapJob:
         body = {
             "url": url,
             "user_prompt": user_prompt,
             "return_sources_limit": return_sources_limit,
-            "max_depth": max_depth,
             "geo_location": geo_location,
             "render_html": render_javascript,
         }
@@ -83,15 +82,13 @@ async def map_async(
         url: str,
         user_prompt: str,
         return_sources_limit: int = 25,
-        max_depth: int = 3,
         geo_location: str | None = None,
         render_javascript: bool = False,
     ) -> AiMapJob:
         body = {
             "url": url,
             "user_prompt": user_prompt,
             "return_sources_limit": return_sources_limit,
-            "max_depth": max_depth,
             "geo_location": geo_location,
             "render_html": render_javascript,
         }