opengradient 0.3.24__py3-none-any.whl → 0.3.26__py3-none-any.whl

opengradient/proto/infer_pb2.py

@@ -4,47 +4,42 @@
 # source: infer.proto
 # Protobuf Python Version: 5.27.2
 """Generated protocol buffer code."""
+
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
 from google.protobuf import runtime_version as _runtime_version
 from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
-_runtime_version.ValidateProtobufRuntimeVersion(
-    _runtime_version.Domain.PUBLIC,
-    5,
-    27,
-    2,
-    '',
-    'infer.proto'
-)
+
+_runtime_version.ValidateProtobufRuntimeVersion(_runtime_version.Domain.PUBLIC, 5, 27, 2, "", "infer.proto")
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
 
 
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0binfer.proto\x12\tinference\"[\n\x10InferenceRequest\x12\n\n\x02tx\x18\x01 \x01(\t\x12;\n\x10image_generation\x18\x06 \x01(\x0b\x32!.inference.ImageGenerationRequest\"u\n\x16ImageGenerationRequest\x12\r\n\x05model\x18\x01 \x01(\t\x12\x0e\n\x06prompt\x18\x02 \x01(\t\x12\x13\n\x06height\x18\x03 \x01(\x05H\x00\x88\x01\x01\x12\x12\n\x05width\x18\x04 \x01(\x05H\x01\x88\x01\x01\x42\t\n\x07_heightB\x08\n\x06_width\"\x1b\n\rInferenceTxId\x12\n\n\x02id\x18\x01 \x01(\t\"\xd4\x01\n\x0fInferenceStatus\x12\x31\n\x06status\x18\x01 \x01(\x0e\x32!.inference.InferenceStatus.Status\x12\x1a\n\rerror_message\x18\x02 \x01(\tH\x00\x88\x01\x01\"`\n\x06Status\x12\x16\n\x12STATUS_UNSPECIFIED\x10\x00\x12\x16\n\x12STATUS_IN_PROGRESS\x10\x01\x12\x14\n\x10STATUS_COMPLETED\x10\x02\x12\x10\n\x0cSTATUS_ERROR\x10\x03\x42\x10\n\x0e_error_message\"\xa4\x01\n\x0fInferenceResult\x12\x43\n\x17image_generation_result\x18\x05 \x01(\x0b\x32\".inference.ImageGenerationResponse\x12\x17\n\npublic_key\x18\x07 \x01(\tH\x00\x88\x01\x01\x12\x16\n\tsignature\x18\x08 \x01(\tH\x01\x88\x01\x01\x42\r\n\x0b_public_keyB\x0c\n\n_signature\"-\n\x17ImageGenerationResponse\x12\x12\n\nimage_data\x18\x01 \x01(\x0c\x32\xf6\x01\n\x10InferenceService\x12J\n\x11RunInferenceAsync\x12\x1b.inference.InferenceRequest\x1a\x18.inference.InferenceTxId\x12J\n\x12GetInferenceStatus\x12\x18.inference.InferenceTxId\x1a\x1a.inference.InferenceStatus\x12J\n\x12GetInferenceResult\x12\x18.inference.InferenceTxId\x1a\x1a.inference.InferenceResultb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
+    b'\n\x0binfer.proto\x12\tinference"[\n\x10InferenceRequest\x12\n\n\x02tx\x18\x01 \x01(\t\x12;\n\x10image_generation\x18\x06 \x01(\x0b\x32!.inference.ImageGenerationRequest"u\n\x16ImageGenerationRequest\x12\r\n\x05model\x18\x01 \x01(\t\x12\x0e\n\x06prompt\x18\x02 \x01(\t\x12\x13\n\x06height\x18\x03 \x01(\x05H\x00\x88\x01\x01\x12\x12\n\x05width\x18\x04 \x01(\x05H\x01\x88\x01\x01\x42\t\n\x07_heightB\x08\n\x06_width"\x1b\n\rInferenceTxId\x12\n\n\x02id\x18\x01 \x01(\t"\xd4\x01\n\x0fInferenceStatus\x12\x31\n\x06status\x18\x01 \x01(\x0e\x32!.inference.InferenceStatus.Status\x12\x1a\n\rerror_message\x18\x02 \x01(\tH\x00\x88\x01\x01"`\n\x06Status\x12\x16\n\x12STATUS_UNSPECIFIED\x10\x00\x12\x16\n\x12STATUS_IN_PROGRESS\x10\x01\x12\x14\n\x10STATUS_COMPLETED\x10\x02\x12\x10\n\x0cSTATUS_ERROR\x10\x03\x42\x10\n\x0e_error_message"\xa4\x01\n\x0fInferenceResult\x12\x43\n\x17image_generation_result\x18\x05 \x01(\x0b\x32".inference.ImageGenerationResponse\x12\x17\n\npublic_key\x18\x07 \x01(\tH\x00\x88\x01\x01\x12\x16\n\tsignature\x18\x08 \x01(\tH\x01\x88\x01\x01\x42\r\n\x0b_public_keyB\x0c\n\n_signature"-\n\x17ImageGenerationResponse\x12\x12\n\nimage_data\x18\x01 \x01(\x0c\x32\xf6\x01\n\x10InferenceService\x12J\n\x11RunInferenceAsync\x12\x1b.inference.InferenceRequest\x1a\x18.inference.InferenceTxId\x12J\n\x12GetInferenceStatus\x12\x18.inference.InferenceTxId\x1a\x1a.inference.InferenceStatus\x12J\n\x12GetInferenceResult\x12\x18.inference.InferenceTxId\x1a\x1a.inference.InferenceResultb\x06proto3'
+)
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'infer_pb2', _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "infer_pb2", _globals)
 if not _descriptor._USE_C_DESCRIPTORS:
