Add toon format options (#7)

* Add toon format options

* bump version

Author: andriuskuk

pyproject.toml

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

src/oxylabs_ai_studio/apps/ai_crawler.py

@@ -31,15 +31,15 @@ def crawl(
         self,
         url: str,
         user_prompt: str,
-        output_format: Literal["json", "markdown", "csv"] = "markdown",
+        output_format: Literal["json", "markdown", "csv", "toon"] = "markdown",
         schema: dict[str, Any] | None = None,
         render_javascript: bool = False,
         return_sources_limit: int = 25,
         geo_location: str | None = None,
     ) -> AiCrawlerJob:
-        if output_format in ["json", "csv"] and schema is None:
+        if output_format in ["json", "csv", "toon"] and schema is None:
             raise ValueError(
-                "openapi_schema is required when output_format is json or csv.",
+                "openapi_schema is required when output_format is json, csv or toon.",
             )
 
         body = {
@@ -121,16 +121,16 @@ async def crawl_async(
         self,
         url: str,
         user_prompt: str = "",
-        output_format: Literal["json", "markdown", "csv"] = "markdown",
+        output_format: Literal["json", "markdown", "csv", "toon"] = "markdown",
         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 in ["json", "csv"] and schema is None:
+        if output_format in ["json", "csv", "toon"] and schema is None:
             raise ValueError(
-                "openapi_schema is required when output_format is json or csv.",
+                "openapi_schema is required when output_format is json, csv or toon.",
             )
 
         body = {

src/oxylabs_ai_studio/apps/ai_scraper.py

@@ -31,14 +31,16 @@ def __init__(self, api_key: str | None = None):
     def scrape(
         self,
         url: str,
-        output_format: Literal["json", "markdown", "csv", "screenshot"] = "markdown",
+        output_format: Literal[
+            "json", "markdown", "csv", "screenshot", "toon"
+        ] = "markdown",
         schema: dict[str, Any] | None = None,
         render_javascript: bool | Literal["auto"] = False,
         geo_location: str | None = None,
     ) -> AiScraperJob:
-        if output_format in ["json", "csv"] and schema is None:
+        if output_format in ["json", "csv", "toon"] and schema is None:
             raise ValueError(
-                "openapi_schema is required when output_format is json or csv.",
+                "openapi_schema is required when output_format is json, csv or toon.",
             )
 
         body = {
@@ -119,15 +121,17 @@ def generate_schema(self, prompt: str) -> dict[str, Any] | None:
     async def scrape_async(
         self,
         url: str,
-        output_format: Literal["json", "markdown", "csv", "screenshot"] = "markdown",
+        output_format: Literal[
+            "json", "markdown", "csv", "screenshot", "toon"
+        ] = "markdown",
         schema: dict[str, Any] | None = None,
         render_javascript: bool = False,
         geo_location: str | None = None,
     ) -> AiScraperJob:
         """Async version of scrape."""
-        if output_format in ["json", "csv"] and schema is None:
+        if output_format in ["json", "csv", "toon"] and schema is None:
             raise ValueError(
-                "openapi_schema is required when output_format is json or csv.",
+                "openapi_schema is required when output_format is json, csv or toon.",
             )
 
         body = {

src/oxylabs_ai_studio/apps/browser_agent.py

@@ -16,7 +16,7 @@
 
 
 class DataModel(BaseModel):
-    type: Literal["json", "markdown", "html", "screenshot", "csv"]
+    type: Literal["json", "markdown", "html", "screenshot", "csv", "toon"]
     content: dict[str, Any] | str | None
 
 
@@ -35,14 +35,14 @@ def run(
         url: str,
         user_prompt: str = "",
         output_format: Literal[
-            "json", "markdown", "html", "screenshot", "csv"
+            "json", "markdown", "html", "screenshot", "csv", "toon"
         ] = "markdown",
         schema: dict[str, Any] | None = None,
         geo_location: str | None = None,
     ) -> BrowserAgentJob:
-        if output_format in ["json", "csv"] and schema is None:
+        if output_format in ["json", "csv", "toon"] and schema is None:
             raise ValueError(
-                "openapi_schema is required when output_format is json or csv.",
+                "openapi_schema is required when output_format is json, csv or toon.",
             )
 
         body = {
@@ -125,14 +125,15 @@ async def run_async(
             "html",
             "screenshot",
             "csv",
+            "toon",
         ] = "markdown",
         schema: dict[str, Any] | None = None,
         geo_location: str | None = None,
     ) -> BrowserAgentJob:
         """Async version of run."""
-        if output_format in ["json", "csv"] and schema is None:
+        if output_format in ["json", "csv", "toon"] and schema is None:
             raise ValueError(
-                "openapi_schema is required when output_format is json or csv.",
+                "openapi_schema is required when output_format is json, csv or toon.",
             )
 
         body = {