opengradient 0.4.12b1__tar.gz → 0.4.13__tar.gz

{opengradient-0.4.12b1/src/opengradient.egg-info → opengradient-0.4.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opengradient
-Version: 0.4.12b1
+Version: 0.4.13
 Summary: Python SDK for OpenGradient decentralized model management & inference services
 Author-email: OpenGradient <oliver@opengradient.ai>
 License: MIT License
@@ -35,10 +35,8 @@ Classifier: Programming Language :: Python :: 3.12
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: eth-utils==2.2.2
-Requires-Dist: eth-account>=0.13.0
-Requires-Dist: web3>=6.11
-Requires-Dist: websockets>=14.1
+Requires-Dist: eth-account>=0.13.4
+Requires-Dist: web3>=7.3.0
 Requires-Dist: click>=8.1.7
 Requires-Dist: firebase-rest-api>=1.11.0
 Requires-Dist: grpcio>=1.66.2

{opengradient-0.4.12b1 → opengradient-0.4.13}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "opengradient"
-version = "0.4.12.beta1"
+version = "0.4.13"
 description = "Python SDK for OpenGradient decentralized model management & inference services"
 authors = [{name = "OpenGradient", email = "oliver@opengradient.ai"}]
 license = {file = "LICENSE"}
@@ -20,10 +20,8 @@ classifiers = [
 ]
 
 dependencies = [
-    "eth-utils==2.2.2",
-    "eth-account>=0.13.0",
-    "web3>=6.11",
-    "websockets>=14.1",
+    "eth-account>=0.13.4",
+    "web3>=7.3.0",
     "click>=8.1.7",
     "firebase-rest-api>=1.11.0",
     "grpcio>=1.66.2",

{opengradient-0.4.12b1 → opengradient-0.4.13}/src/opengradient/__init__.py

@@ -5,7 +5,7 @@ OpenGradient Python SDK for interacting with AI models and infrastructure.
 from typing import Any, Dict, List, Optional, Tuple, Union
 
 from .client import Client
-from .defaults import DEFAULT_INFERENCE_CONTRACT_ADDRESS, DEFAULT_RPC_URL
+from .defaults import DEFAULT_INFERENCE_CONTRACT_ADDRESS, DEFAULT_RPC_URL, DEFAULT_API_URL
 from .types import (
     LLM,
     TEE_LLM,
@@ -32,6 +32,7 @@ def new_client(
     password: Optional[str],
     private_key: str,
     rpc_url=DEFAULT_RPC_URL,
+    api_url=DEFAULT_API_URL,
     contract_address=DEFAULT_INFERENCE_CONTRACT_ADDRESS,
 ) -> Client:
     """
@@ -45,10 +46,10 @@ def new_client(
         contract_address: Optional inference contract address
     """
 
-    return Client(email=email, password=password, private_key=private_key, rpc_url=rpc_url, contract_address=contract_address)
+    return Client(email=email, password=password, private_key=private_key, rpc_url=rpc_url, api_url=api_url, contract_address=contract_address)
 
 
-def init(email: str, password: str, private_key: str, rpc_url=DEFAULT_RPC_URL, contract_address=DEFAULT_INFERENCE_CONTRACT_ADDRESS):
+def init(email: str, password: str, private_key: str, rpc_url=DEFAULT_RPC_URL, api_url=DEFAULT_API_URL, contract_address=DEFAULT_INFERENCE_CONTRACT_ADDRESS):
     """Initialize the OpenGradient SDK with authentication and network settings.
 
     Args:
@@ -56,11 +57,12 @@ def init(email: str, password: str, private_key: str, rpc_url=DEFAULT_RPC_URL, c
         password: User's password for authentication
         private_key: Ethereum private key for blockchain transactions
         rpc_url: Optional RPC URL for the blockchain network, defaults to mainnet
+        api_url: Optional API URL for the OpenGradient API, defaults to mainnet
         contract_address: Optional inference contract address
     """
     global _client
-
-    _client = Client(private_key=private_key, rpc_url=rpc_url, email=email, password=password, contract_address=contract_address)
+    
+    _client = Client(private_key=private_key, rpc_url=rpc_url, api_url=api_url, email=email, password=password, contract_address=contract_address)
     return _client
 
 

opengradient-0.4.13/src/opengradient/abi/InferencePrecompile.abi

@@ -0,0 +1 @@
+[{"anonymous":false,"inputs":[{"indexed":false,"internalType":"string","name":"inferenceID","type":"string"},{"components":[{"internalType":"enum LLMInferenceMode","name":"mode","type":"uint8"},{"internalType":"string","name":"modelCID","type":"string"},{"components":[{"internalType":"string","name":"role","type":"string"},{"internalType":"string","name":"content","type":"string"},{"internalType":"string","name":"name","type":"string"},{"internalType":"string","name":"tool_call_id","type":"string"},{"components":[{"internalType":"string","name":"id","type":"string"},{"internalType":"string","name":"name","type":"string"},{"internalType":"string","name":"arguments","type":"string"}],"internalType":"struct ToolCall[]","name":"tool_calls","type":"tuple[]"}],"internalType":"struct ChatMessage[]","name":"messages","type":"tuple[]"},{"components":[{"internalType":"string","name":"description","type":"string"},{"internalType":"string","name":"name","type":"string"},{"internalType":"string","name":"parameters","type":"string"}],"internalType":"struct ToolDefinition[]","name":"tools","type":"tuple[]"},{"internalType":"string","name":"tool_choice","type":"string"},{"internalType":"uint32","name":"max_tokens","type":"uint32"},{"internalType":"string[]","name":"stop_sequence","type":"string[]"},{"internalType":"uint32","name":"temperature","type":"uint32"}],"indexed":false,"internalType":"struct LLMChatRequest","name":"request","type":"tuple"},{"components":[{"internalType":"string","name":"finish_reason","type":"string"},{"components":[{"internalType":"string","name":"role","type":"string"},{"internalType":"string","name":"content","type":"string"},{"internalType":"string","name":"name","type":"string"},{"internalType":"string","name":"tool_call_id","type":"string"},{"components":[{"internalType":"string","name":"id","type":"string"},{"internalType":"string","name":"name","type":"string"},{"internalType":"string","name":"arguments","type":"string"}],"internalType":"struct ToolCall[]","name":"tool_calls","type":"tuple[]"}],"internalType":"struct ChatMessage","name":"message","type":"tuple"}],"indexed":false,"internalType":"struct LLMChatResponse","name":"response","type":"tuple"}],"name":"LLMChat","type":"event"},{"anonymous":false,"inputs":[{"indexed":false,"internalType":"string","name":"inferenceID","type":"string"},{"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"}],"indexed":false,"internalType":"struct LLMCompletionRequest","name":"request","type":"tuple"},{"components":[{"internalType":"string","name":"answer","type":"string"}],"indexed":false,"internalType":"struct LLMCompletionResponse","name":"response","type":"tuple"}],"name":"LLMCompletionEvent","type":"event"},{"anonymous":false,"inputs":[{"indexed":false,"internalType":"string","name":"inferenceID","type":"string"},{"components":[{"internalType":"enum ModelInferenceMode","name":"mode","type":"uint8"},{"internalType":"string","name":"modelCID","type":"string"},{"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":"input","type":"tuple"}],"indexed":false,"internalType":"struct ModelInferenceRequest","name":"request","type":"tuple"},{"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[]"},{"components":[{"internalType":"string","name":"name","type":"string"},{"internalType":"string","name":"value","type":"string"}],"internalType":"struct TensorLib.JsonScalar[]","name":"jsons","type":"tuple[]"},{"internalType":"bool","name":"is_simulation_result","type":"bool"}],"indexed":false,"internalType":"struct ModelOutput","name":"response","type":"tuple"}],"name":"ModelInferenceEvent","type":"event"},{"inputs":[{"components":[{"internalType":"enum LLMInferenceMode","name":"mode","type":"uint8"},{"internalType":"string","name":"modelCID","type":"string"},{"components":[{"internalType":"string","name":"role","type":"string"},{"internalType":"string","name":"content","type":"string"},{"internalType":"string","name":"name","type":"string"},{"internalType":"string","name":"tool_call_id","type":"string"},{"components":[{"internalType":"string","name":"id","type":"string"},{"internalType":"string","name":"name","type":"string"},{"internalType":"string","name":"arguments","type":"string"}],"internalType":"struct ToolCall[]","name":"tool_calls","type":"tuple[]"}],"internalType":"struct ChatMessage[]","name":"messages","type":"tuple[]"},{"components":[{"internalType":"string","name":"description","type":"string"},{"internalType":"string","name":"name","type":"string"},{"internalType":"string","name":"parameters","type":"string"}],"internalType":"struct ToolDefinition[]","name":"tools","type":"tuple[]"},{"internalType":"string","name":"tool_choice","type":"string"},{"internalType":"uint32","name":"max_tokens","type":"uint32"},{"internalType":"string[]","name":"stop_sequence","type":"string[]"},{"internalType":"uint32","name":"temperature","type":"uint32"}],"internalType":"struct LLMChatRequest","name":"request","type":"tuple"}],"name":"runLLMChat","outputs":[{"components":[{"internalType":"string","name":"finish_reason","type":"string"},{"components":[{"internalType":"string","name":"role","type":"string"},{"internalType":"string","name":"content","type":"string"},{"internalType":"string","name":"name","type":"string"},{"internalType":"string","name":"tool_call_id","type":"string"},{"components":[{"internalType":"string","name":"id","type":"string"},{"internalType":"string","name":"name","type":"string"},{"internalType":"string","name":"arguments","type":"string"}],"internalType":"struct ToolCall[]","name":"tool_calls","type":"tuple[]"}],"internalType":"struct ChatMessage","name":"message","type":"tuple"}],"internalType":"struct LLMChatResponse","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 LLMCompletionRequest","name":"request","type":"tuple"}],"name":"runLLMCompletion","outputs":[{"components":[{"internalType":"string","name":"answer","type":"string"}],"internalType":"struct LLMCompletionResponse","name":"","type":"tuple"}],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"components":[{"internalType":"enum ModelInferenceMode","name":"mode","type":"uint8"},{"internalType":"string","name":"modelCID","type":"string"},{"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":"input","type":"tuple"}],"internalType":"struct ModelInferenceRequest","name":"request","type":"tuple"}],"name":"runModelInference","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[]"},{"components":[{"internalType":"string","name":"name","type":"string"},{"internalType":"string","name":"value","type":"string"}],"internalType":"struct TensorLib.JsonScalar[]","name":"jsons","type":"tuple[]"},{"internalType":"bool","name":"is_simulation_result","type":"bool"}],"internalType":"struct ModelOutput","name":"","type":"tuple"}],"stateMutability":"nonpayable","type":"function"}]