-  DESCRIPTOR._loaded_options = None
-  _globals['_INFERENCEREQUEST']._serialized_start=26
-  _globals['_INFERENCEREQUEST']._serialized_end=117
-  _globals['_IMAGEGENERATIONREQUEST']._serialized_start=119
-  _globals['_IMAGEGENERATIONREQUEST']._serialized_end=236
-  _globals['_INFERENCETXID']._serialized_start=238
-  _globals['_INFERENCETXID']._serialized_end=265
-  _globals['_INFERENCESTATUS']._serialized_start=268
-  _globals['_INFERENCESTATUS']._serialized_end=480
-  _globals['_INFERENCESTATUS_STATUS']._serialized_start=366
-  _globals['_INFERENCESTATUS_STATUS']._serialized_end=462
-  _globals['_INFERENCERESULT']._serialized_start=483
-  _globals['_INFERENCERESULT']._serialized_end=647
-  _globals['_IMAGEGENERATIONRESPONSE']._serialized_start=649
-  _globals['_IMAGEGENERATIONRESPONSE']._serialized_end=694
-  _globals['_INFERENCESERVICE']._serialized_start=697
-  _globals['_INFERENCESERVICE']._serialized_end=943
+    DESCRIPTOR._loaded_options = None
+    _globals["_INFERENCEREQUEST"]._serialized_start = 26
+    _globals["_INFERENCEREQUEST"]._serialized_end = 117
+    _globals["_IMAGEGENERATIONREQUEST"]._serialized_start = 119
+    _globals["_IMAGEGENERATIONREQUEST"]._serialized_end = 236
+    _globals["_INFERENCETXID"]._serialized_start = 238
+    _globals["_INFERENCETXID"]._serialized_end = 265
+    _globals["_INFERENCESTATUS"]._serialized_start = 268
+    _globals["_INFERENCESTATUS"]._serialized_end = 480
+    _globals["_INFERENCESTATUS_STATUS"]._serialized_start = 366
+    _globals["_INFERENCESTATUS_STATUS"]._serialized_end = 462
+    _globals["_INFERENCERESULT"]._serialized_start = 483
+    _globals["_INFERENCERESULT"]._serialized_end = 647
+    _globals["_IMAGEGENERATIONRESPONSE"]._serialized_start = 649
+    _globals["_IMAGEGENERATIONRESPONSE"]._serialized_end = 694
+    _globals["_INFERENCESERVICE"]._serialized_start = 697
+    _globals["_INFERENCESERVICE"]._serialized_end = 943
 # @@protoc_insertion_point(module_scope)

opengradient/proto/infer_pb2_grpc.py

@@ -1,33 +1,33 @@
 # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
 """Client and server classes corresponding to protobuf-defined services."""
+
 import grpc
-import warnings
 
 from . import infer_pb2 as infer__pb2
 
-GRPC_GENERATED_VERSION = '1.66.2'
+GRPC_GENERATED_VERSION = "1.66.2"
 GRPC_VERSION = grpc.__version__
 _version_not_supported = False
 
 try:
     from grpc._utilities import first_version_is_lower
