opengradient 0.3.22__tar.gz → 0.3.24__tar.gz

{opengradient-0.3.22/src/opengradient.egg-info → opengradient-0.3.24}/PKG-INFO

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

{opengradient-0.3.22 → opengradient-0.3.24}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "opengradient"
-version = "0.3.22"
+version = "0.3.24"
 description = "Python SDK for OpenGradient decentralized model management & inference services"
 authors = [{name = "OpenGradient", email = "oliver@opengradient.ai"}]
 license = {file = "LICENSE"}

{opengradient-0.3.22 → opengradient-0.3.24}/src/opengradient/__init__.py

@@ -5,7 +5,7 @@ from typing import Dict, List, Optional, Tuple, Any, Union
 from pathlib import Path
 from .client import Client
 from .defaults import DEFAULT_INFERENCE_CONTRACT_ADDRESS, DEFAULT_RPC_URL
-from .types import InferenceMode, LlmInferenceMode, LLM, TEE_LLM
+from .types import HistoricalInputQuery, InferenceMode, LlmInferenceMode, LLM, TEE_LLM, SchedulerParams
 from . import llm
 from . import mltools
 
@@ -247,23 +247,44 @@ def generate_image(model: str, prompt: str, height: Optional[int] = None, width:
 
 def new_workflow(
     model_cid: str,
-    input_query: Dict[str, Any],
-    input_tensor_name: str
+    input_query: Union[Dict[str, Any], HistoricalInputQuery],
+    input_tensor_name: str,
+    scheduler_params: Optional[Union[Dict[str, int], SchedulerParams]] = None
 ) -> str:
     """
     Deploy a new workflow contract with the specified parameters.
     
+    This function deploys a new workflow contract and optionally registers it with 
+    the scheduler for automated execution. If scheduler_params is not provided, 
+    the workflow will be deployed without automated execution scheduling.
+    
     Args:
         model_cid: IPFS CID of the model
-        input_query: Dictionary containing query parameters
+        input_query: Dictionary or 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
+            If not provided, the workflow will be deployed without scheduling.
     
     Returns:
-        str: Deployed contract address
+        str: Deployed contract address. If scheduler_params was provided, the workflow
+             will be automatically executed according to the specified schedule.
     """
     if _client is None:
         raise RuntimeError("OpenGradient client not initialized. Call og.init(...) first.")
-    return _client.new_workflow(model_cid, input_query, input_tensor_name)
+    
+    # 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
+    )
 
 def read_workflow_result(contract_address: str) -> Dict[str, Union[str, Dict]]:
     """
@@ -314,7 +335,7 @@ __all__ = [
     'upload',
     'init',
     'LLM',
-    'TEE_LLM'
+    'TEE_LLM',
     'new_workflow',
     'read_workflow_result',
     'run_workflow'
@@ -329,6 +350,6 @@ __pdoc__ = {
     'llm': True,
     'mltools': True,
     'proto': False,
-    'types': False,
+    'types': True,
     'utils': False
 }

{opengradient-0.3.22 → opengradient-0.3.24}/src/opengradient/client.py

@@ -20,7 +20,8 @@ from opengradient.types import (
     LlmInferenceMode, 
     LLM, 
     TEE_LLM,
-    ModelOutput
+    ModelOutput,
+    SchedulerParams
 )
 
 import grpc
@@ -812,18 +813,11 @@ class Client:
         self,
         model_cid: str,
         input_query: Union[Dict[str, Any], HistoricalInputQuery],
-        input_tensor_name: str
+        input_tensor_name: str,
+        scheduler_params: Optional[SchedulerParams] = None
     ) -> str:
         """
         Deploy a new workflow contract with the specified parameters.
