opengradient 0.3.22__py3-none-any.whl → 0.3.23__py3-none-any.whl

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'

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/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.dist-info → opengradient-0.3.23.dist-info}/METADATA

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

{opengradient-0.3.22.dist-info → opengradient-0.3.23.dist-info}/RECORD

@@ -1,10 +1,10 @@
-opengradient/__init__.py,sha256=g6uHW_AlVFF_Pdo0v5VFC29iEEPDHQfYl-HzBZVJPDI,11524
+opengradient/__init__.py,sha256=uU_Onxf-4JMQ9gHMQPbJDbU93kbB8qNayBLrnZnRFzg,12742
 opengradient/account.py,sha256=2B7rtCXQDX-yF4U69h8B9-OUreJU4IqoGXG_1Hn9nWs,1150
 opengradient/cli.py,sha256=vYE_zr1oKYOkiAEWRyW3cR6jJQr6xKterjI1grhmROg,26399
-opengradient/client.py,sha256=WUXt-XJygaH6F5oC1LtEC4uk_TJC7-55mCaeYzDJ3aY,40164
+opengradient/client.py,sha256=Dybu7g5kP7rJKQq0NbNjamJUVNTYLdSFdEd3ZA6fiaQ,42668
 opengradient/defaults.py,sha256=6tsW9Z84z6YtITCsULTTgnN0KRUZjfSoeWJZqdWkYCo,384
 opengradient/exceptions.py,sha256=v4VmUGTvvtjhCZAhR24Ga42z3q-DzR1Y5zSqP_yn2Xk,3366
-opengradient/types.py,sha256=Lw-SF2GwTVyMu7_R5dql_2VsPNoGNPL3syfHs18X75s,3658
+opengradient/types.py,sha256=Z8HFmCxNiC4Pr8qVoALF_jYOAqIJuKsiJSgq96q-sxQ,4175
 opengradient/utils.py,sha256=lUDPmyPqLwpZI-owyN6Rm3QvUjOn5pLN5G1QyriVm-E,6994
 opengradient/abi/ModelExecutorHistorical.abi,sha256=Z-2CQRG2BO5R-Ix1OLgQD9GCcHZw8w6-Zth3rzNX0TU,3278
 opengradient/abi/inference.abi,sha256=MR5u9npZ-Yx2EqRW17_M-UnGgFF3mMEMepOwaZ-Bkgc,7040
@@ -18,9 +18,9 @@ opengradient/proto/__init__.py,sha256=AhaSmrqV0TXGzCKaoPV8-XUvqs2fGAJBM2aOmDpkNb
 opengradient/proto/infer.proto,sha256=13eaEMcppxkBF8yChptsX9HooWFwJKze7oLZNl-LEb8,1217
 opengradient/proto/infer_pb2.py,sha256=wg2vjLQCNv6HRhYuIqgj9xivi3nO4IPz6E5wh2dhDqY,3446
 opengradient/proto/infer_pb2_grpc.py,sha256=y5GYwD1EdNs892xx58jdfyA0fO5QC7k3uZOtImTHMiE,6891
-opengradient-0.3.22.dist-info/LICENSE,sha256=xEcvQ3AxZOtDkrqkys2Mm6Y9diEnaSeQRKvxi-JGnNA,1069
-opengradient-0.3.22.dist-info/METADATA,sha256=q2tugCQdfo-675y6Z8ETQIP87Yh6ThNfrbQO4DaGQtw,8754
-opengradient-0.3.22.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-opengradient-0.3.22.dist-info/entry_points.txt,sha256=yUKTaJx8RXnybkob0J62wVBiCp_1agVbgw9uzsmaeJc,54
-opengradient-0.3.22.dist-info/top_level.txt,sha256=oC1zimVLa2Yi1LQz8c7x-0IQm92milb5ax8gHBHwDqU,13
-opengradient-0.3.22.dist-info/RECORD,,
+opengradient-0.3.23.dist-info/LICENSE,sha256=xEcvQ3AxZOtDkrqkys2Mm6Y9diEnaSeQRKvxi-JGnNA,1069
+opengradient-0.3.23.dist-info/METADATA,sha256=EhqCC5WEHRypug84TG0flHH-MsMDKKRN9B5D2DydaJY,8754
+opengradient-0.3.23.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+opengradient-0.3.23.dist-info/entry_points.txt,sha256=yUKTaJx8RXnybkob0J62wVBiCp_1agVbgw9uzsmaeJc,54
+opengradient-0.3.23.dist-info/top_level.txt,sha256=oC1zimVLa2Yi1LQz8c7x-0IQm92milb5ax8gHBHwDqU,13
+opengradient-0.3.23.dist-info/RECORD,,