opengradient 0.2.7__tar.gz → 0.3.0__tar.gz

{opengradient-0.2.7/src/opengradient.egg-info → opengradient-0.3.0}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.1
 Name: opengradient
-Version: 0.2.7
-Summary: A Python SDK for OpenGradient inference services
+Version: 0.3.0
+Summary: Python SDK for OpenGradient decentralized model management & inference services
 Author-email: OpenGradient <oliver@opengradient.ai>
 License: MIT License
         
@@ -135,47 +135,86 @@ Requires-Dist: websockets==13.1
 Requires-Dist: xattr==1.1.0
 Requires-Dist: yarl==1.13.1
 
-# OpenGradient Python SDK
+# OpenGradient SDK
 
-Python SDK for OpenGradient inference services.
+Python SDK for OpenGradient decentralized model management & inference services.
 
 ## Installation
-```
+```python
 pip install opengradient
 ```
 
 ## Quick Start
-```
+```python
 import opengradient as og
-og.init(private_key="x", rpc_url="y", contract_address="z")
-```
-
-### Sign in with Email
-```
-og.login(email="you@opengradient.ai", password="xyz")
+og.init(email="<email>", password="<password>")
 ```
 
 ### Create a Model
+```python
+og.create_model(model_name="<model_name>", model_desc="<model_description>")
 ```
-og.create_model(model_name="test-network-model", model_desc="testing upload to sdk")
+
+### Create a Model (with file upload)
+```python
+og.create_model(model_name="<model_name>", model_desc="<model_description>", model_path="<model_path>")
 ```
 
 ### Create a Version of a Model
-```
-og.create_version(model_name="test-network-model", notes="test notes")
+```python
+og.create_version(model_name="<model_name>", notes="<model_notes>")
 ```
 
 ### Upload Files to a Model
+```python
+og.upload(model_path="<model_path>", model_name="<model_name>", version="<version>")
 ```
-og.upload(model_path="local_path_to_your_model.onnx", model_name="test-network-model", version="0.01")
+
+### List Files of a Model Version
+```python
+og.list_files(model_name="<model_name>", version="<version>")
 ```
 
 ### Run Inference
-```
+```python
 inference_mode = og.InferenceMode.VANILLA
-inference_cid = og.infer(model_cid, model_inputs, inference_mode)
+og.infer(model_cid, model_inputs, inference_mode)
 ```
 
+## Using the CLI
+
+```bash
+export OPENGRADIENT_EMAIL="<email>"
+export OPENGRADIENT_PASSWORD="<password>"
 ```
-og.infer(model_id, inference_mode, model_input)
+
+#### Creating a Model
+```bash
+opengradient create_model "<model_name>" "<description>" 
+```
+- creating a model automatically initializes version `v0.01`
+
+#### Creating a Version
+```bash
+opengradient create_model "<model_name>" "<description>" 
+```
+
+#### Upload a File
+```bash
+opengradient upload "<model_path>" "<model_name>" "<version>" 
+```
+
+#### List Files of a Model Version
+```bash
+opengradient list_files "<model_name>" "<version>"
+```
+
+####  CLI infer using string 
+```bash
+opengradient infer QmbUqS93oc4JTLMHwpVxsE39mhNxy6hpf6Py3r9oANr8aZ VANILLA '{"num_input1":[1.0, 2.0, 3.0], "num_input2":10, "str_input1":["hello", "ONNX"], "str_input2":" world"}'
+```
+
+#### CLI infer using file path input
+```bash
+opengradient infer QmbUqS93oc4JTLMHwpVxsE39mhNxy6hpf6Py3r9oANr8aZ VANILLA --input_file input.json
 ```

opengradient-0.3.0/README.md

