simile 0.3.1__tar.gz → 0.5.4__tar.gz

{simile-0.3.1 → simile-0.5.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: simile
-Version: 0.3.1
+Version: 0.5.4
 Summary: Package for interfacing with Simile AI agents for simulation
 Author-email: Simile AI <cqz@simile.ai>
 License: MIT
@@ -46,3 +46,14 @@ from simile import Simile
 
 client = Simile(api_key="your_api_key")
 ```
+
+## Publishing
+
+1. Bump the version in `pyproject.toml`
+2. Commit and push your changes
+3. Go to the **Actions** tab in GitHub
+4. Select **Build and Publish to PyPI** workflow
+5. Click **Run workflow** and select the branch to run from
+6. The workflow will automatically build and publish to PyPI
+
+The workflow uses the `PYPI_API_TOKEN` secret configured in the repository settings.

simile-0.5.4/README.md

@@ -0,0 +1,33 @@
+# Simile API Python Client
+
+A Python client for interacting with the Simile API server.
+
+## Installation
+
+```bash
+pip install simile
+```
+
+## Dependencies
+
+- `httpx>=0.20.0`
+- `pydantic>=2.0.0`
+
+## Usage
+
+```python
+from simile import Simile
+
+client = Simile(api_key="your_api_key")
+```
+
+## Publishing
+
+1. Bump the version in `pyproject.toml`
+2. Commit and push your changes
+3. Go to the **Actions** tab in GitHub
+4. Select **Build and Publish to PyPI** workflow
+5. Click **Run workflow** and select the branch to run from
+6. The workflow will automatically build and publish to PyPI
+
+The workflow uses the `PYPI_API_TOKEN` secret configured in the repository settings.

{simile-0.3.1 → simile-0.5.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "simile"
-version = "0.3.1"
+version = "0.5.4"
 authors = [
   { name="Simile AI", email="cqz@simile.ai" },
 ]
@@ -34,3 +34,37 @@ dependencies = [
 
 [project.urls]
 "Homepage" = "https://github.com/simile-team/simile-sdk"
+
+[tool.ruff]
+# Set the maximum line length to 100 (default is 88)
+line-length = 100
+target-version = "py311"
+
+[tool.ruff.lint]
+# Enable a comprehensive set of rules
+select = [
+    "E",   # pycodestyle errors
+    "W",   # pycodestyle warnings
+    "F",   # pyflakes
+    "I",   # isort
+    "B",   # flake8-bugbear
+    "C4",  # flake8-comprehensions
+    "UP",  # pyupgrade
+    "ARG", # flake8-unused-arguments
+    "SIM", # flake8-simplify
+]
+ignore = [
+    "E501",  # line too long (handled by formatter)
+    "B008",  # do not perform function calls in argument defaults
+    "B006",  # do not use mutable data structures for argument defaults
+]
+
+[tool.ruff.lint.per-file-ignores]
+"__init__.py" = ["F401"]  # Allow unused imports in __init__.py
+"examples/*" = ["E501"]    # Allow long lines in examples
+
+[tool.ruff.format]
+# Use double quotes for strings
+quote-style = "double"
+# Indent with spaces
+indent-style = "space"

simile-0.5.4/simile/__init__.py

@@ -0,0 +1,61 @@
+from .auth_client import SimileAuth
+from .client import Simile
+from .exceptions import (
+    SimileAPIError,
+    SimileAuthenticationError,
+    SimileBadRequestError,
+    SimileNotFoundError,
+)
+from .models import (
+    Agent,
+    ClosedGenerationRequest,
+    ClosedGenerationResponse,
+    ClosedQuestionMemoryTurn,
+    ContextMemoryTurn,
+    CreateAgentPayload,
+    CreateDataItemPayload,
+    CreatePopulationPayload,
+    DataItem,
+    DeletionResponse,
+    ImageMemoryTurn,
+    MemoryStream,
+    MemoryTurn,
+    MemoryTurnType,
+    OpenGenerationRequest,
+    OpenGenerationResponse,
+    OpenQuestionMemoryTurn,
+    Population,
+    PopulationInfo,
+    UpdateDataItemPayload,
+)
+
+__all__ = [
+    "Simile",
+    "SimileAuth",
+    "Population",
+    "PopulationInfo",
+    "Agent",
+    "DataItem",
+    "CreatePopulationPayload",
+    "CreateAgentPayload",
+    "CreateDataItemPayload",
+    "UpdateDataItemPayload",
+    "DeletionResponse",
+    "OpenGenerationRequest",
+    "OpenGenerationResponse",
+    "ClosedGenerationRequest",
+    "ClosedGenerationResponse",
+    "MemoryStream",
+    "MemoryTurn",
+    "MemoryTurnType",
+    "ContextMemoryTurn",
+    "ImageMemoryTurn",
+    "OpenQuestionMemoryTurn",
+    "ClosedQuestionMemoryTurn",
+    "SimileAPIError",
+    "SimileAuthenticationError",
+    "SimileNotFoundError",
+    "SimileBadRequestError",
+]
+
+__version__ = "0.5.5"

{simile-0.3.1 → simile-0.5.4}/simile/auth_client.py

@@ -5,19 +5,19 @@ This client handles authentication-related operations using Google OAuth tokens
 rather than API keys.
 """
 
-import httpx
-from httpx import AsyncClient
-from typing import List, Optional, Union
 import uuid
 from datetime import datetime
 
-from .models import BaseModel
+import httpx
+from httpx import AsyncClient
+
 from .exceptions import (
     SimileAPIError,
     SimileAuthenticationError,
-    SimileNotFoundError,
     SimileBadRequestError,
+    SimileNotFoundError,
 )
+from .models import BaseModel
 
 
 # Import the models from auth_api_endpoints
@@ -28,7 +28,7 @@ class UserInfo(BaseModel):
     user_id: str
     email: str
     name: str
-    picture: Optional[str] = None
+    picture: str | None = None
     created_at: datetime
     last_login: datetime
 
@@ -40,8 +40,8 @@ class APIKey(BaseModel):
     name: str
     key_prefix: str
     created_at: datetime
-    last_used: Optional[datetime] = None
-    expires_at: Optional[datetime] = None
+    last_used: datetime | None = None
+    expires_at: datetime | None = None
     is_active: bool = True
 
 
@@ -53,7 +53,7 @@ class APIKeyCreateResponse(BaseModel):
     key: str
     key_prefix: str
     created_at: datetime
-    expires_at: Optional[datetime] = None
+    expires_at: datetime | None = None
 
 
 class PopulationAccess(BaseModel):
@@ -61,7 +61,7 @@ class PopulationAccess(BaseModel):
 
     population_id: uuid.UUID
     name: str
-    description: Optional[str] = None
+    description: str | None = None
     role: str
     created_at: datetime
     member_count: int
@@ -75,7 +75,7 @@ class PopulationShareResponse(BaseModel):
     population_id: uuid.UUID
     role: str
     expires_at: datetime
-    max_uses: Optional[int] = None
+    max_uses: int | None = None
     created_at: datetime
 
 
@@ -84,7 +84,7 @@ class PopulationJoinResponse(BaseModel):
 
     population_id: uuid.UUID
     name: str
-    description: Optional[str] = None
+    description: str | None = None
     role: str
     message: str = "Successfully joined population"
 
@@ -118,9 +118,7 @@ class SimileAuth:
             timeout=TIMEOUT_CONFIG,
         )
 
-    async def _request(
-        self, method: str, endpoint: str, **kwargs
-    ) -> Union[httpx.Response, BaseModel]:
+    async def _request(self, method: str, endpoint: str, **kwargs) -> httpx.Response | BaseModel:
         """Make an HTTP request to the API."""
         url = f"{self.base_url}/{endpoint.lstrip('/')}"
         response_model_cls = kwargs.pop("response_model", None)
@@ -142,30 +140,30 @@ class SimileAuth:
                 detail = e.response.text
 
             if status_code == 401:
-                raise SimileAuthenticationError(detail=detail)
+                raise SimileAuthenticationError(detail=detail) from e
             elif status_code == 404:
-                raise SimileNotFoundError(detail=detail)
+                raise SimileNotFoundError(detail=detail) from e
             elif status_code == 400:
-                raise SimileBadRequestError(detail=detail)
+                raise SimileBadRequestError(detail=detail) from e
             else:
                 raise SimileAPIError(
                     f"API request failed: {e}", status_code=status_code, detail=detail
-                )
+                ) from e
         except httpx.RequestError as e:
-            raise SimileAPIError(f"Request error: {type(e).__name__}: {e}")
+            raise SimileAPIError(f"Request error: {type(e).__name__}: {e}") from e
 
     async def get_current_user(self) -> UserInfo:
         """Get current user information."""
         response = await self._request("GET", "auth/me", response_model=UserInfo)
         return response
 
-    async def list_api_keys(self) -> List[APIKey]:
+    async def list_api_keys(self) -> list[APIKey]:
         """List all API keys for the current user."""
         response = await self._request("GET", "auth/api-keys")
         return [APIKey(**key_data) for key_data in response.json()]
 
     async def create_api_key(
-        self, name: str, expires_in_days: Optional[int] = None
+        self, name: str, expires_in_days: int | None = None
     ) -> APIKeyCreateResponse:
         """
         Create a new API key.
@@ -186,7 +184,7 @@ class SimileAuth:
         )
         return response
 
-    async def delete_api_key(self, key_id: Union[str, uuid.UUID]) -> dict:
+    async def delete_api_key(self, key_id: str | uuid.UUID) -> dict:
         """
         Delete an API key.
 
@@ -199,17 +197,17 @@ class SimileAuth:
         response = await self._request("DELETE", f"auth/api-keys/{str(key_id)}")
         return response.json()
 
-    async def list_accessible_populations(self) -> List[PopulationAccess]:
+    async def list_accessible_populations(self) -> list[PopulationAccess]:
         """List all populations the current user has access to."""
         response = await self._request("GET", "auth/populations")
         return [PopulationAccess(**pop_data) for pop_data in response.json()]
 
     async def create_population_share_code(
         self,
-        population_id: Union[str, uuid.UUID],
+        population_id: str | uuid.UUID,
         role: str = "viewer",
-        expires_in_hours: Optional[int] = 24,
-        max_uses: Optional[int] = None,
+        expires_in_hours: int | None = 24,
+        max_uses: int | None = None,
     ) -> PopulationShareResponse:
         """
         Create a share code for a population.
@@ -239,9 +237,7 @@ class SimileAuth:
         )
         return response
 
-    async def join_population_with_share_code(
-        self, share_code: str
-    ) -> PopulationJoinResponse:
+    async def join_population_with_share_code(self, share_code: str) -> PopulationJoinResponse:
         """
         Join a population using a share code.