agentex-sdk 0.1.0a6__py3-none-any.whl

agentex/resources/events.py

@@ -0,0 +1,294 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Optional
+
+import httpx
+
+from ..types import event_list_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 ..types.event import Event
+from .._base_client import make_request_options
+from ..types.event_list_response import EventListResponse
+
+__all__ = ["EventsResource", "AsyncEventsResource"]
+
+
+class EventsResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> EventsResourceWithRawResponse:
+        """
+        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/scaleapi/agentex-python#accessing-raw-response-data-eg-headers
+        """
+        return EventsResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> EventsResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/scaleapi/agentex-python#with_streaming_response
+        """
+        return EventsResourceWithStreamingResponse(self)
+
+    def retrieve(
+        self,
+        event_id: str,
+        *,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> Event:
+        """
+        Get Event
+
+        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 event_id:
+            raise ValueError(f"Expected a non-empty value for `event_id` but received {event_id!r}")
+        return self._get(
+            f"/events/{event_id}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Event,
+        )
+
+    def list(
+        self,
+        *,
+        agent_id: str,
+        task_id: str,
+        last_processed_event_id: Optional[str] | NotGiven = NOT_GIVEN,
+        limit: Optional[int] | 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,
+    ) -> EventListResponse:
+        """
+        List events for a specific task and agent.
+
+        Optionally filter for events after a specific sequence ID. Results are ordered
+        by sequence_id.
+
+        Args:
+          agent_id: The agent ID to filter events by
+
+          task_id: The task ID to filter events by
+
+          last_processed_event_id: Optional event ID to get events after this ID
+
+          limit: Optional limit on number of results
+
+          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(
+            "/events",
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "agent_id": agent_id,
+                        "task_id": task_id,
+                        "last_processed_event_id": last_processed_event_id,
+                        "limit": limit,
+                    },
+                    event_list_params.EventListParams,
+                ),
+            ),
+            cast_to=EventListResponse,
+        )
+
+
+class AsyncEventsResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncEventsResourceWithRawResponse:
+        """
+        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/scaleapi/agentex-python#accessing-raw-response-data-eg-headers
+        """
+        return AsyncEventsResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncEventsResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/scaleapi/agentex-python#with_streaming_response
+        """
+        return AsyncEventsResourceWithStreamingResponse(self)
+
+    async def retrieve(
+        self,
+        event_id: str,
+        *,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> Event:
+        """
+        Get Event
+
+        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 event_id:
+            raise ValueError(f"Expected a non-empty value for `event_id` but received {event_id!r}")
+        return await self._get(
+            f"/events/{event_id}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Event,
+        )
+
+    async def list(
+        self,
+        *,
+        agent_id: str,
+        task_id: str,
+        last_processed_event_id: Optional[str] | NotGiven = NOT_GIVEN,
+        limit: Optional[int] | 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,
+    ) -> EventListResponse:
+        """
+        List events for a specific task and agent.
+
+        Optionally filter for events after a specific sequence ID. Results are ordered
+        by sequence_id.
+
+        Args:
+          agent_id: The agent ID to filter events by
+
+          task_id: The task ID to filter events by
+
+          last_processed_event_id: Optional event ID to get events after this ID
+
+          limit: Optional limit on number of results
+
+          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(
+            "/events",
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=await async_maybe_transform(
+                    {
+                        "agent_id": agent_id,
+                        "task_id": task_id,
+                        "last_processed_event_id": last_processed_event_id,
+                        "limit": limit,
+                    },
+                    event_list_params.EventListParams,
+                ),
+            ),
+            cast_to=EventListResponse,
+        )
+
+
+class EventsResourceWithRawResponse:
+    def __init__(self, events: EventsResource) -> None:
+        self._events = events
+
+        self.retrieve = to_raw_response_wrapper(
+            events.retrieve,
+        )
+        self.list = to_raw_response_wrapper(
+            events.list,
+        )
+
+
+class AsyncEventsResourceWithRawResponse:
+    def __init__(self, events: AsyncEventsResource) -> None:
+        self._events = events
+
+        self.retrieve = async_to_raw_response_wrapper(
+            events.retrieve,
+        )
+        self.list = async_to_raw_response_wrapper(
+            events.list,
+        )
+
+
+class EventsResourceWithStreamingResponse:
+    def __init__(self, events: EventsResource) -> None:
+        self._events = events
+
+        self.retrieve = to_streamed_response_wrapper(
+            events.retrieve,
+        )
+        self.list = to_streamed_response_wrapper(
+            events.list,
+        )
+
+
+class AsyncEventsResourceWithStreamingResponse:
+    def __init__(self, events: AsyncEventsResource) -> None:
+        self._events = events
+
+        self.retrieve = async_to_streamed_response_wrapper(
+            events.retrieve,
+        )
+        self.list = async_to_streamed_response_wrapper(
+            events.list,
+        )

agentex/resources/messages/__init__.py