+
     _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
 except ImportError:
     _version_not_supported = True
 
 if _version_not_supported:
     raise RuntimeError(
-        f'The grpc package installed is at version {GRPC_VERSION},'
-        + f' but the generated code in infer_pb2_grpc.py depends on'
-        + f' grpcio>={GRPC_GENERATED_VERSION}.'
-        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
-        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+        f"The grpc package installed is at version {GRPC_VERSION},"
+        + " but the generated code in infer_pb2_grpc.py depends on"
+        + f" grpcio>={GRPC_GENERATED_VERSION}."
+        + f" Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}"
+        + f" or downgrade your generated code using grpcio-tools<={GRPC_VERSION}."
     )
 
 
 class InferenceServiceStub(object):
-    """The inference service definition
-    """
+    """The inference service definition"""
 
     def __init__(self, channel):
         """Constructor.
@@ -36,89 +36,91 @@ class InferenceServiceStub(object):
             channel: A grpc.Channel.
         """
         self.RunInferenceAsync = channel.unary_unary(
-                '/inference.InferenceService/RunInferenceAsync',
-                request_serializer=infer__pb2.InferenceRequest.SerializeToString,
-                response_deserializer=infer__pb2.InferenceTxId.FromString,
-                _registered_method=True)
+            "/inference.InferenceService/RunInferenceAsync",
+            request_serializer=infer__pb2.InferenceRequest.SerializeToString,
+            response_deserializer=infer__pb2.InferenceTxId.FromString,
+            _registered_method=True,
+        )
         self.GetInferenceStatus = channel.unary_unary(
-                '/inference.InferenceService/GetInferenceStatus',
-                request_serializer=infer__pb2.InferenceTxId.SerializeToString,
-                response_deserializer=infer__pb2.InferenceStatus.FromString,
-                _registered_method=True)
+            "/inference.InferenceService/GetInferenceStatus",
+            request_serializer=infer__pb2.InferenceTxId.SerializeToString,
+            response_deserializer=infer__pb2.InferenceStatus.FromString,
+            _registered_method=True,
+        )
         self.GetInferenceResult = channel.unary_unary(
-                '/inference.InferenceService/GetInferenceResult',
-                request_serializer=infer__pb2.InferenceTxId.SerializeToString,
-                response_deserializer=infer__pb2.InferenceResult.FromString,
-                _registered_method=True)
+            "/inference.InferenceService/GetInferenceResult",
+            request_serializer=infer__pb2.InferenceTxId.SerializeToString,
+            response_deserializer=infer__pb2.InferenceResult.FromString,
+            _registered_method=True,
+        )
 
 
 class InferenceServiceServicer(object):
-    """The inference service definition
-    """
+    """The inference service definition"""
 
     def RunInferenceAsync(self, request, context):
         """Missing associated documentation comment in .proto file."""
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
 
     def GetInferenceStatus(self, request, context):
         """Missing associated documentation comment in .proto file."""
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
 
     def GetInferenceResult(self, request, context):
         """Missing associated documentation comment in .proto file."""
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
 
 
 def add_InferenceServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
-            'RunInferenceAsync': grpc.unary_unary_rpc_method_handler(
-                    servicer.RunInferenceAsync,
-                    request_deserializer=infer__pb2.InferenceRequest.FromString,
-                    response_serializer=infer__pb2.InferenceTxId.SerializeToString,
-            ),
-            'GetInferenceStatus': grpc.unary_unary_rpc_method_handler(
-                    servicer.GetInferenceStatus,
-                    request_deserializer=infer__pb2.InferenceTxId.FromString,
-                    response_serializer=infer__pb2.InferenceStatus.SerializeToString,
-            ),
-            'GetInferenceResult': grpc.unary_unary_rpc_method_handler(
-                    servicer.GetInferenceResult,
-                    request_deserializer=infer__pb2.InferenceTxId.FromString,
-                    response_serializer=infer__pb2.InferenceResult.SerializeToString,
-            ),
+        "RunInferenceAsync": grpc.unary_unary_rpc_method_handler(
+            servicer.RunInferenceAsync,
+            request_deserializer=infer__pb2.InferenceRequest.FromString,
+            response_serializer=infer__pb2.InferenceTxId.SerializeToString,
+        ),
+        "GetInferenceStatus": grpc.unary_unary_rpc_method_handler(
+            servicer.GetInferenceStatus,
+            request_deserializer=infer__pb2.InferenceTxId.FromString,
+            response_serializer=infer__pb2.InferenceStatus.SerializeToString,
+        ),
+        "GetInferenceResult": grpc.unary_unary_rpc_method_handler(
+            servicer.GetInferenceResult,
+            request_deserializer=infer__pb2.InferenceTxId.FromString,
+            response_serializer=infer__pb2.InferenceResult.SerializeToString,
+        ),
     }