@@ -0,0 +1,83 @@
+# OpenGradient SDK
+
+Python SDK for OpenGradient decentralized model management & inference services.
+
+## Installation
+```python
+pip install opengradient
+```
+
+## Quick Start
+```python
+import opengradient as og
+og.init(email="<email>", password="<password>")
+```
+
+### Create a Model
+```python
+og.create_model(model_name="<model_name>", model_desc="<model_description>")
+```
+
+### Create a Model (with file upload)
+```python
+og.create_model(model_name="<model_name>", model_desc="<model_description>", model_path="<model_path>")
+```
+
+### Create a Version of a Model
+```python
+og.create_version(model_name="<model_name>", notes="<model_notes>")
+```
+
+### Upload Files to a Model
+```python
+og.upload(model_path="<model_path>", model_name="<model_name>", version="<version>")
+```
+
+### List Files of a Model Version
+```python
+og.list_files(model_name="<model_name>", version="<version>")
+```
+
+### Run Inference
+```python
+inference_mode = og.InferenceMode.VANILLA
+og.infer(model_cid, model_inputs, inference_mode)
+```
+
+## Using the CLI
+
+```bash
+export OPENGRADIENT_EMAIL="<email>"
+export OPENGRADIENT_PASSWORD="<password>"
+```
+
+#### Creating a Model
+```bash
+opengradient create_model "<model_name>" "<description>" 
+```
+- creating a model automatically initializes version `v0.01`
+
+#### Creating a Version
+```bash
+opengradient create_model "<model_name>" "<description>" 
+```
+
+#### Upload a File
+```bash
+opengradient upload "<model_path>" "<model_name>" "<version>" 
+```
+
+#### List Files of a Model Version
+```bash
+opengradient list_files "<model_name>" "<version>"
+```
+
+####  CLI infer using string 
+```bash
+opengradient infer QmbUqS93oc4JTLMHwpVxsE39mhNxy6hpf6Py3r9oANr8aZ VANILLA '{"num_input1":[1.0, 2.0, 3.0], "num_input2":10, "str_input1":["hello", "ONNX"], "str_input2":" world"}'
+```
+
+#### CLI infer using file path input
+```bash
+opengradient infer QmbUqS93oc4JTLMHwpVxsE39mhNxy6hpf6Py3r9oANr8aZ VANILLA --input_file input.json
+```

{opengradient-0.2.7 → opengradient-0.3.0}/pyproject.toml

@@ -4,8 +4,8 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "opengradient"
-version = "0.2.7"
-description = "A Python SDK for OpenGradient inference services"
+version = "0.3.0"
+description = "Python SDK for OpenGradient decentralized model management & inference services"
 authors = [{name = "OpenGradient", email = "oliver@opengradient.ai"}]
 license = {file = "LICENSE"}
 readme = "README.md"

{opengradient-0.2.7 → opengradient-0.3.0}/src/opengradient/__init__.py

@@ -1,16 +1,16 @@
 from .client import Client
-from .exceptions import OpenGradientError, FileNotFoundError, UploadError, InferenceError, ResultRetrievalError
-from .types import ModelInput, InferenceMode, Number, NumberTensor, StringTensor, ModelOutput
-
-__version__ = "0.2.7"
+from .defaults import *
+from .types import InferenceMode
+from typing import List, Dict
+__version__ = "0.3.0"
 
 _client = None
 
-def init(private_key="cd09980ef6e280afc3900d2d6801f9e9c5d858a5deaeeab74a65643f5ff1a4c1",
-         rpc_url="http://18.218.115.248:8545",
-         contract_address="0x350E0A430b2B1563481833a99523Cfd17a530e4e",
-         email="test@test.com",
-         password="Test-123"):
+def init(email: str,
+         password: str,
+         private_key=DEFAULT_PRIVATE_KEY,
+         rpc_url=DEFAULT_RPC_URL,
+         contract_address=DEFAULT_INFERENCE_CONTRACT_ADDRESS):
     global _client
     _client = Client(private_key=private_key, rpc_url=rpc_url, contract_address=contract_address, email=email, password=password)
 
@@ -19,10 +19,18 @@ def upload(model_path, model_name, version):
         raise RuntimeError("OpenGradient client not initialized. Call og.init() first.")
     return _client.upload(model_path, model_name, version)
 
-def create_model(model_name, model_desc):
+def create_model(model_name: str, model_desc: str, model_path: str = None):
     if _client is None:
         raise RuntimeError("OpenGradient client not initialized. Call og.init() first.")
-    return _client.create_model(model_name, model_desc)
+    
+    result = _client.create_model(model_name, model_desc)
+    
+    if model_path:
+        version = "0.01"
+        upload_result = _client.upload(model_path, model_name, version)
+        result["upload"] = upload_result
+    
+    return result
 
 def create_version(model_name, notes=None, is_major=False):
     if _client is None:
