opperai 0.0.4__tar.gz

opperai-0.0.4/.github/workflows/publish.yml

@@ -0,0 +1,135 @@
+name: Build & Publish
+
+on:
+  push:
+    branches:
+      - main
+    tags:
+      - v*
+  workflow_dispatch:
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Code checkout
+        uses: actions/checkout@v4
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: '3.11'
+          cache: pip
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install .[test]
+      - name: Run lint
+        run: |
+          pip install ruff==0.1.11
+          ruff format --check .
+      - name: Run tests
+        run: |
+          python -m pytest
+      - name: Install pypa/build
+        run: >-
+          python3 -m
+          pip install
+          build
+          --user
+      - name: Build a binary wheel and a source tarball
+        run: python3 -m build
+      - name: Store the distribution packages
+        uses: actions/upload-artifact@v3
+        with:
+          name: python-package-distributions
+          path: dist/
+    
+  publish-test:
+    name: Publish SDK to TestPyPI
+    runs-on: ubuntu-latest
+    needs:
+      - build
+    environment:
+      name: testpypi
+      url: https://test.pypi.org/p/opperai
+    permissions:
+      id-token: write  
+    steps:
+      - name: Download all the dists
+        uses: actions/download-artifact@v3
+        with:
+          name: python-package-distributions
+          path: dist/
+
+      - name: Publish package distributions to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          repository-url: https://test.pypi.org/legacy/
+          password: ${{ secrets.TEST_PYPI_API_TOKEN }}
+          skip-existing: true
+
+  publish-to-pypi:
+    name: >-
+      Publish Python SDK to PyPI
+    if: startsWith(github.ref, 'refs/tags/')  
+    needs:
+    - build
+    runs-on: ubuntu-latest
+    environment:
+      name: pypi
+      url: https://pypi.org/p/opperai
+    permissions:
+      id-token: write  
+
+    steps:
+    - name: Download all the dists
+      uses: actions/download-artifact@v3
+      with:
+        name: python-package-distributions
+        path: dist/
+    - name: Publish distribution 📦 to PyPI
+      uses: pypa/gh-action-pypi-publish@release/v1
+      
+
+  github-release:
+    name: >-
+      GitHub Release
+    needs:
+    - publish-to-pypi
+    runs-on: ubuntu-latest
+
+    permissions:
+      contents: write  
+      id-token: write  
+
+    steps:
+    - name: Download all the dists
+      uses: actions/download-artifact@v3
+      with:
+        name: python-package-distributions
+        path: dist/
+    - name: Sign the dists with Sigstore
+      uses: sigstore/gh-action-sigstore-python@v1.2.3
+      with:
+        inputs: >-
+          ./dist/*.tar.gz
+          ./dist/*.whl
+    - name: Create GitHub Release
+      env:
+        GITHUB_TOKEN: ${{ github.token }}
+      run: >-
+        gh release create
+        '${{ github.ref_name }}'
+        --repo '${{ github.repository }}'
+        --notes ""
+    - name: Upload artifact signatures to GitHub Release
+      env:
+        GITHUB_TOKEN: ${{ github.token }}
+      # Upload to GitHub Release using the `gh` CLI.
+      # `dist/` contains the built packages, and the
+      # sigstore-produced signatures and certificates.
+      run: >-
+        gh release upload
+        '${{ github.ref_name }}' dist/**
+        --repo '${{ github.repository }}'
+

opperai-0.0.4/.gitignore

@@ -0,0 +1,13 @@
+*.pyc
+__pycache__/
+*.pyo
+*.pyd
+.Python
+env/
+venv/
+ENV/
+env.bak/
+venv.bak
+.vscode
+dist
+*.egg-info

opperai-0.0.4/PKG-INFO

@@ -0,0 +1,45 @@
+Metadata-Version: 2.1
+Name: opperai
+Version: 0.0.4
+Summary: Opper Python client
+Project-URL: Homepage, https://opper.ai
+Project-URL: Documentation, https://docs.opper.ai
+Project-URL: Platform, https://platform.opper.ai
+Author-email: Opper <opper@opper.ai>
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Requires-Dist: httpx
+Requires-Dist: pydantic
+Provides-Extra: test
+Requires-Dist: pytest; extra == 'test'
+Requires-Dist: pytest-asyncio; extra == 'test'
+Description-Content-Type: text/markdown
+
+# [Opper](https//opper.ai) Python SDK
+
+## Install
+
+```bash
+pip install opperai
+```
+
+## Functions
+
+To call a function you created at [https://platform.opper.ai](https://platform.opper.ai) you can use the following code:
+
+
+```python
+from opper import Client
+from opper.types import ChatPayload, Message
+
+# Use AsyncClient for async operations
+client = Client(api_key="your-api-key") 
+response = client.functions.chat("your-function-path", 
+ ChatPayload(messages=[Message(role="user", content="hello")])
+)
+
+print(response)
+
+```
+

opperai-0.0.4/README.md

@@ -0,0 +1,27 @@
+# [Opper](https//opper.ai) Python SDK
+
+## Install
+
+```bash
+pip install opperai
+```
+
+## Functions
+
+To call a function you created at [https://platform.opper.ai](https://platform.opper.ai) you can use the following code:
+
+
+```python
+from opper import Client
+from opper.types import ChatPayload, Message
+
+# Use AsyncClient for async operations
+client = Client(api_key="your-api-key") 
+response = client.functions.chat("your-function-path", 
+ ChatPayload(messages=[Message(role="user", content="hello")])
+)
+
+print(response)
+
+```
+

opperai-0.0.4/pyproject.toml

@@ -0,0 +1,39 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "opperai"
+version = "0.0.4"
+description = "Opper Python client"
+authors = [
+  {name = "Opper", email = "opper@opper.ai"},
+]
+
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+]
+dependencies = [
+    "pydantic",
+    "httpx"
+]
+readme = "README.md"
+
+[project.urls]
+Homepage = "https://opper.ai"
+Documentation = "https://docs.opper.ai"
+Platform = "https://platform.opper.ai"
+
+[project.optional-dependencies]
+test = [
+    "pytest",
+    "pytest-asyncio"
+]
+
+
+[tool.pytest.ini_options]
+pythonpath = [
+  "src"
+]

opperai-0.0.4/pytest.ini

@@ -0,0 +1,2 @@
+[pytest]
+pythonpath = src

opperai-0.0.4/src/opperai/__init__.py

@@ -0,0 +1 @@
+from .client import AsyncClient, Client

opperai-0.0.4/src/opperai/client.py

@@ -0,0 +1,42 @@
+import httpx
+from .functions import Functions, AsyncFunctions
+
+DEFAULT_API_URL = "https://api.opper.ai"
+
+
+class AsyncClient:
+    def __init__(self, api_key: str, api_url: str = DEFAULT_API_URL):
+        self.http_client = _async_http_client(api_key, api_url)
+        self.functions = AsyncFunctions(self.http_client)
+
+
+class Client:
+    def __init__(self, api_key: str, api_url: str = DEFAULT_API_URL):
+        self.http_client = _http_client(api_key, api_url)
+        self.functions = Functions(self.http_client)
+
+
+class _async_http_client:
+    def __init__(self, api_key: str, api_url):
+        self.session = httpx.AsyncClient(
+            base_url=api_url, headers={"X-OPPER-API-KEY": f"{api_key}"}
+        )
+
+    async def do_request(self, method: str, path: str, **kwargs):
+        response = await self.session.request(method, path, **kwargs)
+        if response.status_code != 200:
+            raise Exception(f"Request failed with status {response.status_code}")
+        return response.json()
+
+
+class _http_client:
+    def __init__(self, api_key: str, api_url: str):
+        self.session = httpx.Client(
+            base_url=api_url, headers={"X-OPPER-API-KEY": f"{api_key}"}
+        )
+
+    def do_request(self, method: str, path: str, **kwargs):
+        response = self.session.request(method, path, **kwargs)
+        if response.status_code != 200:
+            raise Exception(f"Request failed with status {response.status_code}")
+        return response.json()

opperai-0.0.4/src/opperai/functions.py

@@ -0,0 +1,35 @@
+from .types import ChatPayload, FunctionResponse
+
+
+class Functions:
+    def __init__(self, http_client):
+        self.http_client = http_client
+
+    def chat(self, function_path, data: ChatPayload, stream=False) -> FunctionResponse:
+        serialized_data = data.model_dump()
+
+        data = self.http_client.do_request(
+            "POST",
+            f"/v1/chat/{function_path}",
+            json=serialized_data,
+            params={"stream": "true"} if stream else None,
+        )
+        return FunctionResponse(**data)
+
+
+class AsyncFunctions:
+    def __init__(self, http_client):
+        self.http_client = http_client
+
+    async def chat(
+        self, function_path, data: ChatPayload, stream=False
+    ) -> FunctionResponse:
+        serialized_data = data.model_dump()
+
+        data = await self.http_client.do_request(
+            "POST",
+            f"/v1/chat/{function_path}",
+            json=serialized_data,
+            params={"stream": "true"} if stream else None,
+        )
+        return FunctionResponse(**data)

opperai-0.0.4/src/opperai/types/__init__.py

@@ -0,0 +1,34 @@
+from pydantic import BaseModel, Field
+from typing import List, Optional
+
+
+class Message(BaseModel):
+    role: str
+    content: str
+
+
+class ChatPayload(BaseModel):
+    messages: List[Message]
+
+
+class ContextData(BaseModel):
+    embeddings_id: str
+    embeddings_table: str
+    dataset_id: int
+    content: str
+    score: Optional[float]
+
+
+class DebugData(BaseModel):
+    context: List[ContextData]
+
+
+class StreamingChunk(BaseModel):
+    delta: str
+    error: Optional[str] = None
+
+
+class FunctionResponse(BaseModel):
+    message: str
+    error: Optional[str] = None
+    debug: Optional[DebugData] = None

opperai-0.0.4/tests/test_functions.py

@@ -0,0 +1,32 @@
+import pytest
+from opperai import AsyncClient, Client
+from opperai.types import ChatPayload, Message
+
+from unittest.mock import patch
+
+api_key = "key"
+
+
+@patch("opperai.client._http_client.do_request")
+def test_chat(mock_do_request):
+    mock_do_request.return_value = {"message": "Bonjour"}
+    client = Client(api_key=api_key)
+    resp = client.functions.chat(
+        "french", ChatPayload(messages=[Message(role="user", content="hello")])
+    )
+
+    assert resp.message == "Bonjour"
+    mock_do_request.assert_called_once()
+
+
+@pytest.mark.asyncio
+@patch("opperai.client._async_http_client.do_request")
+async def test_async_chat(mock_do_request):
+    mock_do_request.return_value = {"message": "Bonjour"}
+    client = AsyncClient(api_key="op-dev-api-key", api_url="http://localhost:8000")
+    resp = await client.functions.chat(
+        "french", ChatPayload(messages=[Message(role="user", content="hello")])
+    )
+
+    assert resp.message == "Bonjour"
+    mock_do_request.assert_called_once()