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

deeprails/resources/monitor.py

@@ -0,0 +1,566 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import List
+from typing_extensions import Literal
+
+import httpx
+
+from ..types import monitor_create_params, monitor_update_params, monitor_retrieve_params, monitor_submit_event_params
+from .._types import Body, Omit, Query, Headers, NotGiven, omit, 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.api_response import APIResponse
+from ..types.monitor_retrieve_response import MonitorRetrieveResponse
+from ..types.monitor_submit_event_response import MonitorSubmitEventResponse
+
+__all__ = ["MonitorResource", "AsyncMonitorResource"]
+
+
+class MonitorResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> MonitorResourceWithRawResponse:
+        """
+        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 MonitorResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> MonitorResourceWithStreamingResponse:
+        """
+        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 MonitorResourceWithStreamingResponse(self)
+
+    def create(
+        self,
+        *,
+        name: str,
+        description: str | Omit = omit,
+        # 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,
+    ) -> APIResponse:
+        """
+        Create a new monitor to evaluate model inputs and outputs using guardrails.
+
+        Args:
+          name: Name of the new monitor.
+
+          description: Description of the new monitor.
+
+          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(
+            "/monitor",
+            body=maybe_transform(
+                {
+                    "name": name,
+                    "description": description,
+                },
+                monitor_create_params.MonitorCreateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=APIResponse,
+        )
+
+    def retrieve(
+        self,
+        monitor_id: str,
+        *,
+        limit: int | Omit = omit,
+        # 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,
+    ) -> MonitorRetrieveResponse:
+        """
+        Retrieve the details and evaluations associated with a specific monitor.
+
+        Args:
+          limit: Limit the returned events associated with this monitor. Defaults to 10.
+
+          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 monitor_id:
+            raise ValueError(f"Expected a non-empty value for `monitor_id` but received {monitor_id!r}")
+        return self._get(
+            f"/monitor/{monitor_id}",
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform({"limit": limit}, monitor_retrieve_params.MonitorRetrieveParams),
+            ),
+            cast_to=MonitorRetrieveResponse,
+        )
+
+    def update(
+        self,
+        monitor_id: str,
+        *,
+        description: str | Omit = omit,
+        monitor_status: Literal["active", "inactive"] | Omit = omit,
+        name: str | Omit = omit,
+        # 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,
+    ) -> APIResponse:
+        """
+        Update the name, description, or status of an existing monitor.
+
+        Args:
+          description: Description of the monitor.
+
+          monitor_status: Status of the monitor. Can be `active` or `inactive`. Inactive monitors no
+              longer record and evaluate events.
+
+          name: Name of the monitor.
+
+          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 monitor_id:
+            raise ValueError(f"Expected a non-empty value for `monitor_id` but received {monitor_id!r}")
+        return self._put(
+            f"/monitor/{monitor_id}",
+            body=maybe_transform(
+                {
+                    "description": description,
+                    "monitor_status": monitor_status,
+                    "name": name,
+                },
+                monitor_update_params.MonitorUpdateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=APIResponse,
+        )
+
+    def submit_event(
+        self,
+        monitor_id: str,
+        *,
+        guardrail_metrics: List[
+            Literal[
+                "correctness",
+                "completeness",
+                "instruction_adherence",
+                "context_adherence",
+                "ground_truth_adherence",
+                "comprehensive_safety",
+            ]
+        ],
+        model_input: monitor_submit_event_params.ModelInput,
+        model_output: str,
+        model_used: str | Omit = omit,
+        nametag: str | Omit = omit,
+        run_mode: Literal["precision_plus", "precision", "smart", "economy"] | Omit = omit,
+        # 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,
+    ) -> MonitorSubmitEventResponse:
+        """
+        Submit a model input and output pair to a monitor for evaluation.
+
+        Args:
+          guardrail_metrics: An array of guardrail metrics that the model input and output pair will be
+              evaluated on. For non-enterprise users, these will be limited to `correctness`,
+              `completeness`, `instruction_adherence`, `context_adherence`,
+              `ground_truth_adherence`, and/or `comprehensive_safety`.
+
+          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 monitor 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 monitor_id:
+            raise ValueError(f"Expected a non-empty value for `monitor_id` but received {monitor_id!r}")
+        return self._post(
+            f"/monitor/{monitor_id}/events",
+            body=maybe_transform(
+                {
+                    "guardrail_metrics": guardrail_metrics,
+                    "model_input": model_input,
+                    "model_output": model_output,
+                    "model_used": model_used,
+                    "nametag": nametag,
+                    "run_mode": run_mode,
+                },
+                monitor_submit_event_params.MonitorSubmitEventParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=MonitorSubmitEventResponse,
+        )
+
+
+class AsyncMonitorResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncMonitorResourceWithRawResponse:
+        """
+        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 AsyncMonitorResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncMonitorResourceWithStreamingResponse:
+        """
+        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 AsyncMonitorResourceWithStreamingResponse(self)
+
+    async def create(
+        self,
+        *,
+        name: str,
+        description: str | Omit = omit,
+        # 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,
+    ) -> APIResponse:
+        """
+        Create a new monitor to evaluate model inputs and outputs using guardrails.
+
+        Args:
+          name: Name of the new monitor.
+
+          description: Description of the new monitor.
+
+          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(
+            "/monitor",
+            body=await async_maybe_transform(
+                {
+                    "name": name,
+                    "description": description,
+                },
+                monitor_create_params.MonitorCreateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=APIResponse,
+        )
+
+    async def retrieve(
+        self,
+        monitor_id: str,
+        *,
+        limit: int | Omit = omit,
+        # 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,
+    ) -> MonitorRetrieveResponse:
+        """
+        Retrieve the details and evaluations associated with a specific monitor.
+
+        Args:
+          limit: Limit the returned events associated with this monitor. Defaults to 10.
+
+          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 monitor_id:
+            raise ValueError(f"Expected a non-empty value for `monitor_id` but received {monitor_id!r}")
+        return await self._get(
+            f"/monitor/{monitor_id}",
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=await async_maybe_transform({"limit": limit}, monitor_retrieve_params.MonitorRetrieveParams),
+            ),
+            cast_to=MonitorRetrieveResponse,
+        )
+
+    async def update(
+        self,
+        monitor_id: str,
+        *,
+        description: str | Omit = omit,
+        monitor_status: Literal["active", "inactive"] | Omit = omit,
+        name: str | Omit = omit,
+        # 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,
+    ) -> APIResponse:
+        """
+        Update the name, description, or status of an existing monitor.
+
+        Args:
+          description: Description of the monitor.
+
+          monitor_status: Status of the monitor. Can be `active` or `inactive`. Inactive monitors no
+              longer record and evaluate events.
+
+          name: Name of the monitor.
+
+          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 monitor_id:
+            raise ValueError(f"Expected a non-empty value for `monitor_id` but received {monitor_id!r}")
+        return await self._put(
+            f"/monitor/{monitor_id}",
+            body=await async_maybe_transform(
+                {
+                    "description": description,
+                    "monitor_status": monitor_status,
+                    "name": name,
+                },
+                monitor_update_params.MonitorUpdateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=APIResponse,
+        )
+
+    async def submit_event(
+        self,
+        monitor_id: str,
+        *,
+        guardrail_metrics: List[
+            Literal[
+                "correctness",
+                "completeness",
+                "instruction_adherence",
+                "context_adherence",
+                "ground_truth_adherence",
+                "comprehensive_safety",
+            ]
+        ],
+        model_input: monitor_submit_event_params.ModelInput,
+        model_output: str,
+        model_used: str | Omit = omit,
+        nametag: str | Omit = omit,
+        run_mode: Literal["precision_plus", "precision", "smart", "economy"] | Omit = omit,
+        # 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,
+    ) -> MonitorSubmitEventResponse:
+        """
+        Submit a model input and output pair to a monitor for evaluation.
+
+        Args:
+          guardrail_metrics: An array of guardrail metrics that the model input and output pair will be
+              evaluated on. For non-enterprise users, these will be limited to `correctness`,
+              `completeness`, `instruction_adherence`, `context_adherence`,
+              `ground_truth_adherence`, and/or `comprehensive_safety`.
+
+          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 monitor 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 monitor_id:
+            raise ValueError(f"Expected a non-empty value for `monitor_id` but received {monitor_id!r}")
+        return await self._post(
+            f"/monitor/{monitor_id}/events",
+            body=await async_maybe_transform(
+                {
+                    "guardrail_metrics": guardrail_metrics,
+                    "model_input": model_input,
+                    "model_output": model_output,
+                    "model_used": model_used,
+                    "nametag": nametag,
+                    "run_mode": run_mode,
+                },
+                monitor_submit_event_params.MonitorSubmitEventParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=MonitorSubmitEventResponse,
+        )
+
+
+class MonitorResourceWithRawResponse:
+    def __init__(self, monitor: MonitorResource) -> None:
+        self._monitor = monitor
+
+        self.create = to_raw_response_wrapper(
+            monitor.create,
+        )
+        self.retrieve = to_raw_response_wrapper(
+            monitor.retrieve,
+        )
+        self.update = to_raw_response_wrapper(
+            monitor.update,
+        )
+        self.submit_event = to_raw_response_wrapper(
+            monitor.submit_event,
+        )
+
+
+class AsyncMonitorResourceWithRawResponse:
+    def __init__(self, monitor: AsyncMonitorResource) -> None:
+        self._monitor = monitor
+
+        self.create = async_to_raw_response_wrapper(
+            monitor.create,
+        )
+        self.retrieve = async_to_raw_response_wrapper(
+            monitor.retrieve,
+        )
+        self.update = async_to_raw_response_wrapper(
+            monitor.update,
+        )
+        self.submit_event = async_to_raw_response_wrapper(
+            monitor.submit_event,
+        )
+
+
+class MonitorResourceWithStreamingResponse:
+    def __init__(self, monitor: MonitorResource) -> None:
+        self._monitor = monitor
+
+        self.create = to_streamed_response_wrapper(
+            monitor.create,
+        )
+        self.retrieve = to_streamed_response_wrapper(
+            monitor.retrieve,
+        )
+        self.update = to_streamed_response_wrapper(
+            monitor.update,
+        )
+        self.submit_event = to_streamed_response_wrapper(
+            monitor.submit_event,
+        )
+
+
+class AsyncMonitorResourceWithStreamingResponse:
+    def __init__(self, monitor: AsyncMonitorResource) -> None:
+        self._monitor = monitor
+
+        self.create = async_to_streamed_response_wrapper(
+            monitor.create,
+        )
+        self.retrieve = async_to_streamed_response_wrapper(
+            monitor.retrieve,
+        )
+        self.update = async_to_streamed_response_wrapper(
+            monitor.update,
+        )
+        self.submit_event = async_to_streamed_response_wrapper(
+            monitor.submit_event,
+        )

deeprails/types/__init__.py

@@ -0,0 +1,16 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from .evaluation import Evaluation as Evaluation
+from .api_response import APIResponse as APIResponse
+from .defend_response import DefendResponse as DefendResponse
+from .monitor_create_params import MonitorCreateParams as MonitorCreateParams
+from .monitor_update_params import MonitorUpdateParams as MonitorUpdateParams
+from .evaluate_create_params import EvaluateCreateParams as EvaluateCreateParams
+from .monitor_retrieve_params import MonitorRetrieveParams as MonitorRetrieveParams
+from .monitor_retrieve_response import MonitorRetrieveResponse as MonitorRetrieveResponse
+from .monitor_submit_event_params import MonitorSubmitEventParams as MonitorSubmitEventParams
+from .defend_create_workflow_params import DefendCreateWorkflowParams as DefendCreateWorkflowParams
+from .defend_update_workflow_params import DefendUpdateWorkflowParams as DefendUpdateWorkflowParams
+from .monitor_submit_event_response import MonitorSubmitEventResponse as MonitorSubmitEventResponse

deeprails/types/api_response.py

@@ -0,0 +1,50 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from datetime import datetime
+from typing_extensions import Literal
+
+from .._models import BaseModel
+
+__all__ = ["APIResponse", "Data"]
+
+
+class Data(BaseModel):
+    monitor_id: str
+    """A unique monitor ID."""
+
+    name: str
+    """Name of the monitor."""
+
+    created_at: Optional[datetime] = None
+    """The time the monitor was created in UTC."""
+
+    description: Optional[str] = None
+    """Description of the monitor."""
+
+    monitor_status: Optional[Literal["active", "inactive"]] = None
+    """Status of the monitor.
+
+    Can be `active` or `inactive`. Inactive monitors no longer record and evaluate
+    events.
+    """
+
+    updated_at: Optional[datetime] = None
+    """The most recent time the monitor was modified in UTC."""
+
+    user_id: Optional[str] = None
+    """User ID of the user who created the monitor."""
+
+
+class APIResponse(BaseModel):
+    success: bool
+    """Represents whether the request was completed successfully."""
+
+    data: Optional[Data] = None
+    """Response payload for creating or updating a monitor."""
+
+    message: Optional[str] = None
+    """The accompanying message for the request.
+
+    Includes error details when applicable.
+    """

deeprails/types/defend/__init__.py

@@ -0,0 +1,6 @@
+# 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

deeprails/types/defend/event_submit_event_params.py

@@ -0,0 +1,44 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Dict, Union
+from typing_extensions import Literal, Required, TypeAlias, TypedDict
+
+__all__ = ["EventSubmitEventParams", "ModelInput"]
+
+
+class EventSubmitEventParams(TypedDict, total=False):
+    model_input: Required[ModelInput]
+    """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: Required[str]
+    """Output generated by the LLM to be evaluated."""
+
+    model_used: Required[str]
+    """Model ID used to generate the output, like `gpt-4o` or `o3`."""
+
+    nametag: Required[str]
+    """An optional, user-defined tag for the event."""
+
+    run_mode: Required[Literal["precision_plus", "precision", "smart", "economy"]]
+    """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`.
+    """
+
+
+class ModelInputTyped(TypedDict, total=False):
+    user_prompt: Required[str]
+
+    context: str
+
+
+ModelInput: TypeAlias = Union[ModelInputTyped, Dict[str, object]]