omtx 0.1.0__tar.gz

omtx-0.1.0/LICENSE

@@ -0,0 +1,23 @@
+MIT License
+
+Copyright (c) 2025 OMTX
+
+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.
+
+

omtx-0.1.0/PKG-INFO

@@ -0,0 +1,199 @@
+Metadata-Version: 2.4
+Name: omtx
+Version: 0.1.0
+Summary: Python SDK for the OM Gateway V2
+Home-page: https://github.com/omtx/python-sdk
+Author: OM Therapeutics
+Author-email: OMTX <dev@omtx.ai>
+License: MIT License
+        
+        Copyright (c) 2025 OMTX
+        
+        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.
+        
+        
+        
+Project-URL: Homepage, https://github.com/omtx-ai/om-external
+Project-URL: Issues, https://github.com/omtx-ai/om-external/issues
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Typing :: Typed
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: requests>=2.31.0
+Requires-Dist: httpx>=0.27.0
+Dynamic: author
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: requires-python
+
+# OMTX Python SDK
+
+Simple Python client for the OM API Gateway. Generate molecular diligence reports and perform deep scientific research with just a few lines of code.
+
+## Installation
+
+```bash
+pip install omtx
+```
+
+## Quick Start
+
+```python
+from omtx import Client
+
+# Initialize client (uses OMTX_API_KEY environment variable)
+client = Client()
+
+# Generate a diligence report
+result = client.generate_diligence("BRAF")
+print(result.summary)
+
+# Perform deep research
+research = client.deep_research("CRISPR applications in cancer treatment")
+print(research.final_report)
+
+# Check available credits
+credits = client.credits()
+print(f"Available credits: {credits}")
+```
+
+## Setup
+
+### Get an API Key
+
+1. Sign up at [https://omtx.ai](https://omtx.ai)
+2. Navigate to your dashboard
+3. Generate an API key
+
+### Set Your API Key
+
+#### Option 1: Environment Variable (Recommended)
+
+```bash
+export OMTX_API_KEY="your-api-key-here"
+```
+
+#### Option 2: Pass to Client
+
+```python
+from omtx import Client
+
+client = Client(api_key="your-api-key-here")
+```
+
+## Examples
+
+### Generate Diligence Report
+
+```python
+from omtx import Client
+
+client = Client()
+
+# Generate report for a molecular target
+result = client.generate_diligence("KRAS")
+
+# Access the summary
+print(result.summary)
+
+# Access full report data
+print(result.report_data)
+```
+
+### Deep Research
+
+```python
+from omtx import Client
+
+client = Client()
+
+# Perform deep research on a scientific query
+research = client.deep_research(
+    "CAR-T therapy mechanisms and recent advances",
+    max_iterations=5  # Optional: more iterations = deeper research
+)
+
+# View the final report
+print(research.final_report)
+
+# Access individual findings
+for finding in research.findings:
+    print(f"- {finding['claim']}")
+
+# Check execution time
+print(f"Research completed in {research.execution_time:.1f} seconds")
+```
+
+### Error Handling
+
+```python
+from omtx import Client, OMTXError, InsufficientCreditsError
+
+client = Client()
+
+try:
+    result = client.generate_diligence("BRAF")
+except InsufficientCreditsError as e:
+    print(f"Not enough credits. Please add more credits to your account.")
+except OMTXError as e:
+    print(f"An error occurred: {e}")
+```
+
+### Context Manager
+
+```python
+from omtx import Client
+
+# Automatically cleans up resources
+with Client() as client:
+    result = client.generate_diligence("TP53")
+    print(result.summary)
+```
+
+## Available Methods
+
+- `generate_diligence(target)` - Generate a comprehensive diligence report for a molecular target
+- `deep_research(query, max_iterations=3)` - Perform iterative deep research on a scientific query
+- `credits()` - Check available credits
+
+## Features
+
+- **Simple API** - Just a few methods to remember
+- **Automatic Retries** - Handles transient network errors automatically
+- **Idempotency** - Safe to retry requests without double charges
+- **Type Hints** - Full IDE autocomplete support
+- **Error Handling** - Clear, actionable error messages
+
+## Requirements
+
+- Python 3.8 or higher
+- An OMTX API key
+
+## Support
+
+- Email: support@omtx.ai
+- Documentation: https://docs.omtx.ai
+- Issues: https://github.com/omtx/python-sdk/issues
+
+## License
+
+MIT License - see LICENSE file for details.

omtx-0.1.0/README.md

@@ -0,0 +1,152 @@
+# OMTX Python SDK
+
+Simple Python client for the OM API Gateway. Generate molecular diligence reports and perform deep scientific research with just a few lines of code.
+
+## Installation
+
+```bash
+pip install omtx
+```
+
+## Quick Start
+
+```python
+from omtx import Client
+
+# Initialize client (uses OMTX_API_KEY environment variable)
+client = Client()
+
+# Generate a diligence report
+result = client.generate_diligence("BRAF")
+print(result.summary)
+
+# Perform deep research
+research = client.deep_research("CRISPR applications in cancer treatment")
+print(research.final_report)
+
+# Check available credits
+credits = client.credits()
+print(f"Available credits: {credits}")
+```
+
+## Setup
+
+### Get an API Key
+
+1. Sign up at [https://omtx.ai](https://omtx.ai)
+2. Navigate to your dashboard
+3. Generate an API key
+
+### Set Your API Key
+
+#### Option 1: Environment Variable (Recommended)
+
+```bash
+export OMTX_API_KEY="your-api-key-here"
+```
+
+#### Option 2: Pass to Client
+
+```python
+from omtx import Client
+
+client = Client(api_key="your-api-key-here")
+```
+
+## Examples
+
+### Generate Diligence Report
+
+```python
+from omtx import Client
+
+client = Client()
+
+# Generate report for a molecular target
+result = client.generate_diligence("KRAS")
+
+# Access the summary
+print(result.summary)
+
+# Access full report data
+print(result.report_data)
+```
+
+### Deep Research
+
+```python
+from omtx import Client
+
+client = Client()
+
+# Perform deep research on a scientific query
+research = client.deep_research(
+    "CAR-T therapy mechanisms and recent advances",
+    max_iterations=5  # Optional: more iterations = deeper research
+)
+
+# View the final report
+print(research.final_report)
+
+# Access individual findings
+for finding in research.findings:
+    print(f"- {finding['claim']}")
+
+# Check execution time
+print(f"Research completed in {research.execution_time:.1f} seconds")
+```
+
+### Error Handling
+
+```python
+from omtx import Client, OMTXError, InsufficientCreditsError
+
+client = Client()
+
+try:
+    result = client.generate_diligence("BRAF")
+except InsufficientCreditsError as e:
+    print(f"Not enough credits. Please add more credits to your account.")
+except OMTXError as e:
+    print(f"An error occurred: {e}")
+```
+
+### Context Manager
+
+```python
+from omtx import Client
+
+# Automatically cleans up resources
+with Client() as client:
+    result = client.generate_diligence("TP53")
+    print(result.summary)
+```
+
+## Available Methods
+
+- `generate_diligence(target)` - Generate a comprehensive diligence report for a molecular target
+- `deep_research(query, max_iterations=3)` - Perform iterative deep research on a scientific query
+- `credits()` - Check available credits
+
+## Features
+
+- **Simple API** - Just a few methods to remember
+- **Automatic Retries** - Handles transient network errors automatically
+- **Idempotency** - Safe to retry requests without double charges
+- **Type Hints** - Full IDE autocomplete support
+- **Error Handling** - Clear, actionable error messages
+
+## Requirements
+
+- Python 3.8 or higher
+- An OMTX API key
+
+## Support
+
+- Email: support@omtx.ai
+- Documentation: https://docs.omtx.ai
+- Issues: https://github.com/omtx/python-sdk/issues
+
+## License
+
+MIT License - see LICENSE file for details.

omtx-0.1.0/omtx/__init__.py

@@ -0,0 +1,18 @@
+"""OMTX Python SDK - Clients for OM Gateway V2.
+
+Includes:
+- OMTXClient: Simple, explicit client with idempotency and typed methods
+- Client: Dynamic endpoint registry alternative (advanced)
+"""
+
+from .client import OMTXClient, Client
+from .exceptions import OMTXError, InsufficientCreditsError, AuthenticationError
+
+__version__ = "0.1.0"
+__all__ = [
+    "OMTXClient",
+    "Client",
+    "OMTXError",
+    "InsufficientCreditsError",
+    "AuthenticationError",
+]

omtx-0.1.0/omtx/_internal/__init__.py

@@ -0,0 +1 @@
+# Internal implementation details - not part of public API

omtx-0.1.0/omtx/_internal/http.py

@@ -0,0 +1,91 @@
+"""Internal HTTP client with automatic retry and idempotency"""
+import os
+import time
+import uuid
+from typing import Dict, Any, Optional
+import httpx
+from ..exceptions import OMTXError, AuthenticationError, InsufficientCreditsError, APIError
+
+
+class HTTPClient:
+    """Handles all HTTP communication with the OM API Gateway"""
+    
+    def __init__(self, api_key: str = None, base_url: str = None):
+        self.api_key = api_key or os.environ.get("OMTX_API_KEY")
+        if not self.api_key:
+            raise AuthenticationError(
+                "API key required. Pass api_key to Client() or set OMTX_API_KEY environment variable"
+            )
+        
+        self.base_url = base_url or os.environ.get("OMTX_API_URL", "https://api.omtx.ai")
+        self.client = httpx.Client(timeout=60.0)
+    
+    def request(self, method: str, path: str, json: Dict[str, Any] = None) -> Dict[str, Any]:
+        """Make HTTP request with automatic retry and idempotency"""
+        # Generate idempotency key for non-GET requests
+        headers = {"X-API-Key": self.api_key}
+        if method != "GET":
+            headers["Idempotency-Key"] = str(uuid.uuid4())
+        
+        url = f"{self.base_url}{path}"
+        
+        # Retry logic with exponential backoff
+        last_error = None
+        for attempt in range(3):
+            try:
+                response = self.client.request(
+                    method=method,
+                    url=url,
+                    headers=headers,
+                    json=json
+                )
+                
+                # Handle successful responses
+                if response.status_code == 200:
+                    return response.json()
+                
+                # Handle specific error codes
+                if response.status_code == 401:
+                    raise AuthenticationError("Invalid API key")
+                elif response.status_code == 402:
+                    raise InsufficientCreditsError("Insufficient credits")
+                elif response.status_code == 404:
+                    raise OMTXError(f"Endpoint not found: {path}")
+                elif response.status_code >= 500:
+                    # Server error - retry
+                    last_error = APIError(
+                        f"Server error: {response.text}", 
+                        status_code=response.status_code
+                    )
+                    if attempt < 2:  # Don't sleep on last attempt
+                        time.sleep(2 ** attempt)  # Exponential backoff
+                    continue
+                else:
+                    # Other client errors
+                    try:
+                        error_detail = response.json().get("detail", response.text)
+                    except:
+                        error_detail = response.text
+                    raise APIError(error_detail, status_code=response.status_code)
+                    
+            except httpx.NetworkError as e:
+                # Network error - retry
+                last_error = OMTXError(f"Network error: {str(e)}")
+                if attempt < 2:
+                    time.sleep(2 ** attempt)
+                continue
+        
+        # All retries failed
+        raise last_error or OMTXError("Request failed after 3 attempts")
+    
+    def get(self, path: str) -> Dict[str, Any]:
+        """GET request"""
+        return self.request("GET", path)
+    
+    def post(self, path: str, json: Dict[str, Any]) -> Dict[str, Any]:
+        """POST request"""
+        return self.request("POST", path, json)
+    
+    def close(self):
+        """Close the HTTP client"""
+        self.client.close()