@@ -37,4 +45,9 @@ def infer(model_cid, inference_mode, model_input):
 def login(email: str, password: str):
     if _client is None:
         raise RuntimeError("OpenGradient client not initialized. Call og.init() first.")
-    return _client.login(email, password)
+    return _client.login(email, password)
+
+def list_files(model_name: str, version: str) -> List[Dict]:
+    if _client is None:
+        raise RuntimeError("OpenGradient client not initialized. Call og.init() first.")
+    return _client.list_files(model_name, version)

opengradient-0.3.0/src/opengradient/abi/inference.abi

@@ -0,0 +1 @@
+[{"anonymous":false,"inputs":[{"components":[{"components":[{"internalType":"string","name":"name","type":"string"},{"components":[{"internalType":"int128","name":"value","type":"int128"},{"internalType":"int128","name":"decimals","type":"int128"}],"internalType":"struct TensorLib.Number[]","name":"values","type":"tuple[]"},{"internalType":"uint32[]","name":"shape","type":"uint32[]"}],"internalType":"struct TensorLib.MultiDimensionalNumberTensor[]","name":"numbers","type":"tuple[]"},{"components":[{"internalType":"string","name":"name","type":"string"},{"internalType":"string[]","name":"values","type":"string[]"}],"internalType":"struct TensorLib.StringTensor[]","name":"strings","type":"tuple[]"},{"internalType":"bool","name":"is_simulation_result","type":"bool"}],"indexed":false,"internalType":"struct ModelOutput","name":"output","type":"tuple"}],"name":"InferenceResult","type":"event"},{"inputs":[{"internalType":"string","name":"modelId","type":"string"},{"internalType":"enum ModelInferenceMode","name":"inferenceMode","type":"uint8"},{"components":[{"components":[{"internalType":"string","name":"name","type":"string"},{"components":[{"internalType":"int128","name":"value","type":"int128"},{"internalType":"int128","name":"decimals","type":"int128"}],"internalType":"struct TensorLib.Number[]","name":"values","type":"tuple[]"},{"internalType":"uint32[]","name":"shape","type":"uint32[]"}],"internalType":"struct TensorLib.MultiDimensionalNumberTensor[]","name":"numbers","type":"tuple[]"},{"components":[{"internalType":"string","name":"name","type":"string"},{"internalType":"string[]","name":"values","type":"string[]"}],"internalType":"struct TensorLib.StringTensor[]","name":"strings","type":"tuple[]"}],"internalType":"struct ModelInput","name":"modelInput","type":"tuple"}],"name":"run","outputs":[{"components":[{"components":[{"internalType":"string","name":"name","type":"string"},{"components":[{"internalType":"int128","name":"value","type":"int128"},{"internalType":"int128","name":"decimals","type":"int128"}],"internalType":"struct TensorLib.Number[]","name":"values","type":"tuple[]"},{"internalType":"uint32[]","name":"shape","type":"uint32[]"}],"internalType":"struct TensorLib.MultiDimensionalNumberTensor[]","name":"numbers","type":"tuple[]"},{"components":[{"internalType":"string","name":"name","type":"string"},{"internalType":"string[]","name":"values","type":"string[]"}],"internalType":"struct TensorLib.StringTensor[]","name":"strings","type":"tuple[]"},{"internalType":"bool","name":"is_simulation_result","type":"bool"}],"internalType":"struct ModelOutput","name":"","type":"tuple"}],"stateMutability":"nonpayable","type":"function"}]

opengradient-0.3.0/src/opengradient/abi/llm.abi

