opperai 0.0.4__py2.py3-none-any.whl

opperai/__init__.py

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

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/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/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.dist-info/METADATA

@@ -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.dist-info/RECORD

@@ -0,0 +1,7 @@
+opperai/__init__.py,sha256=FRzbO1SjaKqzpMMeMLLNAXqzp9ljnwiGgHfhQCnGKS4,40
+opperai/client.py,sha256=4sRgGa_1woZEqeE1F3sF2m07podxsyYml6HyfhJunmQ,1474
+opperai/functions.py,sha256=rOQLwzGY3G8yZU5NTgmLyCgJ0xgYJYmdh8SxOF59ggo,1034
+opperai/types/__init__.py,sha256=KGTlAHgDRAzvLqSpHWFo_1OcDIMdRUoE8fLcoQ9iVys,603
+opperai-0.0.4.dist-info/METADATA,sha256=Fm0xxkVYZGnuHugYoZVkzSczqll2Lf9ARnpo_zlqFhE,1121
+opperai-0.0.4.dist-info/WHEEL,sha256=VYAwk8D_V6zmIA2XKK-k7Fem_KAtVk3hugaRru3yjGc,105
+opperai-0.0.4.dist-info/RECORD,,

opperai-0.0.4.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.21.0
+Root-Is-Purelib: true
+Tag: py2-none-any
+Tag: py3-none-any