opengradient 0.2.0__tar.gz → 0.2.1__tar.gz

{opengradient-0.2.0/src/opengradient.egg-info → opengradient-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: opengradient
-Version: 0.2.0
+Version: 0.2.1
 Summary: A Python SDK for OpenGradient inference services
 Author-email: OpenGradient <oliver@opengradient.ai>
 License: MIT License

{opengradient-0.2.0 → opengradient-0.2.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "opengradient"
-version = "0.2.0"
+version = "0.2.1"
 description = "A Python SDK for OpenGradient inference services"
 authors = [{name = "OpenGradient", email = "oliver@opengradient.ai"}]
 license = {file = "LICENSE"}

{opengradient-0.2.0 → opengradient-0.2.1}/src/opengradient/__init__.py

@@ -2,7 +2,7 @@ from .client import Client
 from .exceptions import OpenGradientError, FileNotFoundError, UploadError, InferenceError, ResultRetrievalError
 from .types import ModelInput, InferenceMode, Number, NumberTensor, StringTensor, ModelOutput
 
-__version__ = "0.2.0"
+__version__ = "0.2.1"
 
 _client = None
 
@@ -10,25 +10,25 @@ def init(private_key, rpc_url, contract_address, email="test@test.com", password
     global _client
     _client = Client(private_key=private_key, rpc_url=rpc_url, contract_address=contract_address, email=email, password=password)
 
-def upload(model_path, model_id, version_id):
+def upload(model_path, model_name, version):
     if _client is None:
         raise RuntimeError("OpenGradient client not initialized. Call og.init() first.")
-    return _client.upload(model_path, model_id, version_id)
+    return _client.upload(model_path, model_name, version)
 
 def create_model(model_name, model_desc):
     if _client is None:
         raise RuntimeError("OpenGradient client not initialized. Call og.init() first.")
-    return _client.create_model(model_name,model_desc)
+    return _client.create_model(model_name, model_desc)
 
-def create_version(model_id, notes=None, is_major=False):
+def create_version(model_name, notes=None, is_major=False):
     if _client is None:
         raise RuntimeError("OpenGradient client not initialized. Call og.init() first.")
-    return _client.create_version(model_id, notes, is_major)
+    return _client.create_version(model_name, notes, is_major)
 
-def infer(model_id, inference_mode, model_input):
+def infer(model_cid, inference_mode, model_input):
     if _client is None:
         raise RuntimeError("OpenGradient client not initialized. Call og.init() first.")
-    return _client.infer(model_id, inference_mode, model_input)
+    return _client.infer(model_cid, inference_mode, model_input)
 
 def sign_in_with_email_and_password(email: str, password: str):
     if _client is None:

{opengradient-0.2.0 → opengradient-0.2.1}/src/opengradient/client.py

@@ -67,14 +67,14 @@ class Client:
         else:
             logging.error("No user is currently signed in")
 
-    def create_model(self, model_name: str, model_desc: str, version_id: str = "1.00") -> dict:
+    def create_model(self, model_name: str, model_desc: str, version: str = "1.00") -> dict:
         """
-        Create a new model with the given model_name and model_desc, and a specified version ID.
+        Create a new model with the given model_name and model_desc, and a specified version.
 
         Args:
             model_name (str): The name of the model.
             model_desc (str): The description of the model.
-            version_id (str): The version identifier (default is "1.00").
+            version (str): The version identifier (default is "1.00").
 
         Returns:
             dict: The server response containing model details.
@@ -111,13 +111,13 @@ class Client:
 
             # Create the specified version for the newly created model
             try:
-                version_response = self.create_version(model_id, version_id)
-                logging.info(f"Version creation successful. Version ID: {version_response['version_id']}")
+                version_response = self.create_version(model_id, version)
+                logging.info(f"Version creation successful. Version String: {version_response['versionString']}")
             except Exception as ve:
                 logging.error(f"Version creation failed, but model was created. Error: {str(ve)}")
-                return {"id": model_id, "version_id": None, "version_error": str(ve)}
+                return {"id": model_id, "versionString": None, "version_error": str(ve)}
 
-            return {"id": model_id, "version_id": version_response["version_id"]}
+            return {"id": model_id, "versionString": version_response["versionString"]}
 
         except requests.RequestException as e:
             logging.error(f"Model creation failed: {str(e)}")
@@ -130,12 +130,12 @@ class Client:
             logging.error(f"Unexpected error during model creation: {str(e)}")
             raise
 
-    def create_version(self, model_id: str, notes: str = None, is_major: bool = False) -> dict:
+    def create_version(self, model_name: str, notes: str = None, is_major: bool = False) -> dict:
         """
         Create a new version for the specified model.
 
         Args:
-            model_id (str): The unique identifier for the model.
+            model_name (str): The unique identifier for the model.
             notes (str, optional): Notes for the new version.
             is_major (bool, optional): Whether this is a major version update. Defaults to False.
 
@@ -148,7 +148,7 @@ class Client:
         if not self.user:
             raise ValueError("User not authenticated")
 
-        url = f"https://api.opengradient.ai/api/v0/models/{model_id}/versions"
+        url = f"https://api.opengradient.ai/api/v0/models/{model_name}/versions"
         headers = {
             'Authorization': f'Bearer {self.user["idToken"]}',
             'Content-Type': 'application/json'
@@ -172,14 +172,14 @@ class Client:
 
             if isinstance(json_response, list) and not json_response:
                 logging.info(f"Server returned an empty list. Assuming version was created successfully.")
-                return {"version_id": "Unknown", "note": "Created based on empty response"}
+                return {"versionString": "Unknown", "note": "Created based on empty response"}
             elif isinstance(json_response, dict):
-                version_id = json_response.get('version_id') or json_response.get('id')
-                if not version_id:
-                    logging.warning(f"'version_id' not found in response. Response: {json_response}")
-                    return {"version_id": "Unknown", "note": "Version ID not provided in response"}
-                logging.info(f"Version creation successful. Version ID: {version_id}")
-                return {"version_id": version_id}
+                version_string = json_response.get('versionString') or json_response.get('id')
+                if not version_string:
+                    logging.warning(f"'versionString' not found in response. Response: {json_response}")
+                    return {"versionString": "Unknown", "note": "Version String not provided in response"}
+                logging.info(f"Version creation successful. Version String: {version_string}")
+                return {"versionString": version_string}
             else:
                 logging.error(f"Unexpected response type: {type(json_response)}. Content: {json_response}")
                 raise Exception(f"Unexpected response type: {type(json_response)}")
@@ -195,14 +195,14 @@ class Client:
             logging.error(f"Unexpected error during version creation: {str(e)}")
             raise
 
-    def upload(self, model_path: str, model_id: str, version_id: str) -> dict:
+    def upload(self, model_path: str, model_name: str, version: str) -> dict:
         """
         Upload a model file to the server.
 
         Args:
             model_path (str): The path to the model file.
-            model_id (str): The unique identifier for the model.
-            version_id (str): The version identifier for the model.
+            model_name (str): The unique identifier for the model.
+            version (str): The version identifier for the model.
 
         Returns:
             dict: The server response containing the model CID and size.
@@ -218,7 +218,7 @@ class Client:
         if not os.path.exists(model_path):
             raise FileNotFoundError(f"Model file not found: {model_path}")
 
-        url = f"https://api.opengradient.ai/api/v0/models/{model_id}/versions/{version_id}/files"
+        url = f"https://api.opengradient.ai/api/v0/models/{model_name}/versions/{version}/files"
         headers = {
             'Authorization': f'Bearer {self.user["idToken"]}'
         }
@@ -284,7 +284,7 @@ class Client:
         Perform inference on a model.
 
         Args:
-            model_id (str): The unique identifier for the model.
+            model_cid (str): The unique content identifier for the model from IPFS.
             inference_mode (InferenceMode): The inference mode.
             model_input (Dict[str, Union[str, int, float, List, np.ndarray]]): The input data for the model.
 

{opengradient-0.2.0 → opengradient-0.2.1/src/opengradient.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: opengradient
-Version: 0.2.0
+Version: 0.2.1
 Summary: A Python SDK for OpenGradient inference services
 Author-email: OpenGradient <oliver@opengradient.ai>
 License: MIT License