raccoonai 0.1.0a6__py3-none-any.whl → 0.1.0a8__py3-none-any.whl

raccoonai/resources/tail/tail.py

@@ -0,0 +1,273 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Optional
+from typing_extensions import Literal
+
+import httpx
+
+from .apps import (
+    AppsResource,
+    AsyncAppsResource,
+    AppsResourceWithRawResponse,
+    AsyncAppsResourceWithRawResponse,
+    AppsResourceWithStreamingResponse,
+    AsyncAppsResourceWithStreamingResponse,
+)
+from .auth import (
+    AuthResource,
+    AsyncAuthResource,
+    AuthResourceWithRawResponse,
+    AsyncAuthResourceWithRawResponse,
+    AuthResourceWithStreamingResponse,
+    AsyncAuthResourceWithStreamingResponse,
+)
+from ...types import tail_users_params
+from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from ..._utils import (
+    maybe_transform,
+    async_maybe_transform,
+)
+from ..._compat import cached_property
+from ..._resource import SyncAPIResource, AsyncAPIResource
+from ..._response import (
+    to_raw_response_wrapper,
+    to_streamed_response_wrapper,
+    async_to_raw_response_wrapper,
+    async_to_streamed_response_wrapper,
+)
+from ..._base_client import make_request_options
+from ...types.tail_users_response import TailUsersResponse
+
+__all__ = ["TailResource", "AsyncTailResource"]
+
+
+class TailResource(SyncAPIResource):
+    @cached_property
+    def apps(self) -> AppsResource:
+        return AppsResource(self._client)
+
+    @cached_property
+    def auth(self) -> AuthResource:
+        return AuthResource(self._client)
+
+    @cached_property
+    def with_raw_response(self) -> TailResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/raccoonaihq/raccoonai-python#accessing-raw-response-data-eg-headers
+        """
+        return TailResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> TailResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/raccoonaihq/raccoonai-python#with_streaming_response
+        """
+        return TailResourceWithStreamingResponse(self)
+
+    def users(
+        self,
+        *,
+        email_id: Optional[str] | NotGiven = NOT_GIVEN,
+        limit: Optional[int] | NotGiven = NOT_GIVEN,
+        page: Optional[int] | NotGiven = NOT_GIVEN,
+        raccoon_passcode: Optional[str] | NotGiven = NOT_GIVEN,
+        search_query: Optional[str] | NotGiven = NOT_GIVEN,
+        sort_by: Optional[Literal["createdAt", "name", "email", "raccoonPasscode"]] | NotGiven = NOT_GIVEN,
+        sort_order: Optional[Literal["ascend", "descend"]] | 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,
+    ) -> TailUsersResponse:
+        """
+        Get Users 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._get(
+            "/tail/users",
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "email_id": email_id,
+                        "limit": limit,
+                        "page": page,
+                        "raccoon_passcode": raccoon_passcode,
+                        "search_query": search_query,
+                        "sort_by": sort_by,
+                        "sort_order": sort_order,
+                    },
+                    tail_users_params.TailUsersParams,
+                ),
+            ),
+            cast_to=TailUsersResponse,
+        )
+
+
+class AsyncTailResource(AsyncAPIResource):
+    @cached_property
+    def apps(self) -> AsyncAppsResource:
+        return AsyncAppsResource(self._client)
+
+    @cached_property
+    def auth(self) -> AsyncAuthResource:
+        return AsyncAuthResource(self._client)
+
+    @cached_property
+    def with_raw_response(self) -> AsyncTailResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/raccoonaihq/raccoonai-python#accessing-raw-response-data-eg-headers
+        """
+        return AsyncTailResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncTailResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/raccoonaihq/raccoonai-python#with_streaming_response
+        """
+        return AsyncTailResourceWithStreamingResponse(self)
+
+    async def users(
+        self,
+        *,
+        email_id: Optional[str] | NotGiven = NOT_GIVEN,
+        limit: Optional[int] | NotGiven = NOT_GIVEN,
+        page: Optional[int] | NotGiven = NOT_GIVEN,
+        raccoon_passcode: Optional[str] | NotGiven = NOT_GIVEN,
+        search_query: Optional[str] | NotGiven = NOT_GIVEN,
+        sort_by: Optional[Literal["createdAt", "name", "email", "raccoonPasscode"]] | NotGiven = NOT_GIVEN,
+        sort_order: Optional[Literal["ascend", "descend"]] | 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,
+    ) -> TailUsersResponse:
+        """
+        Get Users 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._get(
+            "/tail/users",
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=await async_maybe_transform(
+                    {
+                        "email_id": email_id,
+                        "limit": limit,
+                        "page": page,
+                        "raccoon_passcode": raccoon_passcode,
+                        "search_query": search_query,
+                        "sort_by": sort_by,
+                        "sort_order": sort_order,
+                    },
+                    tail_users_params.TailUsersParams,
+                ),
+            ),
+            cast_to=TailUsersResponse,
+        )
+
+
+class TailResourceWithRawResponse:
+    def __init__(self, tail: TailResource) -> None:
+        self._tail = tail
+
+        self.users = to_raw_response_wrapper(
+            tail.users,
+        )
+
+    @cached_property
+    def apps(self) -> AppsResourceWithRawResponse:
+        return AppsResourceWithRawResponse(self._tail.apps)
+
+    @cached_property
+    def auth(self) -> AuthResourceWithRawResponse:
+        return AuthResourceWithRawResponse(self._tail.auth)
+
+
+class AsyncTailResourceWithRawResponse:
+    def __init__(self, tail: AsyncTailResource) -> None:
+        self._tail = tail
+
+        self.users = async_to_raw_response_wrapper(
+            tail.users,
+        )
+
+    @cached_property
+    def apps(self) -> AsyncAppsResourceWithRawResponse:
+        return AsyncAppsResourceWithRawResponse(self._tail.apps)
+
+    @cached_property
+    def auth(self) -> AsyncAuthResourceWithRawResponse:
+        return AsyncAuthResourceWithRawResponse(self._tail.auth)
+
+
+class TailResourceWithStreamingResponse:
+    def __init__(self, tail: TailResource) -> None:
+        self._tail = tail
+
+        self.users = to_streamed_response_wrapper(
+            tail.users,
+        )
+
+    @cached_property
+    def apps(self) -> AppsResourceWithStreamingResponse:
+        return AppsResourceWithStreamingResponse(self._tail.apps)
+
+    @cached_property
+    def auth(self) -> AuthResourceWithStreamingResponse:
+        return AuthResourceWithStreamingResponse(self._tail.auth)
+
+
+class AsyncTailResourceWithStreamingResponse:
+    def __init__(self, tail: AsyncTailResource) -> None:
+        self._tail = tail
+
+        self.users = async_to_streamed_response_wrapper(
+            tail.users,
+        )
+
+    @cached_property
+    def apps(self) -> AsyncAppsResourceWithStreamingResponse:
+        return AsyncAppsResourceWithStreamingResponse(self._tail.apps)
+
+    @cached_property
+    def auth(self) -> AsyncAuthResourceWithStreamingResponse:
+        return AsyncAuthResourceWithStreamingResponse(self._tail.auth)

