deeprails 0.1.0__py3-none-any.whl → 0.2.0__py3-none-any.whl

deeprails/__init__.py

@@ -0,0 +1 @@
+from .client import DeepRails

deeprails/client.py

@@ -1,31 +1,101 @@
-import requests
-from typing import Dict, Any
+import httpx
+from typing import List, Optional, Dict, Any
 
-from .evaluate import EvaluateClient
-from .monitor import MonitorClient
+from .schemas import EvaluationResponse
+from .exceptions import DeepRailsAPIError
 
 class DeepRails:
     """
-    Main entry point for the deeprails client.
-    
-    Usage:
-        from deeprails import DeepRails
-        
-        client = DeepRails(token="YOUR_TOKEN")
-        resp = client.evaluate.create({...})
+    Python SDK client for the DeepRails API.
     """
-    def __init__(self, token: str, base_url: str = "https://deeprails.ai"):
+    def __init__(self, token: str, base_url: str = "https://api.deeprails.com"):
         """
-        :param token: Bearer token for authentication.
-        :param base_url: Base URL for the DeepRails API.
+        Initializes the DeepRails client.
+
+        Args:
+            token: Your DeepRails API key (starts with 'sk_').
+            base_url: The base URL of the DeepRails API.
         """
-        self.token = token
-        self.base_url = base_url.rstrip("/")
-        # Common headers for all requests
-        self.headers = {
+        if not token:
+            raise ValueError("A valid DeepRails API token is required.")
+        
+        self._base_url = base_url
+        self._headers = {
+            "Authorization": f"Bearer {token}",
             "Content-Type": "application/json",
-            "Authorization": f"Bearer {self.token}"
+            "User-Agent": "deeprails-python-sdk/0.2.0"
         }
-        # Create client objects
-        self.evaluate = EvaluateClient(self.base_url, self.headers)
-        self.monitor = MonitorClient(self.base_url, self.headers)
+        self._client = httpx.Client(base_url=self._base_url, headers=self._headers, timeout=30.0)
+
+    def _request(self, method: str, endpoint: str, **kwargs) -> httpx.Response:
+        """Helper method to make requests and handle API errors."""
+        try:
+            response = self._client.request(method, endpoint, **kwargs)
+            response.raise_for_status()
+            return response
+        except httpx.HTTPStatusError as e:
+            error_detail = "No detail provided."
+            try:
+                error_detail = e.response.json().get("detail", error_detail)
+            except Exception:
+                error_detail = e.response.text
+            raise DeepRailsAPIError(status_code=e.response.status_code, error_detail=error_detail) from e
+        except httpx.RequestError as e:
+            raise DeepRailsAPIError(status_code=500, error_detail=f"Request failed: {e}") from e
+
+    def create_evaluation(
+        self,
+        *,
+        model_input: Dict[str, Any],
+        model_output: str,
+        model_used: Optional[str] = None,
+        run_mode: Optional[str] = "smart",  # Set default to "smart"
+        guardrail_metrics: Optional[List[str]] = None,
+        nametag: Optional[str] = None,
+        webhook: Optional[str] = None
+    ) -> EvaluationResponse:
+        """
+        Creates a new evaluation and immediately processes it.
+
+        Args:
+            model_input: A dictionary containing the inputs for the model.
+                         Must contain a "user_prompt" key.
+            model_output: The response generated by the model you are evaluating.
+            model_used: The name or identifier of the model being evaluated.
+            run_mode: The evaluation mode (e.g., "smart", "dev").
+            guardrail_metrics: A list of metrics to evaluate.
+            nametag: A user-defined name or tag for the evaluation.
+            webhook: A URL to send a POST request to upon evaluation completion.
+
+        Returns:
+            An EvaluationResponse object with the details of the created evaluation.
+        """
+        if "user_prompt" not in model_input:
+            raise ValueError("`model_input` must contain a 'user_prompt' key.")
+
+        payload = {
+            "model_input": model_input,
+            "model_output": model_output,
+            "model_used": model_used,
+            "run_mode": run_mode,
+            "guardrail_metrics": guardrail_metrics,
+            "nametag": nametag,
+            "webhook": webhook,
+        }
+        json_payload = {k: v for k, v in payload.items() if v is not None}
+
+        response = self._request("POST", "/evaluate", json=json_payload)
+        return EvaluationResponse.parse_obj(response.json())
+
+    def get_evaluation(self, eval_id: str) -> EvaluationResponse:
+        """
+        Retrieves the status and results of a specific evaluation.
+
+        Args:
+            eval_id: The unique identifier of the evaluation.
+
+        Returns:
+            An EvaluationResponse object with the full, up-to-date details of the evaluation.
+        """
+        response = self._request("GET", f"/evaluate/{eval_id}")
+        return EvaluationResponse.parse_obj(response.json())

deeprails/exceptions.py

@@ -0,0 +1,10 @@
+class DeepRailsError(Exception):
+    """Base exception class for the DeepRails SDK."""
+    pass
+
+class DeepRailsAPIError(DeepRailsError):
+    """Raised when the DeepRails API returns an error."""
+    def __init__(self, status_code: int, error_detail: str):
+        self.status_code = status_code
+        self.error_detail = error_detail
+        super().__init__(f"API Error {status_code}: {error_detail}")

deeprails/schemas.py

@@ -0,0 +1,30 @@
+from typing import List, Optional, Dict, Any
+from pydantic import BaseModel, Field
+from datetime import datetime
+
+
+class EvaluationResponse(BaseModel):
+    """Represents the response for an evaluation from the DeepRails API."""
+    eval_id: str
+    evaluation_status: str
+    guardrail_metrics: Optional[List[str]] = None
+    model_used: Optional[str] = None
+    run_mode: Optional[str] = None
+    model_input: Optional[Dict[str, Any]] = None
+    model_output: Optional[str] = None
+    estimated_cost: Optional[float] = None
+    input_tokens: Optional[int] = None
+    output_tokens: Optional[int] = None
+    nametag: Optional[str] = None
+    progress: Optional[int] = Field(None, ge=0, le=100)
+    start_timestamp: Optional[datetime] = None
+    completion_timestamp: Optional[datetime] = None
+    error_message: Optional[str] = None
+    error_timestamp: Optional[datetime] = None
+    evaluation_result: Optional[Dict[str, Any]] = None
+    evaluation_total_cost: Optional[float] = None
+    created_at: Optional[datetime] = None
+    modified_at: Optional[datetime] = None
+
+    class Config:
+        extra = 'ignore'

deeprails-0.2.0.dist-info/METADATA

@@ -0,0 +1,154 @@
+Metadata-Version: 2.4
+Name: deeprails
+Version: 0.2.0
+Summary: Python SDK for interacting with the DeepRails API
+Project-URL: Homepage, https://deeprails.com
+Project-URL: Documentation, https://docs.deeprails.com
+Author-email: Neil Mate <support@deeprails.ai>
+License: MIT License
+        
+        Copyright (c) [2025] [DeepRails Inc.ß]
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+        
+        IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+License-File: LICENSE
+Keywords: ai,deeprails,evaluation,genai,guardrails,sdk
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.8
+Requires-Dist: httpx<0.29.0,>=0.28.1
+Requires-Dist: pydantic<3.0.0,>=2.11.7
+Description-Content-Type: text/markdown
+
+# DeepRails Python SDK
+
+A lightweight, intuitive Python SDK for interacting with the DeepRails API. DeepRails helps you evaluate and improve AI-generated outputs through a comprehensive set of guardrail metrics.
+
+## Installation
+
+```bash
+pip install deeprails
+```
+
+## Quick Start
+
+```python
+from deeprails import DeepRails
+
+# Initialize with your API token
+client = DeepRails(token="YOUR_API_KEY")
+
+# Create an evaluation
+evaluation = client.create_evaluation(
+    model_input={"user_prompt": "Prompt used to generate completion"},
+    model_output="Generated output",
+    model_used="gpt-4o-mini (LLM used to generate completion)",
+    guardrail_metrics=["correctness", "completeness"]
+)
+
+# Print evaluation ID
+print(f"Evaluation created with ID: {evaluation.eval_id}")
+```
+
+## Features
+
+- **Simple API**: Just a few lines of code to integrate evaluation into your workflow
+- **Comprehensive Metrics**: Evaluate outputs on correctness, completeness, and more
+- **Real-time Progress**: Track evaluation progress in real-time
+- **Detailed Results**: Get detailed scores and rationales for each metric
+
+## Authentication
+
+All API requests require authentication using your DeepRails API key. Your API key is a sensitive credential that should be kept secure.
+
+```python
+# Best practice: Load token from environment variable
+import os
+token = os.environ.get("DEEPRAILS_API_KEY")
+client = DeepRails(token=token)
+```
+
+## Creating Evaluations
+
+```python
+try:
+    evaluation = client.create_evaluation(
+        model_input={"user_prompt": "Prompt used to generate completion"},
+        model_output="Generated output",
+        model_used="gpt-4o-mini (LLM used to generate completion)",
+        guardrail_metrics=["correctness", "completeness"]
+    )
+    print(f"ID: {evaluation.eval_id}")
+    print(f"Status: {evaluation.evaluation_status}")
+    print(f"Progress: {evaluation.progress}%")
+except Exception as e:
+    print(f"Error: {e}")
+```
+
+### Parameters
+
+- `model_input`: Dictionary containing the prompt and any context (must include `user_prompt`)
+- `model_output`: The generated output to evaluate
+- `model_used`: (Optional) The model that generated the output
+- `run_mode`: (Optional) Evaluation run mode - defaults to "smart"
+- `guardrail_metrics`: (Optional) List of metrics to evaluate
+- `nametag`: (Optional) Custom identifier for this evaluation
+- `webhook`: (Optional) URL to receive completion notifications
+
+## Retrieving Evaluations
+
+```python
+try:
+    eval_id = "eval-xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx"
+    evaluation = client.get_evaluation(eval_id)
+    
+    print(f"Status: {evaluation.evaluation_status}")
+    
+    if evaluation.evaluation_result:
+        print("\nResults:")
+        for metric, result in evaluation.evaluation_result.items():
+            score = result.get('score', 'N/A')
+            print(f"  {metric}: {score}")
+except Exception as e:
+    print(f"Error: {e}")
+```
+
+## Available Metrics
+
+- `correctness`: Evaluates factual accuracy of the output
+- `completeness`: Checks if the output addresses all aspects of the prompt
+- `harmfulness`: Detects potentially harmful content
+- `bias`: Identifies biased language or reasoning
+- And more...
+
+## Error Handling
+
+The SDK throws `DeepRailsAPIError` for API-related errors, with status code and detailed message.
+
+```python
+from deeprails import DeepRailsAPIError
+
+try:
+    # SDK operations
+except DeepRailsAPIError as e:
+    print(f"API Error: {e.status_code} - {e.error_detail}")
+except Exception as e:
+    print(f"Unexpected error: {e}")
+```
+
+## Support
+
+For questions or support, please contact support@deeprails.ai.

deeprails-0.2.0.dist-info/RECORD

@@ -0,0 +1,8 @@
+deeprails/__init__.py,sha256=7ccTz1heYcCd3DIH3wmHc67FD6CUzM8_J4WmDeq0RZ0,29
+deeprails/client.py,sha256=gm3aq5FQj7qpLbWeyKkew3gbRI0l9QyGhvk7m0N7JxU,4004
+deeprails/exceptions.py,sha256=ipwFq4lROv7XpcBC5h9cGqPf6f68zeOMEyKPVy7H0co,405
+deeprails/schemas.py,sha256=pqZ-J7mIVdYe-q8sPSw07fe3H4YFCYb-PrshA_qU8RU,1113
+deeprails-0.2.0.dist-info/METADATA,sha256=Evweth9DLw-iHgbfZQdhfFG0i6MMXnm0GRkPyhscJe0,5759
+deeprails-0.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+deeprails-0.2.0.dist-info/licenses/LICENSE,sha256=GsV7lN6fihCcDgkJbfs0rq1q9d6IyB0TFQ8HLKUpSXM,1077
+deeprails-0.2.0.dist-info/RECORD,,

{deeprails-0.1.0.dist-info → deeprails-0.2.0.dist-info}/WHEEL

@@ -1,5 +1,4 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.37.0)
+Generator: hatchling 1.27.0
 Root-Is-Purelib: true
 Tag: py3-none-any
