mixpeek 0.8.41__py3-none-any.whl → 0.10.0__py3-none-any.whl

mixpeek/resources/accounts/private.py

@@ -0,0 +1,232 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Iterable, Optional
+
+import httpx
+
+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.accounts import private_update_params
+from ...types.accounts.user import User
+
+__all__ = ["PrivateResource", "AsyncPrivateResource"]
+
+
+class PrivateResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> PrivateResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return the
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/mixpeek/python-client#accessing-raw-response-data-eg-headers
+        """
+        return PrivateResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> PrivateResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/mixpeek/python-client#with_streaming_response
+        """
+        return PrivateResourceWithStreamingResponse(self)
+
+    def update(
+        self,
+        *,
+        api_keys: Optional[Iterable[private_update_params.APIKey]] | NotGiven = NOT_GIVEN,
+        metadata: Optional[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,
+    ) -> User:
+        """
+        Update User
+
+        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._put(
+            "/accounts/private/",
+            body=maybe_transform(
+                {
+                    "api_keys": api_keys,
+                    "metadata": metadata,
+                },
+                private_update_params.PrivateUpdateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=User,
+        )
+
+    def list(
+        self,
+        *,
+        # 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,
+    ) -> User:
+        """Get User"""
+        return self._get(
+            "/accounts/private/",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=User,
+        )
+
+
+class AsyncPrivateResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncPrivateResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return the
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/mixpeek/python-client#accessing-raw-response-data-eg-headers
+        """
+        return AsyncPrivateResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncPrivateResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/mixpeek/python-client#with_streaming_response
+        """
+        return AsyncPrivateResourceWithStreamingResponse(self)
+
+    async def update(
+        self,
+        *,
+        api_keys: Optional[Iterable[private_update_params.APIKey]] | NotGiven = NOT_GIVEN,
+        metadata: Optional[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,
+    ) -> User:
+        """
+        Update User
+
+        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._put(
+            "/accounts/private/",
+            body=await async_maybe_transform(
+                {
+                    "api_keys": api_keys,
+                    "metadata": metadata,
+                },
+                private_update_params.PrivateUpdateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=User,
+        )
+
+    async def list(
+        self,
+        *,
+        # 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,
+    ) -> User:
+        """Get User"""
+        return await self._get(
+            "/accounts/private/",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=User,
+        )
+
+
+class PrivateResourceWithRawResponse:
+    def __init__(self, private: PrivateResource) -> None:
+        self._private = private
+
+        self.update = to_raw_response_wrapper(
+            private.update,
+        )
+        self.list = to_raw_response_wrapper(
+            private.list,
+        )
+
+
+class AsyncPrivateResourceWithRawResponse:
+    def __init__(self, private: AsyncPrivateResource) -> None:
+        self._private = private
+
+        self.update = async_to_raw_response_wrapper(
+            private.update,
+        )
+        self.list = async_to_raw_response_wrapper(
+            private.list,
+        )
+
+
+class PrivateResourceWithStreamingResponse:
+    def __init__(self, private: PrivateResource) -> None:
+        self._private = private
+
+        self.update = to_streamed_response_wrapper(
+            private.update,
+        )
+        self.list = to_streamed_response_wrapper(
+            private.list,
+        )
+
+
+class AsyncPrivateResourceWithStreamingResponse:
+    def __init__(self, private: AsyncPrivateResource) -> None:
+        self._private = private
+
+        self.update = async_to_streamed_response_wrapper(
+            private.update,
+        )
+        self.list = async_to_streamed_response_wrapper(
+            private.list,
+        )

mixpeek/resources/agent/__init__.py

@@ -0,0 +1,33 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from .task import (
+    TaskResource,
+    AsyncTaskResource,
+    TaskResourceWithRawResponse,
+    AsyncTaskResourceWithRawResponse,
+    TaskResourceWithStreamingResponse,
+    AsyncTaskResourceWithStreamingResponse,
+)
+from .agent import (
+    AgentResource,
+    AsyncAgentResource,
+    AgentResourceWithRawResponse,
+    AsyncAgentResourceWithRawResponse,
+    AgentResourceWithStreamingResponse,
+    AsyncAgentResourceWithStreamingResponse,
+)
+
+__all__ = [
+    "TaskResource",
+    "AsyncTaskResource",
+    "TaskResourceWithRawResponse",
+    "AsyncTaskResourceWithRawResponse",
+    "TaskResourceWithStreamingResponse",
+    "AsyncTaskResourceWithStreamingResponse",
+    "AgentResource",
+    "AsyncAgentResource",
+    "AgentResourceWithRawResponse",
+    "AsyncAgentResourceWithRawResponse",
+    "AgentResourceWithStreamingResponse",
+    "AsyncAgentResourceWithStreamingResponse",
+]

mixpeek/resources/agent/agent.py

@@ -0,0 +1,225 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+import httpx
+
+from .task import (
+    TaskResource,
+    AsyncTaskResource,
+    TaskResourceWithRawResponse,
+    AsyncTaskResourceWithRawResponse,
+    TaskResourceWithStreamingResponse,
+    AsyncTaskResourceWithStreamingResponse,
+)
+from ...types import agent_create_params
+from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from ..._utils import (
+    maybe_transform,
+    strip_not_given,
+    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.agentresponse import Agentresponse
+
+__all__ = ["AgentResource", "AsyncAgentResource"]
+
+
+class AgentResource(SyncAPIResource):
+    @cached_property
+    def task(self) -> TaskResource:
+        return TaskResource(self._client)
+
+    @cached_property
+    def with_raw_response(self) -> AgentResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return the
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/mixpeek/python-client#accessing-raw-response-data-eg-headers
+        """
+        return AgentResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AgentResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/mixpeek/python-client#with_streaming_response
+        """
+        return AgentResourceWithStreamingResponse(self)
+
+    def create(
+        self,
+        *,
+        prompt: str,
+        authorization: str | NotGiven = NOT_GIVEN,
+        index_id: str | 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,
+    ) -> Agentresponse:
+        """
+        Start Agent
+
+        Args:
+          index_id: filter by organization
+
+          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
+        """
+        extra_headers = {
+            **strip_not_given(
+                {
+                    "Authorization": authorization,
+                    "index-id": index_id,
+                }
+            ),
+            **(extra_headers or {}),
+        }
+        return self._post(
+            "/agent/",
+            body=maybe_transform({"prompt": prompt}, agent_create_params.AgentCreateParams),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Agentresponse,
+        )
+
+
+class AsyncAgentResource(AsyncAPIResource):
+    @cached_property
+    def task(self) -> AsyncTaskResource:
+        return AsyncTaskResource(self._client)
+
+    @cached_property
+    def with_raw_response(self) -> AsyncAgentResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return the
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/mixpeek/python-client#accessing-raw-response-data-eg-headers
+        """
+        return AsyncAgentResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncAgentResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/mixpeek/python-client#with_streaming_response
+        """
+        return AsyncAgentResourceWithStreamingResponse(self)
+
+    async def create(
+        self,
+        *,
+        prompt: str,
+        authorization: str | NotGiven = NOT_GIVEN,
+        index_id: str | 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,
+    ) -> Agentresponse:
+        """
+        Start Agent
+
+        Args:
+          index_id: filter by organization
+
+          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
+        """
+        extra_headers = {
+            **strip_not_given(
+                {
+                    "Authorization": authorization,
+                    "index-id": index_id,
+                }
+            ),
+            **(extra_headers or {}),
+        }
+        return await self._post(
+            "/agent/",
+            body=await async_maybe_transform({"prompt": prompt}, agent_create_params.AgentCreateParams),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Agentresponse,
+        )
+
+
+class AgentResourceWithRawResponse:
+    def __init__(self, agent: AgentResource) -> None:
+        self._agent = agent
+
+        self.create = to_raw_response_wrapper(
+            agent.create,
+        )
+
+    @cached_property
+    def task(self) -> TaskResourceWithRawResponse:
+        return TaskResourceWithRawResponse(self._agent.task)
+
+
+class AsyncAgentResourceWithRawResponse:
+    def __init__(self, agent: AsyncAgentResource) -> None:
+        self._agent = agent
+
+        self.create = async_to_raw_response_wrapper(
+            agent.create,
+        )
+
+    @cached_property
+    def task(self) -> AsyncTaskResourceWithRawResponse:
+        return AsyncTaskResourceWithRawResponse(self._agent.task)
+
+
+class AgentResourceWithStreamingResponse:
+    def __init__(self, agent: AgentResource) -> None:
+        self._agent = agent
+
+        self.create = to_streamed_response_wrapper(
+            agent.create,
+        )
+
+    @cached_property
+    def task(self) -> TaskResourceWithStreamingResponse:
+        return TaskResourceWithStreamingResponse(self._agent.task)
+
+
+class AsyncAgentResourceWithStreamingResponse:
+    def __init__(self, agent: AsyncAgentResource) -> None:
+        self._agent = agent
+
+        self.create = async_to_streamed_response_wrapper(
+            agent.create,
+        )
+
+    @cached_property
+    def task(self) -> AsyncTaskResourceWithStreamingResponse:
+        return AsyncTaskResourceWithStreamingResponse(self._agent.task)

mixpeek/resources/agent/task.py

@@ -0,0 +1,189 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+import httpx
+
+from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from ..._utils import strip_not_given
+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
+
+__all__ = ["TaskResource", "AsyncTaskResource"]
+
+
+class TaskResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> TaskResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return the
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/mixpeek/python-client#accessing-raw-response-data-eg-headers
+        """
+        return TaskResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> TaskResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/mixpeek/python-client#with_streaming_response
+        """
+        return TaskResourceWithStreamingResponse(self)
+
+    def retrieve(
+        self,
+        task_id: str,
+        *,
+        authorization: str | NotGiven = NOT_GIVEN,
+        index_id: str | 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,
+    ) -> object:
+        """
+        Get Agent Result
+
+        Args:
+          index_id: filter by organization
+
+          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 task_id:
+            raise ValueError(f"Expected a non-empty value for `task_id` but received {task_id!r}")
+        extra_headers = {
+            **strip_not_given(
+                {
+                    "Authorization": authorization,
+                    "index-id": index_id,
+                }
+            ),
+            **(extra_headers or {}),
+        }
+        return self._get(
+            f"/agent/{task_id}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=object,
+        )
+
+
+class AsyncTaskResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncTaskResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return the
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/mixpeek/python-client#accessing-raw-response-data-eg-headers
+        """
+        return AsyncTaskResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncTaskResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/mixpeek/python-client#with_streaming_response
+        """
+        return AsyncTaskResourceWithStreamingResponse(self)
+
+    async def retrieve(
+        self,
+        task_id: str,
+        *,
+        authorization: str | NotGiven = NOT_GIVEN,
+        index_id: str | 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,
+    ) -> object:
+        """
+        Get Agent Result
+
+        Args:
+          index_id: filter by organization
+
+          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 task_id:
+            raise ValueError(f"Expected a non-empty value for `task_id` but received {task_id!r}")
+        extra_headers = {
+            **strip_not_given(
+                {
+                    "Authorization": authorization,
+                    "index-id": index_id,
+                }
+            ),
+            **(extra_headers or {}),
+        }
+        return await self._get(
+            f"/agent/{task_id}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=object,
+        )
+
+
+class TaskResourceWithRawResponse:
+    def __init__(self, task: TaskResource) -> None:
+        self._task = task
+
+        self.retrieve = to_raw_response_wrapper(
+            task.retrieve,
+        )
+
+
+class AsyncTaskResourceWithRawResponse:
+    def __init__(self, task: AsyncTaskResource) -> None:
+        self._task = task
+
+        self.retrieve = async_to_raw_response_wrapper(
+            task.retrieve,
+        )
+
+
+class TaskResourceWithStreamingResponse:
+    def __init__(self, task: TaskResource) -> None:
+        self._task = task
+
+        self.retrieve = to_streamed_response_wrapper(
+            task.retrieve,
+        )
+
+
+class AsyncTaskResourceWithStreamingResponse:
+    def __init__(self, task: AsyncTaskResource) -> None:
+        self._task = task
+
+        self.retrieve = async_to_streamed_response_wrapper(
+            task.retrieve,
+        )