-    generic_handler = grpc.method_handlers_generic_handler(
-            'inference.InferenceService', rpc_method_handlers)
+    generic_handler = grpc.method_handlers_generic_handler("inference.InferenceService", rpc_method_handlers)
     server.add_generic_rpc_handlers((generic_handler,))
-    server.add_registered_method_handlers('inference.InferenceService', rpc_method_handlers)
+    server.add_registered_method_handlers("inference.InferenceService", rpc_method_handlers)
 
 
- # This class is part of an EXPERIMENTAL API.
+# This class is part of an EXPERIMENTAL API.
 class InferenceService(object):
-    """The inference service definition
-    """
+    """The inference service definition"""
 
     @staticmethod
-    def RunInferenceAsync(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
+    def RunInferenceAsync(
+        request,
+        target,
+        options=(),
+        channel_credentials=None,
+        call_credentials=None,
+        insecure=False,
+        compression=None,
+        wait_for_ready=None,
+        timeout=None,
+        metadata=None,
+    ):
         return grpc.experimental.unary_unary(
             request,
             target,
-            '/inference.InferenceService/RunInferenceAsync',
+            "/inference.InferenceService/RunInferenceAsync",
             infer__pb2.InferenceRequest.SerializeToString,
             infer__pb2.InferenceTxId.FromString,
             options,
@@ -129,23 +131,26 @@ class InferenceService(object):
             wait_for_ready,
             timeout,
             metadata,
-            _registered_method=True)
+            _registered_method=True,
+        )
 
     @staticmethod
-    def GetInferenceStatus(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
+    def GetInferenceStatus(
+        request,
+        target,
+        options=(),
+        channel_credentials=None,
+        call_credentials=None,
+        insecure=False,
+        compression=None,
+        wait_for_ready=None,
+        timeout=None,
+        metadata=None,
+    ):
         return grpc.experimental.unary_unary(
             request,
             target,
-            '/inference.InferenceService/GetInferenceStatus',
+            "/inference.InferenceService/GetInferenceStatus",
             infer__pb2.InferenceTxId.SerializeToString,
             infer__pb2.InferenceStatus.FromString,
             options,
@@ -156,23 +161,26 @@ class InferenceService(object):
             wait_for_ready,
             timeout,
             metadata,
-            _registered_method=True)
+            _registered_method=True,
+        )
 
     @staticmethod
-    def GetInferenceResult(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
+    def GetInferenceResult(
+        request,
+        target,
+        options=(),
+        channel_credentials=None,
+        call_credentials=None,
+        insecure=False,
+        compression=None,
+        wait_for_ready=None,
+        timeout=None,
+        metadata=None,
+    ):
         return grpc.experimental.unary_unary(
             request,
             target,
-            '/inference.InferenceService/GetInferenceResult',
+            "/inference.InferenceService/GetInferenceResult",
             infer__pb2.InferenceTxId.SerializeToString,
             infer__pb2.InferenceResult.FromString,
             options,
@@ -183,4 +191,5 @@ class InferenceService(object):
             wait_for_ready,
             timeout,
             metadata,
-            _registered_method=True)
+            _registered_method=True,
+        )

opengradient/types.py

@@ -1,18 +1,21 @@
+import time
 from dataclasses import dataclass
-from typing import List, Tuple, Union, Dict, Optional
 from enum import Enum, IntEnum
-import time
+from typing import Dict, List, Optional, Tuple, Union
+
 
 class CandleOrder(IntEnum):
     ASCENDING = 0
     DESCENDING = 1
 
+
 class CandleType(IntEnum):
     HIGH = 0
     LOW = 1
     OPEN = 2
     CLOSE = 3
 
+
 @dataclass
 class HistoricalInputQuery:
     currency_pair: str
@@ -28,65 +31,74 @@ class HistoricalInputQuery:
             self.total_candles,
             self.candle_duration_in_mins,
             int(self.order),
-            [int(ct) for ct in self.candle_types]
+            [int(ct) for ct in self.candle_types],
         )
 
     @classmethod
