opengradient 0.5.11__py3-none-any.whl → 0.5.13__py3-none-any.whl

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

@@ -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.dist-info → opengradient-0.5.13.dist-info}/RECORD

@@ -2,12 +2,12 @@ opengradient/__init__.py,sha256=hRGex2VGwAQ-lqBNphW93D4R3wPHQpQrTkwgGLe2MoA,1025
 opengradient/account.py,sha256=5wrYpws_1lozjOFjLCTHtxgoxK-LmObDAaVy9eDcJY4,1145
 opengradient/alpha.py,sha256=WAtL1GGbEpoeLO89rOMd8-YAgAFJYM1UJlICm6YGsPs,15195
 opengradient/cli.py,sha256=pfgyLfD1MIDifKmGLFsJqBlgvqIcnsIh3zzg7PaIeH4,33670
-opengradient/client.py,sha256=KQZI1WZaMqyIHRWFH4HA1fFd1JtSdkrUYa7RX_bIwJA,65289
+opengradient/client.py,sha256=yGQ7JngirqPjS0BDOrIeSGVYLrDQpL59tkIz9cL61Jk,65333
 opengradient/defaults.py,sha256=8faLPwvp_BQdErY_SEjBzvmGVuOBdZ2zKcoryD8SCXk,797
 opengradient/exceptions.py,sha256=88tfegboGtlehQcwhxsl6ZzhLJWZWlkf_bkHTiCtXpo,3391
 opengradient/types.py,sha256=bADakUM6WwdMORGC5HvQvWCezNwIlVc7l0zodPapbhQ,14622
 opengradient/utils.py,sha256=ZUq4OBIml2vsC0tRqus4Zwb_e3g4woo00apByrafuVw,8058
-opengradient/x402_auth.py,sha256=fuSsgFmvrlZ3X6uAlUnqtkQ485NECxHfBr3IFADpKAg,3309
+opengradient/x402_auth.py,sha256=DDY8DPt1VlWXYJBrwsQ9vRTkX_nW0kHPw0M3IFuG1lM,3272
 opengradient/abi/InferencePrecompile.abi,sha256=reepTHg6Q01UrFP0Gexc-JayplsvOLPfG7jrEZ-cV28,10197
 opengradient/abi/PriceHistoryInference.abi,sha256=ZB3fZdx1kaFlp2wt1vTbTZZG1k8HPvmNtkG5Q8Bnajw,5098
 opengradient/abi/WorkflowScheduler.abi,sha256=yEGs76qO4S1z980KL5hBdfyXiJ6k-kERcB1O_o73AEU,416
@@ -29,9 +29,9 @@ opengradient/workflow_models/constants.py,sha256=viIkb_LGcfVprqQNaA80gBTj6cfYam0
 opengradient/workflow_models/types.py,sha256=Z22hF6c8Y4D2GlzVEIBODGwsqSjSrQvUcpZ7R-mIJdI,409
 opengradient/workflow_models/utils.py,sha256=aL2-Hp5J5qlJft6-wx4GnZNOXZ1vjYaTF1uEgYavWdI,1509
 opengradient/workflow_models/workflow_models.py,sha256=d4C_gs39DAfy4cdY9Ee6GMXpPfzwvKFpmxzK1A7LNgU,3900
-opengradient-0.5.11.dist-info/licenses/LICENSE,sha256=xEcvQ3AxZOtDkrqkys2Mm6Y9diEnaSeQRKvxi-JGnNA,1069
-opengradient-0.5.11.dist-info/METADATA,sha256=qJZT1jnWcmpeBQsLN1YrmPRCfMzD4nk7saMguFYNmyc,5437
-opengradient-0.5.11.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-opengradient-0.5.11.dist-info/entry_points.txt,sha256=yUKTaJx8RXnybkob0J62wVBiCp_1agVbgw9uzsmaeJc,54
-opengradient-0.5.11.dist-info/top_level.txt,sha256=oC1zimVLa2Yi1LQz8c7x-0IQm92milb5ax8gHBHwDqU,13
-opengradient-0.5.11.dist-info/RECORD,,
+opengradient-0.5.13.dist-info/licenses/LICENSE,sha256=xEcvQ3AxZOtDkrqkys2Mm6Y9diEnaSeQRKvxi-JGnNA,1069
+opengradient-0.5.13.dist-info/METADATA,sha256=kpXm-b7HB09GDchtUHwHNQ8OrNUDHQhUCHvlp07ENDw,5437
+opengradient-0.5.13.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+opengradient-0.5.13.dist-info/entry_points.txt,sha256=yUKTaJx8RXnybkob0J62wVBiCp_1agVbgw9uzsmaeJc,54
+opengradient-0.5.13.dist-info/top_level.txt,sha256=oC1zimVLa2Yi1LQz8c7x-0IQm92milb5ax8gHBHwDqU,13
+opengradient-0.5.13.dist-info/RECORD,,