nimble_python 0.14.0__py3-none-any.whl → 0.16.0__py3-none-any.whl

nimble_python/_qs.py

@@ -101,7 +101,10 @@ class Querystring:
                     items.extend(self._stringify_item(key, item, opts))
                 return items
             elif array_format == "indices":
-                raise NotImplementedError("The array indices format is not supported yet")
+                items = []
+                for i, item in enumerate(value):
+                    items.extend(self._stringify_item(f"{key}[{i}]", item, opts))
+                return items
             elif array_format == "brackets":
                 items = []
                 key = key + "[]"

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.14.0"  # x-release-please-version
+__version__ = "0.16.0"  # x-release-please-version

nimble_python/resources/agent.py

@@ -2,14 +2,21 @@
 
 from __future__ import annotations
 
-from typing import Dict, Optional
-from typing_extensions import Literal
+from typing import Dict, List, Iterable, Optional
+from typing_extensions import Literal, overload
 
 import httpx
 
-from ..types import agent_run_params, agent_list_params, agent_run_async_params
+from ..types import (
+    agent_run_params,
+    agent_list_params,
+    agent_publish_params,
+    agent_generate_params,
+    agent_run_async_params,
+    agent_run_batch_params,
+)
 from .._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
-from .._utils import path_template, maybe_transform, async_maybe_transform
+from .._utils import path_template, required_args, maybe_transform, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
 from .._response import (
@@ -22,7 +29,11 @@ from .._base_client import make_request_options
 from ..types.agent_get_response import AgentGetResponse
 from ..types.agent_run_response import AgentRunResponse
 from ..types.agent_list_response import AgentListResponse
+from ..types.agent_publish_response import AgentPublishResponse
+from ..types.agent_generate_response import AgentGenerateResponse
 from ..types.agent_run_async_response import AgentRunAsyncResponse
+from ..types.agent_run_batch_response import AgentRunBatchResponse
+from ..types.agent_get_generation_response import AgentGetGenerationResponse
 
 __all__ = ["AgentResource", "AsyncAgentResource"]
 
@@ -63,7 +74,7 @@ class AgentResource(SyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> AgentListResponse:
         """
-        List Templates
+        List Agent Templates
 
         Args:
           limit: Number of results per page
@@ -105,6 +116,102 @@ class AgentResource(SyncAPIResource):
             cast_to=AgentListResponse,
         )
 
+    @overload
+    def generate(
+        self,
+        *,
+        agent_name: str,
+        prompt: str,
+        url: str,
+        input_schema: object | Omit = omit,
+        metadata: Optional[object] | Omit = omit,
+        output_schema: 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.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> AgentGenerateResponse:
+        """
+        Create Agent Generation
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        ...
+
+    @overload
+    def generate(
+        self,
+        *,
+        from_agent: str,
+        prompt: str,
+        # 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.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> AgentGenerateResponse:
+        """
+        Create Agent Generation
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        ...
+
+    @required_args(["agent_name", "prompt", "url"], ["from_agent", "prompt"])
+    def generate(
+        self,
+        *,
+        agent_name: str | Omit = omit,
+        prompt: str,
+        url: str | Omit = omit,
+        input_schema: object | Omit = omit,
+        metadata: Optional[object] | Omit = omit,
+        output_schema: 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.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> AgentGenerateResponse:
+        return self._post(
+            "/v1/agents/generations",
+            body=maybe_transform(
+                {
+                    "agent_name": agent_name,
+                    "prompt": prompt,
+                    "url": url,
+                    "input_schema": input_schema,
+                    "metadata": metadata,
+                    "output_schema": output_schema,
+                    "from_agent": from_agent,
+                },
+                agent_generate_params.AgentGenerateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=AgentGenerateResponse,
+        )
+
     def get(
         self,
         template_name: str,
@@ -117,7 +224,7 @@ class AgentResource(SyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> AgentGetResponse:
         """
-        Get Template
+        Get Agent Template
 
         Args:
           extra_headers: Send extra headers
@@ -138,11 +245,80 @@ class AgentResource(SyncAPIResource):
             cast_to=AgentGetResponse,
         )
 
+    def get_generation(
+        self,
+        generation_id: str,
+        *,
+        # 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.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> AgentGetGenerationResponse:
+        """
+        Get Agent Generation
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not generation_id:
+            raise ValueError(f"Expected a non-empty value for `generation_id` but received {generation_id!r}")
+        return self._get(
+            path_template("/v1/agents/generations/{generation_id}", generation_id=generation_id),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=AgentGetGenerationResponse,
+        )
+
+    def publish(
+        self,
+        agent_name: str,
+        *,
+        version_id: str,
+        # 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.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> AgentPublishResponse:
+        """
+        Publish Agent Version
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not agent_name:
+            raise ValueError(f"Expected a non-empty value for `agent_name` but received {agent_name!r}")
+        return self._post(
+            path_template("/v1/agents/{agent_name}/publish", agent_name=agent_name),
+            body=maybe_transform({"version_id": version_id}, agent_publish_params.AgentPublishParams),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=AgentPublishResponse,
+        )
+
     def run(
         self,
         *,
         agent: str,
         params: Dict[str, object],
+        formats: List[Literal["html", "markdown", "screenshot", "headers"]] | Omit = omit,
         localization: bool | 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.
@@ -151,10 +327,13 @@ class AgentResource(SyncAPIResource):
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> AgentRunResponse:
-        """
-        Execute WSA Realtime Endpoint
+        """Execute WSA Realtime Endpoint
 
         Args:
+          formats: Response formats to include.
+
+        All disabled by default.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -169,6 +348,7 @@ class AgentResource(SyncAPIResource):
                 {
                     "agent": agent,
                     "params": params,
+                    "formats": formats,
                     "localization": localization,
                 },
                 agent_run_params.AgentRunParams,
@@ -185,6 +365,7 @@ class AgentResource(SyncAPIResource):
         agent: str,
         params: Dict[str, object],
         callback_url: str | Omit = omit,
+        formats: List[Literal["html", "markdown", "screenshot", "headers"]] | Omit = omit,
         localization: bool | Omit = omit,
         storage_compress: bool | Omit = omit,
         storage_object_name: str | Omit = omit,
@@ -203,6 +384,8 @@ class AgentResource(SyncAPIResource):
         Args:
           callback_url: URL to call back when async operation completes
 
+          formats: Response formats to include. All disabled by default.
+
           storage_compress: Whether to compress stored data
 
           storage_object_name: Custom name for the stored object
@@ -226,6 +409,7 @@ class AgentResource(SyncAPIResource):
                     "agent": agent,
                     "params": params,
                     "callback_url": callback_url,
+                    "formats": formats,
                     "localization": localization,
                     "storage_compress": storage_compress,
                     "storage_object_name": storage_object_name,
@@ -240,6 +424,45 @@ class AgentResource(SyncAPIResource):
             cast_to=AgentRunAsyncResponse,
         )
 
+    def run_batch(
+        self,
+        *,
+        inputs: Iterable[agent_run_batch_params.Input],
+        shared_inputs: agent_run_batch_params.SharedInputs,
+        # 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.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> AgentRunBatchResponse:
+        """
+        Execute WSA Batch Endpoint
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return self._post(
+            "/v1/agents/batch",
+            body=maybe_transform(
+                {
+                    "inputs": inputs,
+                    "shared_inputs": shared_inputs,
+                },
+                agent_run_batch_params.AgentRunBatchParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=AgentRunBatchResponse,
+        )
+
 
 class AsyncAgentResource(AsyncAPIResource):
     @cached_property
@@ -277,7 +500,7 @@ class AsyncAgentResource(AsyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> AgentListResponse:
         """
-        List Templates
+        List Agent Templates
 
         Args:
           limit: Number of results per page
@@ -319,6 +542,102 @@ class AsyncAgentResource(AsyncAPIResource):
             cast_to=AgentListResponse,
         )
 
+    @overload
+    async def generate(
+        self,
+        *,
+        agent_name: str,
+        prompt: str,
+        url: str,
+        input_schema: object | Omit = omit,
+        metadata: Optional[object] | Omit = omit,
+        output_schema: 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.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> AgentGenerateResponse:
+        """
+        Create Agent Generation
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        ...
+
+    @overload
+    async def generate(
+        self,
+        *,
+        from_agent: str,
+        prompt: str,
+        # 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.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> AgentGenerateResponse:
+        """
+        Create Agent Generation
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        ...
+
+    @required_args(["agent_name", "prompt", "url"], ["from_agent", "prompt"])
+    async def generate(
+        self,
+        *,
+        agent_name: str | Omit = omit,
+        prompt: str,
+        url: str | Omit = omit,
+        input_schema: object | Omit = omit,
+        metadata: Optional[object] | Omit = omit,
+        output_schema: 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.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> AgentGenerateResponse:
+        return await self._post(
+            "/v1/agents/generations",
+            body=await async_maybe_transform(
+                {
+                    "agent_name": agent_name,
+                    "prompt": prompt,
+                    "url": url,
+                    "input_schema": input_schema,
+                    "metadata": metadata,
+                    "output_schema": output_schema,
+                    "from_agent": from_agent,
+                },
+                agent_generate_params.AgentGenerateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=AgentGenerateResponse,
+        )
+
     async def get(
         self,
         template_name: str,
@@ -331,7 +650,7 @@ class AsyncAgentResource(AsyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> AgentGetResponse:
         """
-        Get Template
+        Get Agent Template
 
         Args:
           extra_headers: Send extra headers
@@ -352,11 +671,80 @@ class AsyncAgentResource(AsyncAPIResource):
             cast_to=AgentGetResponse,
         )
 
+    async def get_generation(
+        self,
+        generation_id: str,
+        *,
+        # 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.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> AgentGetGenerationResponse:
+        """
+        Get Agent Generation
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not generation_id:
+            raise ValueError(f"Expected a non-empty value for `generation_id` but received {generation_id!r}")
+        return await self._get(
+            path_template("/v1/agents/generations/{generation_id}", generation_id=generation_id),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=AgentGetGenerationResponse,
+        )
+
+    async def publish(
+        self,
+        agent_name: str,
+        *,
+        version_id: str,
+        # 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.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> AgentPublishResponse:
+        """
+        Publish Agent Version
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not agent_name:
+            raise ValueError(f"Expected a non-empty value for `agent_name` but received {agent_name!r}")
+        return await self._post(
+            path_template("/v1/agents/{agent_name}/publish", agent_name=agent_name),
+            body=await async_maybe_transform({"version_id": version_id}, agent_publish_params.AgentPublishParams),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=AgentPublishResponse,
+        )
+
     async def run(
         self,
         *,
         agent: str,
         params: Dict[str, object],
+        formats: List[Literal["html", "markdown", "screenshot", "headers"]] | Omit = omit,
         localization: bool | 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.
@@ -365,10 +753,13 @@ class AsyncAgentResource(AsyncAPIResource):
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> AgentRunResponse:
-        """
-        Execute WSA Realtime Endpoint
+        """Execute WSA Realtime Endpoint
 
         Args:
+          formats: Response formats to include.
+
+        All disabled by default.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -383,6 +774,7 @@ class AsyncAgentResource(AsyncAPIResource):
                 {
                     "agent": agent,
                     "params": params,
+                    "formats": formats,
                     "localization": localization,
                 },
                 agent_run_params.AgentRunParams,
@@ -399,6 +791,7 @@ class AsyncAgentResource(AsyncAPIResource):
         agent: str,
         params: Dict[str, object],
         callback_url: str | Omit = omit,
+        formats: List[Literal["html", "markdown", "screenshot", "headers"]] | Omit = omit,
         localization: bool | Omit = omit,
         storage_compress: bool | Omit = omit,
         storage_object_name: str | Omit = omit,
@@ -417,6 +810,8 @@ class AsyncAgentResource(AsyncAPIResource):
         Args:
           callback_url: URL to call back when async operation completes
 
+          formats: Response formats to include. All disabled by default.
+
           storage_compress: Whether to compress stored data
 
           storage_object_name: Custom name for the stored object
@@ -440,6 +835,7 @@ class AsyncAgentResource(AsyncAPIResource):
                     "agent": agent,
                     "params": params,
                     "callback_url": callback_url,
+                    "formats": formats,
                     "localization": localization,
                     "storage_compress": storage_compress,
                     "storage_object_name": storage_object_name,
@@ -454,6 +850,45 @@ class AsyncAgentResource(AsyncAPIResource):
             cast_to=AgentRunAsyncResponse,
         )
 
+    async def run_batch(
+        self,
+        *,
+        inputs: Iterable[agent_run_batch_params.Input],
+        shared_inputs: agent_run_batch_params.SharedInputs,
+        # 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.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> AgentRunBatchResponse:
+        """
+        Execute WSA Batch Endpoint
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return await self._post(
+            "/v1/agents/batch",
+            body=await async_maybe_transform(
+                {
+                    "inputs": inputs,
+                    "shared_inputs": shared_inputs,
+                },
+                agent_run_batch_params.AgentRunBatchParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=AgentRunBatchResponse,
+        )
+
 
 class AgentResourceWithRawResponse:
     def __init__(self, agent: AgentResource) -> None:
@@ -462,15 +897,27 @@ class AgentResourceWithRawResponse:
         self.list = to_raw_response_wrapper(
             agent.list,
         )
+        self.generate = to_raw_response_wrapper(
+            agent.generate,
+        )
         self.get = to_raw_response_wrapper(
             agent.get,
         )
+        self.get_generation = to_raw_response_wrapper(
+            agent.get_generation,
+        )
+        self.publish = to_raw_response_wrapper(
+            agent.publish,
+        )
         self.run = to_raw_response_wrapper(
             agent.run,
         )
         self.run_async = to_raw_response_wrapper(
             agent.run_async,
         )
+        self.run_batch = to_raw_response_wrapper(
+            agent.run_batch,
+        )
 
 
 class AsyncAgentResourceWithRawResponse:
@@ -480,15 +927,27 @@ class AsyncAgentResourceWithRawResponse:
         self.list = async_to_raw_response_wrapper(
             agent.list,
         )
+        self.generate = async_to_raw_response_wrapper(
+            agent.generate,
+        )
         self.get = async_to_raw_response_wrapper(
             agent.get,
         )
+        self.get_generation = async_to_raw_response_wrapper(
+            agent.get_generation,
+        )
+        self.publish = async_to_raw_response_wrapper(
+            agent.publish,
+        )
         self.run = async_to_raw_response_wrapper(
             agent.run,
         )
         self.run_async = async_to_raw_response_wrapper(
             agent.run_async,
         )
+        self.run_batch = async_to_raw_response_wrapper(
+            agent.run_batch,
+        )
 
 
 class AgentResourceWithStreamingResponse:
@@ -498,15 +957,27 @@ class AgentResourceWithStreamingResponse:
         self.list = to_streamed_response_wrapper(
             agent.list,
         )
+        self.generate = to_streamed_response_wrapper(
+            agent.generate,
+        )
         self.get = to_streamed_response_wrapper(
             agent.get,
         )
+        self.get_generation = to_streamed_response_wrapper(
+            agent.get_generation,
+        )
+        self.publish = to_streamed_response_wrapper(
+            agent.publish,
+        )
         self.run = to_streamed_response_wrapper(
             agent.run,
         )
         self.run_async = to_streamed_response_wrapper(
             agent.run_async,
         )
+        self.run_batch = to_streamed_response_wrapper(
+            agent.run_batch,
+        )
 
 
 class AsyncAgentResourceWithStreamingResponse:
@@ -516,12 +987,24 @@ class AsyncAgentResourceWithStreamingResponse:
         self.list = async_to_streamed_response_wrapper(
             agent.list,
         )
+        self.generate = async_to_streamed_response_wrapper(
+            agent.generate,
+        )
         self.get = async_to_streamed_response_wrapper(
             agent.get,
         )
+        self.get_generation = async_to_streamed_response_wrapper(
+            agent.get_generation,
+        )
+        self.publish = async_to_streamed_response_wrapper(
+            agent.publish,
+        )
         self.run = async_to_streamed_response_wrapper(
             agent.run,
         )
         self.run_async = async_to_streamed_response_wrapper(
             agent.run_async,
         )
+        self.run_batch = async_to_streamed_response_wrapper(
+            agent.run_batch,
+        )

nimble_python/types/__init__.py

@@ -34,15 +34,22 @@ from .crawl_run_response import CrawlRunResponse as CrawlRunResponse
 from .task_list_response import TaskListResponse as TaskListResponse
 from .agent_list_response import AgentListResponse as AgentListResponse
 from .crawl_list_response import CrawlListResponse as CrawlListResponse
+from .agent_publish_params import AgentPublishParams as AgentPublishParams
 from .client_search_params import ClientSearchParams as ClientSearchParams
+from .agent_generate_params import AgentGenerateParams as AgentGenerateParams
 from .client_extract_params import ClientExtractParams as ClientExtractParams
 from .crawl_status_response import CrawlStatusResponse as CrawlStatusResponse
 from .task_results_response import TaskResultsResponse as TaskResultsResponse
+from .agent_publish_response import AgentPublishResponse as AgentPublishResponse
 from .agent_run_async_params import AgentRunAsyncParams as AgentRunAsyncParams
+from .agent_run_batch_params import AgentRunBatchParams as AgentRunBatchParams
 from .extract_async_response import ExtractAsyncResponse as ExtractAsyncResponse
 from .extract_batch_response import ExtractBatchResponse as ExtractBatchResponse
+from .agent_generate_response import AgentGenerateResponse as AgentGenerateResponse
 from .batch_progress_response import BatchProgressResponse as BatchProgressResponse
 from .agent_run_async_response import AgentRunAsyncResponse as AgentRunAsyncResponse
+from .agent_run_batch_response import AgentRunBatchResponse as AgentRunBatchResponse
 from .crawl_terminate_response import CrawlTerminateResponse as CrawlTerminateResponse
 from .client_extract_async_params import ClientExtractAsyncParams as ClientExtractAsyncParams
 from .client_extract_batch_params import ClientExtractBatchParams as ClientExtractBatchParams
+from .agent_get_generation_response import AgentGetGenerationResponse as AgentGetGenerationResponse

nimble_python/types/agent_generate_params.py

@@ -0,0 +1,31 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Union, Optional
+from typing_extensions import Required, TypeAlias, TypedDict
+
+__all__ = ["AgentGenerateParams", "CreateAgentGenerationRequest", "CreateAgentRefinementRequest"]
+
+
+class CreateAgentGenerationRequest(TypedDict, total=False):
+    agent_name: Required[str]
+
+    prompt: Required[str]
+
+    url: Required[str]
+
+    input_schema: object
+
+    metadata: Optional[object]
+
+    output_schema: object
+
+
+class CreateAgentRefinementRequest(TypedDict, total=False):
+    from_agent: Required[str]
+
+    prompt: Required[str]
+
+
+AgentGenerateParams: TypeAlias = Union[CreateAgentGenerationRequest, CreateAgentRefinementRequest]

nimble_python/types/agent_generate_response.py

@@ -0,0 +1,32 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from datetime import datetime
+
+from .._models import BaseModel
+
+__all__ = ["AgentGenerateResponse"]
+
+
+class AgentGenerateResponse(BaseModel):
+    id: str
+
+    status: str
+
+    agent_name: Optional[str] = None
+
+    completed_at: Optional[datetime] = None
+
+    created_at: Optional[datetime] = None
+
+    error: Optional[str] = None
+
+    generated_version: Optional[object] = None
+
+    generated_version_id: Optional[str] = None
+
+    source_version_id: Optional[str] = None
+
+    started_at: Optional[datetime] = None
+
+    summary: Optional[str] = None

nimble_python/types/agent_get_generation_response.py

@@ -0,0 +1,32 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from datetime import datetime
+
+from .._models import BaseModel
+
+__all__ = ["AgentGetGenerationResponse"]
+
+
+class AgentGetGenerationResponse(BaseModel):
+    id: str
+
+    status: str
+
+    agent_name: Optional[str] = None
+
+    completed_at: Optional[datetime] = None
+
+    created_at: Optional[datetime] = None
+
+    error: Optional[str] = None
+
+    generated_version: Optional[object] = None
+
+    generated_version_id: Optional[str] = None
+
+    source_version_id: Optional[str] = None
+
+    started_at: Optional[datetime] = None
+
+    summary: Optional[str] = None

nimble_python/types/agent_publish_params.py

@@ -0,0 +1,11 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Required, TypedDict
+
+__all__ = ["AgentPublishParams"]
+
+
+class AgentPublishParams(TypedDict, total=False):
+    version_id: Required[str]

nimble_python/types/agent_publish_response.py

@@ -0,0 +1,11 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from .._models import BaseModel
+
+__all__ = ["AgentPublishResponse"]
+
+
+class AgentPublishResponse(BaseModel):
+    agent_name: str
+
+    published_version_id: str

nimble_python/types/agent_run_async_params.py

@@ -2,8 +2,8 @@
 
 from __future__ import annotations
 
-from typing import Dict
-from typing_extensions import Required, TypedDict
+from typing import Dict, List
+from typing_extensions import Literal, Required, TypedDict
 
 __all__ = ["AgentRunAsyncParams"]
 
@@ -16,6 +16,9 @@ class AgentRunAsyncParams(TypedDict, total=False):
     callback_url: str
     """URL to call back when async operation completes"""
 
+    formats: List[Literal["html", "markdown", "screenshot", "headers"]]
+    """Response formats to include. All disabled by default."""
+
     localization: bool
 
     storage_compress: bool

nimble_python/types/agent_run_batch_params.py

@@ -0,0 +1,34 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Dict, List, Iterable
+from typing_extensions import Literal, Required, TypedDict
+
+__all__ = ["AgentRunBatchParams", "Input", "SharedInputs"]
+
+
+class AgentRunBatchParams(TypedDict, total=False):
+    inputs: Required[Iterable[Input]]
+
+    shared_inputs: Required[SharedInputs]
+
+
+class Input(TypedDict, total=False):
+    formats: List[Literal["html", "markdown", "screenshot", "headers"]]
+    """Response formats to include. All disabled by default."""
+
+    localization: bool
+
+    params: Dict[str, object]
+
+
+class SharedInputs(TypedDict, total=False):
+    agent: Required[str]
+
+    formats: List[Literal["html", "markdown", "screenshot", "headers"]]
+    """Response formats to include. All disabled by default."""
+
+    localization: bool
+
+    params: Dict[str, object]

nimble_python/types/agent_run_batch_response.py

@@ -0,0 +1,68 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+from typing_extensions import Literal
+
+from pydantic import Field as FieldInfo
+
+from .._models import BaseModel
+
+__all__ = ["AgentRunBatchResponse", "Task"]
+
+
+class Task(BaseModel):
+    id: str
+    """Unique task identifier."""
+
+    api_query: object = FieldInfo(alias="_query")
+
+    created_at: str
+    """Timestamp when the task was created."""
+
+    input: object
+    """Original input data for the task."""
+
+    state: Literal["pending", "success", "error"]
+    """Current state of the task."""
+
+    status_url: str
+    """URL for checking the task status."""
+
+    account_name: Optional[str] = None
+    """Account name that owns the task."""
+
+    api_type: Optional[Literal["web", "serp", "ecommerce", "social", "media", "agent", "extract"]] = None
+
+    batch_id: Optional[str] = None
+    """Batch ID if this task is part of a batch."""
+
+    download_url: Optional[str] = None
+    """URL for downloading the task results."""
+
+    error: Optional[str] = None
+    """Error message if the task failed."""
+
+    error_type: Optional[str] = None
+    """Classification of the error type."""
+
+    modified_at: Optional[str] = None
+    """Timestamp when the task was last modified."""
+
+    output_url: Optional[str] = None
+    """Storage location of the output data."""
+
+    status_code: Optional[float] = None
+    """HTTP status code from the task execution."""
+
+
+class AgentRunBatchResponse(BaseModel):
+    """Response when a batch of extract tasks is created successfully."""
+
+    batch_id: str
+    """Unique identifier for the batch."""
+
+    batch_size: float
+    """Number of tasks in the batch."""
+
+    tasks: List[Task]
+    """List of created tasks."""

nimble_python/types/agent_run_params.py

@@ -2,8 +2,8 @@
 
 from __future__ import annotations
 
-from typing import Dict
-from typing_extensions import Required, TypedDict
+from typing import Dict, List
+from typing_extensions import Literal, Required, TypedDict
 
 __all__ = ["AgentRunParams"]
 
@@ -13,4 +13,7 @@ class AgentRunParams(TypedDict, total=False):
 
     params: Required[Dict[str, object]]
 
+    formats: List[Literal["html", "markdown", "screenshot", "headers"]]
+    """Response formats to include. All disabled by default."""
+
     localization: bool

nimble_python/types/agent_run_response.py

@@ -246,6 +246,9 @@ class Data(BaseModel):
     network_capture: Optional[List[DataNetworkCapture]] = None
     """The network capture data collected during the task."""
 
+    pages_html: Optional[List[str]] = None
+    """Individual HTML content of each pagination page, before merging."""
+
     parsing: Optional[DataParsing] = None
     """The parsing results extracted from the HTML & network content."""
 

nimble_python/types/extract_response.py

@@ -246,6 +246,9 @@ class Data(BaseModel):
     network_capture: Optional[List[DataNetworkCapture]] = None
     """The network capture data collected during the task."""
 
+    pages_html: Optional[List[str]] = None
+    """Individual HTML content of each pagination page, before merging."""
+
     parsing: Optional[DataParsing] = None
     """The parsing results extracted from the HTML & network content."""
 

{nimble_python-0.14.0.dist-info → nimble_python-0.16.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: nimble_python
-Version: 0.14.0
+Version: 0.16.0
 Summary: The official Python library for the nimble API
 Project-URL: Homepage, https://github.com/Nimbleway/nimble-python
 Project-URL: Repository, https://github.com/Nimbleway/nimble-python

{nimble_python-0.14.0.dist-info → nimble_python-0.16.0.dist-info}/RECORD

@@ -6,12 +6,12 @@ nimble_python/_constants.py,sha256=giUwuSKDnpjRLaHvgzErnXB3CSYjKMaI5xnBQJI-23s,4
 nimble_python/_exceptions.py,sha256=x1M0s_wI9jkJ5ARR07APUuDZfxl7BjQ53snlFU-jAoQ,3220
 nimble_python/_files.py,sha256=KnEzGi_O756MvKyJ4fOCW_u3JhOeWPQ4RsmDvqihDQU,3545
 nimble_python/_models.py,sha256=R3MpO2z4XhTAnD3ObEks32suRXleF1g7BEgQTOLIxTs,32112
-nimble_python/_qs.py,sha256=craIKyvPktJ94cvf9zn8j8ekG9dWJzhWv0ob34lIOv4,4828
+nimble_python/_qs.py,sha256=ExvESqZz_a5ZALZG1aR2rkzE35o9Rmk0yPMQtKN0DTo,4924
 nimble_python/_resource.py,sha256=TDtzjUMbNHRGQNq-4HJqce3T3THBD5Q7ESrONYPvKdI,1100
 nimble_python/_response.py,sha256=hrLOCn3D0gCapH31eR9BVOgBpTYliUTVUsdeWc1J76A,28983
 nimble_python/_streaming.py,sha256=StqGzUAIG9gQNCkRQnEAbN3IyiX4hDdCp2QR7_DANw8,10550
 nimble_python/_types.py,sha256=ZhbxWgjPQAqNY4ePn8aUJV26mm5oDBrZqskA1XyjlQ0,7601
-nimble_python/_version.py,sha256=FySRe9QcDTqiUANTKc2AucieJn5w2ihGxRF2CsNCAIY,166
+nimble_python/_version.py,sha256=Lb-CsoIL_IraYv5EV1eucIYPOAIePAbpXjQtYHRX5vQ,166
 nimble_python/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nimble_python/_utils/__init__.py,sha256=DJEWihdzRWrZI23vwocBkgvZCwn9iG7yV0c_9vKaysU,2355
 nimble_python/_utils/_compat.py,sha256=33246eDcl3pwL6kWsEhVuT4Akrd8gZEW9LPTm465ohk,1231
@@ -29,18 +29,25 @@ nimble_python/_utils/_typing.py,sha256=N_5PPuFNsaygbtA_npZd98SVN1LQQvFTKL6bkWPBZ
 nimble_python/_utils/_utils.py,sha256=ugfUaneOK7I8h9b3656flwf5u_kthY0gvNuqvgOLoSU,12252
 nimble_python/lib/.keep,sha256=wuNrz-5SXo3jJaJOJgz4vFHM41YH_g20F5cRQo0vLes,224
 nimble_python/resources/__init__.py,sha256=Wh5Ns-DfhwV5B1NL1sauYogLEctZiqtJxLuEVQzUvVw,1876
-nimble_python/resources/agent.py,sha256=HDsKDK7zYFSplZcG9pVFAdvEptZJ2t2A5XRqx3IT7s4,19088
+nimble_python/resources/agent.py,sha256=lA2lZIr6D_yOXtyBUVM8nEdR1vsVtRQCw138Z6cqU-4,37893
 nimble_python/resources/batches.py,sha256=LxPXo7mpDcXmtJAv7prVnDpJHkhHypix6ZNSxh9kpQU,11688
 nimble_python/resources/crawl.py,sha256=GOMDGANIPAfay-J3ujW9ghlK-h-aIy99ChIKRN-pZK8,20651
 nimble_python/resources/tasks.py,sha256=8rV6YZNpx8rq_83L8UZvt7HlPlRhY8SGrbVQAFDjBtM,13021
-nimble_python/types/__init__.py,sha256=al8UTqmM7bIeUUKbqDTHc6-23McbQAmVjF12tfN8r5g,2745
+nimble_python/types/__init__.py,sha256=IMIgUvouuZ_JeP5p7HBJEnwQJdnSBt9YopnYPY8y2Tw,3327
+nimble_python/types/agent_generate_params.py,sha256=2YZjjZy7EDa-ISBftfjtgDh3wOba8UmmIVimYkILrG8,766
+nimble_python/types/agent_generate_response.py,sha256=NaLHpAtIMssf5o094YfpsphnQXRtdIyAbM7RiVa5ij8,664
+nimble_python/types/agent_get_generation_response.py,sha256=phzi3gsbrruea_eNcZIYJu9f8jYw26Nc_1sOyBsxIss,674
 nimble_python/types/agent_get_response.py,sha256=qQtjNztWHGWGJ_LKLn2-VtQlzv2AfkPvu0QaADRXqIg,1131
 nimble_python/types/agent_list_params.py,sha256=yUwgtiXrzfEXd3I--X_ISovY2BSHKXDvFZ2XUCUR9ew,670
 nimble_python/types/agent_list_response.py,sha256=kBEVOD9nlby5QsfaDkjPLXDJyQotph3HBo321jP8rb4,599
-nimble_python/types/agent_run_async_params.py,sha256=vaDEeQx9X1XWJlUcLSa0Bht0c65obrSDIb-AfHPWQ1w,725
+nimble_python/types/agent_publish_params.py,sha256=GNnBk5qgq1T1JMotVJSVrY_bW6CJHRRuF5H-9ofj9zE,289
+nimble_python/types/agent_publish_response.py,sha256=VEyf8cMKVqgWRhOGPDuYevVhwPcE-oynoBc8UPoXVs8,247
+nimble_python/types/agent_run_async_params.py,sha256=kma0zTwV2RK4BIDnOAneQ1Vx1sLZqSezBs403ODdoaI,877
 nimble_python/types/agent_run_async_response.py,sha256=aawcjhtW6FaWxIPcg-l-1yIhsQDLUoYOaoSf_MLJ67s,321
-nimble_python/types/agent_run_params.py,sha256=WbmK8oa-Iv0QS1I8X6AAXhrbbMPzNACV5IHSq8rdwGg,365
-nimble_python/types/agent_run_response.py,sha256=xvHQbGDhgaWyZy4gcNMfYEQ4YLSUVRiF54Qm2SwIE2g,8036
+nimble_python/types/agent_run_batch_params.py,sha256=R6yqYKx4w4-WFitwU1uPUu4DskYIc32c9NeKccm8UQ4,909
+nimble_python/types/agent_run_batch_response.py,sha256=0dtYQ6epddVqO2sdbeTDJyGGGAob24Cl3y1qC2k2AT4,1776
+nimble_python/types/agent_run_params.py,sha256=hC7CkPKvFoJf-KKIifJNKwDv-O_dS6Mk65k-A7up_RY,517
+nimble_python/types/agent_run_response.py,sha256=0kPTccMFP3kBu-9legG_dRP5HSDNiafqDgZkaYxnqFA,8155
 nimble_python/types/batch_get_response.py,sha256=5MIDHw3xt4UPdZY7eeYMDTJHbvyP4fCWNFMOLmuxcGM,2117
 nimble_python/types/batch_progress_response.py,sha256=oHEHXclc0NnI3TiGMalfElhZ15rfvR7nZRub9Gg_nqs,726
 nimble_python/types/client_extract_async_params.py,sha256=P26kzOH_UudChy5ZcCfoVz5YE-RVNCYifMAIAAeXErI,19339
@@ -56,7 +63,7 @@ nimble_python/types/crawl_status_response.py,sha256=rCNZkRNV65KzZpQCqbSX2FIW6kLh
 nimble_python/types/crawl_terminate_response.py,sha256=Kw0udMxFAfBZcpKfIpK8Dt1wKyjFM5zUgGTiBNl1K4c,271
 nimble_python/types/extract_async_response.py,sha256=__3OdwaSstOvDzaQnMv2w4cWeD4GN0-2zZ7YxdNDufs,1786
 nimble_python/types/extract_batch_response.py,sha256=AjiTdf7eHP_kE3TCX7nDr2OdzdKWZyLDFkh3duZlNyg,1774
-nimble_python/types/extract_response.py,sha256=3PFokSkbonRZoaDR8OhgDmDAB2VzYUpQuyAQCiY0BVg,8034
+nimble_python/types/extract_response.py,sha256=B1Md11774eeSLinT7Z-a6ImP_X-WrUGoYSD1mbZMPjo,8153
 nimble_python/types/map_response.py,sha256=rNCav8_9GXdbdNIaIQEWi6f8c_YduImddoCzxV4i4Os,607
 nimble_python/types/search_response.py,sha256=lykUe_ICfs-dvtcz9nnrcFrV80FkSkWrR3ljumCyH4Y,2210
 nimble_python/types/task_get_response.py,sha256=YMLwfZv2ejUu-4VePPCUkojzXDZ_4tYtYCem9JfQh0U,1563
@@ -91,7 +98,7 @@ nimble_python/types/shared_params/scroll_action.py,sha256=6PJMWVc9JhjyrXeAsFnvHR
 nimble_python/types/shared_params/wait_action.py,sha256=2jlVgNNZcC2M9tka3xbmLkDocCJuoX_1qIGOmd6a2DU,1046
 nimble_python/types/shared_params/wait_for_element_action.py,sha256=cKhvFe09G2aQBFKBXg00S43ykQqgNyTLYR1DoJmDvTA,1364
 nimble_python/types/shared_params/wait_for_navigation_action.py,sha256=oPAK3OxZObqJjDEh9v0_q1ntHfWqgtXOp_k-oX8ZiHE,1257
-nimble_python-0.14.0.dist-info/METADATA,sha256=hCQzxTLHFSpfR9tdiU8hhLX9wb-qzqytAWAbNIZmWGg,15816
-nimble_python-0.14.0.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-nimble_python-0.14.0.dist-info/licenses/LICENSE,sha256=khtIpaYdgNbXXcmzsxlSL_iKmh160uTREZTiMQKCB24,11336
-nimble_python-0.14.0.dist-info/RECORD,,
+nimble_python-0.16.0.dist-info/METADATA,sha256=be0K4zHYggpMamVI94U0JkITCx187hFZMdUYzxaXdrM,15816
+nimble_python-0.16.0.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+nimble_python-0.16.0.dist-info/licenses/LICENSE,sha256=khtIpaYdgNbXXcmzsxlSL_iKmh160uTREZTiMQKCB24,11336
+nimble_python-0.16.0.dist-info/RECORD,,