mira-network 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

mira_network-0.1.1.dist-info/METADATA

@@ -0,0 +1,107 @@
+Metadata-Version: 2.1
+Name: mira-network
+Version: 0.1.1
+Summary: Python SDK for Mira Network API
+Author-Email: sarim2000 <sarimbleedblue@gmail.com>
+License: MIT
+Requires-Python: ==3.10.*
+Requires-Dist: httpx>=0.28.1
+Requires-Dist: pydantic>=2.10.4
+Requires-Dist: typing-extensions>=4.8.0
+Requires-Dist: requests>=2.32.3
+Description-Content-Type: text/markdown
+
+# Mira SDK
+
+A Python SDK for interacting with the Mira Network API. This SDK provides a simple interface to access all Mira API endpoints including model inference, flow management, and credit system.
+
+## Installation
+
+```bash
+pip install mira-sdk
+```
+
+## Quick Start
+
+```python
+import asyncio
+from mira_sdk import MiraClient, Message, AiRequest
+
+async def main():
+    # Initialize client
+    client = MiraClient(
+        base_url="https://api.mira.example.com",
+        api_token="your-api-token"
+    )
+    
+    # List available models
+    models = await client.list_models()
+    print("Available models:", models)
+    
+    # Generate text
+    request = AiRequest(
+        model="mira/llama3.1",
+        messages=[
+            Message(role="system", content="You are a helpful assistant."),
+            Message(role="user", content="Hello!")
+        ],
+        model_provider=None
+    )
+    
+    response = await client.generate(request)
+    print("Response:", response)
+
+if __name__ == "__main__":
+    asyncio.run(main())
+```
+
+## Features
+
+- Asynchronous API using `httpx`
+- Full type hints support
+- Pydantic models for request/response validation
+- Support for all Mira API endpoints:
+  - Model inference
+  - Flow management
+  - API token management
+  - Credit system
+
+## API Reference
+
+### Models
+
+- `Message`: Represents a chat message
+- `ModelProvider`: Configuration for custom model providers
+- `AiRequest`: Request for model inference
+- `FlowChatCompletion`: Request for flow-based chat completion
+- `FlowRequest`: Request for creating/updating flows
+- `ApiTokenRequest`: Request for creating API tokens
+- `AddCreditRequest`: Request for adding credits
+
+### Client Methods
+
+#### Model Operations
+- `list_models()`: List available models
+- `generate(request: AiRequest)`: Generate text using specified model
+
+#### Flow Operations
+- `list_flows()`: List all flows
+- `get_flow(flow_id: str)`: Get flow details
+- `create_flow(request: FlowRequest)`: Create new flow
+- `update_flow(flow_id: str, request: FlowRequest)`: Update flow
+- `delete_flow(flow_id: str)`: Delete flow
+- `generate_with_flow(flow_id: str, request: FlowChatCompletion)`: Generate using flow
+
+#### Token Operations
+- `create_api_token(request: ApiTokenRequest)`: Create API token
+- `list_api_tokens()`: List API tokens
+- `delete_api_token(token: str)`: Delete API token
+
+#### Credit Operations
+- `get_user_credits()`: Get credit information
+- `add_credit(request: AddCreditRequest)`: Add credits
+- `get_credits_history()`: Get credit history
+
+## License
+
+MIT License

mira_network-0.1.1.dist-info/RECORD

@@ -0,0 +1,7 @@
+mira_network-0.1.1.dist-info/METADATA,sha256=qN1Q6pwqsmsGOab06qp0qGl-nv0Hlpw1c4eoOSH7ZKY,2903
+mira_network-0.1.1.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+mira_network-0.1.1.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+mira_sdk/__init__.py,sha256=KVFj7e20pprJuOWCG9Ky_w2JqKBu8TFPCilPDKRhkyI,364
+mira_sdk/client.py,sha256=Qf8zSQRDmIhR2N9SAXq_ddz-z2iwu07S_w-UoLOMI0I,5773
+mira_sdk/models.py,sha256=FQfxjK_Cs7Nt0dB1qYHd7rov3VJSVr6rNAqZqOarU48,1749
+mira_network-0.1.1.dist-info/RECORD,,

{mira_network-0.1.0.dist-info → mira_network-0.1.1.dist-info}/WHEEL

@@ -1,5 +1,4 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.6.0)
+Generator: pdm-backend (2.4.3)
 Root-Is-Purelib: true
 Tag: py3-none-any
-

mira_network-0.1.1.dist-info/entry_points.txt

@@ -0,0 +1,4 @@
+[console_scripts]
+
+[gui_scripts]
+

mira_sdk/__init__.py

