nimble_python 0.20.0__py3-none-any.whl → 0.22.0__py3-none-any.whl

nimble_python/_client.py

@@ -294,6 +294,7 @@ class Nimble(SyncAPIClient):
         self,
         *,
         url: str,
+        auto_driver_configuration: Dict[str, int] | Omit = omit,
         body: object | Omit = omit,
         browser: client_extract_params.Browser | Omit = omit,
         browser_actions: Iterable[client_extract_params.BrowserAction] | Omit = omit,
@@ -555,7 +556,10 @@ class Nimble(SyncAPIClient):
         ]
         | Omit = omit,
         device: Literal["desktop", "mobile", "tablet"] | Omit = omit,
-        driver: Literal["vx6", "vx8", "vx8-pro", "vx10", "vx10-pro", "vx12", "vx12-pro", "media-vx6"] | Omit = omit,
+        driver: Literal[
+            "auto", "vx6", "vx8", "vx8-pro", "vx10", "vx10-pro", "vx12", "vx12-pro", "media-vx6", "fast-vx6"
+        ]
+        | Omit = omit,
         expected_status_codes: Iterable[int] | Omit = omit,
         formats: List[Literal["html", "markdown", "screenshot", "headers", "links"]] | Omit = omit,
         headers: Dict[str, Union[str, SequenceNotStr[str], None]] | Omit = omit,
@@ -1184,6 +1188,11 @@ class Nimble(SyncAPIClient):
         Args:
           url: Target URL to scrape
 
+          auto_driver_configuration: Custom flow for the optimization engine: maps candidate names to the number of
+              attempts to spend on each candidate before advancing (0 skips it). Key order
+              defines the flow order. Providing it opts the request into 'auto' driver
+              selection.
+
           body: Request body for POST, PUT, PATCH methods
 
           browser: Browser type to emulate