raccoonai/types/__init__.py

@@ -4,12 +4,18 @@ 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 .lam_tasks_params import LamTasksParams as LamTasksParams
+from .tail_users_params import TailUsersParams as TailUsersParams
+from .lam_tasks_response import LamTasksResponse as LamTasksResponse
 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 .tail_users_response import TailUsersResponse as TailUsersResponse
 from .fleet_create_response import FleetCreateResponse as FleetCreateResponse
+from .fleet_sessions_params import FleetSessionsParams as FleetSessionsParams
 from .fleet_status_response import FleetStatusResponse as FleetStatusResponse
+from .extension_all_response import ExtensionAllResponse as ExtensionAllResponse
+from .extension_get_response import ExtensionGetResponse as ExtensionGetResponse
+from .extension_upload_params import ExtensionUploadParams as ExtensionUploadParams
+from .fleet_sessions_response import FleetSessionsResponse as FleetSessionsResponse
 from .fleet_terminate_response import FleetTerminateResponse as FleetTerminateResponse
-from .lam_integration_run_params import LamIntegrationRunParams as LamIntegrationRunParams
-from .lam_integration_run_response import LamIntegrationRunResponse as LamIntegrationRunResponse
+from .extension_upload_response import ExtensionUploadResponse as ExtensionUploadResponse

