raccoonai 0.1.0a5__py3-none-any.whl → 0.1.0a7__py3-none-any.whl

raccoonai/resources/lam.py

@@ -7,7 +7,7 @@ from typing_extensions import Literal, overload
 
 import httpx
 
-from ..types import lam_run_params, lam_extract_params, lam_integration_run_params
+from ..types import lam_run_params, lam_integration_run_params
 from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
 from .._utils import (
     required_args,
@@ -25,7 +25,6 @@ from .._response import (
 from .._streaming import Stream, AsyncStream
 from .._base_client import make_request_options
 from ..types.lam_run_response import LamRunResponse
-from ..types.lam_extract_response import LamExtractResponse
 from ..types.lam_integration_run_response import LamIntegrationRunResponse
 
 __all__ = ["LamResource", "AsyncLamResource"]
@@ -51,210 +50,6 @@ class LamResource(SyncAPIResource):
         """
         return LamResourceWithStreamingResponse(self)
 
-    @overload
-    def extract(
-        self,
-        *,
-        query: str,
-        raccoon_passcode: str,
-        advanced: Optional[lam_extract_params.Advanced] | NotGiven = NOT_GIVEN,
-        app_url: Optional[str] | NotGiven = NOT_GIVEN,
-        chat_history: Optional[Iterable[object]] | NotGiven = NOT_GIVEN,
-        max_count: Optional[int] | NotGiven = NOT_GIVEN,
-        schema: object | NotGiven = NOT_GIVEN,
-        stream: Optional[Literal[False]] | NotGiven = NOT_GIVEN,
-        # 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,
-    ) -> LamExtractResponse:
-        """Lam Extract Endpoint
-
-        Args:
-          query: The input query string for the request.
-
-        This is typically the main prompt.
-
-          raccoon_passcode: The raccoon passcode associated with the end user on behalf of which the call is
-              being made.
-
-          advanced: Advanced configuration options for the session, such as ad-blocking and CAPTCHA
-              solving.
-
-          app_url: This is the entrypoint URL for the web agent.
-
-          chat_history: The history of the conversation as a list of messages or objects you might use
-              while building a chat app to give the model context of the past conversation.
-
-          max_count: The maximum number of results to extract.
-
-          schema: The expected schema for the response. This is a dictionary where the keys
-              describe the fields and the values describe their purposes.
-
-          stream: Whether the response should be streamed back or not.
-
-          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 extract(
-        self,
-        *,
-        query: str,
-        raccoon_passcode: str,
-        stream: Literal[True],
-        advanced: Optional[lam_extract_params.Advanced] | NotGiven = NOT_GIVEN,
-        app_url: Optional[str] | NotGiven = NOT_GIVEN,
-        chat_history: Optional[Iterable[object]] | NotGiven = NOT_GIVEN,
-        max_count: Optional[int] | NotGiven = NOT_GIVEN,
-        schema: object | NotGiven = NOT_GIVEN,
-        # 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,
-    ) -> Stream[LamExtractResponse]:
-        """Lam Extract Endpoint
-
-        Args:
-          query: The input query string for the request.
-
-        This is typically the main prompt.
-
-          raccoon_passcode: The raccoon passcode associated with the end user on behalf of which the call is
-              being made.
-
-          stream: Whether the response should be streamed back or not.
-
-          advanced: Advanced configuration options for the session, such as ad-blocking and CAPTCHA
-              solving.
-
-          app_url: This is the entrypoint URL for the web agent.
-
-          chat_history: The history of the conversation as a list of messages or objects you might use
-              while building a chat app to give the model context of the past conversation.
-
-          max_count: The maximum number of results to extract.
-
-          schema: The expected schema for the response. This is a dictionary where the keys
-              describe the fields and the values describe their purposes.
-
-          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 extract(
-        self,
-        *,
-        query: str,
-        raccoon_passcode: str,
-        stream: bool,
-        advanced: Optional[lam_extract_params.Advanced] | NotGiven = NOT_GIVEN,
-        app_url: Optional[str] | NotGiven = NOT_GIVEN,
-        chat_history: Optional[Iterable[object]] | NotGiven = NOT_GIVEN,
-        max_count: Optional[int] | NotGiven = NOT_GIVEN,
-        schema: object | NotGiven = NOT_GIVEN,
-        # 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,
-    ) -> LamExtractResponse | Stream[LamExtractResponse]:
-        """Lam Extract Endpoint
-
-        Args:
-          query: The input query string for the request.
-
-        This is typically the main prompt.
-
-          raccoon_passcode: The raccoon passcode associated with the end user on behalf of which the call is
-              being made.
-
-          stream: Whether the response should be streamed back or not.
-
-          advanced: Advanced configuration options for the session, such as ad-blocking and CAPTCHA
-              solving.
-
-          app_url: This is the entrypoint URL for the web agent.
-
-          chat_history: The history of the conversation as a list of messages or objects you might use
-              while building a chat app to give the model context of the past conversation.
-
-          max_count: The maximum number of results to extract.
-
-          schema: The expected schema for the response. This is a dictionary where the keys
-              describe the fields and the values describe their purposes.
-
-          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(["query", "raccoon_passcode"], ["query", "raccoon_passcode", "stream"])
-    def extract(
-        self,
-        *,
-        query: str,
-        raccoon_passcode: str,
-        advanced: Optional[lam_extract_params.Advanced] | NotGiven = NOT_GIVEN,
-        app_url: Optional[str] | NotGiven = NOT_GIVEN,
-        chat_history: Optional[Iterable[object]] | NotGiven = NOT_GIVEN,
-        max_count: Optional[int] | NotGiven = NOT_GIVEN,
-        schema: object | NotGiven = NOT_GIVEN,
-        stream: Optional[Literal[False]] | Literal[True] | NotGiven = NOT_GIVEN,
-        # 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,
-    ) -> LamExtractResponse | Stream[LamExtractResponse]:
-        return self._post(
-            "/lam/extract",
-            body=maybe_transform(
-                {
-                    "query": query,
-                    "raccoon_passcode": raccoon_passcode,
-                    "advanced": advanced,
-                    "app_url": app_url,
-                    "chat_history": chat_history,
-                    "max_count": max_count,
-                    "schema": schema,
-                    "stream": stream,
-                },
-                lam_extract_params.LamExtractParams,
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=LamExtractResponse,
-            stream=stream or False,
-            stream_cls=Stream[LamExtractResponse],
-        )
-
     @overload
     def integration_run(
         self,
@@ -437,6 +232,9 @@ class LamResource(SyncAPIResource):
         advanced: Optional[lam_run_params.Advanced] | NotGiven = NOT_GIVEN,
         app_url: Optional[str] | NotGiven = NOT_GIVEN,
         chat_history: Optional[Iterable[object]] | NotGiven = NOT_GIVEN,
+        max_count: Optional[int] | NotGiven = NOT_GIVEN,
+        mode: Optional[Literal["deepsearch", "default"]] | NotGiven = NOT_GIVEN,
+        schema: object | NotGiven = NOT_GIVEN,
         stream: Optional[Literal[False]] | NotGiven = NOT_GIVEN,
         # 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.
@@ -463,6 +261,13 @@ class LamResource(SyncAPIResource):
           chat_history: The history of the conversation as a list of messages or objects you might use
               while building a chat app to give the model context of the past conversation.
 
+          max_count: The maximum number of results to extract.
+
+          mode: Mode of execution.
+
+          schema: The expected schema for the response. This is a dictionary where the keys
+              describe the fields and the values describe their purposes.
+
           stream: Whether the response should be streamed back or not.
 
           extra_headers: Send extra headers
@@ -485,6 +290,9 @@ class LamResource(SyncAPIResource):
         advanced: Optional[lam_run_params.Advanced] | NotGiven = NOT_GIVEN,
         app_url: Optional[str] | NotGiven = NOT_GIVEN,
         chat_history: Optional[Iterable[object]] | NotGiven = NOT_GIVEN,
+        max_count: Optional[int] | NotGiven = NOT_GIVEN,
+        mode: Optional[Literal["deepsearch", "default"]] | NotGiven = NOT_GIVEN,
+        schema: object | NotGiven = NOT_GIVEN,
         # 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,
@@ -512,6 +320,13 @@ class LamResource(SyncAPIResource):
           chat_history: The history of the conversation as a list of messages or objects you might use
               while building a chat app to give the model context of the past conversation.
 
+          max_count: The maximum number of results to extract.
+
+          mode: Mode of execution.
+
+          schema: The expected schema for the response. This is a dictionary where the keys
+              describe the fields and the values describe their purposes.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -532,6 +347,9 @@ class LamResource(SyncAPIResource):
         advanced: Optional[lam_run_params.Advanced] | NotGiven = NOT_GIVEN,
         app_url: Optional[str] | NotGiven = NOT_GIVEN,
         chat_history: Optional[Iterable[object]] | NotGiven = NOT_GIVEN,
+        max_count: Optional[int] | NotGiven = NOT_GIVEN,
+        mode: Optional[Literal["deepsearch", "default"]] | NotGiven = NOT_GIVEN,
+        schema: object | NotGiven = NOT_GIVEN,
         # 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,
@@ -559,6 +377,13 @@ class LamResource(SyncAPIResource):
           chat_history: The history of the conversation as a list of messages or objects you might use
               while building a chat app to give the model context of the past conversation.
 
+          max_count: The maximum number of results to extract.
+
+          mode: Mode of execution.
+
+          schema: The expected schema for the response. This is a dictionary where the keys
+              describe the fields and the values describe their purposes.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -578,6 +403,9 @@ class LamResource(SyncAPIResource):
         advanced: Optional[lam_run_params.Advanced] | NotGiven = NOT_GIVEN,
         app_url: Optional[str] | NotGiven = NOT_GIVEN,
         chat_history: Optional[Iterable[object]] | NotGiven = NOT_GIVEN,
+        max_count: Optional[int] | NotGiven = NOT_GIVEN,
+        mode: Optional[Literal["deepsearch", "default"]] | NotGiven = NOT_GIVEN,
+        schema: object | NotGiven = NOT_GIVEN,
         stream: Optional[Literal[False]] | Literal[True] | NotGiven = NOT_GIVEN,
         # 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.
@@ -595,6 +423,9 @@ class LamResource(SyncAPIResource):
                     "advanced": advanced,
                     "app_url": app_url,
                     "chat_history": chat_history,
+                    "max_count": max_count,
+                    "mode": mode,
+                    "schema": schema,
                     "stream": stream,
                 },
                 lam_run_params.LamRunParams,
@@ -628,210 +459,6 @@ class AsyncLamResource(AsyncAPIResource):
         """
         return AsyncLamResourceWithStreamingResponse(self)
 
-    @overload
-    async def extract(
-        self,
-        *,
-        query: str,
-        raccoon_passcode: str,
-        advanced: Optional[lam_extract_params.Advanced] | NotGiven = NOT_GIVEN,
-        app_url: Optional[str] | NotGiven = NOT_GIVEN,
-        chat_history: Optional[Iterable[object]] | NotGiven = NOT_GIVEN,
-        max_count: Optional[int] | NotGiven = NOT_GIVEN,
-        schema: object | NotGiven = NOT_GIVEN,
-        stream: Optional[Literal[False]] | NotGiven = NOT_GIVEN,
-        # 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,
-    ) -> LamExtractResponse:
-        """Lam Extract Endpoint
-
-        Args:
-          query: The input query string for the request.
-
-        This is typically the main prompt.
-
-          raccoon_passcode: The raccoon passcode associated with the end user on behalf of which the call is
-              being made.
-
-          advanced: Advanced configuration options for the session, such as ad-blocking and CAPTCHA
-              solving.
-
-          app_url: This is the entrypoint URL for the web agent.
-
-          chat_history: The history of the conversation as a list of messages or objects you might use
-              while building a chat app to give the model context of the past conversation.
-
-          max_count: The maximum number of results to extract.
-
-          schema: The expected schema for the response. This is a dictionary where the keys
-              describe the fields and the values describe their purposes.
-
-          stream: Whether the response should be streamed back or not.
-
-          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 extract(
-        self,
-        *,
-        query: str,
-        raccoon_passcode: str,
-        stream: Literal[True],
-        advanced: Optional[lam_extract_params.Advanced] | NotGiven = NOT_GIVEN,
-        app_url: Optional[str] | NotGiven = NOT_GIVEN,
-        chat_history: Optional[Iterable[object]] | NotGiven = NOT_GIVEN,
-        max_count: Optional[int] | NotGiven = NOT_GIVEN,
-        schema: object | NotGiven = NOT_GIVEN,
-        # 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,
-    ) -> AsyncStream[LamExtractResponse]:
-        """Lam Extract Endpoint
-
-        Args:
-          query: The input query string for the request.
-
-        This is typically the main prompt.
-
-          raccoon_passcode: The raccoon passcode associated with the end user on behalf of which the call is
-              being made.
-
-          stream: Whether the response should be streamed back or not.
-
-          advanced: Advanced configuration options for the session, such as ad-blocking and CAPTCHA
-              solving.
-
-          app_url: This is the entrypoint URL for the web agent.
-
-          chat_history: The history of the conversation as a list of messages or objects you might use
-              while building a chat app to give the model context of the past conversation.
-
-          max_count: The maximum number of results to extract.
-
-          schema: The expected schema for the response. This is a dictionary where the keys
-              describe the fields and the values describe their purposes.
-
-          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 extract(
-        self,
-        *,
-        query: str,
-        raccoon_passcode: str,
-        stream: bool,
-        advanced: Optional[lam_extract_params.Advanced] | NotGiven = NOT_GIVEN,
-        app_url: Optional[str] | NotGiven = NOT_GIVEN,
-        chat_history: Optional[Iterable[object]] | NotGiven = NOT_GIVEN,
-        max_count: Optional[int] | NotGiven = NOT_GIVEN,
-        schema: object | NotGiven = NOT_GIVEN,
-        # 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,
-    ) -> LamExtractResponse | AsyncStream[LamExtractResponse]:
-        """Lam Extract Endpoint
-
-        Args:
-          query: The input query string for the request.
-
-        This is typically the main prompt.
-
-          raccoon_passcode: The raccoon passcode associated with the end user on behalf of which the call is
-              being made.
-
-          stream: Whether the response should be streamed back or not.
-
-          advanced: Advanced configuration options for the session, such as ad-blocking and CAPTCHA
-              solving.
-
-          app_url: This is the entrypoint URL for the web agent.
-
-          chat_history: The history of the conversation as a list of messages or objects you might use
-              while building a chat app to give the model context of the past conversation.
-
-          max_count: The maximum number of results to extract.
-
-          schema: The expected schema for the response. This is a dictionary where the keys
-              describe the fields and the values describe their purposes.
-
-          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(["query", "raccoon_passcode"], ["query", "raccoon_passcode", "stream"])
-    async def extract(
-        self,
-        *,
-        query: str,
-        raccoon_passcode: str,
-        advanced: Optional[lam_extract_params.Advanced] | NotGiven = NOT_GIVEN,
-        app_url: Optional[str] | NotGiven = NOT_GIVEN,
-        chat_history: Optional[Iterable[object]] | NotGiven = NOT_GIVEN,
-        max_count: Optional[int] | NotGiven = NOT_GIVEN,
-        schema: object | NotGiven = NOT_GIVEN,
-        stream: Optional[Literal[False]] | Literal[True] | NotGiven = NOT_GIVEN,
-        # 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,
-    ) -> LamExtractResponse | AsyncStream[LamExtractResponse]:
-        return await self._post(
-            "/lam/extract",
-            body=await async_maybe_transform(
-                {
-                    "query": query,
-                    "raccoon_passcode": raccoon_passcode,
-                    "advanced": advanced,
-                    "app_url": app_url,
-                    "chat_history": chat_history,
-                    "max_count": max_count,
-                    "schema": schema,
-                    "stream": stream,
-                },
-                lam_extract_params.LamExtractParams,
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=LamExtractResponse,
-            stream=stream or False,
-            stream_cls=AsyncStream[LamExtractResponse],
-        )
-
     @overload
     async def integration_run(
         self,
@@ -1014,6 +641,9 @@ class AsyncLamResource(AsyncAPIResource):
         advanced: Optional[lam_run_params.Advanced] | NotGiven = NOT_GIVEN,
         app_url: Optional[str] | NotGiven = NOT_GIVEN,
         chat_history: Optional[Iterable[object]] | NotGiven = NOT_GIVEN,
+        max_count: Optional[int] | NotGiven = NOT_GIVEN,
+        mode: Optional[Literal["deepsearch", "default"]] | NotGiven = NOT_GIVEN,
+        schema: object | NotGiven = NOT_GIVEN,
         stream: Optional[Literal[False]] | NotGiven = NOT_GIVEN,
         # 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.
@@ -1040,6 +670,13 @@ class AsyncLamResource(AsyncAPIResource):
           chat_history: The history of the conversation as a list of messages or objects you might use
               while building a chat app to give the model context of the past conversation.
 
+          max_count: The maximum number of results to extract.
+
+          mode: Mode of execution.
+
+          schema: The expected schema for the response. This is a dictionary where the keys
+              describe the fields and the values describe their purposes.
+
           stream: Whether the response should be streamed back or not.
 
           extra_headers: Send extra headers
@@ -1062,6 +699,9 @@ class AsyncLamResource(AsyncAPIResource):
         advanced: Optional[lam_run_params.Advanced] | NotGiven = NOT_GIVEN,
         app_url: Optional[str] | NotGiven = NOT_GIVEN,
         chat_history: Optional[Iterable[object]] | NotGiven = NOT_GIVEN,
+        max_count: Optional[int] | NotGiven = NOT_GIVEN,
+        mode: Optional[Literal["deepsearch", "default"]] | NotGiven = NOT_GIVEN,
+        schema: object | NotGiven = NOT_GIVEN,
         # 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,
@@ -1089,6 +729,13 @@ class AsyncLamResource(AsyncAPIResource):
           chat_history: The history of the conversation as a list of messages or objects you might use
               while building a chat app to give the model context of the past conversation.
 
+          max_count: The maximum number of results to extract.
+
+          mode: Mode of execution.
+
+          schema: The expected schema for the response. This is a dictionary where the keys
+              describe the fields and the values describe their purposes.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -1109,6 +756,9 @@ class AsyncLamResource(AsyncAPIResource):
         advanced: Optional[lam_run_params.Advanced] | NotGiven = NOT_GIVEN,
         app_url: Optional[str] | NotGiven = NOT_GIVEN,
         chat_history: Optional[Iterable[object]] | NotGiven = NOT_GIVEN,
+        max_count: Optional[int] | NotGiven = NOT_GIVEN,
+        mode: Optional[Literal["deepsearch", "default"]] | NotGiven = NOT_GIVEN,
+        schema: object | NotGiven = NOT_GIVEN,
         # 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,
@@ -1136,6 +786,13 @@ class AsyncLamResource(AsyncAPIResource):
           chat_history: The history of the conversation as a list of messages or objects you might use
               while building a chat app to give the model context of the past conversation.
 
+          max_count: The maximum number of results to extract.
+
+          mode: Mode of execution.
+
+          schema: The expected schema for the response. This is a dictionary where the keys
+              describe the fields and the values describe their purposes.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -1155,6 +812,9 @@ class AsyncLamResource(AsyncAPIResource):
         advanced: Optional[lam_run_params.Advanced] | NotGiven = NOT_GIVEN,
         app_url: Optional[str] | NotGiven = NOT_GIVEN,
         chat_history: Optional[Iterable[object]] | NotGiven = NOT_GIVEN,
+        max_count: Optional[int] | NotGiven = NOT_GIVEN,
+        mode: Optional[Literal["deepsearch", "default"]] | NotGiven = NOT_GIVEN,
+        schema: object | NotGiven = NOT_GIVEN,
         stream: Optional[Literal[False]] | Literal[True] | NotGiven = NOT_GIVEN,
         # 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.
@@ -1172,6 +832,9 @@ class AsyncLamResource(AsyncAPIResource):
                     "advanced": advanced,
                     "app_url": app_url,
                     "chat_history": chat_history,
+                    "max_count": max_count,
+                    "mode": mode,
+                    "schema": schema,
                     "stream": stream,
                 },
                 lam_run_params.LamRunParams,
@@ -1189,9 +852,6 @@ class LamResourceWithRawResponse:
     def __init__(self, lam: LamResource) -> None:
         self._lam = lam
 
-        self.extract = to_raw_response_wrapper(
-            lam.extract,
-        )
         self.integration_run = to_raw_response_wrapper(
             lam.integration_run,
         )
@@ -1204,9 +864,6 @@ class AsyncLamResourceWithRawResponse:
     def __init__(self, lam: AsyncLamResource) -> None:
         self._lam = lam
 
-        self.extract = async_to_raw_response_wrapper(
-            lam.extract,
-        )
         self.integration_run = async_to_raw_response_wrapper(
             lam.integration_run,
         )
@@ -1219,9 +876,6 @@ class LamResourceWithStreamingResponse:
     def __init__(self, lam: LamResource) -> None:
         self._lam = lam
 
-        self.extract = to_streamed_response_wrapper(
-            lam.extract,
-        )
         self.integration_run = to_streamed_response_wrapper(
             lam.integration_run,
         )
@@ -1234,9 +888,6 @@ class AsyncLamResourceWithStreamingResponse:
     def __init__(self, lam: AsyncLamResource) -> None:
         self._lam = lam
 
-        self.extract = async_to_streamed_response_wrapper(
-            lam.extract,
-        )
         self.integration_run = async_to_streamed_response_wrapper(
             lam.integration_run,
         )

raccoonai/types/__init__.py

@@ -4,10 +4,8 @@ from __future__ import annotations
 
 from .lam_run_params import LamRunParams as LamRunParams
 from .lam_run_response import LamRunResponse as LamRunResponse
-from .lam_extract_params import LamExtractParams as LamExtractParams
 from .fleet_create_params import FleetCreateParams as FleetCreateParams
 from .fleet_logs_response import FleetLogsResponse as FleetLogsResponse
-from .lam_extract_response import LamExtractResponse as LamExtractResponse
 from .fleet_create_response import FleetCreateResponse as FleetCreateResponse
 from .fleet_status_response import FleetStatusResponse as FleetStatusResponse
 from .fleet_terminate_response import FleetTerminateResponse as FleetTerminateResponse