opengradient 0.2.6__tar.gz → 0.2.8__tar.gz

opengradient-0.2.8/PKG-INFO

@@ -0,0 +1,209 @@
+Metadata-Version: 2.1
+Name: opengradient
+Version: 0.2.8
+Summary: A Python SDK for OpenGradient inference services
+Author-email: OpenGradient <oliver@opengradient.ai>
+License: MIT License
+        
+        Copyright (c) 2024 OpenGradient
+        
+        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.
+        
+Project-URL: Homepage, https://opengradient.ai
+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.7
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.7
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: aiohappyeyeballs==2.4.3
+Requires-Dist: aiohttp==3.10.8
+Requires-Dist: aiosignal==1.3.1
+Requires-Dist: annotated-types==0.7.0
+Requires-Dist: attrs==24.2.0
+Requires-Dist: bitarray==2.9.2
+Requires-Dist: CacheControl==0.14.0
+Requires-Dist: cachetools==5.5.0
+Requires-Dist: certifi==2024.8.30
+Requires-Dist: cffi==1.17.1
+Requires-Dist: charset-normalizer==3.3.2
+Requires-Dist: ckzg==2.0.1
+Requires-Dist: cleo==2.1.0
+Requires-Dist: click==8.1.7
+Requires-Dist: cramjam==2.8.4
+Requires-Dist: crashtest==0.4.1
+Requires-Dist: cryptography==43.0.1
+Requires-Dist: cytoolz==0.12.3
+Requires-Dist: distlib==0.3.8
+Requires-Dist: dulwich==0.21.7
+Requires-Dist: eth-account==0.13.4
+Requires-Dist: eth-hash==0.7.0
+Requires-Dist: eth-keyfile==0.8.1
+Requires-Dist: eth-keys==0.5.1
+Requires-Dist: eth-rlp==2.1.0
+Requires-Dist: eth-typing==5.0.0
+Requires-Dist: eth-utils==5.0.0
+Requires-Dist: eth_abi==5.1.0
+Requires-Dist: fastjsonschema==2.20.0
+Requires-Dist: fastparquet==2024.5.0
+Requires-Dist: filelock==3.16.1
+Requires-Dist: firebase-rest-api==1.11.0
+Requires-Dist: frozenlist==1.4.1
+Requires-Dist: fsspec==2024.9.0
+Requires-Dist: google-api-core==2.20.0
+Requires-Dist: google-auth==2.35.0
+Requires-Dist: google-cloud-core==2.4.1
+Requires-Dist: google-cloud-firestore==2.19.0
+Requires-Dist: google-cloud-storage==2.18.2
+Requires-Dist: google-crc32c==1.6.0
+Requires-Dist: google-resumable-media==2.7.2
+Requires-Dist: googleapis-common-protos==1.65.0
+Requires-Dist: grpcio==1.66.2
+Requires-Dist: grpcio-status==1.66.2
+Requires-Dist: hexbytes==1.2.1
+Requires-Dist: idna==3.10
+Requires-Dist: jaraco.classes==3.4.0
+Requires-Dist: jwcrypto==1.5.6
+Requires-Dist: keyring==24.3.1
+Requires-Dist: more-itertools==10.5.0
+Requires-Dist: msgpack==1.1.0
+Requires-Dist: multidict==6.1.0
+Requires-Dist: numpy==2.1.1
+Requires-Dist: packaging==24.1
+Requires-Dist: pandas==2.2.3
+Requires-Dist: parsimonious==0.10.0
+Requires-Dist: pathlib==1.0.1
+Requires-Dist: pexpect==4.9.0
+Requires-Dist: pkce==1.0.3
+Requires-Dist: pkginfo==1.11.1
+Requires-Dist: platformdirs==4.3.6
+Requires-Dist: proto-plus==1.24.0
+Requires-Dist: protobuf==5.28.2
+Requires-Dist: ptyprocess==0.7.0
+Requires-Dist: pyarrow==17.0.0
+Requires-Dist: pyasn1==0.6.1
+Requires-Dist: pyasn1_modules==0.4.1
+Requires-Dist: pycparser==2.22
+Requires-Dist: pycryptodome==3.21.0
+Requires-Dist: pydantic==2.9.2
+Requires-Dist: pydantic_core==2.23.4
+Requires-Dist: pyproject_hooks==1.2.0
+Requires-Dist: python-dateutil==2.9.0.post0
+Requires-Dist: python-jwt==4.1.0
+Requires-Dist: pytz==2024.2
+Requires-Dist: pyunormalize==16.0.0
+Requires-Dist: RapidFuzz==3.10.0
+Requires-Dist: regex==2024.9.11
+Requires-Dist: requests==2.32.3
+Requires-Dist: requests-toolbelt==1.0.0
+Requires-Dist: rlp==4.0.1
+Requires-Dist: rsa==4.9
+Requires-Dist: shellingham==1.5.4
+Requires-Dist: six==1.16.0
+Requires-Dist: tomlkit==0.13.2
+Requires-Dist: toolz==0.12.1
+Requires-Dist: trove-classifiers==2024.9.12
+Requires-Dist: types-requests==2.32.0.20240914
+Requires-Dist: typing_extensions==4.12.2
+Requires-Dist: tzdata==2024.2
+Requires-Dist: urllib3==2.2.3
+Requires-Dist: web3==7.3.0
+Requires-Dist: websockets==13.1
+Requires-Dist: xattr==1.1.0
+Requires-Dist: yarl==1.13.1
+
+# OpenGradient Python SDK
+
+Python SDK for OpenGradient 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
+```python
+og.login(email="you@opengradient.ai", password="xyz")
+```
+
+### Create a Model
+```python
+og.create_model(model_name="test-network-model", model_desc="testing upload to sdk")
+```
+
+### Create a Version of a Model
+```python
+og.create_version(model_name="test-network-model", notes="test notes")
+```
+
+### Upload Files to a Model
+```python
+og.upload(model_path="local_path_to_your_model.onnx", model_name="test-network-model", version="0.01")
+```
+
+### Run Inference
+```python
+inference_mode = og.InferenceMode.VANILLA
+inference_cid = og.infer(model_cid, model_inputs, inference_mode)
+```
+
+```python
+og.infer(model_id, inference_mode, model_input)
+```
+
+## Using the CLI
+
+#### 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 "path/to/model.onnx" "<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.8/README.md