@@ -1253,6 +1262,7 @@ class Nimble(SyncAPIClient):
             body=maybe_transform(
                 {
                     "url": url,
+                    "auto_driver_configuration": auto_driver_configuration,
                     "body": body,
                     "browser": browser,
                     "browser_actions": browser_actions,
@@ -1294,6 +1304,7 @@ class Nimble(SyncAPIClient):
         self,
         *,
         url: str,
+        auto_driver_configuration: Dict[str, int] | Omit = omit,
         body: object | Omit = omit,
         browser: client_extract_async_params.Browser | Omit = omit,
         browser_actions: Iterable[client_extract_async_params.BrowserAction] | Omit = omit,
@@ -1556,7 +1567,10 @@ class Nimble(SyncAPIClient):
         ]
         | Omit = omit,
         device: Literal["desktop", "mobile", "tablet"] | Omit = omit,
-        driver: Literal["vx6", "vx8", "vx8-pro", "vx10", "vx10-pro", "vx12", "vx12-pro", "media-vx6"] | Omit = omit,
+        driver: Literal[
+            "auto", "vx6", "vx8", "vx8-pro", "vx10", "vx10-pro", "vx12", "vx12-pro", "media-vx6", "fast-vx6"
+        ]
+        | Omit = omit,
         expected_status_codes: Iterable[int] | Omit = omit,
         formats: List[Literal["html", "markdown", "screenshot", "headers", "links"]] | Omit = omit,
         headers: Dict[str, Union[str, SequenceNotStr[str], None]] | Omit = omit,
@@ -2189,6 +2203,11 @@ class Nimble(SyncAPIClient):
         Args:
           url: Target URL to scrape
 
+          auto_driver_configuration: Custom flow for the optimization engine: maps candidate names to the number of
+              attempts to spend on each candidate before advancing (0 skips it). Key order
+              defines the flow order. Providing it opts the request into 'auto' driver
+              selection.
+
           body: Request body for POST, PUT, PATCH methods
 
           browser: Browser type to emulate
@@ -2268,6 +2287,7 @@ class Nimble(SyncAPIClient):
             body=maybe_transform(
                 {
                     "url": url,
+                    "auto_driver_configuration": auto_driver_configuration,
                     "body": body,
                     "browser": browser,
                     "browser_actions": browser_actions,
@@ -3568,6 +3588,7 @@ class AsyncNimble(AsyncAPIClient):
         self,
         *,
         url: str,
+        auto_driver_configuration: Dict[str, int] | Omit = omit,
         body: object | Omit = omit,
         browser: client_extract_params.Browser | Omit = omit,
         browser_actions: Iterable[client_extract_params.BrowserAction] | Omit = omit,
@@ -3829,7 +3850,10 @@ class AsyncNimble(AsyncAPIClient):
         ]
         | Omit = omit,
         device: Literal["desktop", "mobile", "tablet"] | Omit = omit,
-        driver: Literal["vx6", "vx8", "vx8-pro", "vx10", "vx10-pro", "vx12", "vx12-pro", "media-vx6"] | Omit = omit,
+        driver: Literal[
+            "auto", "vx6", "vx8", "vx8-pro", "vx10", "vx10-pro", "vx12", "vx12-pro", "media-vx6", "fast-vx6"
+        ]
+        | Omit = omit,
         expected_status_codes: Iterable[int] | Omit = omit,
         formats: List[Literal["html", "markdown", "screenshot", "headers", "links"]] | Omit = omit,
         headers: Dict[str, Union[str, SequenceNotStr[str], None]] | Omit = omit,
@@ -4458,6 +4482,11 @@ class AsyncNimble(AsyncAPIClient):
         Args:
           url: Target URL to scrape
 
+          auto_driver_configuration: Custom flow for the optimization engine: maps candidate names to the number of
+              attempts to spend on each candidate before advancing (0 skips it). Key order
+              defines the flow order. Providing it opts the request into 'auto' driver
+              selection.
+
           body: Request body for POST, PUT, PATCH methods
 
           browser: Browser type to emulate
@@ -4527,6 +4556,7 @@ class AsyncNimble(AsyncAPIClient):
             body=await async_maybe_transform(
                 {
                     "url": url,
+                    "auto_driver_configuration": auto_driver_configuration,
                     "body": body,
                     "browser": browser,
                     "browser_actions": browser_actions,
@@ -4568,6 +4598,7 @@ class AsyncNimble(AsyncAPIClient):
         self,
         *,
         url: str,
+        auto_driver_configuration: Dict[str, int] | Omit = omit,
         body: object | Omit = omit,
         browser: client_extract_async_params.Browser | Omit = omit,
         browser_actions: Iterable[client_extract_async_params.BrowserAction] | Omit = omit,
@@ -4830,7 +4861,10 @@ class AsyncNimble(AsyncAPIClient):
         ]
         | Omit = omit,
         device: Literal["desktop", "mobile", "tablet"] | Omit = omit,
-        driver: Literal["vx6", "vx8", "vx8-pro", "vx10", "vx10-pro", "vx12", "vx12-pro", "media-vx6"] | Omit = omit,
+        driver: Literal[
+            "auto", "vx6", "vx8", "vx8-pro", "vx10", "vx10-pro", "vx12", "vx12-pro", "media-vx6", "fast-vx6"
+        ]
+        | Omit = omit,
         expected_status_codes: Iterable[int] | Omit = omit,
         formats: List[Literal["html", "markdown", "screenshot", "headers", "links"]] | Omit = omit,
         headers: Dict[str, Union[str, SequenceNotStr[str], None]] | Omit = omit,
@@ -5463,6 +5497,11 @@ class AsyncNimble(AsyncAPIClient):
         Args:
           url: Target URL to scrape
 
+          auto_driver_configuration: Custom flow for the optimization engine: maps candidate names to the number of
+              attempts to spend on each candidate before advancing (0 skips it). Key order
+              defines the flow order. Providing it opts the request into 'auto' driver
+              selection.
+
           body: Request body for POST, PUT, PATCH methods
 
           browser: Browser type to emulate
@@ -5542,6 +5581,7 @@ class AsyncNimble(AsyncAPIClient):
             body=await async_maybe_transform(
                 {
                     "url": url,
+                    "auto_driver_configuration": auto_driver_configuration,
                     "body": body,
                     "browser": browser,
                     "browser_actions": browser_actions,

nimble_python/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "nimble_python"
-__version__ = "0.20.0"  # x-release-please-version
+__version__ = "0.22.0"  # x-release-please-version

nimble_python/resources/agent.py

@@ -120,9 +120,9 @@ class AgentResource(SyncAPIResource):
         *,
         prompt: str,
         url: str,
-        agent_name: Optional[str] | Omit = omit,
         input_schema: Dict[str, object] | Omit = omit,
         metadata: Optional[agent_generate_params.CreateAgentGenerationRequestMetadata] | Omit = omit,
+        name: Optional[str] | Omit = omit,
         output_schema: Dict[str, object] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -178,9 +178,9 @@ class AgentResource(SyncAPIResource):
         *,
         prompt: str,
         url: str | Omit = omit,
-        agent_name: Optional[str] | Omit = omit,
         input_schema: Dict[str, object] | Omit = omit,
         metadata: Optional[agent_generate_params.CreateAgentGenerationRequestMetadata] | Omit = omit,
+        name: Optional[str] | Omit = omit,
         output_schema: Dict[str, object] | Omit = omit,
         from_agent: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -196,9 +196,9 @@ class AgentResource(SyncAPIResource):
                 {
                     "prompt": prompt,
                     "url": url,
-                    "agent_name": agent_name,
                     "input_schema": input_schema,
                     "metadata": metadata,
+                    "name": name,
                     "output_schema": output_schema,
                     "from_agent": from_agent,
                 },
@@ -511,9 +511,9 @@ class AsyncAgentResource(AsyncAPIResource):
         *,
         prompt: str,
         url: str,
-        agent_name: Optional[str] | Omit = omit,
         input_schema: Dict[str, object] | Omit = omit,
         metadata: Optional[agent_generate_params.CreateAgentGenerationRequestMetadata] | Omit = omit,
+        name: Optional[str] | Omit = omit,
         output_schema: Dict[str, object] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -569,9 +569,9 @@ class AsyncAgentResource(AsyncAPIResource):
         *,
         prompt: str,
         url: str | Omit = omit,
-        agent_name: Optional[str] | Omit = omit,
         input_schema: Dict[str, object] | Omit = omit,
         metadata: Optional[agent_generate_params.CreateAgentGenerationRequestMetadata] | Omit = omit,
+        name: Optional[str] | Omit = omit,
         output_schema: Dict[str, object] | Omit = omit,
         from_agent: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -587,9 +587,9 @@ class AsyncAgentResource(AsyncAPIResource):
                 {
                     "prompt": prompt,
                     "url": url,
-                    "agent_name": agent_name,
                     "input_schema": input_schema,
                     "metadata": metadata,
+                    "name": name,
                     "output_schema": output_schema,
                     "from_agent": from_agent,
                 },

nimble_python/resources/jobs/jobs.py

@@ -80,6 +80,18 @@ class JobsResource(SyncAPIResource):
         Create Job
 
         Args:
+          agent_name: Name of the agent to run.
+
+          name: Job name.
+
+          destination: Where a job writes its results.
+
+          display_name: Human-friendly job name shown in the UI.
+
+          inputs: Configuration for the input data a job processes.
+
+          schedule: Cron-based schedule controlling when a job runs automatically.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -128,6 +140,14 @@ class JobsResource(SyncAPIResource):
         Update Job
 
         Args:
+          destination: Where a job writes its results.
+
+          display_name: New display name.
+
+          inputs: Configuration for the input data a job processes.
+
+          schedule: Cron-based schedule controlling when a job runs automatically.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -352,6 +372,18 @@ class AsyncJobsResource(AsyncAPIResource):
         Create Job
 
         Args:
+          agent_name: Name of the agent to run.
+
+          name: Job name.
+
+          destination: Where a job writes its results.
+
+          display_name: Human-friendly job name shown in the UI.
+
+          inputs: Configuration for the input data a job processes.
+
+          schedule: Cron-based schedule controlling when a job runs automatically.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -400,6 +432,14 @@ class AsyncJobsResource(AsyncAPIResource):
         Update Job
 
         Args:
+          destination: Where a job writes its results.
+
+          display_name: New display name.
+
+          inputs: Configuration for the input data a job processes.
+
+          schedule: Cron-based schedule controlling when a job runs automatically.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request

nimble_python/types/agent_generate_params.py

@@ -20,12 +20,12 @@ class CreateAgentGenerationRequest(TypedDict, total=False):
 
     url: Required[str]
 
-    agent_name: Optional[str]
-
     input_schema: Dict[str, object]
 
     metadata: Optional[CreateAgentGenerationRequestMetadata]
 
+    name: Optional[str]
+
     output_schema: Dict[str, object]
 
 

nimble_python/types/agent_run_batch_response.py

@@ -22,7 +22,7 @@ class Task(BaseModel):
     input: object
     """Original input data for the task."""
 
-    state: Literal["pending", "success", "error"]
+    state: Literal["pending", "queued", "in_progress", "success", "error"]
     """Current state of the task."""
 
     status_url: str
@@ -31,7 +31,9 @@ class Task(BaseModel):
     account_name: Optional[str] = None
     """Account name that owns the task."""
 
-    api_type: Optional[Literal["web", "serp", "ecommerce", "social", "media", "agent", "extract", "fast-serp"]] = None
+    api_type: Optional[
+        Literal["web", "serp", "ecommerce", "social", "media", "agent", "extract", "fast-serp", "labs"]
+    ] = None
 
     batch_id: Optional[str] = None
     """Batch ID if this task is part of a batch."""
@@ -51,6 +53,9 @@ class Task(BaseModel):
     output_url: Optional[str] = None
     """Storage location of the output data."""
 
+    queue: Optional[str] = None
+    """Queue name the task was submitted to."""
+
     status_code: Optional[float] = None
     """HTTP status code from the task execution."""
 

nimble_python/types/batch_get_response.py

@@ -22,7 +22,7 @@ class Task(BaseModel):
     input: object
     """Original input data for the task."""
 
-    state: Literal["pending", "success", "error"]
+    state: Literal["pending", "queued", "in_progress", "success", "error"]
     """Current state of the task."""
 
     status_url: str
@@ -31,7 +31,9 @@ class Task(BaseModel):
     account_name: Optional[str] = None
     """Account name that owns the task."""
 
-    api_type: Optional[Literal["web", "serp", "ecommerce", "social", "media", "agent", "extract", "fast-serp"]] = None
+    api_type: Optional[
+        Literal["web", "serp", "ecommerce", "social", "media", "agent", "extract", "fast-serp", "labs"]
+    ] = None
 
     batch_id: Optional[str] = None
     """Batch ID if this task is part of a batch."""
@@ -51,6 +53,9 @@ class Task(BaseModel):
     output_url: Optional[str] = None
     """Storage location of the output data."""
 
+    queue: Optional[str] = None
+    """Queue name the task was submitted to."""
+
     status_code: Optional[float] = None
     """HTTP status code from the task execution."""
 

nimble_python/types/client_extract_async_params.py

@@ -37,6 +37,14 @@ class ClientExtractAsyncParams(TypedDict, total=False):
     url: Required[str]
     """Target URL to scrape"""
 
+    auto_driver_configuration: Dict[str, int]
+    """
+    Custom flow for the optimization engine: maps candidate names to the number of
+    attempts to spend on each candidate before advancing (0 skips it). Key order
+    defines the flow order. Providing it opts the request into 'auto' driver
+    selection.
+    """
+
     body: object
     """Request body for POST, PUT, PATCH methods"""
 
@@ -316,7 +324,7 @@ class ClientExtractAsyncParams(TypedDict, total=False):
     device: Literal["desktop", "mobile", "tablet"]
     """Device type for browser emulation"""
 
-    driver: Literal["vx6", "vx8", "vx8-pro", "vx10", "vx10-pro", "vx12", "vx12-pro", "media-vx6"]
+    driver: Literal["auto", "vx6", "vx8", "vx8-pro", "vx10", "vx10-pro", "vx12", "vx12-pro", "media-vx6", "fast-vx6"]
     """Browser driver to use"""
 
     expected_status_codes: Iterable[int]
@@ -1009,7 +1017,11 @@ BrowserAction: TypeAlias = Union[
 ]
 
 
-class CookiesUnionMember0(TypedDict, total=False, extra_items=object):  # type: ignore[call-arg]
+class CookiesUnionMember0(  # type: ignore[call-arg]
+    TypedDict,
+    total=False,
+    extra_items=object,  # pyright: ignore[reportGeneralTypeIssues]
+):
     creation: Optional[str]
 
     domain: Optional[str]

nimble_python/types/client_extract_batch_params.py

@@ -82,7 +82,11 @@ InputBrowserAction: TypeAlias = Union[
 ]
 
 
-class InputCookiesUnionMember0(TypedDict, total=False, extra_items=object):  # type: ignore[call-arg]
+class InputCookiesUnionMember0(  # type: ignore[call-arg]
+    TypedDict,
+    total=False,
+    extra_items=object,  # pyright: ignore[reportGeneralTypeIssues]
+):
     creation: Optional[str]
 
     domain: Optional[str]
@@ -146,6 +150,14 @@ class InputSession(TypedDict, total=False):
 
 
 class Input(TypedDict, total=False):
+    auto_driver_configuration: Dict[str, int]
+    """
+    Custom flow for the optimization engine: maps candidate names to the number of
+    attempts to spend on each candidate before advancing (0 skips it). Key order
+    defines the flow order. Providing it opts the request into 'auto' driver
+    selection.
+    """
+
     body: object
     """Request body for POST, PUT, PATCH methods"""
 
@@ -425,8 +437,11 @@ class Input(TypedDict, total=False):
     device: Literal["desktop", "mobile", "tablet"]
     """Device type for browser emulation"""
 
-    driver: Literal["vx6", "vx8", "vx8-pro", "vx10", "vx10-pro", "vx12", "vx12-pro", "media-vx6"]
-    """Browser driver to use"""
+    driver: Literal["auto", "vx6", "vx8", "vx8-pro", "vx10", "vx10-pro", "vx12", "vx12-pro", "media-vx6", "fast-vx6"]
+    """Browser driver to use.
+
+    Use 'auto' to let the engine select the candidate config per domain.
+    """
 
     expected_status_codes: Iterable[int]
     """Expected HTTP status codes for successful requests"""
@@ -1124,7 +1139,11 @@ SharedInputsBrowserAction: TypeAlias = Union[
 ]
 
 
-class SharedInputsCookiesUnionMember0(TypedDict, total=False, extra_items=object):  # type: ignore[call-arg]
+class SharedInputsCookiesUnionMember0(  # type: ignore[call-arg]
+    TypedDict,
+    total=False,
+    extra_items=object,  # pyright: ignore[reportGeneralTypeIssues]
+):
     creation: Optional[str]
 
     domain: Optional[str]
@@ -1193,6 +1212,14 @@ class SharedInputs(TypedDict, total=False):
     Can include extraction parameters and async/storage settings.
     """
 
+    auto_driver_configuration: Dict[str, int]
+    """
+    Custom flow for the optimization engine: maps candidate names to the number of
+    attempts to spend on each candidate before advancing (0 skips it). Key order
+    defines the flow order. Providing it opts the request into 'auto' driver
+    selection.
+    """
+
     body: object
     """Request body for POST, PUT, PATCH methods"""
 
@@ -1472,8 +1499,11 @@ class SharedInputs(TypedDict, total=False):
     device: Literal["desktop", "mobile", "tablet"]
     """Device type for browser emulation"""
 
-    driver: Literal["vx6", "vx8", "vx8-pro", "vx10", "vx10-pro", "vx12", "vx12-pro", "media-vx6"]
-    """Browser driver to use"""
+    driver: Literal["auto", "vx6", "vx8", "vx8-pro", "vx10", "vx10-pro", "vx12", "vx12-pro", "media-vx6", "fast-vx6"]
+    """Browser driver to use.
+
+    Use 'auto' to let the engine select the candidate config per domain.
+    """
 
     expected_status_codes: Iterable[int]
     """Expected HTTP status codes for successful requests"""

nimble_python/types/client_extract_params.py

@@ -37,6 +37,14 @@ class ClientExtractParams(TypedDict, total=False):
     url: Required[str]
     """Target URL to scrape"""
 
+    auto_driver_configuration: Dict[str, int]
+    """
+    Custom flow for the optimization engine: maps candidate names to the number of
+    attempts to spend on each candidate before advancing (0 skips it). Key order
+    defines the flow order. Providing it opts the request into 'auto' driver
+    selection.
+    """
+
     body: object
     """Request body for POST, PUT, PATCH methods"""
 
@@ -313,7 +321,7 @@ class ClientExtractParams(TypedDict, total=False):
     device: Literal["desktop", "mobile", "tablet"]
     """Device type for browser emulation"""
 
-    driver: Literal["vx6", "vx8", "vx8-pro", "vx10", "vx10-pro", "vx12", "vx12-pro", "media-vx6"]
+    driver: Literal["auto", "vx6", "vx8", "vx8-pro", "vx10", "vx10-pro", "vx12", "vx12-pro", "media-vx6", "fast-vx6"]
     """Browser driver to use"""
 
     expected_status_codes: Iterable[int]
@@ -994,7 +1002,11 @@ BrowserAction: TypeAlias = Union[
 ]
 
 
-class CookiesUnionMember0(TypedDict, total=False, extra_items=object):  # type: ignore[call-arg]
+class CookiesUnionMember0(  # type: ignore[call-arg]
+    TypedDict,
+    total=False,
+    extra_items=object,  # pyright: ignore[reportGeneralTypeIssues]
+):
     creation: Optional[str]
 
     domain: Optional[str]

nimble_python/types/crawl_run_params.py

@@ -118,7 +118,11 @@ ExtractOptionsBrowserAction: TypeAlias = Union[
 ]
 
 
-class ExtractOptionsCookiesUnionMember0(TypedDict, total=False, extra_items=object):  # type: ignore[call-arg]
+class ExtractOptionsCookiesUnionMember0(  # type: ignore[call-arg]
+    TypedDict,
+    total=False,
+    extra_items=object,  # pyright: ignore[reportGeneralTypeIssues]
+):
     creation: Optional[str]
 
     domain: Optional[str]
@@ -182,6 +186,14 @@ class ExtractOptionsSession(TypedDict, total=False):
 
 
 class ExtractOptions(TypedDict, total=False):
+    auto_driver_configuration: Dict[str, int]
+    """
+    Custom flow for the optimization engine: maps candidate names to the number of
+    attempts to spend on each candidate before advancing (0 skips it). Key order
+    defines the flow order. Providing it opts the request into 'auto' driver
+    selection.
+    """
+
     body: object
     """Request body for POST, PUT, PATCH methods"""
 
@@ -458,7 +470,7 @@ class ExtractOptions(TypedDict, total=False):
     device: Literal["desktop", "mobile", "tablet"]
     """Device type for browser emulation"""
 
-    driver: Literal["vx6", "vx8", "vx8-pro", "vx10", "vx10-pro", "vx12", "vx12-pro", "media-vx6"]
+    driver: Literal["auto", "vx6", "vx8", "vx8-pro", "vx10", "vx10-pro", "vx12", "vx12-pro", "media-vx6", "fast-vx6"]
     """Browser driver to use"""
 
     expected_status_codes: Iterable[int]

nimble_python/types/extract_async_response.py

@@ -24,7 +24,7 @@ class Task(BaseModel):
     input: object
     """Original input data for the task."""
 
-    state: Literal["pending", "success", "error"]
+    state: Literal["pending", "queued", "in_progress", "success", "error"]
     """Current state of the task."""
 
     status_url: str
@@ -33,7 +33,9 @@ class Task(BaseModel):
     account_name: Optional[str] = None
     """Account name that owns the task."""
 
-    api_type: Optional[Literal["web", "serp", "ecommerce", "social", "media", "agent", "extract", "fast-serp"]] = None
+    api_type: Optional[
+        Literal["web", "serp", "ecommerce", "social", "media", "agent", "extract", "fast-serp", "labs"]
+    ] = None
 
     batch_id: Optional[str] = None
     """Batch ID if this task is part of a batch."""
@@ -53,6 +55,9 @@ class Task(BaseModel):
     output_url: Optional[str] = None
     """Storage location of the output data."""
 
+    queue: Optional[str] = None
+    """Queue name the task was submitted to."""
+
     status_code: Optional[float] = None
     """HTTP status code from the task execution."""
 

nimble_python/types/extract_batch_response.py

@@ -22,7 +22,7 @@ class Task(BaseModel):
     input: object
     """Original input data for the task."""
 
-    state: Literal["pending", "success", "error"]
+    state: Literal["pending", "queued", "in_progress", "success", "error"]
     """Current state of the task."""
 
     status_url: str
@@ -31,7 +31,9 @@ class Task(BaseModel):
     account_name: Optional[str] = None
     """Account name that owns the task."""
 
-    api_type: Optional[Literal["web", "serp", "ecommerce", "social", "media", "agent", "extract", "fast-serp"]] = None
+    api_type: Optional[
+        Literal["web", "serp", "ecommerce", "social", "media", "agent", "extract", "fast-serp", "labs"]
+    ] = None
 
     batch_id: Optional[str] = None
     """Batch ID if this task is part of a batch."""
@@ -51,6 +53,9 @@ class Task(BaseModel):
     output_url: Optional[str] = None
     """Storage location of the output data."""
 
+    queue: Optional[str] = None
+    """Queue name the task was submitted to."""
+
     status_code: Optional[float] = None
     """HTTP status code from the task execution."""