opengradient 0.3.4__tar.gz → 0.3.5__tar.gz

{opengradient-0.3.4/src/opengradient.egg-info → opengradient-0.3.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: opengradient
-Version: 0.3.4
+Version: 0.3.5
 Summary: Python SDK for OpenGradient decentralized model management & inference services
 Author-email: OpenGradient <oliver@opengradient.ai>
 License: MIT License

{opengradient-0.3.4 → opengradient-0.3.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "opengradient"
-version = "0.3.4"
+version = "0.3.5"
 description = "Python SDK for OpenGradient decentralized model management & inference services"
 authors = [{name = "OpenGradient", email = "oliver@opengradient.ai"}]
 license = {file = "LICENSE"}
@@ -132,11 +132,11 @@ include-package-data = true
 
 [tool.setuptools.packages.find]
 where = ["src"]
-exclude = ["tests*"]
+exclude = ["tests*", "stresstest*"]
 
 [tool.setuptools.package-data]
 "*" = ["*.py", "*.abi"]
 "opengradient" = ["abi/*.abi"]
 
 [tool.setuptools.exclude-package-data]
-"*" = ["*.ipynb", "*.pyc", "*.pyo", ".gitignore", "requirements.txt", "conftest.py"]
+"*" = ["*.ipynb", "*.pyc", "*.pyo", ".gitignore", "requirements.txt", "conftest.py"]

{opengradient-0.3.4 → opengradient-0.3.5}/src/opengradient/__init__.py

@@ -2,7 +2,10 @@ from .client import Client
 from .defaults import *
 from .types import InferenceMode
 from typing import List, Dict, Optional, Tuple
-__version__ = "0.3.4"
+import os
+
+__version__ = "0.3.5"
+__all__ = ['init', 'upload', 'create_model', 'create_version', 'infer', 'infer_llm', 'login', 'list_files', 'InferenceMode']
 
 _client = None
 
@@ -59,4 +62,14 @@ def login(email: str, password: str):
 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)
+    return _client.list_files(model_name, version)
+
+def create_model_from_huggingface(repo_id: str, model_name: str, model_desc: str):
+    if _client is None:
+        raise RuntimeError("OpenGradient client not initialized. Call og.init() first.")
+
+    with tempfile.TemporaryDirectory() as temp_dir:
+        snapshot_download(repo_id, local_dir=temp_dir)
+        result = create_model(model_name, model_desc, temp_dir)
+    
+    return result

{opengradient-0.3.4 → opengradient-0.3.5}/src/opengradient/cli.py

@@ -118,7 +118,7 @@ def cli(ctx):
         try:
             ctx.obj['client'] = Client(private_key=ctx.obj['private_key'], 
                                         rpc_url=ctx.obj['rpc_url'],
-                                        contract_address=ctx.obj['contract_address'],
+                                        contract_address=DEFAULT_INFERENCE_CONTRACT_ADDRESS,
                                         email=ctx.obj.get('email'),
                                         password=ctx.obj.get('password'))
         except Exception as e:
@@ -301,12 +301,23 @@ def infer(ctx, model_cid: str, inference_mode: str, input_data, input_file: Path
             with input_file.open('r') as file:
                 model_input = json.load(file)
             
-        click.echo(f"Running {inference_mode} inference for model \"{model_cid}\"\n")
+        click.echo(f"Running {inference_mode} inference for model \"{model_cid}\"")
         tx_hash, model_output = client.infer(model_cid=model_cid, inference_mode=InferenceModes[inference_mode], model_input=model_input)
 
-        click.secho("Success!", fg="green")
-        click.echo(f"Transaction hash: {tx_hash}")
-        click.echo(f"Inference result:\n{pformat(model_output, indent=2, width=120)}")
+        click.echo()  # Add a newline for better spacing
+        click.secho("✅ Transaction successful", fg="green", bold=True)
+        click.echo("──────────────────────────────────────")
+        click.echo(f"Transaction hash: ", nl=False)
+        click.secho(tx_hash, fg="cyan", bold=True)
+
+        block_explorer_link = f"{DEFAULT_BLOCKCHAIN_EXPLORER}0x{tx_hash}"
+        click.echo(f"Block explorer link: ", nl=False)
+        click.secho(block_explorer_link, fg="blue", underline=True)
+        click.echo()
+
+        click.secho("Inference result:", fg="green")
+        formatted_output = json.dumps(model_output, indent=2, default=lambda x: x.tolist() if hasattr(x, 'tolist') else str(x))
+        click.echo(formatted_output)
     except json.JSONDecodeError as e:
         click.echo(f"Error decoding JSON: {e}", err=True)
         click.echo(f"Error occurred on line {e.lineno}, column {e.colno}", err=True)
@@ -343,12 +354,26 @@ def llm(ctx, model_cid: str, prompt: str, max_tokens: int, stop_sequence: List[s
             temperature=temperature
         )
 
-        click.secho("Success!", fg="green")
-        click.echo(f"Transaction hash: {tx_hash}")
-        click.echo(f"LLM output:\n{llm_output}")
+        print_llm_inference_result(model_cid, tx_hash, llm_output)
     except Exception as e:
         click.echo(f"Error running LLM inference: {str(e)}")
 
+def print_llm_inference_result(model_cid, tx_hash, llm_output):
+    click.secho(f"✅ LLM Inference Successful", fg="green", bold=True)
+    click.echo("──────────────────────────────────────")
+    click.echo(f"Model CID: ", nl=False)
+    click.secho(model_cid, fg="cyan", bold=True)
+    click.echo(f"Transaction hash: ", nl=False)
+    click.secho(tx_hash, fg="cyan", bold=True)
+    block_explorer_link = f"{DEFAULT_BLOCKCHAIN_EXPLORER}0x{tx_hash}"
+    click.echo(f"Block explorer link: ", nl=False)
+    click.secho(block_explorer_link, fg="blue", underline=True)
+    click.echo("──────────────────────────────────────")
+    click.secho("LLM Output:", fg="yellow", bold=True)
+    click.echo()
+    click.echo(llm_output)
+    click.echo()
+
 @cli.command()
 def create_account():
     """Create a new test account for OpenGradient inference and model management"""

{opengradient-0.3.4 → opengradient-0.3.5}/src/opengradient/defaults.py

@@ -3,4 +3,5 @@
 DEFAULT_RPC_URL="http://18.218.115.248:8545"
 DEFAULT_OG_FAUCET_URL="http://18.218.115.248:8080/?address="
 DEFAULT_HUB_SIGNUP_URL="https://hub.opengradient.ai/signup"
-DEFAULT_INFERENCE_CONTRACT_ADDRESS="0x350E0A430b2B1563481833a99523Cfd17a530e4e"
+DEFAULT_INFERENCE_CONTRACT_ADDRESS="0x24Ec56879245C707220Af7234d2fF3F22cA9Aa63"
+DEFAULT_BLOCKCHAIN_EXPLORER="http://3.145.62.2/tx/"

{opengradient-0.3.4 → opengradient-0.3.5/src/opengradient.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: opengradient
-Version: 0.3.4
+Version: 0.3.5
 Summary: Python SDK for OpenGradient decentralized model management & inference services
 Author-email: OpenGradient <oliver@opengradient.ai>
 License: MIT License