@@ -0,0 +1,72 @@
+# OpenGradient Python SDK
+
+Python SDK for OpenGradient 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
+```python
+og.login(email="you@opengradient.ai", password="xyz")
+```
+
+### Create a Model
+```python
+og.create_model(model_name="test-network-model", model_desc="testing upload to sdk")
+```
+
+### Create a Version of a Model
+```python
+og.create_version(model_name="test-network-model", notes="test notes")
+```
+
+### Upload Files to a Model
+```python
+og.upload(model_path="local_path_to_your_model.onnx", model_name="test-network-model", version="0.01")
+```
+
+### Run Inference
+```python
+inference_mode = og.InferenceMode.VANILLA
+inference_cid = og.infer(model_cid, model_inputs, inference_mode)
+```
+
+```python
+og.infer(model_id, inference_mode, model_input)
+```
+
+## Using the CLI
+
+#### 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 "path/to/model.onnx" "<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.8/pyproject.toml

@@ -0,0 +1,142 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "opengradient"
+version = "0.2.8"
+description = "A Python SDK for OpenGradient inference services"
+authors = [{name = "OpenGradient", email = "oliver@opengradient.ai"}]
+license = {file = "LICENSE"}
+readme = "README.md"
+requires-python = ">=3.7"
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.7",
+    "Programming Language :: Python :: 3.8",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+]
+dependencies = [
+    "aiohappyeyeballs==2.4.3",
+    "aiohttp==3.10.8",
+    "aiosignal==1.3.1",
+    "annotated-types==0.7.0",
+    "attrs==24.2.0",
+    "bitarray==2.9.2",
+    "CacheControl==0.14.0",
+    "cachetools==5.5.0",
+    "certifi==2024.8.30",
+    "cffi==1.17.1",
+    "charset-normalizer==3.3.2",
+    "ckzg==2.0.1",
+    "cleo==2.1.0",
+    "click==8.1.7",
+    "cramjam==2.8.4",
+    "crashtest==0.4.1",
+    "cryptography==43.0.1",
+    "cytoolz==0.12.3",
+    "distlib==0.3.8",
+    "dulwich==0.21.7",
+    "eth-account==0.13.4",
+    "eth-hash==0.7.0",
+    "eth-keyfile==0.8.1",
+    "eth-keys==0.5.1",
+    "eth-rlp==2.1.0",
+    "eth-typing==5.0.0",
+    "eth-utils==5.0.0",
+    "eth_abi==5.1.0",
+    "fastjsonschema==2.20.0",
+    "fastparquet==2024.5.0",
+    "filelock==3.16.1",
+    "firebase-rest-api==1.11.0",
+    "frozenlist==1.4.1",
+    "fsspec==2024.9.0",
+    "google-api-core==2.20.0",
+    "google-auth==2.35.0",
+    "google-cloud-core==2.4.1",
+    "google-cloud-firestore==2.19.0",
+    "google-cloud-storage==2.18.2",
+    "google-crc32c==1.6.0",
+    "google-resumable-media==2.7.2",
+    "googleapis-common-protos==1.65.0",
+    "grpcio==1.66.2",
+    "grpcio-status==1.66.2",
+    "hexbytes==1.2.1",
+    "idna==3.10",
+    "jaraco.classes==3.4.0",
+    "jwcrypto==1.5.6",
+    "keyring==24.3.1",
+    "more-itertools==10.5.0",
+    "msgpack==1.1.0",
+    "multidict==6.1.0",
+    "numpy==2.1.1",
+    "packaging==24.1",
+    "pandas==2.2.3",
+    "parsimonious==0.10.0",
+    "pathlib==1.0.1",
+    "pexpect==4.9.0",
+    "pkce==1.0.3",
+    "pkginfo==1.11.1",
+    "platformdirs==4.3.6",
+    "proto-plus==1.24.0",
+    "protobuf==5.28.2",
+    "ptyprocess==0.7.0",
+    "pyarrow==17.0.0",
+    "pyasn1==0.6.1",
+    "pyasn1_modules==0.4.1",
+    "pycparser==2.22",
+    "pycryptodome==3.21.0",
+    "pydantic==2.9.2",
+    "pydantic_core==2.23.4",
+    "pyproject_hooks==1.2.0",
+    "python-dateutil==2.9.0.post0",
+    "python-jwt==4.1.0",
+    "pytz==2024.2",
+    "pyunormalize==16.0.0",
+    "RapidFuzz==3.10.0",
+    "regex==2024.9.11",
+    "requests==2.32.3",
+    "requests-toolbelt==1.0.0",
+    "rlp==4.0.1",
+    "rsa==4.9",
+    "shellingham==1.5.4",
+    "six==1.16.0",
+    "tomlkit==0.13.2",
+    "toolz==0.12.1",
+    "trove-classifiers==2024.9.12",
+    "types-requests==2.32.0.20240914",
+    "typing_extensions==4.12.2",
+    "tzdata==2024.2",
+    "urllib3==2.2.3",
+    "web3==7.3.0",
+    "websockets==13.1",
+    "xattr==1.1.0",
+    "yarl==1.13.1",
+]
+
+[project.scripts]
+opengradient = "opengradient.cli:cli"
+
+[project.urls]
+Homepage = "https://opengradient.ai"
+
+[tool.setuptools]
+package-dir = {"" = "src"}
+include-package-data = true
+
+[tool.setuptools.packages.find]
+where = ["src"]
+exclude = ["tests*"]
+
+[tool.setuptools.package-data]
+"*" = ["*.py", "*.abi"]
+"opengradient" = ["abi/*.abi"]
+
+[tool.setuptools.exclude-package-data]
+"*" = ["*.ipynb", "*.pyc", "*.pyo", ".gitignore", "requirements.txt", "conftest.py"]

