daisi-sdk 0.1.0__tar.gz

daisi_sdk-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Daisi Contributors
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

daisi_sdk-0.1.0/PKG-INFO

@@ -0,0 +1,322 @@
+Metadata-Version: 2.4
+Name: daisi-sdk
+Version: 0.1.0
+Summary: Python API wrapper for the Daisi SDK
+Author-email: Daisi Contributors <dev@daisi.net>
+License: MIT
+Keywords: daisi,sdk,api,wrapper
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: grpcio>=1.60.0
+Requires-Dist: grpcio-tools>=1.60.0
+Requires-Dist: protobuf>=4.25.0
+Requires-Dist: pydantic>=2.0.0
+Requires-Dist: python-dotenv>=1.0.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.4.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.1.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"
+Requires-Dist: pytest-grpc>=0.8.0; extra == "dev"
+Requires-Dist: black>=23.0.0; extra == "dev"
+Requires-Dist: isort>=5.12.0; extra == "dev"
+Requires-Dist: flake8>=6.0.0; extra == "dev"
+Requires-Dist: mypy>=1.4.0; extra == "dev"
+Requires-Dist: mypy-protobuf>=3.5.0; extra == "dev"
+Requires-Dist: pylint>=2.17.0; extra == "dev"
+Requires-Dist: grpc-stubs>=1.53.0; extra == "dev"
+Provides-Extra: docs
+Requires-Dist: sphinx>=7.0.0; extra == "docs"
+Requires-Dist: sphinx-rtd-theme>=1.3.0; extra == "docs"
+Requires-Dist: sphinx-autodoc-typehints>=1.23.0; extra == "docs"
+Dynamic: license-file
+
+# Daisi SDK - Python gRPC Wrapper
+
+A modern Python wrapper for the Daisi SDK, providing a clean and Pythonic interface to interact with the Daisi distributed AI network via gRPC.
+
+## Features
+
+- **gRPC-based**: Full gRPC implementation matching the .NET SDK
+- **Type-safe**: Full type hints for IDE autocomplete and type checking
+- **Streaming support**: Native support for streaming inference responses
+- **Session management**: Complete session lifecycle management
+- **Comprehensive**: All Daisi services (Auth, Sessions, Inference, Models, Hosts, Peers, Settings, Commands)
+- **Production-ready**: Follows PEP standards and best practices
+
+## Installation
+
+```bash
+pip install daisi-sdk
+```
+
+### Development Installation
+
+```bash
+git clone https://github.com/daisinet/daisi-sdk-python.git
+cd daisi-sdk-python
+pip install -e ".[dev]"
+```
+
+## Quick Start
+
+```python
+from daisi import DaisiClient
+
+# Initialize client with API key
+client = DaisiClient(client_key="your-client-key")
+
+# Create a session
+session_id, host = client.session.create(model_name="llama-3")
+
+# Create an inference
+inference_id = client.inference.create(
+    session_id=session_id,
+    initialization_prompt="You are a helpful assistant."
+)
+
+# Send inference request with streaming response
+for response in client.inference.send("Hello, how are you?"):
+    if response.Type == InferenceResponseTypes.InferenceText:
+        print(response.Content, end="", flush=True)
+
+# Get statistics
+stats = client.inference.stats()
+print(f"\nTokens used: {stats.SessionTokenCount}")
+
+# Clean up
+client.inference.close_inference()
+client.session.close()
+client.close()
+```
+
+## Configuration
+
+Set your configuration via environment variables:
+
+```bash
+export DAISI_CLIENT_KEY="your-client-key"
+export DAISI_ORC_ADDRESS="orc.daisi.net"
+export DAISI_ORC_PORT="443"
+```
+
+Or configure in code:
+
+```python
+from daisi import DaisiClient, DaisiConfig
+
+config = DaisiConfig(
+    client_key="your-client-key",
+    orc_address="orc.daisi.net",
+    orc_port=443,
+    use_ssl=True
+)
+
+client = DaisiClient(
+    client_key=config.client_key,
+    orc_address=config.orc_address,
+    orc_port=config.orc_port
+)
+```
+
+## Available Clients
+
+The SDK provides specialized clients for each service:
+
+- **`client.auth`**: Authentication operations (create/validate keys, auth codes)
+- **`client.session`**: Session management (create, claim, close, connect)
+- **`client.inference`**: Inference operations (create, send streaming, stats, close)
+- **`client.models`**: Model discovery and requirements
+- **`client.hosts`**: Host registration and management
+- **`client.peers`**: Peer operations
+- **`client.settings`**: Settings management
+- **`client.host_commands`**: Host command streaming
+- **`client.app_commands`**: App command streaming
+
+## Usage Examples
+
+### Authentication
+
+```python
+from daisi import DaisiClient
+
+client = DaisiClient(client_key="initial-key")
+
+# Send auth code
+client.auth.send_auth_code("user@example.com")
+
+# Validate auth code and get new client key
+client_key, user_name, account_name, account_id = client.auth.validate_auth_code(
+    secret_key="secret",
+    email_or_phone="user@example.com",
+    auth_code="123456",
+    app_id="my-app"
+)
+```
+
+### Session Management
+
+```python
+# Create session with preferences
+session_id, host = client.session.create(
+    model_name="llama-3",
+    direct_connect_required=False,
+    preferred_region="us-west",
+    private_network_only=False
+)
+
+# Connect to session
+sid, has_capacity, already_connected = client.session.connect(session_id)
+```
+
+### Inference with Tool Groups
+
+```python
+from protos.v1.models.InferenceModels_pb2 import (
+    ThinkChainOfThought,
+    InferenceCodingTools,
+    InferenceFileTools
+)
+
+# Create inference with tools
+session_id, inference_id = client.inference.create(
+    session_id=session_id,
+    think_level=ThinkChainOfThought,
+    tool_groups=[InferenceCodingTools, InferenceFileTools]
+)
+
+# Send with custom parameters
+for response in client.inference.send(
+    text="Write a Python function to calculate fibonacci",
+    temperature=0.7,
+    top_p=0.9,
+    max_tokens=2000
+):
+    print(response.Content, end="")
+```
+
+### Models
+
+```python
+# Get required models
+models_response = client.models.get_required_models()
+```
+
+### Hosts
+
+```python
+# Get available hosts
+hosts = client.hosts.get_hosts()
+
+for host in hosts:
+    print(f"{host.Name}: {host.Status} - {host.Region}")
+```
+
+## Development
+
+### Generating Protobuf Code
+
+If you modify the .proto files:
+
+```bash
+python generate_protos.py
+```
+
+### Testing
+
+```bash
+# Run all tests
+pytest
+
+# Run with coverage
+pytest --cov=src/daisi
+
+# Run specific test file
+pytest tests/test_client.py
+```
+
+### Code Quality
+
+```bash
+# Format code
+black src/ tests/
+
+# Sort imports
+isort src/ tests/
+
+# Lint
+flake8 src/ tests/
+
+# Type checking
+mypy src/
+
+# Run all checks
+make lint
+```
+
+## Project Structure
+
+```
+daisi-sdk-python/
+├── src/daisi/              # Main package
+│   ├── __init__.py
+│   ├── client.py           # Main unified client
+│   ├── config.py           # Configuration
+│   ├── base_client.py      # Base gRPC client
+│   ├── exceptions.py       # Custom exceptions
+│   └── clients/            # Service-specific clients
+│       ├── auth.py
+│       ├── session.py
+│       ├── inference.py
+│       ├── models.py
+│       ├── hosts.py
+│       ├── peers.py
+│       ├── settings.py
+│       └── commands.py
+├── protos/v1/              # Protobuf definitions
+│   ├── *.proto             # Service definitions
+│   └── models/             # Model definitions
+├── src/protos/             # Generated Python code
+├── tests/                  # Test suite
+├── pyproject.toml          # Project configuration
+├── generate_protos.py      # Proto generation script
+└── README.md
+```
+
+## Architecture
+
+The SDK is built on gRPC and Protocol Buffers, matching the .NET SDK architecture:
+
+- **gRPC**: Efficient binary protocol with streaming support
+- **Protocol Buffers**: Strongly-typed message definitions
+- **Session-based**: Orchestrator (Orc) manages session lifecycle
+- **Streaming**: Real-time token-by-token inference responses
+- **Direct Connect**: Optional peer-to-peer mode
+
+## Contributing
+
+Contributions are welcome! Ensure:
+
+1. Code follows PEP 8 with Black formatting
+2. All tests pass and coverage is maintained
+3. Type hints are included
+4. Documentation is updated
+
+## License
+
+MIT License - see LICENSE file for details
+
+## Links
+
+- [Daisi Website](https://daisi.net)
+- [.NET SDK](https://github.com/daisinet/daisi-sdk-dotnet)
+- [Documentation](https://daisi.ai/Learn/SDK)

daisi_sdk-0.1.0/README.md

@@ -0,0 +1,280 @@
+# Daisi SDK - Python gRPC Wrapper
+
+A modern Python wrapper for the Daisi SDK, providing a clean and Pythonic interface to interact with the Daisi distributed AI network via gRPC.
+
+## Features
+
+- **gRPC-based**: Full gRPC implementation matching the .NET SDK
+- **Type-safe**: Full type hints for IDE autocomplete and type checking
+- **Streaming support**: Native support for streaming inference responses
+- **Session management**: Complete session lifecycle management
+- **Comprehensive**: All Daisi services (Auth, Sessions, Inference, Models, Hosts, Peers, Settings, Commands)
+- **Production-ready**: Follows PEP standards and best practices
+
+## Installation
+
+```bash
+pip install daisi-sdk
+```
+
+### Development Installation
+
+```bash
+git clone https://github.com/daisinet/daisi-sdk-python.git
+cd daisi-sdk-python
+pip install -e ".[dev]"
+```
+
+## Quick Start
+
+```python
+from daisi import DaisiClient
+
+# Initialize client with API key
+client = DaisiClient(client_key="your-client-key")
+
+# Create a session
+session_id, host = client.session.create(model_name="llama-3")
+
+# Create an inference
+inference_id = client.inference.create(
+    session_id=session_id,
+    initialization_prompt="You are a helpful assistant."
+)
+
+# Send inference request with streaming response
+for response in client.inference.send("Hello, how are you?"):
+    if response.Type == InferenceResponseTypes.InferenceText:
+        print(response.Content, end="", flush=True)
+
+# Get statistics
+stats = client.inference.stats()
+print(f"\nTokens used: {stats.SessionTokenCount}")
+
+# Clean up
+client.inference.close_inference()
+client.session.close()
+client.close()
+```
+
+## Configuration
+
+Set your configuration via environment variables:
+
+```bash
+export DAISI_CLIENT_KEY="your-client-key"
+export DAISI_ORC_ADDRESS="orc.daisi.net"
+export DAISI_ORC_PORT="443"
+```
+
+Or configure in code:
+
+```python
+from daisi import DaisiClient, DaisiConfig
+
+config = DaisiConfig(
+    client_key="your-client-key",
+    orc_address="orc.daisi.net",
+    orc_port=443,
+    use_ssl=True
+)
+
+client = DaisiClient(
+    client_key=config.client_key,
+    orc_address=config.orc_address,
+    orc_port=config.orc_port
+)
+```
+
+## Available Clients
+
+The SDK provides specialized clients for each service:
+
+- **`client.auth`**: Authentication operations (create/validate keys, auth codes)
+- **`client.session`**: Session management (create, claim, close, connect)
+- **`client.inference`**: Inference operations (create, send streaming, stats, close)
+- **`client.models`**: Model discovery and requirements
+- **`client.hosts`**: Host registration and management
+- **`client.peers`**: Peer operations
+- **`client.settings`**: Settings management
+- **`client.host_commands`**: Host command streaming
+- **`client.app_commands`**: App command streaming
+
+## Usage Examples
+
+### Authentication
+
+```python
+from daisi import DaisiClient
+
+client = DaisiClient(client_key="initial-key")
+
+# Send auth code
+client.auth.send_auth_code("user@example.com")
+
+# Validate auth code and get new client key
+client_key, user_name, account_name, account_id = client.auth.validate_auth_code(
+    secret_key="secret",
+    email_or_phone="user@example.com",
+    auth_code="123456",
+    app_id="my-app"
+)
+```
+
+### Session Management
+
+```python
+# Create session with preferences
+session_id, host = client.session.create(
+    model_name="llama-3",
+    direct_connect_required=False,
+    preferred_region="us-west",
+    private_network_only=False
+)
+
+# Connect to session
+sid, has_capacity, already_connected = client.session.connect(session_id)
+```
+
+### Inference with Tool Groups
+
+```python
+from protos.v1.models.InferenceModels_pb2 import (
+    ThinkChainOfThought,
+    InferenceCodingTools,
+    InferenceFileTools
+)
+
+# Create inference with tools
+session_id, inference_id = client.inference.create(
+    session_id=session_id,
+    think_level=ThinkChainOfThought,
+    tool_groups=[InferenceCodingTools, InferenceFileTools]
+)
+
+# Send with custom parameters
+for response in client.inference.send(
+    text="Write a Python function to calculate fibonacci",
+    temperature=0.7,
+    top_p=0.9,
+    max_tokens=2000
+):
+    print(response.Content, end="")
+```
+
+### Models
+
+```python
+# Get required models
+models_response = client.models.get_required_models()
+```
+
+### Hosts
+
+```python
+# Get available hosts
+hosts = client.hosts.get_hosts()
+
+for host in hosts:
+    print(f"{host.Name}: {host.Status} - {host.Region}")
+```
+
+## Development
+
+### Generating Protobuf Code
+
+If you modify the .proto files:
+
+```bash
+python generate_protos.py
+```
+
+### Testing
+
+```bash
+# Run all tests
+pytest
+
+# Run with coverage
+pytest --cov=src/daisi
+
+# Run specific test file
+pytest tests/test_client.py
+```
+
+### Code Quality
+
+```bash
+# Format code
+black src/ tests/
+
+# Sort imports
+isort src/ tests/
+
+# Lint
+flake8 src/ tests/
+
+# Type checking
+mypy src/
+
+# Run all checks
+make lint
+```
+
+## Project Structure
+
+```
+daisi-sdk-python/
+├── src/daisi/              # Main package
+│   ├── __init__.py
+│   ├── client.py           # Main unified client
+│   ├── config.py           # Configuration
+│   ├── base_client.py      # Base gRPC client
+│   ├── exceptions.py       # Custom exceptions
+│   └── clients/            # Service-specific clients
+│       ├── auth.py
+│       ├── session.py
+│       ├── inference.py
+│       ├── models.py
+│       ├── hosts.py
+│       ├── peers.py
+│       ├── settings.py
+│       └── commands.py
+├── protos/v1/              # Protobuf definitions
+│   ├── *.proto             # Service definitions
+│   └── models/             # Model definitions
+├── src/protos/             # Generated Python code
+├── tests/                  # Test suite
+├── pyproject.toml          # Project configuration
+├── generate_protos.py      # Proto generation script
+└── README.md
+```
+
+## Architecture
+
+The SDK is built on gRPC and Protocol Buffers, matching the .NET SDK architecture:
+
+- **gRPC**: Efficient binary protocol with streaming support
+- **Protocol Buffers**: Strongly-typed message definitions
+- **Session-based**: Orchestrator (Orc) manages session lifecycle
+- **Streaming**: Real-time token-by-token inference responses
+- **Direct Connect**: Optional peer-to-peer mode
+
+## Contributing
+
+Contributions are welcome! Ensure:
+
+1. Code follows PEP 8 with Black formatting
+2. All tests pass and coverage is maintained
+3. Type hints are included
+4. Documentation is updated
+
+## License
+
+MIT License - see LICENSE file for details
+
+## Links
+
+- [Daisi Website](https://daisi.net)
+- [.NET SDK](https://github.com/daisinet/daisi-sdk-dotnet)
+- [Documentation](https://daisi.ai/Learn/SDK)