opengradient 0.3.3__py3-none-any.whl → 0.3.5__py3-none-any.whl

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.3"
+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/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/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.3.dist-info → opengradient-0.3.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: opengradient
-Version: 0.3.3
+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.3.dist-info → opengradient-0.3.5.dist-info}/RECORD

@@ -1,16 +1,16 @@
-opengradient/__init__.py,sha256=91_tgWyaYIeOUxbE6h3F1IreQbFd9-zuRft3rnsgylY,2447
+opengradient/__init__.py,sha256=sRZyD9zh_QZFC8H9P8T8pOL5l_ebe3I7hn_gTbTzr6I,2976
 opengradient/account.py,sha256=s1C4hAtc8vcHObWjwxwlYJA041S6DTbr7-rK6qiWPsQ,1149
-opengradient/cli.py,sha256=T59Z2S3AsMVS6TLgsSgxW9esssvYP5ZmVJrYE6p4oW4,16105
+opengradient/cli.py,sha256=gbYJcfLWSZRQ8-xZ2OM0PbicutVGclMiCvCFIQxkvD0,17548
 opengradient/client.py,sha256=DCDp2EWPF62ZQnx2_cM0wPghRxgn213VnR65R8yZBVY,23964
-opengradient/defaults.py,sha256=pDfsmPoUzdLG55n-hwh0CMBFxKR2rdNcjqCcwTWc6iw,267
+opengradient/defaults.py,sha256=YI84_wWTvWxPMQIuKiSair2wffATnhitE3Ll2P1jHMU,319
 opengradient/exceptions.py,sha256=v4VmUGTvvtjhCZAhR24Ga42z3q-DzR1Y5zSqP_yn2Xk,3366
 opengradient/types.py,sha256=EoJN-DkQrJ2WTUv8OenlrlWJWFY2jPGTl-T8C_OVjp8,1849
 opengradient/utils.py,sha256=F1Nj-GMNFQFxCtbGgWQq1RP4TSurbpQxJV3yKeEo1b0,6482
 opengradient/abi/inference.abi,sha256=u8FsW0s1YeRjUb9eLS1k_qh_5f_cwOdr0bii-tAdxh0,2683
 opengradient/abi/llm.abi,sha256=zhiPFyBT09EI3QU5DVoKHo7e8T9PFcfIQ3RHDYetm4M,3609
-opengradient-0.3.3.dist-info/LICENSE,sha256=xEcvQ3AxZOtDkrqkys2Mm6Y9diEnaSeQRKvxi-JGnNA,1069
-opengradient-0.3.3.dist-info/METADATA,sha256=80TVQCE8tC5JX7ZuFdqXu5IdFbuCmo5hi1g96eGrCQA,7805
-opengradient-0.3.3.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-opengradient-0.3.3.dist-info/entry_points.txt,sha256=yUKTaJx8RXnybkob0J62wVBiCp_1agVbgw9uzsmaeJc,54
-opengradient-0.3.3.dist-info/top_level.txt,sha256=oC1zimVLa2Yi1LQz8c7x-0IQm92milb5ax8gHBHwDqU,13
-opengradient-0.3.3.dist-info/RECORD,,
+opengradient-0.3.5.dist-info/LICENSE,sha256=xEcvQ3AxZOtDkrqkys2Mm6Y9diEnaSeQRKvxi-JGnNA,1069
+opengradient-0.3.5.dist-info/METADATA,sha256=jqfpyRml-74KsXuOK-b0nA9RyU9jOS4vUbizBTbr38k,7805
+opengradient-0.3.5.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
+opengradient-0.3.5.dist-info/entry_points.txt,sha256=yUKTaJx8RXnybkob0J62wVBiCp_1agVbgw9uzsmaeJc,54
+opengradient-0.3.5.dist-info/top_level.txt,sha256=oC1zimVLa2Yi1LQz8c7x-0IQm92milb5ax8gHBHwDqU,13
+opengradient-0.3.5.dist-info/RECORD,,

{opengradient-0.3.3.dist-info → opengradient-0.3.5.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.1.0)
+Generator: setuptools (75.2.0)
 Root-Is-Purelib: true
 Tag: py3-none-any