@@ -0,0 +1,21 @@
+from .client import MiraClient
+from .models import (
+    Message,
+    ModelProvider,
+    AiRequest,
+    FlowChatCompletion,
+    FlowRequest,
+    ApiTokenRequest,
+    AddCreditRequest,
+)
+
+__all__ = [
+    "MiraClient",
+    "Message",
+    "ModelProvider",
+    "AiRequest",
+    "FlowChatCompletion",
+    "FlowRequest",
+    "ApiTokenRequest",
+    "AddCreditRequest",
+]

mira_sdk/client.py

@@ -0,0 +1,180 @@
+from typing import Optional, List, Dict, AsyncGenerator, Union
+import httpx
+from src.mira_sdk.models import (
+    Message,
+    ModelProvider,
+    AiRequest,
+    FlowChatCompletion,
+    FlowRequest,
+    ApiTokenRequest,
+    AddCreditRequest,
+)
+
+
+class MiraClient:
+    def __init__(self, base_url: str, api_token: Optional[str] = None):
+        """Initialize Mira client.
+
+        Args:
+            base_url: Base URL of the Mira API
+            api_token: Optional API token for authentication
+        """
+        self.base_url = base_url
+        self.api_token = api_token
+        self._client = httpx.AsyncClient()
+
+    async def __aenter__(self):
+        return self
+
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        await self._client.aclose()
+
+    def _get_headers(self) -> Dict[str, str]:
+        headers = {"Content-Type": "application/json"}
+        if self.api_token:
+            headers["Authorization"] = f"Bearer {self.api_token}"
+        return headers
+
+    async def list_models(self) -> List[str]:
+        """List available models."""
+        response = await self._client.get(
+            f"{self.base_url}/v1/models",
+            headers=self._get_headers(),
+        )
+        response.raise_for_status()
+        return response.json()
+
+    async def generate(
+        self, request: AiRequest
+    ) -> Union[str, AsyncGenerator[str, None]]:
+        """Generate text using the specified model."""
+        response = await self._client.post(
+            f"{self.base_url}/v1/chat/completions",
+            headers=self._get_headers(),
+            json=request.model_dump(),
+        )
+
+        response.raise_for_status()
+
+        if request.stream:
+
+            async def stream_response():
+                async for chunk in response.aiter_text():
+                    yield chunk
+
+            return stream_response()
+        else:
+            return response.json()
+
+    async def generate_with_flow(
+        self, flow_id: str, request: FlowChatCompletion
+    ) -> Union[str, AsyncGenerator[str, None]]:
+        """Generate text using a specific flow."""
+        response = await self._client.post(
+            f"{self.base_url}/v1/flows/{flow_id}/chat/completions",
+            headers=self._get_headers(),
+            json=request.model_dump(),
+        )
+        response.raise_for_status()
+        return response.json()
+
+    async def list_flows(self) -> List[Dict]:
+        """List all flows."""
+        response = await self._client.get(
+            f"{self.base_url}/flows",
+            headers=self._get_headers(),
+        )
+        response.raise_for_status()
+        return response.json()
+
+    async def get_flow(self, flow_id: str) -> Dict:
+        """Get details of a specific flow."""
+        response = await self._client.get(
+            f"{self.base_url}/flows/{flow_id}",
+            headers=self._get_headers(),
+        )
+        response.raise_for_status()
+        return response.json()
+
+    async def create_flow(self, request: FlowRequest) -> Dict:
+        """Create a new flow."""
+        response = await self._client.post(
+            f"{self.base_url}/flows",
+            headers=self._get_headers(),
+            json=request.model_dump(),
+        )
+        response.raise_for_status()
+        return response.json()
+
+    async def update_flow(self, flow_id: str, request: FlowRequest) -> Dict:
+        """Update an existing flow."""
+        response = await self._client.put(
+            f"{self.base_url}/flows/{flow_id}",
+            headers=self._get_headers(),
+            json=request.model_dump(),
+        )
+        response.raise_for_status()
+        return response.json()
+
+    async def delete_flow(self, flow_id: str) -> None:
+        """Delete a flow."""
+        response = await self._client.delete(
+            f"{self.base_url}/flows/{flow_id}",
+            headers=self._get_headers(),
+        )
+        response.raise_for_status()
+
+    async def create_api_token(self, request: ApiTokenRequest) -> Dict:
+        """Create a new API token."""
+        response = await self._client.post(
+            f"{self.base_url}/tokens",
+            headers=self._get_headers(),
+            json=request.model_dump(),
+        )
+        response.raise_for_status()
+        return response.json()
+
+    async def list_api_tokens(self) -> List[Dict]:
+        """List all API tokens."""
+        response = await self._client.get(
+            f"{self.base_url}/tokens",
+            headers=self._get_headers(),
+        )
+        response.raise_for_status()
+        return response.json()
+
+    async def delete_api_token(self, token: str) -> None:
+        """Delete an API token."""
+        response = await self._client.delete(
+            f"{self.base_url}/tokens/{token}",
+            headers=self._get_headers(),
+        )
+        response.raise_for_status()
+
+    async def get_user_credits(self) -> Dict:
+        """Get user credits information."""
+        response = await self._client.get(
+            f"{self.base_url}/credits",
+            headers=self._get_headers(),
+        )
+        response.raise_for_status()
+        return response.json()
+
+    async def add_credit(self, request: AddCreditRequest) -> Dict:
+        """Add credits to a user account."""
+        response = await self._client.post(
+            f"{self.base_url}/credits",
+            headers=self._get_headers(),
+            json=request.model_dump(),
+        )
+        response.raise_for_status()
+        return response.json()
+
+    async def get_credits_history(self) -> List[Dict]:
+        """Get user credits history."""
+        response = await self._client.get(
+            f"{self.base_url}/credits/history",
+            headers=self._get_headers(),
+        )
+        response.raise_for_status()
+        return response.json()