raccoonai/types/extension_all_response.py

@@ -0,0 +1,22 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List
+
+from pydantic import Field as FieldInfo
+
+from .._models import BaseModel
+
+__all__ = ["ExtensionAllResponse", "Extension"]
+
+
+class Extension(BaseModel):
+    extension_id: str = FieldInfo(alias="extensionId")
+    """Unique identifier for the extension."""
+
+    filename: str
+    """Name of the extension file."""
+
+
+class ExtensionAllResponse(BaseModel):
+    extensions: List[Extension]
+    """List of extensions."""

raccoonai/types/extension_get_response.py

@@ -0,0 +1,16 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+
+from pydantic import Field as FieldInfo
+
+from .._models import BaseModel
+
+__all__ = ["ExtensionGetResponse"]
+
+
+class ExtensionGetResponse(BaseModel):
+    extension_id: str = FieldInfo(alias="extensionId")
+    """Unique identifier for the extension."""
+
+    filename: str
+    """Name of the extension file."""

raccoonai/types/extension_upload_params.py

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

raccoonai/types/extension_upload_response.py

@@ -0,0 +1,16 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+
+from pydantic import Field as FieldInfo
+
+from .._models import BaseModel
+
+__all__ = ["ExtensionUploadResponse"]
+
+
+class ExtensionUploadResponse(BaseModel):
+    extension_id: str = FieldInfo(alias="extensionId")
+    """Unique identifier for the extension."""
+
+    filename: str
+    """Name of the extension file."""

raccoonai/types/fleet_create_params.py

@@ -2,10 +2,10 @@
 
 from __future__ import annotations
 
-from typing import Dict, List, Optional
-from typing_extensions import TypedDict
+from typing import Dict, List, Iterable, Optional
+from typing_extensions import Literal, TypedDict
 
-__all__ = ["FleetCreateParams", "Advanced", "Settings"]
+__all__ = ["FleetCreateParams", "Advanced", "AdvancedProxy", "Settings"]
 
 
 class FleetCreateParams(TypedDict, total=False):
@@ -15,13 +15,7 @@ class FleetCreateParams(TypedDict, total=False):
     solving.
     """
 
-    app_name: Optional[str]
-    """
-    The name of the app for which the session is going to run for, used for
-    streamlining authentication.
-    """
-
-    browser_type: Optional[str]
+    browser_type: Optional[Literal["chromium", "firefox", "webkit"]]
     """The type of browser to use.
 
     Supported values include 'chromium', 'firefox', and 'webkit'.
@@ -49,12 +43,32 @@ class FleetCreateParams(TypedDict, total=False):
     """The entrypoint url for the session."""
 
 
+class AdvancedProxy(TypedDict, total=False):
+    city: Optional[str]
+    """Target city."""
+
+    country: Optional[str]
+    """Target country (2-letter ISO code)."""
+
+    enable: bool
+    """Whether to use a proxy for the browser session."""
+
+    state: Optional[str]
+    """Target state (2-letter code)."""
+
+    zip: Optional[int]
+    """Target postal code."""
+
+
 class Advanced(TypedDict, total=False):
     block_ads: Optional[bool]
     """Whether to block advertisements during the browser session."""
 
-    proxy: Optional[bool]
-    """Whether to use a proxy for the browser session."""
+    extension_ids: Optional[Iterable[object]]
+    """list of extension ids"""
+
+    proxy: Optional[AdvancedProxy]
+    """Proxy details for the browser session."""
 
     solve_captchas: Optional[bool]
     """Whether to attempt automatic CAPTCHA solving."""