-    def from_dict(cls, data: dict) -> 'HistoricalInputQuery':
+    def from_dict(cls, data: dict) -> "HistoricalInputQuery":
         """Create HistoricalInputQuery from dictionary format"""
-        order = CandleOrder[data['order'].upper()]
-        candle_types = [CandleType[ct.upper()] for ct in data['candle_types']]
-        
+        order = CandleOrder[data["order"].upper()]
+        candle_types = [CandleType[ct.upper()] for ct in data["candle_types"]]
+
         return cls(
-            currency_pair=data['currency_pair'],
-            total_candles=int(data['total_candles']),
-            candle_duration_in_mins=int(data['candle_duration_in_mins']),
+            currency_pair=data["currency_pair"],
+            total_candles=int(data["total_candles"]),
+            candle_duration_in_mins=int(data["candle_duration_in_mins"]),
             order=order,
-            candle_types=candle_types
+            candle_types=candle_types,
         )
 
+
 @dataclass
 class Number:
     value: int
     decimals: int
 
+
 @dataclass
 class NumberTensor:
     name: str
     values: List[Tuple[int, int]]
 
+
 @dataclass
 class StringTensor:
     name: str
     values: List[str]
 
+
 @dataclass
 class ModelInput:
     numbers: List[NumberTensor]
     strings: List[StringTensor]
 
+
 class InferenceMode:
     VANILLA = 0
     ZKML = 1
     TEE = 2
 
+
 class LlmInferenceMode:
     VANILLA = 0
     TEE = 1
 
+
 @dataclass
 class ModelOutput:
     numbers: List[NumberTensor]
     strings: List[StringTensor]
     is_simulation_result: bool
 
+
 @dataclass
 class AbiFunction:
     name: str
-    inputs: List[Union[str, 'AbiFunction']]
-    outputs: List[Union[str, 'AbiFunction']]
+    inputs: List[Union[str, "AbiFunction"]]
+    outputs: List[Union[str, "AbiFunction"]]
     state_mutability: str
 
+
 @dataclass
 class Abi:
     functions: List[AbiFunction]
@@ -95,32 +107,25 @@ class Abi:
     def from_json(cls, abi_json):
         functions = []
         for item in abi_json:
-            if item['type'] == 'function':
-                inputs = cls._parse_inputs_outputs(item['inputs'])
-                outputs = cls._parse_inputs_outputs(item['outputs'])
-                functions.append(AbiFunction(
-                    name=item['name'],
-                    inputs=inputs,
-                    outputs=outputs,
-                    state_mutability=item['stateMutability']
-                ))
+            if item["type"] == "function":
+                inputs = cls._parse_inputs_outputs(item["inputs"])
+                outputs = cls._parse_inputs_outputs(item["outputs"])
+                functions.append(AbiFunction(name=item["name"], inputs=inputs, outputs=outputs, state_mutability=item["stateMutability"]))
         return cls(functions=functions)
 
     @staticmethod
     def _parse_inputs_outputs(items):
         result = []
         for item in items:
-            if 'components' in item:
-                result.append(AbiFunction(
-                    name=item['name'],
-                    inputs=Abi._parse_inputs_outputs(item['components']),
-                    outputs=[],
-                    state_mutability=''
-                ))
+            if "components" in item:
+                result.append(
+                    AbiFunction(name=item["name"], inputs=Abi._parse_inputs_outputs(item["components"]), outputs=[], state_mutability="")
+                )
             else:
                 result.append(f"{item['name']}:{item['type']}")
         return result
-    
+
+
 class LLM(str, Enum):
     """Enum for available LLM models"""
 
@@ -130,11 +135,13 @@ class LLM(str, Enum):
     HERMES_3_LLAMA_3_1_70B = "NousResearch/Hermes-3-Llama-3.1-70B"
     META_LLAMA_3_1_70B_INSTRUCT = "meta-llama/Llama-3.1-70B-Instruct"
 
+
 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"
 
+
 @dataclass
 class SchedulerParams:
     frequency: int
@@ -145,10 +152,7 @@ class SchedulerParams:
         return int(time.time()) + (self.duration_hours * 60 * 60)
 
     @staticmethod
-    def from_dict(data: Optional[Dict[str, int]]) -> Optional['SchedulerParams']:
+    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)
-        )
+        return SchedulerParams(frequency=data.get("frequency", 600), duration_hours=data.get("duration_hours", 2))