{opengradient-0.2.6 → opengradient-0.2.8}/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
+from .defaults import *
+from .types import InferenceMode
 
-__version__ = "0.2.6"
+__version__ = "0.2.8"
 
 _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(private_key=DEFAULT_PRIVATE_KEY,
+         rpc_url=DEFAULT_RPC_URL,
+         contract_address=DEFAULT_INFERENCE_CONTRACT_ADDRESS,
+         email=DEFAULT_HUB_EMAIL,
+         password=DEFAULT_HUB_PASSWORD):
     global _client
     _client = Client(private_key=private_key, rpc_url=rpc_url, contract_address=contract_address, email=email, password=password)
 

opengradient-0.2.8/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.2.6 → opengradient-0.2.8}/src/opengradient/cli.py

@@ -5,10 +5,11 @@ import json
 import ast
 from pathlib import Path
 from .client import Client
-from opengradient.types import InferenceMode, ModelInput
+from .defaults import *
+from .types import InferenceMode, ModelInput
 
 # Environment variable names
-API_KEY_ENV = 'OPENGRADIENT_API_KEY'
+PRIVATE_KEY_ENV = 'OPENGRADIENT_PRIVATE_KEY'
 RPC_URL_ENV = 'OPENGRADIENT_RPC_URL'
 CONTRACT_ADDRESS_ENV = 'OPENGRADIENT_CONTRACT_ADDRESS'
 EMAIL_ENV = 'OPENGRADIENT_EMAIL'