-

deeprails/evaluate.py

@@ -1,46 +0,0 @@
-import requests
-from typing import Dict, Any
-
-class EvaluateClient:
-    """
-    Client for the Evaluate endpoints:
-      - POST / (create a new evaluation)
-      - GET /{eval_id} (fetch an existing evaluation)
-    """
-    def __init__(self, base_url: str, headers: Dict[str, str]):
-        self.base_url = f"{base_url}/evaluate"
-        self.headers = headers
-
-    def create(self, evaluation_data: Dict[str, Any]) -> Dict[str, Any]:
-        """
-        Create a new evaluation by POSTing to /evaluate.
-        
-        :param evaluation_data: Dict matching the EvaluateCreate model:
-            {
-              "model_input": {...},
-              "model_output": "...",
-              "type": "...",
-              "guardrails_metrics": [...],
-              "score_format": "...",
-              "webhook": "..."
-            }
-        :return: The JSON response from the API as a Python dict
-                 matching EvaluationResponse.
-        """
-        url = self.base_url  # e.g. https://deeprails.ai/evaluate
-        response = requests.post(url, json=evaluation_data, headers=self.headers)
-        response.raise_for_status()
-        return response.json()
-
-    def fetch(self, eval_id: str) -> Dict[str, Any]:
-        """
-        Fetch an existing evaluation by GETting /evaluate/{eval_id}.
-        
-        :param eval_id: The ID of the evaluation to retrieve
-        :return: The JSON response as a Python dict
-                 matching EvaluationResponse.
-        """
-        url = f"{self.base_url}/{eval_id}"
-        response = requests.get(url, headers=self.headers)
-        response.raise_for_status()
-        return response.json()