{opengradient-0.4.12b1 → opengradient-0.4.13}/src/opengradient/cli.py

@@ -17,6 +17,7 @@ from .defaults import (
     DEFAULT_INFERENCE_CONTRACT_ADDRESS,
     DEFAULT_OG_FAUCET_URL,
     DEFAULT_RPC_URL,
+    DEFAULT_API_URL,
 )
 from .types import InferenceMode, LlmInferenceMode, LLM, TEE_LLM
 
@@ -132,6 +133,7 @@ def cli(ctx):
             ctx.obj["client"] = Client(
                 private_key=ctx.obj["private_key"],
                 rpc_url=DEFAULT_RPC_URL,
+                api_url=DEFAULT_API_URL,
                 contract_address=DEFAULT_INFERENCE_CONTRACT_ADDRESS,
                 email=ctx.obj.get("email"),
                 password=ctx.obj.get("password"),

{opengradient-0.4.12b1 → opengradient-0.4.13}/src/opengradient/client.py

@@ -2,6 +2,7 @@ import json
 import logging
 import os
 import time
+import base64
 from pathlib import Path
 from typing import Any, Dict, List, Optional, Union, Callable
 
@@ -12,6 +13,7 @@ from eth_account.account import LocalAccount
 from web3 import Web3
 from web3.exceptions import ContractLogicError
 from web3.logs import DISCARD
+import urllib.parse
 
 from .exceptions import OpenGradientError
 from .proto import infer_pb2, infer_pb2_grpc
@@ -49,6 +51,7 @@ REGULAR_TX_TIMEOUT = 30
 DEFAULT_MAX_RETRY = 5
 DEFAULT_RETRY_DELAY_SEC = 1
 
+PRECOMPILE_CONTRACT_ADDRESS = "0x00000000000000000000000000000000000000F4"
 
 class Client:
     _inference_hub_contract_address: str
@@ -56,9 +59,10 @@ class Client:
     _wallet_account: LocalAccount
 
     _hub_user: Optional[Dict]
+    _api_url: str
     _inference_abi: Dict
-
-    def __init__(self, private_key: str, rpc_url: str, contract_address: str, email: Optional[str], password: Optional[str]):
+    _precompile_abi: Dict
+    def __init__(self, private_key: str, rpc_url: str, api_url: str, contract_address: str, email: Optional[str], password: Optional[str]):
         """
         Initialize the Client with private key, RPC URL, and contract address.
 
@@ -71,12 +75,17 @@ class Client:
         """
         self._inference_hub_contract_address = contract_address
         self._blockchain = Web3(Web3.HTTPProvider(rpc_url))
+        self._api_url = api_url
         self._wallet_account = self._blockchain.eth.account.from_key(private_key)
 
         abi_path = Path(__file__).parent / "abi" / "inference.abi"
         with open(abi_path, "r") as abi_file:
             self._inference_abi = json.load(abi_file)
 
+        abi_path = Path(__file__).parent / "abi" / "InferencePrecompile.abi"
+        with open(abi_path, "r") as abi_file:
+            self._precompile_abi = json.load(abi_file)
+
         if email is not None:
             self._hub_user = self._login_to_hub(email, password)
         else:
@@ -292,6 +301,7 @@ class Client:
 
         def execute_transaction():
             contract = self._blockchain.eth.contract(address=self._inference_hub_contract_address, abi=self._inference_abi)
+            precompile_contract = self._blockchain.eth.contract(address=PRECOMPILE_CONTRACT_ADDRESS, abi=self._precompile_abi)
 
             inference_mode_uint8 = inference_mode.value
             converted_model_input = convert_to_model_input(model_input)
@@ -305,6 +315,12 @@ class Client:
 
             # TODO: This should return a ModelOutput class object
             model_output = convert_to_model_output(parsed_logs[0]["args"])
+            if len(model_output) == 0:
+                # check inference directly from node
+                parsed_logs = precompile_contract.events.ModelInferenceEvent().process_receipt(tx_receipt, errors=DISCARD)
+                inference_id = parsed_logs[0]["args"]["inferenceID"]
+                inference_result = self._get_inference_result_from_node(inference_id, inference_mode)
+                model_output = convert_to_model_output(inference_result)
 
             return InferenceResult(tx_hash.hex(), model_output)
 
@@ -956,6 +972,87 @@ class Client:
         return [convert_array_to_model_output(result) for result in results]
 
 
+    def _get_inference_result_from_node(self, inference_id: str, inference_mode: InferenceMode) -> Dict:
+        """
+        Get the inference result from node.
+        
+        Args:
+            inference_id (str): Inference id for a inference request
+            
+        Returns:
+            Dict: The inference result as returned by the node
+            
+        Raises:
+            OpenGradientError: If the request fails or returns an error
+        """
+        try:
+            encoded_id = urllib.parse.quote(inference_id, safe='')
+            url = f"{self._api_url}/artela-network/artela-rollkit/inference/tx/{encoded_id}"
+        
+            response = requests.get(url)
+            if response.status_code == 200:
+                resp = response.json()
+                inference_result = resp.get("inference_results", {})
+                if inference_result:
+                    decoded_bytes = base64.b64decode(inference_result[0])
+                    decoded_string = decoded_bytes.decode('utf-8')
+                    output = json.loads(decoded_string).get("InferenceResult",{})
+                    if output is None:
+                        raise OpenGradientError("Missing InferenceResult in inference output")
+                    
+                    match inference_mode:
+                        case InferenceMode.VANILLA:
+                            if "VanillaResult" not in output:
+                                raise OpenGradientError("Missing VanillaResult in inference output")
+                            if "model_output" not in output["VanillaResult"]:
+                                raise OpenGradientError("Missing model_output in VanillaResult")
+                            return {
+                                "output": output["VanillaResult"]["model_output"]
+                            }
+                    
+                        case InferenceMode.TEE:
+                            if "TeeNodeResult" not in output:
+                                raise OpenGradientError("Missing TeeNodeResult in inference output")
+                            if "Response" not in output["TeeNodeResult"]:
+                                raise OpenGradientError("Missing Response in TeeNodeResult")
+                            if "VanillaResponse" in output["TeeNodeResult"]["Response"]:
+                                if "model_output" not in output["TeeNodeResult"]["Response"]["VanillaResponse"]:
+                                    raise OpenGradientError("Missing model_output in VanillaResponse")
+                                return {
+                                    "output": output["TeeNodeResult"]["Response"]["VanillaResponse"]["model_output"]
+                                }
+                    
+                            else:
+                                raise OpenGradientError("Missing VanillaResponse in TeeNodeResult Response")
+                                
+                        case InferenceMode.ZKML:
+                            if "ZkmlResult" not in output:
+                                raise OpenGradientError("Missing ZkmlResult in inference output")
+                            if "model_output" not in output["ZkmlResult"]:
+                                raise OpenGradientError("Missing model_output in ZkmlResult")
+                            return {
+                                "output": output["ZkmlResult"]["model_output"]
+                            }
+                    
+                        case _:
+                            raise OpenGradientError(f"Invalid inference mode: {inference_mode}")
+                else:
+                    return None
+                    
+            else:
+                error_message = f"Failed to get inference result: HTTP {response.status_code}"
+                if response.text:
+                    error_message += f" - {response.text}"
+                logging.error(error_message)
+                raise OpenGradientError(error_message)
+                
+        except requests.RequestException as e:
+            logging.error(f"Request exception when getting inference result: {str(e)}")
+            raise OpenGradientError(f"Failed to get inference result: {str(e)}")
+        except Exception as e:
+            logging.error(f"Unexpected error when getting inference result: {str(e)}", exc_info=True)
+            raise OpenGradientError(f"Failed to get inference result: {str(e)}")
+
 def run_with_retry(txn_function: Callable, max_retries=DEFAULT_MAX_RETRY, retry_delay=DEFAULT_RETRY_DELAY_SEC):
     """
     Execute a blockchain transaction with retry logic.

{opengradient-0.4.12b1 → opengradient-0.4.13}/src/opengradient/defaults.py

@@ -1,9 +1,10 @@
 # Default variables
 DEFAULT_RPC_URL = "https://eth-devnet.opengradient.ai"
+DEFAULT_API_URL = "https://sdk-devnet.opengradient.ai"
 DEFAULT_OG_FAUCET_URL = "https://faucet.opengradient.ai/?address="
 DEFAULT_HUB_SIGNUP_URL = "https://hub.opengradient.ai/signup"
 DEFAULT_INFERENCE_CONTRACT_ADDRESS = "0x8383C9bD7462F12Eb996DD02F78234C0421A6FaE"
 DEFAULT_SCHEDULER_ADDRESS = "0x7179724De4e7FF9271FA40C0337c7f90C0508eF6"
 DEFAULT_BLOCKCHAIN_EXPLORER = "https://explorer.opengradient.ai/tx/"
 DEFAULT_IMAGE_GEN_HOST = "18.217.25.69"
-DEFAULT_IMAGE_GEN_PORT = 5125
+DEFAULT_IMAGE_GEN_PORT = 5125

{opengradient-0.4.12b1 → opengradient-0.4.13}/src/opengradient/llm/og_langchain.py

@@ -19,7 +19,7 @@ from langchain_core.runnables import Runnable
 from langchain_core.language_models.base import LanguageModelInput
 
 from opengradient import Client, LlmInferenceMode, LLM
-from opengradient.defaults import DEFAULT_INFERENCE_CONTRACT_ADDRESS, DEFAULT_RPC_URL
+from opengradient.defaults import DEFAULT_INFERENCE_CONTRACT_ADDRESS, DEFAULT_RPC_URL, DEFAULT_API_URL
 
 
 class OpenGradientChatModel(BaseChatModel):
@@ -34,7 +34,7 @@ class OpenGradientChatModel(BaseChatModel):
         super().__init__()
 
         self._client = Client(
-            private_key=private_key, rpc_url=DEFAULT_RPC_URL, contract_address=DEFAULT_INFERENCE_CONTRACT_ADDRESS, email=None, password=None
+            private_key=private_key, rpc_url=DEFAULT_RPC_URL, api_url=DEFAULT_API_URL, contract_address=DEFAULT_INFERENCE_CONTRACT_ADDRESS, email=None, password=None
         )
         self._model_cid = model_cid
         self._max_tokens = max_tokens

{opengradient-0.4.12b1 → opengradient-0.4.13}/src/opengradient/utils.py

@@ -122,24 +122,18 @@ def convert_to_model_output(event_data: AttributeDict) -> Dict[str, np.ndarray]:
     We need to reshape each output array using the shape parameter in order to get the array
     back into its original shape.
     """
-    logging.debug(f"Parsing event data: {event_data}")
-
     output_dict = {}
-
     output = event_data.get("output", {})
-    logging.debug(f"Output data: {output}")
 
-    if isinstance(output, AttributeDict):
-        # Parse numbers
+    if isinstance(output, (AttributeDict, dict)):
         for tensor in output.get("numbers", []):
-            logging.debug(f"Processing number tensor: {tensor}")
-            if isinstance(tensor, AttributeDict):
+            if isinstance(tensor, (AttributeDict, dict)):
                 name = tensor.get("name")
                 shape = tensor.get("shape")
                 values = []
                 # Convert from fixed point back into np.float32
                 for v in tensor.get("values", []):
-                    if isinstance(v, AttributeDict):
+                    if isinstance(v, (AttributeDict, dict)):
                         values.append(convert_to_float32(value=int(v.get("value")), decimals=int(v.get("decimals"))))
                     else:
                         logging.warning(f"Unexpected number type: {type(v)}")
@@ -149,8 +143,7 @@ def convert_to_model_output(event_data: AttributeDict) -> Dict[str, np.ndarray]:
 
         # Parse strings
         for tensor in output.get("strings", []):
-            logging.debug(f"Processing string tensor: {tensor}")
-            if isinstance(tensor, AttributeDict):
+            if isinstance(tensor, (AttributeDict, dict)):
                 name = tensor.get("name")
                 shape = tensor.get("shape")
                 values = tensor.get("values", [])
@@ -160,8 +153,7 @@ def convert_to_model_output(event_data: AttributeDict) -> Dict[str, np.ndarray]:
 
         # Parse JSON dicts
         for tensor in output.get("jsons", []):
-            logging.debug(f"Processing JSON tensor: {tensor}")
-            if isinstance(tensor, AttributeDict):
+            if isinstance(tensor, (AttributeDict, dict)):
                 name = tensor.get("name")
                 value = tensor.get("value")
                 output_dict[name] = np.array(json.loads(value))
@@ -172,7 +164,6 @@ def convert_to_model_output(event_data: AttributeDict) -> Dict[str, np.ndarray]:
         logging.warning(f"Unexpected output type: {type(output)}")
 
     logging.debug(f"Parsed output: {output_dict}")
-
     return output_dict
 
 

{opengradient-0.4.12b1 → opengradient-0.4.13/src/opengradient.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opengradient
-Version: 0.4.12b1
+Version: 0.4.13
 Summary: Python SDK for OpenGradient decentralized model management & inference services
 Author-email: OpenGradient <oliver@opengradient.ai>
 License: MIT License
@@ -35,10 +35,8 @@ Classifier: Programming Language :: Python :: 3.12
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: eth-utils==2.2.2
-Requires-Dist: eth-account>=0.13.0
-Requires-Dist: web3>=6.11
-Requires-Dist: websockets>=14.1
+Requires-Dist: eth-account>=0.13.4
+Requires-Dist: web3>=7.3.0
 Requires-Dist: click>=8.1.7
 Requires-Dist: firebase-rest-api>=1.11.0
 Requires-Dist: grpcio>=1.66.2

{opengradient-0.4.12b1 → opengradient-0.4.13}/src/opengradient.egg-info/SOURCES.txt

@@ -15,6 +15,7 @@ src/opengradient.egg-info/dependency_links.txt
 src/opengradient.egg-info/entry_points.txt
 src/opengradient.egg-info/requires.txt
 src/opengradient.egg-info/top_level.txt
+src/opengradient/abi/InferencePrecompile.abi
 src/opengradient/abi/PriceHistoryInference.abi
 src/opengradient/abi/WorkflowScheduler.abi
 src/opengradient/abi/inference.abi

{opengradient-0.4.12b1 → opengradient-0.4.13}/src/opengradient.egg-info/requires.txt

@@ -1,7 +1,5 @@
-eth-utils==2.2.2
-eth-account>=0.13.0
-web3>=6.11
-websockets>=14.1
+eth-account>=0.13.4
+web3>=7.3.0
 click>=8.1.7
 firebase-rest-api>=1.11.0
 grpcio>=1.66.2