@@ -39,48 +40,48 @@ Dict = DictParamType()
 
 # Support inference modes
 InferenceModes = {
-    "VANILLA": opengradient.InferenceMode.VANILLA,
-    "ZKML": opengradient.InferenceMode.ZKML,
-    "TEE": opengradient.InferenceMode.TEE,
+    "VANILLA": InferenceMode.VANILLA,
+    "ZKML": InferenceMode.ZKML,
+    "TEE": InferenceMode.TEE,
 }
 
 # TODO (Kyle): Once we're farther into development, we should remove the defaults for these options
 @click.group()
-@click.option('--api_key', 
-              envvar=API_KEY_ENV, 
+@click.option('--private_key', 
+              envvar=PRIVATE_KEY_ENV, 
               help='Your OpenGradient private key', 
-              default="cd09980ef6e280afc3900d2d6801f9e9c5d858a5deaeeab74a65643f5ff1a4c1")
+              default=DEFAULT_PRIVATE_KEY)
 @click.option('--rpc_url', 
               envvar=RPC_URL_ENV, 
               help='OpenGradient RPC URL address', 
-              default="http://18.218.115.248:8545")
+              default=DEFAULT_RPC_URL)
 @click.option('--contract_address', 
               envvar=CONTRACT_ADDRESS_ENV, 
-              help='OpenGradient contract address', 
-              default="0x350E0A430b2B1563481833a99523Cfd17a530e4e")
+              help='OpenGradient inference contract address', 
+              default=DEFAULT_INFERENCE_CONTRACT_ADDRESS)
 @click.option('--email', 
               envvar=EMAIL_ENV,
-              help='Your OpenGradient hub email address -- not required for inference', 
-              default="test@test.com")
+              help='Your OpenGradient Hub email address -- not required for inference', 
+              default=DEFAULT_HUB_EMAIL)
 @click.option('--password', 
               envvar=PASSWORD_ENV, 
-              help='Your OpenGradient hub password -- not required for inference', 
-              default="Test-123")
+              help='Your OpenGradient Hub password -- not required for inference', 
+              default=DEFAULT_HUB_PASSWORD)
 @click.pass_context
-def cli(ctx, api_key, rpc_url, contract_address, email, password):
+def cli(ctx, private_key, rpc_url, contract_address, email, password):
     """CLI for OpenGradient SDK"""
-    if not api_key:
-        click.echo("Please provide an API key via flag or setting environment variable OPENGRADIENT_API_KEY")
+    if not private_key:
+        click.echo("Please provide a private key via flag or setting environment variable OPENGRADIENT_PRIVATE_KEY")
     if not rpc_url:
         click.echo("Please provide a RPC URL via flag or setting environment variable OPENGRADIENT_RPC_URL")
     if not contract_address:
         click.echo("Please provide a contract address via flag or setting environment variable OPENGRADIENT_CONTRACT_ADDRESS")
