opengradient 0.5.11__tar.gz → 0.5.13__tar.gz

{opengradient-0.5.11/src/opengradient.egg-info → opengradient-0.5.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opengradient
-Version: 0.5.11
+Version: 0.5.13
 Summary: Python SDK for OpenGradient decentralized model management & inference services
 Author-email: OpenGradient <kyle@vannalabs.ai>
 License-Expression: MIT

{opengradient-0.5.11 → opengradient-0.5.13}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "opengradient"
-version = "0.5.11"
+version = "0.5.13"
 description = "Python SDK for OpenGradient decentralized model management & inference services"
 authors = [{name = "OpenGradient", email = "kyle@vannalabs.ai"}]
 readme = "README.md"

{opengradient-0.5.11 → opengradient-0.5.13}/src/opengradient/client.py

@@ -512,8 +512,8 @@ class Client:
         # Check if this is a local model or external
         # TODO (Kyle): separate TEE and Vanilla completion requests
         if inference_mode == LlmInferenceMode.TEE:
-            if model_cid not in TEE_LLM:
-                return OpenGradientError("That model CID is not supported yet for TEE inference")
+            if model_cid not in [llm.value for llm in TEE_LLM]:
+                raise OpenGradientError("That model CID is not supported yet for TEE inference")
 
             return self._external_llm_completion(
                 model=model_cid.split("/")[1],
@@ -726,8 +726,8 @@ class Client:
         # Check if this is a local model or external
         # TODO (Kyle): separate TEE and Vanilla completion requests
         if inference_mode == LlmInferenceMode.TEE:
-            if model_cid not in TEE_LLM:
-                return OpenGradientError("That model CID is not supported yet for TEE inference")
+            if model_cid not in [llm.value for llm in TEE_LLM]:
+                raise OpenGradientError("That model CID is not supported yet for TEE inference")
 
             if stream:
                 # Use threading bridge for true sync streaming

{opengradient-0.5.11 → opengradient-0.5.13}/src/opengradient/x402_auth.py

@@ -43,6 +43,7 @@ class X402Auth(httpx.Auth):
             ]
         ] = None,
         network_filter: typing.Optional[str] = None,
+        scheme_filter: typing.Optional[str] = None,
     ):
         """
         Initialize X402Auth with an Ethereum account for signing payments.
@@ -50,15 +51,14 @@ class X402Auth(httpx.Auth):
         Args:
             account: eth_account LocalAccount instance for signing payments
             max_value: Optional maximum allowed payment amount in base units
-            network_filter: Optional network filter for selecting payment requirements
-            scheme_filter: Optional scheme filter for selecting payment requirements
         """
+        self.network_filter = network_filter
+        self.scheme_filter = scheme_filter
         self.x402_client = x402Client(
             account,
             max_value=max_value,
             payment_requirements_selector=payment_requirements_selector,  # type: ignore
         )
-        self.network_filter = network_filter
 
     async def async_auth_flow(
         self, request: httpx.Request
@@ -84,6 +84,7 @@ class X402Auth(httpx.Auth):
                 selected_requirements = self.x402_client.select_payment_requirements(
                     payment_response.accepts,
                     self.network_filter,
+                    self.scheme_filter,
                 )
 
                 payment_header = self.x402_client.create_payment_header(

{opengradient-0.5.11 → opengradient-0.5.13/src/opengradient.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opengradient
-Version: 0.5.11
+Version: 0.5.13
 Summary: Python SDK for OpenGradient decentralized model management & inference services
 Author-email: OpenGradient <kyle@vannalabs.ai>
 License-Expression: MIT