structifyai 1.158.0__py3-none-any.whl → 1.176.1__py3-none-any.whl

structify/resources/sandbox.py

@@ -3,11 +3,10 @@
 from __future__ import annotations
 
 from typing import Optional
-from typing_extensions import Literal
 
 import httpx
 
-from ..types import sandbox_get_params, sandbox_create_params, sandbox_update_status_params
+from ..types import sandbox_get_params, sandbox_update_status_params
 from .._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
 from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
@@ -45,52 +44,6 @@ class SandboxResource(SyncAPIResource):
         """
         return SandboxResourceWithStreamingResponse(self)
 
-    def create(
-        self,
-        chat_id: str,
-        *,
-        chat_session_id: str,
-        modal_id: str,
-        modal_url: str,
-        status: Literal["alive", "terminated"],
-        latest_node: Optional[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,
-    ) -> Sandbox:
-        """
-        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 chat_id:
-            raise ValueError(f"Expected a non-empty value for `chat_id` but received {chat_id!r}")
-        return self._post(
-            f"/sandbox/{chat_id}",
-            body=maybe_transform(
-                {
-                    "chat_session_id": chat_session_id,
-                    "modal_id": modal_id,
-                    "modal_url": modal_url,
-                    "status": status,
-                    "latest_node": latest_node,
-                },
-                sandbox_create_params.SandboxCreateParams,
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=Sandbox,
-        )
-
     def list(
         self,
         chat_id: str,
@@ -126,7 +79,7 @@ class SandboxResource(SyncAPIResource):
         self,
         chat_id: str,
         *,
-        sandbox_url_override: Optional[str] | Omit = omit,
+        modal_control_service_url_override: Optional[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,
@@ -136,6 +89,8 @@ class SandboxResource(SyncAPIResource):
     ) -> Sandbox:
         """
         Args:
+          modal_control_service_url_override: Override URL for the modal control service (for testing/development)
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -148,7 +103,10 @@ class SandboxResource(SyncAPIResource):
             raise ValueError(f"Expected a non-empty value for `chat_id` but received {chat_id!r}")
         return self._post(
             f"/sandbox/live/{chat_id}",
-            body=maybe_transform({"sandbox_url_override": sandbox_url_override}, sandbox_get_params.SandboxGetParams),
+            body=maybe_transform(
+                {"modal_control_service_url_override": modal_control_service_url_override},
+                sandbox_get_params.SandboxGetParams,
+            ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
@@ -209,52 +167,6 @@ class AsyncSandboxResource(AsyncAPIResource):
         """
         return AsyncSandboxResourceWithStreamingResponse(self)
 
-    async def create(
-        self,
-        chat_id: str,
-        *,
-        chat_session_id: str,
-        modal_id: str,
-        modal_url: str,
-        status: Literal["alive", "terminated"],
-        latest_node: Optional[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,
-    ) -> Sandbox:
-        """
-        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 chat_id:
-            raise ValueError(f"Expected a non-empty value for `chat_id` but received {chat_id!r}")
-        return await self._post(
-            f"/sandbox/{chat_id}",
-            body=await async_maybe_transform(
-                {
-                    "chat_session_id": chat_session_id,
-                    "modal_id": modal_id,
-                    "modal_url": modal_url,
-                    "status": status,
-                    "latest_node": latest_node,
-                },
-                sandbox_create_params.SandboxCreateParams,
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=Sandbox,
-        )
-
     async def list(
         self,
         chat_id: str,
@@ -290,7 +202,7 @@ class AsyncSandboxResource(AsyncAPIResource):
         self,
         chat_id: str,
         *,
-        sandbox_url_override: Optional[str] | Omit = omit,
+        modal_control_service_url_override: Optional[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,
@@ -300,6 +212,8 @@ class AsyncSandboxResource(AsyncAPIResource):
     ) -> Sandbox:
         """
         Args:
+          modal_control_service_url_override: Override URL for the modal control service (for testing/development)
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -313,7 +227,8 @@ class AsyncSandboxResource(AsyncAPIResource):
         return await self._post(
             f"/sandbox/live/{chat_id}",
             body=await async_maybe_transform(
-                {"sandbox_url_override": sandbox_url_override}, sandbox_get_params.SandboxGetParams
+                {"modal_control_service_url_override": modal_control_service_url_override},
+                sandbox_get_params.SandboxGetParams,
             ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
@@ -361,9 +276,6 @@ class SandboxResourceWithRawResponse:
     def __init__(self, sandbox: SandboxResource) -> None:
         self._sandbox = sandbox
 
-        self.create = to_raw_response_wrapper(
-            sandbox.create,
-        )
         self.list = to_raw_response_wrapper(
             sandbox.list,
         )
@@ -379,9 +291,6 @@ class AsyncSandboxResourceWithRawResponse:
     def __init__(self, sandbox: AsyncSandboxResource) -> None:
         self._sandbox = sandbox
 
-        self.create = async_to_raw_response_wrapper(
-            sandbox.create,
-        )
         self.list = async_to_raw_response_wrapper(
             sandbox.list,
         )
@@ -397,9 +306,6 @@ class SandboxResourceWithStreamingResponse:
     def __init__(self, sandbox: SandboxResource) -> None:
         self._sandbox = sandbox
 
-        self.create = to_streamed_response_wrapper(
-            sandbox.create,
-        )
         self.list = to_streamed_response_wrapper(
             sandbox.list,
         )
@@ -415,9 +321,6 @@ class AsyncSandboxResourceWithStreamingResponse:
     def __init__(self, sandbox: AsyncSandboxResource) -> None:
         self._sandbox = sandbox
 
-        self.create = async_to_streamed_response_wrapper(
-            sandbox.create,
-        )
         self.list = async_to_streamed_response_wrapper(
             sandbox.list,
         )

structify/resources/scrape.py

@@ -55,7 +55,6 @@ class ScrapeResource(SyncAPIResource):
         table_name: str,
         dataset_name: Optional[str] | Omit = omit,
         node_id: Optional[str] | Omit = omit,
-        stop_config: Optional[scrape_list_params.StopConfig] | Omit = omit,
         use_proxy: Optional[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.
@@ -71,8 +70,6 @@ class ScrapeResource(SyncAPIResource):
               A dataset is a complete namespace where all references between schemas are held
               within the dataset.
 
-          stop_config: Configuration parameters for the StopChecker
-
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -90,7 +87,6 @@ class ScrapeResource(SyncAPIResource):
                     "table_name": table_name,
                     "dataset_name": dataset_name,
                     "node_id": node_id,
-                    "stop_config": stop_config,
                     "use_proxy": use_proxy,
                 },
                 scrape_list_params.ScrapeListParams,
@@ -109,7 +105,6 @@ class ScrapeResource(SyncAPIResource):
         url: str,
         node_id: Optional[str] | Omit = omit,
         seeded_kg: Optional[KnowledgeGraphParam] | Omit = omit,
-        stop_config: Optional[scrape_scrape_params.StopConfig] | Omit = omit,
         use_proxy: Optional[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.
@@ -124,8 +119,6 @@ class ScrapeResource(SyncAPIResource):
               that the LLM outputs. Also the first representation of an LLM output in the
               pipeline from raw tool output to being merged into a DB
 
-          stop_config: Configuration parameters for the StopChecker
-
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -143,7 +136,6 @@ class ScrapeResource(SyncAPIResource):
                     "url": url,
                     "node_id": node_id,
                     "seeded_kg": seeded_kg,
-                    "stop_config": stop_config,
                     "use_proxy": use_proxy,
                 },
                 scrape_scrape_params.ScrapeScrapeParams,
@@ -183,7 +175,6 @@ class AsyncScrapeResource(AsyncAPIResource):
         table_name: str,
         dataset_name: Optional[str] | Omit = omit,
         node_id: Optional[str] | Omit = omit,
-        stop_config: Optional[scrape_list_params.StopConfig] | Omit = omit,
         use_proxy: Optional[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.
@@ -199,8 +190,6 @@ class AsyncScrapeResource(AsyncAPIResource):
               A dataset is a complete namespace where all references between schemas are held
               within the dataset.
 
-          stop_config: Configuration parameters for the StopChecker
-
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -218,7 +207,6 @@ class AsyncScrapeResource(AsyncAPIResource):
                     "table_name": table_name,
                     "dataset_name": dataset_name,
                     "node_id": node_id,
-                    "stop_config": stop_config,
                     "use_proxy": use_proxy,
                 },
                 scrape_list_params.ScrapeListParams,
@@ -237,7 +225,6 @@ class AsyncScrapeResource(AsyncAPIResource):
         url: str,
         node_id: Optional[str] | Omit = omit,
         seeded_kg: Optional[KnowledgeGraphParam] | Omit = omit,
-        stop_config: Optional[scrape_scrape_params.StopConfig] | Omit = omit,
         use_proxy: Optional[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.
@@ -252,8 +239,6 @@ class AsyncScrapeResource(AsyncAPIResource):
               that the LLM outputs. Also the first representation of an LLM output in the
               pipeline from raw tool output to being merged into a DB
 
-          stop_config: Configuration parameters for the StopChecker
-
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -271,7 +256,6 @@ class AsyncScrapeResource(AsyncAPIResource):
                     "url": url,
                     "node_id": node_id,
                     "seeded_kg": seeded_kg,
-                    "stop_config": stop_config,
                     "use_proxy": use_proxy,
                 },
                 scrape_scrape_params.ScrapeScrapeParams,