Instant search function (#8)

* Instant search endpoint introduced
* Bump version

Author: andriuskuk

pyproject.toml

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

readme.md

@@ -132,6 +132,8 @@ print(result.data)
 - `return_content` (bool): Whether to return markdown contents in results (default: True)
 - `geo_location` (str): search proxy location in ISO2 format.
 
+> **Note:** When `limit <= 10` and `return_content=False`, the search automatically uses the instant endpoint (`/search/instant`) which returns results immediately without polling, providing faster response times.
+
 ### Map (`AiMap.map`)
 ```python
 from oxylabs_ai_studio.apps.ai_map import AiMap

src/oxylabs_ai_studio/apps/ai_search.py

@@ -50,6 +50,24 @@ def search(
             "return_content": return_content,
             "geo_location": geo_location,
         }
+        # Use instant endpoint if limit <= 10 and return_content is False
+        if limit <= 10 and not return_content:
+            client = self.get_client()
+            response = self.call_api(
+                client=client, url="/search/instant", method="POST", body=body
+            )
+            status_code = response.status_code
+            if status_code != 200:
+                raise Exception(f"Failed to perform instant search: `{response.text}`")
+
+            resp_body = response.json()
+            return AiSearchJob(
+                run_id=resp_body["run_id"],
+                message=resp_body.get("message", None),
+                data=resp_body.get("data", None),
+            )
+
+        # Use regular polling endpoint
         client = self.get_client()
         create_response = self.call_api(
             client=client, url="/search/run", method="POST", body=body
@@ -118,6 +136,25 @@ async def search_async(
             "geo_location": geo_location,
         }
         async with self.async_client() as client:
+            # Use instant endpoint if limit <= 10 and return_content is False
+            if limit <= 10 and not return_content:
+                response = await self.call_api_async(
+                    client=client, url="/search/instant", method="POST", body=body
+                )
+                status_code = response.status_code
+                if status_code != 200:
+                    raise Exception(
+                        f"Failed to perform instant search: `{response.text}`"
+                    )
+
+                resp_body = response.json()
+                return AiSearchJob(
+                    run_id=resp_body["run_id"],
+                    message=resp_body.get("message", None),
+                    data=resp_body.get("data", None),
+                )
+
+            # Use regular polling endpoint
             create_response = await self.call_api_async(
                 client=client, url="/search/run", method="POST", body=body
             )