deeprails/monitor.py

@@ -1,50 +0,0 @@
-import requests
-from typing import Dict, Any
-
-class MonitorClient:
-    """
-    Client for the Monitor endpoints:
-      - POST / (create a new monitor)
-      - POST /{monitor_id}/event (log an event)
-    """
-    def __init__(self, base_url: str, headers: Dict[str, str]):
-        self.base_url = f"{base_url}/monitor"
-        self.headers = headers
-
-    def create(self, monitor_data: Dict[str, Any]) -> Dict[str, Any]:
-        """
-        Create a new monitor.
-        
-        :param monitor_data: Dict matching the MonitorCreate model:
-            {
-              "name": "...",
-              "description": "...",
-              "metrics": [...]
-            }
-        :return: Dictionary with e.g. {"monitor_id": "..."} from the API.
-        """
-        url = self.base_url  # e.g. https://deeprails.ai/monitor
-        response = requests.post(url, json=monitor_data, headers=self.headers)
-        response.raise_for_status()
-        return response.json()
-
-    def log(self, monitor_id: str, event_data: Dict[str, Any]) -> Dict[str, Any]:
-        """
-        Log an event under a specific monitor.
-        
-        :param monitor_id: ID of the monitor to which we log the event
-        :param event_data: Dict matching MonitorEventCreate:
-            {
-              "model_input": {...},
-              "model_output": {...},
-              "temperature": 0.7,
-              "top_p": 1.0,
-              "model": "gpt-3.5-turbo",
-              ...
-            }
-        :return: Dictionary with e.g. {"event_id": "..."} from the API.
-        """
-        url = f"{self.base_url}/{monitor_id}/event"
-        response = requests.post(url, json=event_data, headers=self.headers)
-        response.raise_for_status()
-        return response.json()

