mira-network 0.1.2__tar.gz → 0.1.3__tar.gz

{mira_network-0.1.2 → mira_network-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mira-network
-Version: 0.1.2
+Version: 0.1.3
 Summary: Python SDK for Mira Network API
 Author-Email: sarim2000 <sarimbleedblue@gmail.com>
 License: MIT
@@ -9,6 +9,7 @@ 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
+Requires-Dist: pytest-cov>=6.0.0
 Description-Content-Type: text/markdown
 
 # Mira Network SDK
@@ -40,7 +41,7 @@ async def main():
     
     # Generate text
     request = AiRequest(
-        model="mira/llama3.1",
+        model="gpt-4o",
         messages=[
             Message(role="system", content="You are a helpful assistant."),
             Message(role="user", content="Hello!")

{mira_network-0.1.2 → mira_network-0.1.3}/README.md

@@ -27,7 +27,7 @@ async def main():
     
     # Generate text
     request = AiRequest(
-        model="mira/llama3.1",
+        model="gpt-4o",
         messages=[
             Message(role="system", content="You are a helpful assistant."),
             Message(role="user", content="Hello!")

{mira_network-0.1.2 → mira_network-0.1.3}/pyproject.toml

@@ -1,7 +1,4 @@
 [project]
-name = "mira-network"
-version = "0.1.2"
-description = "Python SDK for Mira Network API"
 authors = [
     { name = "sarim2000", email = "sarimbleedblue@gmail.com" },
 ]
@@ -10,18 +7,22 @@ dependencies = [
     "pydantic>=2.10.4",
     "typing-extensions>=4.8.0",
     "requests>=2.32.3",
+    "pytest-cov>=6.0.0",
 ]
-requires-python = "==3.10.*"
+description = "Python SDK for Mira Network API"
+name = "mira-network"
 readme = "README.md"
+requires-python = "==3.10.*"
+version = "0.1.3"
 
 [project.license]
 text = "MIT"
 
 [build-system]
+build-backend = "pdm.backend"
 requires = [
     "pdm-backend",
 ]
-build-backend = "pdm.backend"
 
 [tool.pdm]
 distribution = true

{mira_network-0.1.2 → mira_network-0.1.3}/src/mira_sdk/client.py

@@ -1,6 +1,6 @@
-from typing import Optional, List, Dict, AsyncGenerator, Union
+from typing import AsyncIterator, Optional, List, Dict, AsyncGenerator, Union
 import httpx
-from mira_sdk.models import (
+from .models import (
     Message,
     ModelProvider,
     AiRequest,
@@ -12,7 +12,12 @@ from mira_sdk.models import (
 
 
 class MiraClient:
-    def __init__(self, base_url: str, api_token: Optional[str] = None):
+
+    def __init__(
+        self,
+        base_url: str = "https://mira-network.alts.dev/",
+        api_token: Optional[str] = None,
+    ):
         """Initialize Mira client.
 
         Args:
@@ -44,9 +49,7 @@ class MiraClient:
         response.raise_for_status()
         return response.json()
 
-    async def generate(
-        self, request: AiRequest
-    ) -> Union[str, AsyncGenerator[str, None]]:
+    async def generate(self, request: AiRequest) -> Union[str, AsyncIterator[str]]:
         """Generate text using the specified model."""
         response = await self._client.post(
             f"{self.base_url}/v1/chat/completions",
@@ -57,12 +60,7 @@ class MiraClient:
         response.raise_for_status()
 
         if request.stream:
-
-            async def stream_response():
-                async for chunk in response.aiter_text():
-                    yield chunk
-
-            return stream_response()
+            return response.aiter_lines()
         else:
             return response.json()
 
@@ -127,7 +125,7 @@ class MiraClient:
     async def create_api_token(self, request: ApiTokenRequest) -> Dict:
         """Create a new API token."""
         response = await self._client.post(
-            f"{self.base_url}/tokens",
+            f"{self.base_url}/api-tokens",
             headers=self._get_headers(),
             json=request.model_dump(),
         )
@@ -154,7 +152,7 @@ class MiraClient:
     async def get_user_credits(self) -> Dict:
         """Get user credits information."""
         response = await self._client.get(
-            f"{self.base_url}/credits",
+            f"{self.base_url}/user-credits",
             headers=self._get_headers(),
         )
         response.raise_for_status()

{mira_network-0.1.2 → mira_network-0.1.3}/tests/test_client.py

@@ -13,8 +13,8 @@ from src.mira_sdk.models import (
 @pytest.fixture
 def client():
     return MiraClient(
-        base_url="https://mira-client-balancer.alts.dev",
-        api_token="sk-mira-9eebf90946b4ff710580cafd1c651517e3decd07bd810402",
+        base_url="https://mira-network.alts.dev",
+        api_token="sk-mira-b9ecd5f43ef0363e691322df3295c2b98bebd1c1edb0b6d8",
     )
 
 
@@ -22,13 +22,11 @@ def client():
 async def test_list_models(client):
     result = await client.list_models()
     assert isinstance(result, dict)
-    assert result.get("object") == "list"
-    assert isinstance(result.get("data"), list)
-    for model in result.get("data", []):
-        assert isinstance(model, dict)
-        assert "id" in model
-        assert "object" in model
-        assert model["object"] == "model"
+    assert result["object"] == "list"
+    assert isinstance(result["data"], list)
+    assert len(result["data"]) > 0
+    assert all(isinstance(model, dict) for model in result["data"])
+    assert all("id" in model and "object" in model for model in result["data"])
 
 
 @pytest.mark.asyncio
@@ -48,16 +46,21 @@ async def test_generate(client):
 @pytest.mark.asyncio
 async def test_generate_stream(client):
     request = AiRequest(
-        model="mira/llama3.1",
+        model="gpt-4o",
         messages=[Message(role="user", content="Hi!")],
         stream=True,
         model_provider=None,
     )
-
-    stream = await client.generate(request)
+    print("Making generate request with streaming...")
+    response = await client.generate(request=request)
     chunks = []
-    async for chunk in stream:
+    print("Starting to receive stream chunks...")
+    async for chunk in response:
+        print(f"Received chunk: {chunk}")
+        assert isinstance(chunk, str)
+        assert len(chunk) > 0
         chunks.append(chunk)
+    print(f"Received {len(chunks)} total chunks")
     assert len(chunks) > 0
 
 

mira_network-0.1.2/tests/test_integration.py

@@ -1,48 +0,0 @@
-import pytest
-from src.mira_sdk.client import MiraClient
-from src.mira_sdk.models import Message, AiRequest, ModelProvider
-
-
-@pytest.mark.asyncio
-async def test_real_generate():
-    """This test makes a real API call to generate text."""
-    client = MiraClient(
-        base_url="https://mira-client-balancer.alts.dev",
-        api_token="sk-mira-8ac810228d32ff68fc93266fb9a0ba612724119ffab16dcc"
-    )
-
-    request = AiRequest(
-        model="mira/llama3.1",
-        messages=[Message(role="user", content="Hi Who are you!")],
-        stream=False,
-        model_provider=None
-    )
-
-    async with client:
-        result = await client.generate(request)
-        print("Real API Response:", result)
-        # assert len(result) > 0
-
-
-# @pytest.mark.asyncio
-# async def test_real_generate_stream():
-#     """This test makes a real API call with streaming enabled."""
-#     client = MiraClient(
-#         base_url="https://mira-client-balancer.alts.dev",
-#         api_token="sk-mira-8ac810228d32ff68fc93266fb9a0ba612724119ffab16dcc"
-#     )
-
-#     request = AiRequest(
-#         messages=[Message(role="user", content="Count from 1 to 5")],
-#         stream=True
-#     )
-
-#     async with client:
-#         stream = await client.generate(request)
-#         response = ""
-#         async for chunk in stream:
-#             print("Chunk:", chunk)
-#             response += chunk
-#         print("Final Response:", response)
-#         assert isinstance(response, str)
-#         assert len(response) > 0

mira_network-0.1.2/tests/test_models.py

@@ -1,109 +0,0 @@
-import pytest
-from src.mira_sdk.models import (
-    Message,
-    ModelProvider,
-    AiRequest,
-    FlowChatCompletion,
-    FlowRequest,
-    ApiTokenRequest,
-    AddCreditRequest,
-)
-
-
-def test_message_model():
-    message = Message(role="user", content="Hello!")
-    assert message.role == "user"
-    assert message.content == "Hello!"
-
-
-def test_model_provider():
-    provider = ModelProvider(base_url="https://mira-client-balancer.alts.dev", api_key="sk-mira-8ac810228d32ff68fc93266fb9a0ba612724119ffab16dcc")
-    assert provider.base_url == "https://mira-client-balancer.alts.dev"
-    assert provider.api_key == "sk-mira-8ac810228d32ff68fc93266fb9a0ba612724119ffab16dcc"
-
-
-def test_ai_request():
-    messages = [Message(role="user", content="Hello!")]
-    provider = ModelProvider(base_url="https://mira-client-balancer.alts.dev", api_key="sk-mira-8ac810228d32ff68fc93266fb9a0ba612724119ffab16dcc")
-    
-    # Test with default values
-    request = AiRequest(messages=messages)
-    assert request.model == "mira/llama3.1"
-    assert request.stream is False
-    assert request.model_provider is None
-    assert request.messages == messages
-    
-    # Test with custom values
-    request = AiRequest(
-        model="custom/model",
-        messages=messages,
-        model_provider=provider,
-        stream=True
-    )
-    assert request.model == "custom/model"
-    assert request.stream is True
-    assert request.model_provider == provider
-    assert request.messages == messages
-
-
-def test_flow_chat_completion():
-    # Test with no variables
-    completion = FlowChatCompletion()
-    assert completion.variables is None
-    
-    # Test with variables
-    variables = {"key": "value"}
-    completion = FlowChatCompletion(variables=variables)
-    assert completion.variables == variables
-
-
-def test_flow_request():
-    request = FlowRequest(
-        system_prompt="You are a helpful assistant",
-        name="test-flow"
-    )
-    assert request.system_prompt == "You are a helpful assistant"
-    assert request.name == "test-flow"
-
-
-def test_api_token_request():
-    # Test with no description
-    request = ApiTokenRequest()
-    assert request.description is None
-    
-    # Test with description
-    request = ApiTokenRequest(description="Test token")
-    assert request.description == "Test token"
-
-
-def test_add_credit_request():
-    # Test required fields
-    request = AddCreditRequest(user_id="user123", amount=100.0)
-    assert request.user_id == "user123"
-    assert request.amount == 100.0
-    assert request.description is None
-    
-    # Test with description
-    request = AddCreditRequest(
-        user_id="user123",
-        amount=100.0,
-        description="Test credit"
-    )
-    assert request.user_id == "user123"
-    assert request.amount == 100.0
-    assert request.description == "Test credit"
-
-
-def test_invalid_message():
-    with pytest.raises(ValueError):
-        Message(role="invalid", content="")  # Invalid role
-
-
-def test_invalid_ai_request():
-    with pytest.raises(ValueError):
-        AiRequest(messages=[])  # Empty messages list
-
-
-def test_invalid_add_credit_request():
-    with pytest.raises(ValueError):
-        AddCreditRequest(user_id="", amount=-100)  # Negative amount