@@ -0,0 +1,33 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from .batch import (
+    BatchResource,
+    AsyncBatchResource,
+    BatchResourceWithRawResponse,
+    AsyncBatchResourceWithRawResponse,
+    BatchResourceWithStreamingResponse,
+    AsyncBatchResourceWithStreamingResponse,
+)
+from .messages import (
+    MessagesResource,
+    AsyncMessagesResource,
+    MessagesResourceWithRawResponse,
+    AsyncMessagesResourceWithRawResponse,
+    MessagesResourceWithStreamingResponse,
+    AsyncMessagesResourceWithStreamingResponse,
+)
+
+__all__ = [
+    "BatchResource",
+    "AsyncBatchResource",
+    "BatchResourceWithRawResponse",
+    "AsyncBatchResourceWithRawResponse",
+    "BatchResourceWithStreamingResponse",
+    "AsyncBatchResourceWithStreamingResponse",
+    "MessagesResource",
+    "AsyncMessagesResource",
+    "MessagesResourceWithRawResponse",
+    "AsyncMessagesResourceWithRawResponse",
+    "MessagesResourceWithStreamingResponse",
+    "AsyncMessagesResourceWithStreamingResponse",
+]

agentex/resources/messages/batch.py

@@ -0,0 +1,271 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Dict, Iterable
+
+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.messages import batch_create_params, batch_update_params
+from ...types.task_message_content_param import TaskMessageContentParam
+from ...types.messages.batch_create_response import BatchCreateResponse
+from ...types.messages.batch_update_response import BatchUpdateResponse
+
+__all__ = ["BatchResource", "AsyncBatchResource"]
+
+
+class BatchResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> BatchResourceWithRawResponse:
+        """
+        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/scaleapi/agentex-python#accessing-raw-response-data-eg-headers
+        """
+        return BatchResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> BatchResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/scaleapi/agentex-python#with_streaming_response
+        """
+        return BatchResourceWithStreamingResponse(self)
+
+    def create(
+        self,
+        *,
+        contents: Iterable[TaskMessageContentParam],
+        task_id: str,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> BatchCreateResponse:
+        """
+        Batch Create Messages
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return self._post(
+            "/messages/batch",
+            body=maybe_transform(
+                {
+                    "contents": contents,
+                    "task_id": task_id,
+                },
+                batch_create_params.BatchCreateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=BatchCreateResponse,
+        )
+
+    def update(
+        self,
+        *,
+        task_id: str,
+        updates: Dict[str, TaskMessageContentParam],
+        # 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,
+    ) -> BatchUpdateResponse:
+        """
+        Batch Update Messages
+
+        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(
+            "/messages/batch",
+            body=maybe_transform(
+                {
+                    "task_id": task_id,
+                    "updates": updates,
+                },
+                batch_update_params.BatchUpdateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=BatchUpdateResponse,
+        )
+
+
+class AsyncBatchResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncBatchResourceWithRawResponse:
+        """
+        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/scaleapi/agentex-python#accessing-raw-response-data-eg-headers
+        """
+        return AsyncBatchResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncBatchResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/scaleapi/agentex-python#with_streaming_response
+        """
+        return AsyncBatchResourceWithStreamingResponse(self)
+
+    async def create(
+        self,
+        *,
+        contents: Iterable[TaskMessageContentParam],
+        task_id: str,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> BatchCreateResponse:
+        """
+        Batch Create Messages
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return await self._post(
+            "/messages/batch",
+            body=await async_maybe_transform(
+                {
+                    "contents": contents,
+                    "task_id": task_id,
+                },
+                batch_create_params.BatchCreateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=BatchCreateResponse,
+        )
+
+    async def update(
+        self,
+        *,
+        task_id: str,
+        updates: Dict[str, TaskMessageContentParam],
+        # 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,
+    ) -> BatchUpdateResponse:
+        """
+        Batch Update Messages
+
+        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(
+            "/messages/batch",
+            body=await async_maybe_transform(
+                {
+                    "task_id": task_id,
+                    "updates": updates,
+                },
+                batch_update_params.BatchUpdateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=BatchUpdateResponse,
+        )
+
+
+class BatchResourceWithRawResponse:
+    def __init__(self, batch: BatchResource) -> None:
+        self._batch = batch
+
+        self.create = to_raw_response_wrapper(
+            batch.create,
+        )
+        self.update = to_raw_response_wrapper(
+            batch.update,
+        )
+
+
+class AsyncBatchResourceWithRawResponse:
+    def __init__(self, batch: AsyncBatchResource) -> None:
+        self._batch = batch
+
+        self.create = async_to_raw_response_wrapper(
+            batch.create,
+        )
+        self.update = async_to_raw_response_wrapper(
+            batch.update,
+        )
+
+
+class BatchResourceWithStreamingResponse:
+    def __init__(self, batch: BatchResource) -> None:
+        self._batch = batch
+
+        self.create = to_streamed_response_wrapper(
+            batch.create,
+        )
+        self.update = to_streamed_response_wrapper(
+            batch.update,
+        )
+
+
+class AsyncBatchResourceWithStreamingResponse:
+    def __init__(self, batch: AsyncBatchResource) -> None:
+        self._batch = batch
+
+        self.create = async_to_streamed_response_wrapper(
+            batch.create,
+        )
+        self.update = async_to_streamed_response_wrapper(
+            batch.update,
+        )