deeprails 1.0.0__py3-none-any.whl → 1.2.0__py3-none-any.whl

deeprails/resources/defend/events.py

@@ -1,311 +0,0 @@
-# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
-
-from __future__ import annotations
-
-from typing_extensions import Literal
-
-import httpx
-
-from ..._types import Body, Query, Headers, NotGiven, not_given
-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.defend import event_submit_event_params
-from ...types.defend.workflow_event_response import WorkflowEventResponse
-
-__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/deeprails/deeprails-python-sdk#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/deeprails/deeprails-python-sdk#with_streaming_response
-        """
-        return EventsResourceWithStreamingResponse(self)
-
-    def retrieve_event(
-        self,
-        event_id: str,
-        *,
-        workflow_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,
-    ) -> WorkflowEventResponse:
-        """
-        Retrieve a specific event of a guardrail workflow.
-
-        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 workflow_id:
-            raise ValueError(f"Expected a non-empty value for `workflow_id` but received {workflow_id!r}")
-        if not event_id:
-            raise ValueError(f"Expected a non-empty value for `event_id` but received {event_id!r}")
-        return self._get(
-            f"/defend/{workflow_id}/events/{event_id}",
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=WorkflowEventResponse,
-        )
-
-    def submit_event(
-        self,
-        workflow_id: str,
-        *,
-        model_input: event_submit_event_params.ModelInput,
-        model_output: str,
-        model_used: str,
-        nametag: str,
-        run_mode: Literal["precision_plus", "precision", "smart", "economy"],
-        # 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,
-    ) -> WorkflowEventResponse:
-        """
-        Submit a model input and output pair to a workflow for evaluation.
-
-        Args:
-          model_input: A dictionary of inputs sent to the LLM to generate output. This must contain a
-              `user_prompt` field and an optional `context` field. Additional properties are
-              allowed.
-
-          model_output: Output generated by the LLM to be evaluated.
-
-          model_used: Model ID used to generate the output, like `gpt-4o` or `o3`.
-
-          nametag: An optional, user-defined tag for the event.
-
-          run_mode: Run mode for the workflow event. The run mode allows the user to optimize for
-              speed, accuracy, and cost by determining which models are used to evaluate the
-              event. Available run modes include `precision_plus`, `precision`, `smart`, and
-              `economy`. Defaults to `smart`.
-
-          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 workflow_id:
-            raise ValueError(f"Expected a non-empty value for `workflow_id` but received {workflow_id!r}")
-        return self._post(
-            f"/defend/{workflow_id}/events",
-            body=maybe_transform(
-                {
-                    "model_input": model_input,
-                    "model_output": model_output,
-                    "model_used": model_used,
-                    "nametag": nametag,
-                    "run_mode": run_mode,
-                },
-                event_submit_event_params.EventSubmitEventParams,
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=WorkflowEventResponse,
-        )
-
-
-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/deeprails/deeprails-python-sdk#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/deeprails/deeprails-python-sdk#with_streaming_response
-        """
-        return AsyncEventsResourceWithStreamingResponse(self)
-
-    async def retrieve_event(
-        self,
-        event_id: str,
-        *,
-        workflow_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,
-    ) -> WorkflowEventResponse:
-        """
-        Retrieve a specific event of a guardrail workflow.
-
-        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 workflow_id:
-            raise ValueError(f"Expected a non-empty value for `workflow_id` but received {workflow_id!r}")
-        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"/defend/{workflow_id}/events/{event_id}",
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=WorkflowEventResponse,
-        )
-
-    async def submit_event(
-        self,
-        workflow_id: str,
-        *,
-        model_input: event_submit_event_params.ModelInput,
-        model_output: str,
-        model_used: str,
-        nametag: str,
-        run_mode: Literal["precision_plus", "precision", "smart", "economy"],
-        # 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,
-    ) -> WorkflowEventResponse:
-        """
-        Submit a model input and output pair to a workflow for evaluation.
-
-        Args:
-          model_input: A dictionary of inputs sent to the LLM to generate output. This must contain a
-              `user_prompt` field and an optional `context` field. Additional properties are
-              allowed.
-
-          model_output: Output generated by the LLM to be evaluated.
-
-          model_used: Model ID used to generate the output, like `gpt-4o` or `o3`.
-
-          nametag: An optional, user-defined tag for the event.
-
-          run_mode: Run mode for the workflow event. The run mode allows the user to optimize for
-              speed, accuracy, and cost by determining which models are used to evaluate the
-              event. Available run modes include `precision_plus`, `precision`, `smart`, and
-              `economy`. Defaults to `smart`.
-
-          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 workflow_id:
-            raise ValueError(f"Expected a non-empty value for `workflow_id` but received {workflow_id!r}")
-        return await self._post(
-            f"/defend/{workflow_id}/events",
-            body=await async_maybe_transform(
-                {
-                    "model_input": model_input,
-                    "model_output": model_output,
-                    "model_used": model_used,
-                    "nametag": nametag,
-                    "run_mode": run_mode,
-                },
-                event_submit_event_params.EventSubmitEventParams,
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=WorkflowEventResponse,
-        )
-
-
-class EventsResourceWithRawResponse:
-    def __init__(self, events: EventsResource) -> None:
-        self._events = events
-
-        self.retrieve_event = to_raw_response_wrapper(
-            events.retrieve_event,
-        )
-        self.submit_event = to_raw_response_wrapper(
-            events.submit_event,
-        )
-
-
-class AsyncEventsResourceWithRawResponse:
-    def __init__(self, events: AsyncEventsResource) -> None:
-        self._events = events
-
-        self.retrieve_event = async_to_raw_response_wrapper(
-            events.retrieve_event,
-        )
-        self.submit_event = async_to_raw_response_wrapper(
-            events.submit_event,
-        )
-
-
-class EventsResourceWithStreamingResponse:
-    def __init__(self, events: EventsResource) -> None:
-        self._events = events
-
-        self.retrieve_event = to_streamed_response_wrapper(
-            events.retrieve_event,
-        )
-        self.submit_event = to_streamed_response_wrapper(
-            events.submit_event,
-        )
-
-
-class AsyncEventsResourceWithStreamingResponse:
-    def __init__(self, events: AsyncEventsResource) -> None:
-        self._events = events
-
-        self.retrieve_event = async_to_streamed_response_wrapper(
-            events.retrieve_event,
-        )
-        self.submit_event = async_to_streamed_response_wrapper(
-            events.submit_event,
-        )

deeprails/types/defend/__init__.py

@@ -1,6 +0,0 @@
-# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
-
-from __future__ import annotations
-
-from .workflow_event_response import WorkflowEventResponse as WorkflowEventResponse
-from .event_submit_event_params import EventSubmitEventParams as EventSubmitEventParams