@@ -0,0 +1 @@
+[{"anonymous":false,"inputs":[{"components":[{"components":[{"internalType":"string","name":"name","type":"string"},{"components":[{"internalType":"int128","name":"value","type":"int128"},{"internalType":"int128","name":"decimals","type":"int128"}],"internalType":"struct TensorLib.Number[]","name":"values","type":"tuple[]"},{"internalType":"uint32[]","name":"shape","type":"uint32[]"}],"internalType":"struct TensorLib.MultiDimensionalNumberTensor[]","name":"numbers","type":"tuple[]"},{"components":[{"internalType":"string","name":"name","type":"string"},{"internalType":"string[]","name":"values","type":"string[]"}],"internalType":"struct TensorLib.StringTensor[]","name":"strings","type":"tuple[]"},{"internalType":"bool","name":"is_simulation_result","type":"bool"}],"indexed":false,"internalType":"struct ModelOutput","name":"output","type":"tuple"}],"name":"InferenceResult","type":"event"},{"anonymous":false,"inputs":[{"components":[{"internalType":"string","name":"answer","type":"string"}],"indexed":false,"internalType":"struct LlmResponse","name":"response","type":"tuple"}],"name":"LLMResult","type":"event"},{"inputs":[{"internalType":"string","name":"modelId","type":"string"},{"internalType":"enum ModelInferenceMode","name":"inferenceMode","type":"uint8"},{"components":[{"components":[{"internalType":"string","name":"name","type":"string"},{"components":[{"internalType":"int128","name":"value","type":"int128"},{"internalType":"int128","name":"decimals","type":"int128"}],"internalType":"struct TensorLib.Number[]","name":"values","type":"tuple[]"},{"internalType":"uint32[]","name":"shape","type":"uint32[]"}],"internalType":"struct TensorLib.MultiDimensionalNumberTensor[]","name":"numbers","type":"tuple[]"},{"components":[{"internalType":"string","name":"name","type":"string"},{"internalType":"string[]","name":"values","type":"string[]"}],"internalType":"struct TensorLib.StringTensor[]","name":"strings","type":"tuple[]"}],"internalType":"struct ModelInput","name":"modelInput","type":"tuple"}],"name":"run","outputs":[{"components":[{"components":[{"internalType":"string","name":"name","type":"string"},{"components":[{"internalType":"int128","name":"value","type":"int128"},{"internalType":"int128","name":"decimals","type":"int128"}],"internalType":"struct TensorLib.Number[]","name":"values","type":"tuple[]"},{"internalType":"uint32[]","name":"shape","type":"uint32[]"}],"internalType":"struct TensorLib.MultiDimensionalNumberTensor[]","name":"numbers","type":"tuple[]"},{"components":[{"internalType":"string","name":"name","type":"string"},{"internalType":"string[]","name":"values","type":"string[]"}],"internalType":"struct TensorLib.StringTensor[]","name":"strings","type":"tuple[]"},{"internalType":"bool","name":"is_simulation_result","type":"bool"}],"internalType":"struct ModelOutput","name":"","type":"tuple"}],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"components":[{"internalType":"enum LlmInferenceMode","name":"mode","type":"uint8"},{"internalType":"string","name":"modelCID","type":"string"},{"internalType":"string","name":"prompt","type":"string"},{"internalType":"uint32","name":"max_tokens","type":"uint32"},{"internalType":"string[]","name":"stop_sequence","type":"string[]"},{"internalType":"uint32","name":"temperature","type":"uint32"}],"internalType":"struct LlmInferenceRequest","name":"request","type":"tuple"}],"name":"runLLM","outputs":[{"components":[{"internalType":"string","name":"answer","type":"string"}],"internalType":"struct LlmResponse","name":"","type":"tuple"}],"stateMutability":"nonpayable","type":"function"}]

opengradient-0.3.0/src/opengradient/account.py

@@ -0,0 +1,37 @@
+from eth_account import Account
+import secrets
+from collections import namedtuple
+import os
+import hashlib
+
+EthAccount = namedtuple('EthAccount', ['address', 'private_key'])
+
+
+def generate_eth_account() -> EthAccount:
+    user_seed = _get_user_random_seed()
+    private_key = _generate_secure_private_key(user_seed)
+
+    # derive account
+    account = Account.from_key(private_key)
+
+    # get the public key (address)
+    public_key = account.address
+
+    return EthAccount(address=public_key, private_key=private_key)
+
+def _get_user_random_seed():
+    print("Please type a random string of characters (the longer and more random, the better):")
+    print("> ", end="")  # Add a '>' prompt on a new line
+    return input().encode()
+
+def _generate_secure_private_key(user_input):
+    # Combine multiple sources of entropy
+    system_random = secrets.token_bytes(32)
+    os_urandom = os.urandom(32)
+    timestamp = str(secrets.randbits(256)).encode()
+    
+    # Add user input to the entropy sources
+    combined = system_random + os_urandom + timestamp + user_input
+    
+    # Hash the combined entropy
+    return hashlib.sha256(combined).hexdigest()