mira_sdk/models.py

@@ -0,0 +1,67 @@
+from typing import Optional, List, Dict
+from pydantic import BaseModel, Field, field_validator
+
+
+class Message(BaseModel):
+    role: str
+    content: str
+
+    @field_validator("role")
+    @classmethod
+    def validate_role(cls, v: str) -> str:
+        valid_roles = ["system", "user", "assistant"]
+        if v not in valid_roles:
+            raise ValueError(f"Invalid role. Must be one of: {valid_roles}")
+        return v
+
+
+class ModelProvider(BaseModel):
+    base_url: str
+    api_key: str
+
+
+class AiRequest(BaseModel):
+    model: str = Field("mira/llama3.1", title="Model")
+    model_provider: Optional[ModelProvider] = Field(None, title="Model Provider (optional)")
+    messages: List[Message] = Field([], title="Messages")
+    stream: Optional[bool] = Field(False, title="Stream")
+
+    @field_validator("messages")
+    @classmethod
+    def validate_messages(cls, v: List[Message]) -> List[Message]:
+        if not v:
+            raise ValueError("Messages list cannot be empty")
+        return v
+
+
+class FlowChatCompletion(BaseModel):
+    variables: Optional[Dict] = Field(None, title="Variables")
+
+
+class FlowRequest(BaseModel):
+    system_prompt: str
+    name: str
+
+
+class ApiTokenRequest(BaseModel):
+    description: Optional[str] = None
+
+
+class AddCreditRequest(BaseModel):
+    user_id: str
+    amount: float
+    description: Optional[str] = None
+
+    @field_validator("amount")
+    @classmethod
+    def validate_amount(cls, v: float) -> float:
+        if v <= 0:
+            raise ValueError("Amount must be greater than 0")
+        return v
+
+    @field_validator("user_id")
+    @classmethod
+    def validate_user_id(cls, v: str) -> str:
+        if not v.strip():
+            raise ValueError("User ID cannot be empty")
+        return v

mira_network/client.py

@@ -1,8 +0,0 @@
-class MiraNetwork:
-    def __init__(self, api_key: str) -> None:
-        self.api_key = api_key
-
-    def generate(self) -> dict:
-        # check if the api key is valid
-        if not self.api_key:
-            raise ValueError("API key is required")

mira_network-0.1.0.dist-info/METADATA

@@ -1,10 +0,0 @@
-Metadata-Version: 2.1
-Name: mira-network
-Version: 0.1.0
-Summary: Default template for PDM package
-Author-email: Hamid Raza <contact@hamidraza.com>
-License: MIT
-Requires-Python: >=3.11
-Description-Content-Type: text/markdown
-
-# mira-network

mira_network-0.1.0.dist-info/RECORD

@@ -1,6 +0,0 @@
-mira_network/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mira_network/client.py,sha256=-GSldxzoJrEhLDvF6gd80U6SiEYuUtSnIhJQ6EpeD8w,250
-mira_network-0.1.0.dist-info/METADATA,sha256=OIrrTjqa2NXGkodyvHrg7YNxFRSs1IOHQzQ2EdZ8mJs,240
-mira_network-0.1.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-mira_network-0.1.0.dist-info/top_level.txt,sha256=lqyBtm4UdvqoeCYxm2P-ygcfmHhHurxg_Ue7dTChKEo,13
-mira_network-0.1.0.dist-info/RECORD,,

mira_network-0.1.0.dist-info/top_level.txt

@@ -1 +0,0 @@
-mira_network