-        
-        Args:
-            model_cid: IPFS CID of the model
-            input_query: Either a HistoricalInputQuery object or dictionary containing query parameters
-            input_tensor_name: Name of the input tensor
-        
-        Returns:
-            str: Deployed contract address
         """
         if isinstance(input_query, dict):
             input_query = HistoricalInputQuery.from_dict(input_query)
@@ -835,6 +829,8 @@ class Client:
         with open(bin_path, 'r') as f:
             bytecode = f.read().strip()
         
+        print("📦 Deploying workflow contract...")
+        
         # Create contract instance
         contract = self._w3.eth.contract(abi=abi, bytecode=bytecode)
         
@@ -851,12 +847,66 @@ class Client:
             'gasPrice': self._w3.eth.gas_price,
             'chainId': self._w3.eth.chain_id
         })
-        
+
         signed_txn = self._w3.eth.account.sign_transaction(transaction, self.private_key)
         tx_hash = self._w3.eth.send_raw_transaction(signed_txn.raw_transaction)
         tx_receipt = self._w3.eth.wait_for_transaction_receipt(tx_hash)
+        contract_address = tx_receipt.contractAddress
         
-        return tx_receipt.contractAddress
+        print(f"✅ Workflow contract deployed at: {contract_address}")
+
+        # Register with scheduler if params provided
+        if scheduler_params:
+            print("\n⏰ Setting up automated execution schedule...")
+            print(f"   • Frequency: Every {scheduler_params.frequency} seconds")
+            print(f"   • Duration: {scheduler_params.duration_hours} hours")
+            print(f"   • End Time: {time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(scheduler_params.end_time))}")
+            
+            scheduler_abi = [{
+                "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"
+            }]
+
+            scheduler_address = "0xE81a54399CFDf551bB917d0427464fE54537D245"
+            scheduler_contract = self._w3.eth.contract(
+                address=scheduler_address,
+                abi=scheduler_abi
+            )
+
+            try:
+                # Register the workflow with the scheduler
+                scheduler_tx = scheduler_contract.functions.registerTask(
+                    contract_address,
+                    scheduler_params.end_time,
+                    scheduler_params.frequency
+                ).build_transaction({
+                    'from': self.wallet_address,
+                    'gas': 300000,
+                    'gasPrice': self._w3.eth.gas_price,
+                    'nonce': self._w3.eth.get_transaction_count(self.wallet_address, 'pending'),
+                    'chainId': self._w3.eth.chain_id
+                })
+
+                signed_scheduler_tx = self._w3.eth.account.sign_transaction(scheduler_tx, self.private_key)
+                scheduler_tx_hash = self._w3.eth.send_raw_transaction(signed_scheduler_tx.raw_transaction)
+                self._w3.eth.wait_for_transaction_receipt(scheduler_tx_hash)
+                
+                print("✅ Automated execution schedule set successfully!")
+                print(f"   Transaction hash: {scheduler_tx_hash.hex()}")
+                
+            except Exception as e:
+                print("❌ Failed to set up automated execution schedule")
+                print(f"   Error: {str(e)}")
+                print("   The workflow contract is still deployed and can be executed manually.")
+
+        return contract_address
 
     def read_workflow_result(self, contract_address: str) -> Any:
         """

{opengradient-0.3.22 → opengradient-0.3.24}/src/opengradient/types.py

@@ -1,6 +1,7 @@
 from dataclasses import dataclass
-from typing import List, Tuple, Union
+from typing import List, Tuple, Union, Dict, Optional
 from enum import Enum, IntEnum
+import time
 
 class CandleOrder(IntEnum):
     ASCENDING = 0
@@ -132,4 +133,22 @@ class LLM(str, Enum):
 class TEE_LLM(str, Enum):
     """Enum for LLM models available for TEE execution"""
 
-    META_LLAMA_3_1_70B_INSTRUCT = "meta-llama/Llama-3.1-70B-Instruct"
+    META_LLAMA_3_1_70B_INSTRUCT = "meta-llama/Llama-3.1-70B-Instruct"
+
+@dataclass
+class SchedulerParams:
+    frequency: int
+    duration_hours: int
+
+    @property
+    def end_time(self) -> int:
+        return int(time.time()) + (self.duration_hours * 60 * 60)
+
+    @staticmethod
+    def from_dict(data: Optional[Dict[str, int]]) -> Optional['SchedulerParams']:
+        if data is None:
+            return None
+        return SchedulerParams(
+            frequency=data.get('frequency', 600),
+            duration_hours=data.get('duration_hours', 2)
+        )

{opengradient-0.3.22 → opengradient-0.3.24/src/opengradient.egg-info}/PKG-INFO

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