opengradient 0.4.6__tar.gz → 0.4.7__tar.gz

opengradient-0.4.7/PKG-INFO

@@ -0,0 +1,159 @@
+Metadata-Version: 2.2
+Name: opengradient
+Version: 0.4.7
+Summary: Python SDK for OpenGradient decentralized model management & 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.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+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
+Requires-Dist: numpy>=1.26.4
+Requires-Dist: requests>=2.32.3
+Requires-Dist: langchain>=0.3.7
+Requires-Dist: openai>=1.58.1
+Requires-Dist: pydantic>=2.9.2
+
+# OpenGradient Python SDK
+
+A Python SDK for decentralized model management and inference services on the OpenGradient platform. The SDK enables programmatic access to our model repository and decentralized AI infrastructure.
+
+## Key Features
+
+- Model management and versioning
+- Decentralized model inference
+- Support for LLM inference with various models
+- End-to-end verified AI execution
+- Command-line interface (CLI) for direct access
+
+## Model Hub
+
+Browse and discover AI models on our [Model Hub](https://hub.opengradient.ai/). The Hub provides:
+- Registry of models and LLMs
+- Easy model discovery and deployment
+- Direct integration with the SDK
+
+## Installation
+
+```bash
+pip install opengradient
+```
+
+Note: Windows users should temporarily enable WSL when installing `opengradient` (fix in progress).
+
+## Getting Started
+
+### 1. Account Setup
+
+You'll need two accounts to use the SDK:
+- **Model Hub account**: Create one at [Hub Sign Up](https://hub.opengradient.ai/signup)
+- **OpenGradient account**: Use an existing Ethereum-compatible wallet or create a new one via SDK
+
+The easiest way to set up your accounts is through our configuration wizard:
+
+```bash
+opengradient config init
+```
+
+This wizard will:
+- Guide you through account creation
+- Help you set up credentials
+- Direct you to our Test Faucet for devnet tokens
+
+### 2. Initialize the SDK
+
+```python
+import opengradient as og
+og.init(private_key="<private_key>", email="<email>", password="<password>")
+```
+
+### 3. Basic Usage
+
+Browse available models on our [Model Hub](https://hub.opengradient.ai/) or create and upload your own:
+
+
+```python
+# Create and upload a model
+og.create_model(
+    model_name="my-model",
+    model_desc="Model description",
+    model_path="/path/to/model"
+)
+
+# Run inference
+inference_mode = og.InferenceMode.VANILLA
+result = og.infer(
+    model_cid="your-model-cid",
+    model_inputs={"input": "value"},
+    inference_mode=inference_mode
+)
+```
+
+### 4. Examples
+
+See code examples under [examples](./examples).
+
+## CLI Usage
+
+The SDK includes a command-line interface for quick operations. First, verify your configuration:
+
+```bash
+opengradient config show
+```
+
+Run a test inference:
+
+```bash
+opengradient infer -m QmbUqS93oc4JTLMHwpVxsE39mhNxy6hpf6Py3r9oANr8aZ \
+    --input '{"num_input1":[1.0, 2.0, 3.0], "num_input2":10}'
+```
+
+## Use Cases
+
+1. **Off-chain Applications**: Use OpenGradient as a decentralized alternative to centralized AI providers like HuggingFace and OpenAI.
+
+2. **Model Development**: Manage models on the Model Hub and integrate directly into your development workflow.
+
+## Documentation
+
+For comprehensive documentation, API reference, and examples, visit:
+- [OpenGradient Documentation](https://docs.opengradient.ai/)
+- [API Reference](https://docs.opengradient.ai/api_reference/python_sdk/)
+
+## Support
+
+- Run `opengradient --help` for CLI command reference
+- Visit our [documentation](https://docs.opengradient.ai/) for detailed guides
+- Join our [community](https://.opengradient.ai/) for support

opengradient-0.4.7/README.md

@@ -0,0 +1,111 @@
+# OpenGradient Python SDK
+
+A Python SDK for decentralized model management and inference services on the OpenGradient platform. The SDK enables programmatic access to our model repository and decentralized AI infrastructure.
+
+## Key Features
+
+- Model management and versioning
+- Decentralized model inference
+- Support for LLM inference with various models
+- End-to-end verified AI execution
+- Command-line interface (CLI) for direct access
+
+## Model Hub
+
+Browse and discover AI models on our [Model Hub](https://hub.opengradient.ai/). The Hub provides:
+- Registry of models and LLMs
+- Easy model discovery and deployment
+- Direct integration with the SDK
+
+## Installation
+
+```bash
+pip install opengradient
+```
+
+Note: Windows users should temporarily enable WSL when installing `opengradient` (fix in progress).
+
+## Getting Started
+
+### 1. Account Setup
+
+You'll need two accounts to use the SDK:
+- **Model Hub account**: Create one at [Hub Sign Up](https://hub.opengradient.ai/signup)
+- **OpenGradient account**: Use an existing Ethereum-compatible wallet or create a new one via SDK
+
+The easiest way to set up your accounts is through our configuration wizard:
+
+```bash
+opengradient config init
+```
+
+This wizard will:
+- Guide you through account creation
+- Help you set up credentials
+- Direct you to our Test Faucet for devnet tokens
+
+### 2. Initialize the SDK
+
+```python
+import opengradient as og
+og.init(private_key="<private_key>", email="<email>", password="<password>")
+```
+
+### 3. Basic Usage
+
+Browse available models on our [Model Hub](https://hub.opengradient.ai/) or create and upload your own:
+
+
+```python
+# Create and upload a model
+og.create_model(
+    model_name="my-model",
+    model_desc="Model description",
+    model_path="/path/to/model"
+)
+
+# Run inference
+inference_mode = og.InferenceMode.VANILLA
+result = og.infer(
+    model_cid="your-model-cid",
+    model_inputs={"input": "value"},
+    inference_mode=inference_mode
+)
+```
+
+### 4. Examples
+
+See code examples under [examples](./examples).
+
+## CLI Usage
+
+The SDK includes a command-line interface for quick operations. First, verify your configuration:
+
+```bash
+opengradient config show
+```
+
+Run a test inference:
+
+```bash
+opengradient infer -m QmbUqS93oc4JTLMHwpVxsE39mhNxy6hpf6Py3r9oANr8aZ \
+    --input '{"num_input1":[1.0, 2.0, 3.0], "num_input2":10}'
+```
+
+## Use Cases
+
+1. **Off-chain Applications**: Use OpenGradient as a decentralized alternative to centralized AI providers like HuggingFace and OpenAI.
+
+2. **Model Development**: Manage models on the Model Hub and integrate directly into your development workflow.
+
+## Documentation
+
+For comprehensive documentation, API reference, and examples, visit:
+- [OpenGradient Documentation](https://docs.opengradient.ai/)
+- [API Reference](https://docs.opengradient.ai/api_reference/python_sdk/)
+
+## Support
+
+- Run `opengradient --help` for CLI command reference
+- Visit our [documentation](https://docs.opengradient.ai/) for detailed guides
+- Join our [community](https://.opengradient.ai/) for support

{opengradient-0.4.6 → opengradient-0.4.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "opengradient"
-version = "0.4.6"
+version = "0.4.7"
 description = "Python SDK for OpenGradient decentralized model management & inference services"
 authors = [{name = "OpenGradient", email = "oliver@opengradient.ai"}]
 license = {file = "LICENSE"}
@@ -53,6 +53,7 @@ exclude = ["tests*", "stresstest*"]
 [tool.setuptools.package-data]
 "opengradient" = [
     "abi/*.abi",
+    "bin/*.bin",
     "proto/*.proto",
     "**/*.py"
 ]

{opengradient-0.4.6 → opengradient-0.4.7}/src/opengradient/__init__.py

@@ -6,7 +6,18 @@ from typing import Any, Dict, List, Optional, Tuple, Union
 
 from .client import Client
 from .defaults import DEFAULT_INFERENCE_CONTRACT_ADDRESS, DEFAULT_RPC_URL
-from .types import LLM, TEE_LLM, HistoricalInputQuery, InferenceMode, LlmInferenceMode, SchedulerParams
+from .types import (
+    LLM,
+    TEE_LLM,
+    HistoricalInputQuery,
+    SchedulerParams,
+    CandleType,
+    CandleOrder,
+    InferenceMode,
+    LlmInferenceMode,
+    TextGenerationOutput,
+    ModelOutput,
+)
 
 from . import llm, alphasense
 
@@ -14,15 +25,19 @@ _client = None
 
 
 def new_client(
-    email: str, password: str, private_key: str, rpc_url=DEFAULT_RPC_URL, contract_address=DEFAULT_INFERENCE_CONTRACT_ADDRESS
+    email: Optional[str],
+    password: Optional[str],
+    private_key: str,
+    rpc_url=DEFAULT_RPC_URL,
+    contract_address=DEFAULT_INFERENCE_CONTRACT_ADDRESS,
 ) -> Client:
     """
     Creates a unique OpenGradient client instance with the given authentication and network settings.
 
     Args:
-        email: User's email address for authentication
-        password: User's password for authentication
-        private_key: Ethereum private key for blockchain transactions
+        email: User's email address for authentication with Model Hub
+        password: User's password for authentication with Model Hub
+        private_key: Private key for OpenGradient transactions
         rpc_url: Optional RPC URL for the blockchain network, defaults to mainnet
         contract_address: Optional inference contract address
     """
@@ -65,7 +80,7 @@ def upload(model_path, model_name, version):
     return _client.upload(model_path, model_name, version)
 
 
-def create_model(model_name: str, model_desc: str, model_path: str = None):
+def create_model(model_name: str, model_desc: str, model_path: Optional[str] = None):
     """Create a new model repository.
 
     Args:
@@ -121,7 +136,7 @@ def infer(model_cid, inference_mode, model_input, max_retries: Optional[int] = N
         max_retries: Maximum number of retries for failed transactions
 
     Returns:
-        Tuple[str, Any]: Transaction hash and model output
+        InferenceResult: Transaction hash and model output
 
     Raises:
         RuntimeError: If SDK is not initialized
@@ -134,12 +149,12 @@ def infer(model_cid, inference_mode, model_input, max_retries: Optional[int] = N
 def llm_completion(
     model_cid: LLM,
     prompt: str,
-    inference_mode: str = LlmInferenceMode.VANILLA,
+    inference_mode: LlmInferenceMode = LlmInferenceMode.VANILLA,
     max_tokens: int = 100,
     stop_sequence: Optional[List[str]] = None,
     temperature: float = 0.0,
     max_retries: Optional[int] = None,
-) -> Tuple[str, str]:
+) -> TextGenerationOutput:
     """Generate text completion using an LLM.
 
     Args:
@@ -152,7 +167,7 @@ def llm_completion(
         max_retries: Maximum number of retries for failed transactions
 
     Returns:
-        Tuple[str, str]: Transaction hash and generated text
+        TextGenerationOutput: Transaction hash and generated text
 
     Raises:
         RuntimeError: If SDK is not initialized
@@ -173,14 +188,14 @@ def llm_completion(
 def llm_chat(
     model_cid: LLM,
     messages: List[Dict],
-    inference_mode: str = LlmInferenceMode.VANILLA,
+    inference_mode: LlmInferenceMode = LlmInferenceMode.VANILLA,
     max_tokens: int = 100,
     stop_sequence: Optional[List[str]] = None,
     temperature: float = 0.0,
     tools: Optional[List[Dict]] = None,
     tool_choice: Optional[str] = None,
     max_retries: Optional[int] = None,
-) -> Tuple[str, str, Dict]:
+) -> TextGenerationOutput:
     """Have a chat conversation with an LLM.
 
     Args:
@@ -195,7 +210,7 @@ def llm_chat(
         max_retries: Maximum number of retries for failed transactions
 
     Returns:
-        Tuple[str, str, Dict]: Transaction hash, model response, and metadata
+        TextGenerationOutput
 
     Raises:
         RuntimeError: If SDK is not initialized
@@ -215,24 +230,6 @@ def llm_chat(
     )
 
 
-def login(email: str, password: str):
-    """Login to OpenGradient.
-
-    Args:
-        email: User's email address
-        password: User's password
-
-    Returns:
-        dict: Login response with authentication tokens
-
-    Raises:
-        RuntimeError: If SDK is not initialized
-    """
-    if _client is None:
-        raise RuntimeError("OpenGradient client not initialized. Call og.init() first.")
-    return _client.login(email, password)
-
-
 def list_files(model_name: str, version: str) -> List[Dict]:
     """List files in a model repository version.
 
@@ -251,32 +248,11 @@ def list_files(model_name: str, version: str) -> List[Dict]:
     return _client.list_files(model_name, version)
 
 
-def generate_image(model: str, prompt: str, height: Optional[int] = None, width: Optional[int] = None) -> bytes:
-    """Generate an image from a text prompt.
-
-    Args:
-        model: Model identifier (e.g. "stabilityai/stable-diffusion-xl-base-1.0")
-        prompt: Text description of the desired image
-        height: Optional height of the generated image in pixels
-        width: Optional width of the generated image in pixels
-
-    Returns:
-        bytes: Raw image data as bytes
-
-    Raises:
-        RuntimeError: If SDK is not initialized
-        OpenGradientError: If image generation fails
-    """
-    if _client is None:
-        raise RuntimeError("OpenGradient client not initialized. Call og.init() first.")
-    return _client.generate_image(model, prompt, height=height, width=width)
-
-
 def new_workflow(
     model_cid: str,
-    input_query: Union[Dict[str, Any], HistoricalInputQuery],
+    input_query: HistoricalInputQuery,
     input_tensor_name: str,
-    scheduler_params: Optional[Union[Dict[str, int], SchedulerParams]] = None,
+    scheduler_params: Optional[SchedulerParams] = None,
 ) -> str:
     """
     Deploy a new workflow contract with the specified parameters.
@@ -287,13 +263,9 @@ def new_workflow(
 
     Args:
         model_cid: IPFS CID of the model
-        input_query: Dictionary or HistoricalInputQuery containing query parameters
+        input_query: HistoricalInputQuery containing query parameters
         input_tensor_name: Name of the input tensor
-        scheduler_params: Optional scheduler configuration:
-            - Can be a dictionary with:
-                - frequency: Execution frequency in seconds (default: 600)
-                - duration_hours: How long to run in hours (default: 2)
-            - Or a SchedulerParams instance
+        scheduler_params: Optional scheduler configuration as SchedulerParams instance
             If not provided, the workflow will be deployed without scheduling.
 
     Returns:
@@ -303,15 +275,12 @@ def new_workflow(
     if _client is None:
         raise RuntimeError("OpenGradient client not initialized. Call og.init(...) first.")
 
-    # Convert scheduler_params if it's a dict, otherwise use as is
-    scheduler = SchedulerParams.from_dict(scheduler_params) if isinstance(scheduler_params, dict) else scheduler_params
-
     return _client.new_workflow(
-        model_cid=model_cid, input_query=input_query, input_tensor_name=input_tensor_name, scheduler_params=scheduler
+        model_cid=model_cid, input_query=input_query, input_tensor_name=input_tensor_name, scheduler_params=scheduler_params
     )
 
 
-def read_workflow_result(contract_address: str) -> Dict[str, Union[str, Dict]]:
+def read_workflow_result(contract_address: str) -> ModelOutput:
     """
     Reads the latest inference result from a deployed workflow contract.
 
@@ -335,7 +304,7 @@ def read_workflow_result(contract_address: str) -> Dict[str, Union[str, Dict]]:
     return _client.read_workflow_result(contract_address)
 
 
-def run_workflow(contract_address: str) -> Dict[str, Union[str, Dict]]:
+def run_workflow(contract_address: str) -> ModelOutput:
     """
     Executes the workflow by calling run() on the contract to pull latest data and perform inference.
 
@@ -350,8 +319,23 @@ def run_workflow(contract_address: str) -> Dict[str, Union[str, Dict]]:
     return _client.run_workflow(contract_address)
 
 
+def read_workflow_history(contract_address: str, num_results: int) -> List[Dict]:
+    """
+    Gets historical inference results from a workflow contract.
+
+    Args:
+        contract_address (str): Address of the deployed workflow contract
+        num_results (int): Number of historical results to retrieve
+
+    Returns:
+        List[Dict]: List of historical inference results
+    """
+    if _client is None:
+        raise RuntimeError("OpenGradient client not initialized. Call og.init() first.")
+    return _client.read_workflow_history(contract_address, num_results)
+
+
 __all__ = [
-    "generate_image",
     "list_files",
     "login",
     "llm_chat",
@@ -366,6 +350,14 @@ __all__ = [
     "new_workflow",
     "read_workflow_result",
     "run_workflow",
+    "read_workflow_history",
+    "InferenceMode",
+    "LlmInferenceMode",
+    "HistoricalInputQuery",
+    "SchedulerParams",
+    "CandleType",
+    "CandleOrder",
+    "InferenceMode",
     "llm",
     "alphasense",
 ]

opengradient-0.4.7/src/opengradient/abi/PriceHistoryInference.abi

@@ -0,0 +1 @@
+[{"inputs":[{"internalType":"string","name":"_modelId","type":"string"},{"internalType":"string","name":"_inputName","type":"string"},{"components":[{"internalType":"string","name":"base","type":"string"},{"internalType":"string","name":"quote","type":"string"},{"internalType":"uint32","name":"total_candles","type":"uint32"},{"internalType":"uint32","name":"candle_duration_in_mins","type":"uint32"},{"internalType":"enum CandleOrder","name":"order","type":"uint8"},{"internalType":"enum CandleType[]","name":"candle_types","type":"uint8[]"}],"internalType":"struct HistoricalInputQuery","name":"_query","type":"tuple"}],"stateMutability":"nonpayable","type":"constructor"},{"anonymous":false,"inputs":[{"indexed":true,"internalType":"address","name":"caller","type":"address"},{"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":"result","type":"tuple"}],"name":"InferenceResultEmitted","type":"event"},{"inputs":[],"name":"getInferenceResult","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":"view","type":"function"},{"inputs":[{"internalType":"uint256","name":"num","type":"uint256"}],"name":"getLastInferenceResults","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":"view","type":"function"},{"inputs":[],"name":"historicalContract","outputs":[{"internalType":"contract OGHistorical","name":"","type":"address"}],"stateMutability":"view","type":"function"},{"inputs":[],"name":"inputName","outputs":[{"internalType":"string","name":"","type":"string"}],"stateMutability":"view","type":"function"},{"inputs":[],"name":"inputQuery","outputs":[{"internalType":"string","name":"base","type":"string"},{"internalType":"string","name":"quote","type":"string"},{"internalType":"uint32","name":"total_candles","type":"uint32"},{"internalType":"uint32","name":"candle_duration_in_mins","type":"uint32"},{"internalType":"enum CandleOrder","name":"order","type":"uint8"}],"stateMutability":"view","type":"function"},{"inputs":[],"name":"modelId","outputs":[{"internalType":"string","name":"","type":"string"}],"stateMutability":"view","type":"function"},{"inputs":[],"name":"run","outputs":[],"stateMutability":"nonpayable","type":"function"}]

opengradient-0.4.7/src/opengradient/abi/WorkflowScheduler.abi

@@ -0,0 +1,13 @@
+[ 
+    {
+        "inputs": [
+            {"internalType": "address", "name": "contractAddress", "type": "address"},
+            {"internalType": "uint256", "name": "endTime", "type": "uint256"},
+            {"internalType": "uint256", "name": "frequency", "type": "uint256"}
+        ],
+        "name": "registerTask",
+        "outputs": [],
+        "stateMutability": "nonpayable",
+        "type": "function"
+    }
+]

{opengradient-0.4.6 → opengradient-0.4.7}/src/opengradient/alphasense/read_workflow_tool.py

@@ -12,7 +12,7 @@ def create_read_workflow_tool(
     tool_name: str,
     tool_description: str,
     output_formatter: Callable[..., str] = lambda x: x,
-) -> BaseTool:
+) -> BaseTool | Callable:
     """
     Creates a tool that reads results from a workflow contract on OpenGradient.
 

{opengradient-0.4.6 → opengradient-0.4.7}/src/opengradient/alphasense/run_model_tool.py

@@ -1,5 +1,5 @@
 from enum import Enum
-from typing import Any, Callable, Dict, Type
+from typing import Any, Callable, Dict, Type, Optional
 
 from langchain_core.tools import BaseTool, StructuredTool
 from pydantic import BaseModel
@@ -14,10 +14,10 @@ def create_run_model_tool(
     tool_name: str,
     input_getter: Callable,
     output_formatter: Callable[..., str] = lambda x: x,
-    input_schema: Type[BaseModel] = None,
+    input_schema: Optional[Type[BaseModel]] = None,
     tool_description: str = "Executes the given ML model",
     inference_mode: og.InferenceMode = og.InferenceMode.VANILLA,
-) -> BaseTool:
+) -> BaseTool | Callable:
     """
     Creates a tool that wraps an OpenGradient model for inference.