-    if not api_key or not rpc_url or not contract_address:
+    if not private_key or not rpc_url or not contract_address:
         ctx.exit(1)
         return
 
     try:
-        ctx.obj = Client(private_key=api_key, 
+        ctx.obj = Client(private_key=private_key, 
                          rpc_url=rpc_url,
                          contract_address=contract_address,
                          email=email,
@@ -98,7 +99,7 @@ def client_settings(ctx):
         ctx.exit(1)
         
     click.echo("Settings for OpenGradient client:")
-    click.echo(f"\tAPI key ({API_KEY_ENV}): {client.private_key}")
+    click.echo(f"\tPrivate key ({PRIVATE_KEY_ENV}): {client.private_key}")
     click.echo(f"\tRPC URL ({RPC_URL_ENV}): {client.rpc_url}")
     click.echo(f"\tContract address ({CONTRACT_ADDRESS_ENV}): {client.contract_address}")
     if client.user:
@@ -185,5 +186,10 @@ def infer(ctx, model_cid, inference_mode, input_data, input_file):
     except Exception as e:
         click.echo(f"Error running inference: {str(e)}")
 
+
+@cli.command()
+def version():
+    click.echo(f"OpenGradient CLI version: {opengradient.__version__}")
+
 if __name__ == '__main__':
     cli()

{opengradient-0.2.6 → opengradient-0.2.8}/src/opengradient/client.py

@@ -43,6 +43,12 @@ class Client:
         self.firebase_app = firebase.initialize_app(self.FIREBASE_CONFIG)
         self.auth = self.firebase_app.auth()
         self.user = None
+        
+        logging.debug("Initialized client with parameters:\n"
+                      "private key: %s\n"
+                      "RPC URL: %s\n"
+                      "Contract Address: %s\n",
+                      private_key, rpc_url, contract_address)
 
         abi_path = os.path.join(os.path.dirname(__file__), 'abi', 'inference.abi')
         with open(abi_path, 'r') as abi_file:
@@ -104,20 +110,20 @@ class Client:
             response.raise_for_status()
 
             json_response = response.json()
-            model_id = json_response.get('id')
-            if not model_id:
-                raise Exception(f"Model creation response missing 'id'. Full response: {json_response}")
-            logging.info(f"Model creation successful. Model ID: {model_id}")
+            model_name = json_response.get('name')
+            if not model_name:
+                raise Exception(f"Model creation response missing 'name'. Full response: {json_response}")
+            logging.info(f"Model creation successful. Model name: {model_name}")
 
             # Create the specified version for the newly created model
             try:
-                version_response = self.create_version(model_id, version)
-                logging.info(f"Version creation successful. Version ID: {version_response['versionString']}")
+                version_response = self.create_version(model_name, version)
+                logging.info(f"Version creation successful. Version string: {version_response['versionString']}")
             except Exception as ve:
                 logging.error(f"Version creation failed, but model was created. Error: {str(ve)}")
-                return {"id": model_id, "versionString": None, "version_error": str(ve)}
+                return {"name": model_name, "versionString": None, "version_error": str(ve)}
 
-            return {"id": model_id, "versionString": version_response["versionString"]}
+            return {"name": model_name, "versionString": version_response["versionString"]}
 
         except requests.RequestException as e:
             logging.error(f"Model creation failed: {str(e)}")

opengradient-0.2.8/src/opengradient/defaults.py

@@ -0,0 +1,7 @@
+
+# Default variables
+DEFAULT_PRIVATE_KEY="cd09980ef6e280afc3900d2d6801f9e9c5d858a5deaeeab74a65643f5ff1a4c1"
+DEFAULT_RPC_URL="http://18.218.115.248:8545"
+DEFAULT_INFERENCE_CONTRACT_ADDRESS="0x75D0266DAb643417e9FFD828A1A31C1E039a966c"
+DEFAULT_HUB_EMAIL="test@test.com"
+DEFAULT_HUB_PASSWORD="Test-123"