deeprails-0.1.0.dist-info/METADATA

@@ -1,86 +0,0 @@
-Metadata-Version: 2.1
-Name: deeprails
-Version: 0.1.0
-Summary: Python client library for deeprails.ai
-Home-page: https://github.com/yourusername/deeprails
-Author: Your Name or Organization
-Author-email: your.email@example.com
-License: UNKNOWN
-Platform: UNKNOWN
-Classifier: Programming Language :: Python :: 3
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.7
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: requests (>=2.0.0)
-
-# DeepRails
-
-A Python client library for interacting with the [DeepRails.ai](https://deeprails.ai) API. This package exposes high-level methods for creating and fetching evaluations, as well as creating monitors and logging monitor events.
-
-## Features
-
-- **Evaluation**:  
-  - `client.evaluate.create(data)` – Create an evaluation
-  - `client.evaluate.fetch(eval_id)` – Retrieve the status and results of an evaluation
-
-- **Monitor**:  
-  - `client.monitor.create(data)` – Create a monitor
-  - `client.monitor.log(monitor_id, data)` – Log an event under a monitor
-
-## Installation
-
-Install via [PyPI](https://pypi.org/) using:
-
-```bash
-pip install deeprails
-```
-
-## Quick Start
-```python
-from deeprails import DeepRails
-
-# Initialize the client (replace "YOUR_TOKEN" with a valid token)
-client = DeepRails(token="YOUR_TOKEN")
-
-# 1) Create an evaluation
-eval_request_data = {
-    "model_input": {"user_prompt": "Write a poem about winter"},
-    "model_output": "A frosty greeting on a snowy morn...",
-    "type": "text",
-    "guardrails_metrics": ["correctness", "completeness"],
-    "score_format": "continuous",
-    "webhook": None
-}
-eval_response = client.evaluate.create(eval_request_data)
-print("Evaluation created:", eval_response)
-
-# 2) Fetch the evaluation by ID
-evaluation = client.evaluate.fetch(eval_response["eval_id"])
-print("Fetched evaluation:", evaluation)
-
-# 3) Create a new monitor
-monitor_data = {
-    "name": "My Example Monitor",
-    "description": "Track LLM usage in production",
-    "metrics": ["correctness", "completeness"]
-}
-monitor_resp = client.monitor.create(monitor_data)
-monitor_id = monitor_resp["monitor_id"]
-print("Monitor created:", monitor_resp)
-
-# 4) Log an event under that monitor
-event_data = {
-    "model_input": {"user_prompt": "Tell me a joke"},
-    "model_output": {"response": "Why did the chicken cross the road..."},
-    "temperature": 0.7,
-    "top_p": 1.0,
-    "model": "gpt-3.5-turbo"
-}
-event_resp = client.monitor.log(monitor_id, event_data)
-print("Monitor event logged:", event_resp)
-```
-
-
-

deeprails-0.1.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-deeprails/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-deeprails/client.py,sha256=28vSTSp2tWzFLJq4racWW4lXMXOyF3W9hEvROhZd5VU,983
-deeprails/evaluate.py,sha256=HJ3QtGiCho169r8dx1LXV6U7xjbreTr1X2BjDEfFwuw,1647
-deeprails/monitor.py,sha256=XYtl5TdNqRHtrUetSYHxAOJNisFE_uLmbFSa__q08kA,1743
-deeprails-0.1.0.dist-info/LICENSE,sha256=GsV7lN6fihCcDgkJbfs0rq1q9d6IyB0TFQ8HLKUpSXM,1077
-deeprails-0.1.0.dist-info/METADATA,sha256=kmaiXXtnBuZQhT_tV0vKJH1P3Z3RVU3Npk-dRkg4cj8,2578
-deeprails-0.1.0.dist-info/WHEEL,sha256=ewwEueio1C2XeHTvT17n8dZUJgOvyCWCt0WVNLClP9o,92
-deeprails-0.1.0.dist-info/top_level.txt,sha256=8-3gictPOXqg2k5S_yW66DdO8EqZ-YFab0LNarWUv4Q,10
-deeprails-0.1.0.dist-info/RECORD,,

deeprails-0.1.0.dist-info/top_level.txt

@@ -1 +0,0 @@
-deeprails