raccoonai/types/fleet_create_response.py

@@ -1,5 +1,6 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
+from typing_extensions import Literal
 
 from .._models import BaseModel
 
@@ -13,11 +14,8 @@ class FleetCreateResponse(BaseModel):
     session_id: str
     """A unique identifier for the created session."""
 
-    status: str
-    """The current status of the session.
-
-    Possible values include 'running', 'unknown', or 'terminated'.
-    """
+    status: Literal["starting", "running", "terminated", "completed", "unknown"]
+    """The current status of the session."""
 
     websocket_url: str
     """The WebSocket URL for interacting with the session."""

raccoonai/types/fleet_sessions_params.py

@@ -0,0 +1,42 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import List, Optional
+from typing_extensions import Literal, Annotated, TypedDict
+
+from .._utils import PropertyInfo
+
+__all__ = ["FleetSessionsParams"]
+
+
+class FleetSessionsParams(TypedDict, total=False):
+    end_time: Optional[int]
+    """Filter sessions created before this Unix timestamp (in milliseconds)."""
+
+    execution_type: Annotated[Optional[List[Literal["run", "extract", "fleet"]]], PropertyInfo(alias="executionType")]
+    """Filter sessions by execution type (e.g., 'run', 'extract')."""
+
+    limit: Optional[int]
+    """Number of sessions per page (maximum 100)."""
+
+    page: Optional[int]
+    """Page number for pagination."""
+
+    raccoon_passcode: Optional[str]
+    """Filter sessions by Raccoon passcode."""
+
+    session_id: Optional[str]
+    """Filter sessions by a specific session ID."""
+
+    sort_by: Optional[Literal["timestamp", "executionTime", "taskId", "status", "executionType"]]
+    """Field to sort sessions by (e.g., 'timestamp', 'executionTime')."""
+
+    sort_order: Optional[Literal["ascend", "descend"]]
+    """Sort order ('ascend' or 'descend')."""
+
+    start_time: Optional[int]
+    """Filter sessions created after this Unix timestamp (in milliseconds)."""
+
+    task_id: Optional[str]
+    """Filter sessions by a specific task ID."""

raccoonai/types/fleet_sessions_response.py

@@ -0,0 +1,58 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List
+from typing_extensions import Literal
+
+from pydantic import Field as FieldInfo
+
+from .._models import BaseModel
+
+__all__ = ["FleetSessionsResponse", "Meta", "Session"]
+
+
+class Meta(BaseModel):
+    current_page: int = FieldInfo(alias="currentPage")
+    """The current page number."""
+
+    total_pages: int = FieldInfo(alias="totalPages")
+    """Total number of pages available."""
+
+    total_records: int = FieldInfo(alias="totalRecords")
+    """Total number of records across all pages."""
+
+
+class Session(BaseModel):
+    execution_time: int = FieldInfo(alias="executionTime")
+    """Time taken for the session execution (in milliseconds)."""
+
+    execution_type: Literal["run", "extract", "fleet"] = FieldInfo(alias="executionType")
+    """The type of execution performed (e.g., 'run', 'extract')."""
+
+    inputs: object
+    """Input parameters used for the session."""
+
+    output: List[object]
+    """Output generated by the session."""
+
+    raccoon_passcode: str = FieldInfo(alias="raccoonPasscode")
+    """Passcode associated with the user."""
+
+    session_id: str = FieldInfo(alias="sessionId")
+    """Unique identifier for the session."""
+
+    status: Literal["success", "failure", "running"]
+    """Current status of the session."""
+
+    task_id: str = FieldInfo(alias="taskId")
+    """Unique identifier for the associated task."""
+
+    timestamp: int
+    """Unix timestamp (in milliseconds) indicating when the session was created."""
+
+
+class FleetSessionsResponse(BaseModel):
+    meta: Meta
+    """Metadata about the session list."""
+
+    sessions: List[Session]
+    """List of sessions."""

