opengradient 0.5.1__py3-none-any.whl → 0.5.2__py3-none-any.whl

opengradient/client.py

@@ -465,7 +465,7 @@ class Client:
                 return OpenGradientError("That model CID is not supported yet for TEE inference")
 
             return self._external_llm_completion(
-                model=model_cid,
+                model=model_cid.split('/')[1],
                 prompt=prompt,
                 max_tokens=max_tokens,
                 stop_sequence=stop_sequence,
@@ -479,12 +479,16 @@ class Client:
 
             if model_cid not in [llm.value for llm in LLM]:
                 raise OpenGradientError("That model CID is not yet supported for inference")
+            
+            model_name = model_cid
+            if model_cid in [llm.value for llm in TEE_LLM]:
+                model_name = model_cid.split('/')[1]
 
             contract = self._blockchain.eth.contract(address=self._inference_hub_contract_address, abi=self._inference_abi)
 
             llm_request = {
                 "mode": inference_mode.value,
-                "modelCID": model_cid,
+                "modelCID": model_name,
                 "prompt": prompt,
                 "max_tokens": max_tokens,
                 "stop_sequence": stop_sequence or [],
@@ -670,7 +674,7 @@ class Client:
                 return OpenGradientError("That model CID is not supported yet for TEE inference")
 
             return self._external_llm_chat(
-                model=model_cid,
+                model=model_cid.split('/')[1],
                 messages=messages,
                 max_tokens=max_tokens,
                 stop_sequence=stop_sequence,
@@ -686,6 +690,10 @@ class Client:
             
             if model_cid not in [llm.value for llm in LLM]:
                 raise OpenGradientError("That model CID is not yet supported for inference")
+            
+            model_name = model_cid
+            if model_cid in [llm.value for llm in TEE_LLM]:
+                model_name = model_cid.split('/')[1]
 
             contract = self._blockchain.eth.contract(address=self._inference_hub_contract_address, abi=self._inference_abi)
 
@@ -713,7 +721,7 @@ class Client:
 
             llm_request = {
                 "mode": inference_mode.value,
-                "modelCID": model_cid,
+                "modelCID": model_name,
                 "messages": messages,
                 "max_tokens": max_tokens,
                 "stop_sequence": stop_sequence or [],

opengradient/types.py

@@ -202,27 +202,27 @@ class LLM(str, Enum):
     DOBBY_LEASHED_3_1_8B = "SentientAGI/Dobby-Mini-Leashed-Llama-3.1-8B"
     
     # OpenAI models via TEE
-    GPT_4_1_2025_04_14 = "gpt-4.1-2025-04-14"
-    GPT_4O = "gpt-4o"
-    O4_MINI = "o4-mini"
+    GPT_4_1_2025_04_14 = "OpenAI/gpt-4.1-2025-04-14"
+    GPT_4O = "OpenAI/gpt-4o"
+    O4_MINI = "OpenAI/o4-mini"
     
     # Anthropic models via TEE
-    CLAUDE_3_7_SONNET = "claude-3.7-sonnet"
-    CLAUDE_3_5_HAIKU = "claude-3.5-haiku"
-    CLAUDE_4_0_SONNET = "claude-4.0-sonnet"
+    CLAUDE_3_7_SONNET = "Anthropic/claude-3.7-sonnet"
+    CLAUDE_3_5_HAIKU = "Anthropic/claude-3.5-haiku"
+    CLAUDE_4_0_SONNET = "Anthropic/claude-4.0-sonnet"
     
     # Google models via TEE
-    GEMINI_2_5_FLASH = "gemini-2.5-flash"
-    GEMINI_2_5_PRO = "gemini-2.5-pro"
-    GEMINI_2_0_FLASH = "gemini-2.0-flash"
+    GEMINI_2_5_FLASH = "Google/gemini-2.5-flash"
+    GEMINI_2_5_PRO = "Google/gemini-2.5-pro"
+    GEMINI_2_0_FLASH = "Google/gemini-2.0-flash"
     
     # xAI Grok models via TEE
-    GROK_3_MINI_BETA = "grok-3-mini-beta"
-    GROK_3_BETA = "grok-3-beta"
+    GROK_3_MINI_BETA = "xAI/grok-3-mini-beta"
+    GROK_3_BETA = "xAI/grok-3-beta"
     GROK_2_1212 = "grok-2-1212"
-    GROK_2_VISION_LATEST = "grok-2-vision-latest"
-    GROK_4_1_FAST = "grok-4.1-fast"
-    GROK_4_1_FAST_NON_REASONING = "grok-4-1-fast-non-reasoning"
+    GROK_2_VISION_LATEST = "xAI/grok-2-vision-latest"
+    GROK_4_1_FAST = "xAI/grok-4.1-fast"
+    GROK_4_1_FAST_NON_REASONING = "xAI/grok-4-1-fast-non-reasoning"
 
 class TEE_LLM(str, Enum):
     """Enum for LLM models available for TEE execution"""
@@ -231,27 +231,27 @@ class TEE_LLM(str, Enum):
     # META_LLAMA_3_1_70B_INSTRUCT = "meta-llama/Llama-3.1-70B-Instruct"
     
     # OpenAI models via TEE
-    GPT_4_1_2025_04_14 = "gpt-4.1-2025-04-14"
-    GPT_4O = "gpt-4o"
-    O4_MINI = "o4-mini"
+    GPT_4_1_2025_04_14 = "OpenAI/gpt-4.1-2025-04-14"
+    GPT_4O = "OpenAI/gpt-4o"
+    O4_MINI = "OpenAI/o4-mini"
     
     # Anthropic models via TEE
-    CLAUDE_3_7_SONNET = "claude-3.7-sonnet"
-    CLAUDE_3_5_HAIKU = "claude-3.5-haiku"
-    CLAUDE_4_0_SONNET = "claude-4.0-sonnet"
+    CLAUDE_3_7_SONNET = "Anthropic/claude-3.7-sonnet"
+    CLAUDE_3_5_HAIKU = "Anthropic/claude-3.5-haiku"
+    CLAUDE_4_0_SONNET = "Anthropic/claude-4.0-sonnet"
     
     # Google models via TEE
-    GEMINI_2_5_FLASH = "gemini-2.5-flash"
-    GEMINI_2_5_PRO = "gemini-2.5-pro"
-    GEMINI_2_0_FLASH = "gemini-2.0-flash"
+    GEMINI_2_5_FLASH = "Google/gemini-2.5-flash"
+    GEMINI_2_5_PRO = "Google/gemini-2.5-pro"
+    GEMINI_2_0_FLASH = "Google/gemini-2.0-flash"
     
     # xAI Grok models via TEE
-    GROK_3_MINI_BETA = "grok-3-mini-beta"
-    GROK_3_BETA = "grok-3-beta"
-    GROK_2_1212 = "grok-2-1212"
-    GROK_2_VISION_LATEST = "grok-2-vision-latest"
-    GROK_4_1_FAST = "grok-4.1-fast"
-    GROK_4_1_FAST_NON_REASONING = "grok-4-1-fast-non-reasoning"
+    GROK_3_MINI_BETA = "xAI/grok-3-mini-beta"
+    GROK_3_BETA = "xAI/grok-3-beta"
+    GROK_2_1212 = "xAI/grok-2-1212"
+    GROK_2_VISION_LATEST = "xAI/grok-2-vision-latest"
+    GROK_4_1_FAST = "xAI/grok-4.1-fast"
+    GROK_4_1_FAST_NON_REASONING = "xAI/grok-4-1-fast-non-reasoning"
 
 
 @dataclass

{opengradient-0.5.1.dist-info → opengradient-0.5.2.dist-info}/METADATA

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

{opengradient-0.5.1.dist-info → opengradient-0.5.2.dist-info}/RECORD

@@ -1,10 +1,10 @@
 opengradient/__init__.py,sha256=wVg0KTFNBl7RnZF9huR5-m_q1E7tO-YyQwY7AD9JFoc,12635
 opengradient/account.py,sha256=5wrYpws_1lozjOFjLCTHtxgoxK-LmObDAaVy9eDcJY4,1145
 opengradient/cli.py,sha256=QzjH_KS6TF8gm_L1otFWA-oHkJ5SSfizFoRn0xR0b70,29162
-opengradient/client.py,sha256=46pj3l-JkQfdojoLh94na0kJIYIVwusTcDP5qi_nKwk,62004
+opengradient/client.py,sha256=Seid3raffwAcrc3qVIINu259iO-BdaGXGT5O2VaT6ac,62356
 opengradient/defaults.py,sha256=w8-dr5ciF2TGnqbm_ib0Yz4U0YL5ikpNqkcPVpmXzP8,673
 opengradient/exceptions.py,sha256=88tfegboGtlehQcwhxsl6ZzhLJWZWlkf_bkHTiCtXpo,3391
-opengradient/types.py,sha256=f4HiDnsNBHtsyLD6AlWihtSB93qTFmj72zOQFZRNK5k,7360
+opengradient/types.py,sha256=ygnQXoGJPv9i3daS0oduUsmUNoPGx6Oozkt-Yy7Nn6s,7548
 opengradient/utils.py,sha256=ZUq4OBIml2vsC0tRqus4Zwb_e3g4woo00apByrafuVw,8058
 opengradient/abi/InferencePrecompile.abi,sha256=reepTHg6Q01UrFP0Gexc-JayplsvOLPfG7jrEZ-cV28,10197
 opengradient/abi/PriceHistoryInference.abi,sha256=ZB3fZdx1kaFlp2wt1vTbTZZG1k8HPvmNtkG5Q8Bnajw,5098
@@ -27,9 +27,9 @@ opengradient/workflow_models/constants.py,sha256=viIkb_LGcfVprqQNaA80gBTj6cfYam0
 opengradient/workflow_models/types.py,sha256=Z22hF6c8Y4D2GlzVEIBODGwsqSjSrQvUcpZ7R-mIJdI,409
 opengradient/workflow_models/utils.py,sha256=ySfpuiOBqLTlfto6ZxZf2vc7K6RGIja0l4eaVm5AOzY,1503
 opengradient/workflow_models/workflow_models.py,sha256=d4C_gs39DAfy4cdY9Ee6GMXpPfzwvKFpmxzK1A7LNgU,3900
-opengradient-0.5.1.dist-info/licenses/LICENSE,sha256=xEcvQ3AxZOtDkrqkys2Mm6Y9diEnaSeQRKvxi-JGnNA,1069
-opengradient-0.5.1.dist-info/METADATA,sha256=_Gi-TVU_xljBPwwS6J3olNQWul73Hy3Ak0tErnP1Ed0,3992
-opengradient-0.5.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-opengradient-0.5.1.dist-info/entry_points.txt,sha256=yUKTaJx8RXnybkob0J62wVBiCp_1agVbgw9uzsmaeJc,54
-opengradient-0.5.1.dist-info/top_level.txt,sha256=oC1zimVLa2Yi1LQz8c7x-0IQm92milb5ax8gHBHwDqU,13
-opengradient-0.5.1.dist-info/RECORD,,
+opengradient-0.5.2.dist-info/licenses/LICENSE,sha256=xEcvQ3AxZOtDkrqkys2Mm6Y9diEnaSeQRKvxi-JGnNA,1069
+opengradient-0.5.2.dist-info/METADATA,sha256=vKpBpoNs4RhhXcz275SIoYrIpL550Z7ykk_9zS0-CdA,3992
+opengradient-0.5.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+opengradient-0.5.2.dist-info/entry_points.txt,sha256=yUKTaJx8RXnybkob0J62wVBiCp_1agVbgw9uzsmaeJc,54
+opengradient-0.5.2.dist-info/top_level.txt,sha256=oC1zimVLa2Yi1LQz8c7x-0IQm92milb5ax8gHBHwDqU,13
+opengradient-0.5.2.dist-info/RECORD,,