raccoonai/types/fleet_status_response.py

@@ -1,5 +1,6 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
+from typing_extensions import Literal
 
 from .._models import BaseModel
 
@@ -10,8 +11,5 @@ class FleetStatusResponse(BaseModel):
     session_id: str
     """A unique identifier for the session."""
 
-    status: str
-    """The current status of the session.
-
-    Possible values include 'running', 'unknown', or 'terminated'.
-    """
+    status: Literal["starting", "running", "terminated", "completed", "unknown"]
+    """The current status of the session."""

raccoonai/types/fleet_terminate_response.py

@@ -1,5 +1,6 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
+from typing_extensions import Literal
 
 from .._models import BaseModel
 
@@ -10,8 +11,5 @@ class FleetTerminateResponse(BaseModel):
     session_id: str
     """A unique identifier for the session."""
 
-    status: str
-    """The current status of the session.
-
-    Possible values include 'running', 'unknown', or 'terminated'.
-    """
+    status: Literal["starting", "running", "terminated", "completed", "unknown"]
+    """The current status of the session."""

raccoonai/types/lam_run_params.py

@@ -5,7 +5,7 @@ from __future__ import annotations
 from typing import Union, Iterable, Optional
 from typing_extensions import Literal, Required, TypedDict
 
-__all__ = ["LamRunParamsBase", "Advanced", "LamRunParamsNonStreaming", "LamRunParamsStreaming"]
+__all__ = ["LamRunParamsBase", "Advanced", "AdvancedProxy", "LamRunParamsNonStreaming", "LamRunParamsStreaming"]
 
 
 class LamRunParamsBase(TypedDict, total=False):
@@ -33,13 +33,46 @@ class LamRunParamsBase(TypedDict, total=False):
     while building a chat app to give the model context of the past conversation.
     """
 
+    max_count: Optional[int]
+    """The maximum number of results to extract."""
+
+    mode: Optional[Literal["deepsearch", "default"]]
+    """Mode of execution."""
+
+    schema: object
+    """The expected schema for the response.
+
+    This is a dictionary where the keys describe the fields and the values describe
+    their purposes.
+    """
+
+
+class AdvancedProxy(TypedDict, total=False):
+    city: Optional[str]
+    """Target city."""
+
+    country: Optional[str]
+    """Target country (2-letter ISO code)."""
+
+    enable: bool
+    """Whether to use a proxy for the browser session."""
+
+    state: Optional[str]
+    """Target state (2-letter code)."""
+
+    zip: Optional[int]
+    """Target postal code."""
+
 
 class Advanced(TypedDict, total=False):
     block_ads: Optional[bool]
     """Whether to block advertisements during the browser session."""
 
-    proxy: Optional[bool]
-    """Whether to use a proxy for the browser session."""
+    extension_ids: Optional[Iterable[object]]
+    """list of extension ids"""
+
+    proxy: Optional[AdvancedProxy]
+    """Proxy details for the browser session."""
 
     solve_captchas: Optional[bool]
     """Whether to attempt automatic CAPTCHA solving."""

raccoonai/types/lam_run_response.py

@@ -1,5 +1,7 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
+from typing import List
+from typing_extensions import Literal
 
 from .._models import BaseModel
 
@@ -7,6 +9,12 @@ __all__ = ["LamRunResponse"]
 
 
 class LamRunResponse(BaseModel):
+    data: List[object]
+    """The extracted data as a list of objects when the status is DONE.
+
+    Each object represents an extracted entity.
+    """
+
     livestream_url: str
     """The Livestream URL"""
 
@@ -16,7 +24,7 @@ class LamRunResponse(BaseModel):
     properties: object
     """Additional metadata or details related to the run task."""
 
-    task_status: str
+    task_status: Literal["STARTING", "PROCESSING", "DONE", "HUMAN_INTERACTION", "FAILURE"]
     """The current status of the extraction task.
 
     For example: 'STARTING', 'PROCESSING', 'DONE', 'HUMAN_INTERACTION', or