opengradient 0.2.5__py3-none-any.whl → 0.2.6__py3-none-any.whl

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.5"
+__version__ = "0.2.6"
 
 _client = None
 

opengradient/client.py

@@ -112,12 +112,12 @@ class Client:
             # Create the specified version for the newly created model
             try:
                 version_response = self.create_version(model_id, version)
-                logging.info(f"Version creation successful. Version ID: {version_response['version_id']}")
+                logging.info(f"Version creation successful. Version ID: {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)}")
@@ -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}
+                versionString = json_response.get('versionString')
+                if not versionString:
+                    logging.warning(f"'versionString' not found in response. Response: {json_response}")
+                    return {"versionString": "Unknown", "note": "Version ID not provided in response"}
+                logging.info(f"Version creation successful. Version ID: {versionString}")
+                return {"versionString": versionString}
             else:
                 logging.error(f"Unexpected response type: {type(json_response)}. Content: {json_response}")
                 raise Exception(f"Unexpected response type: {type(json_response)}")
@@ -205,7 +205,7 @@ class Client:
             version (str): The version identifier for the model.
 
         Returns:
-            dict: The server response containing the model CID and size.
+            dict: The processed result.
 
         Raises:
             OpenGradientError: If the upload fails.
@@ -247,17 +247,17 @@ class Client:
                 response = requests.post(url, data=monitor, headers=headers, timeout=3600)  # 1 hour timeout
                 
                 logging.info(f"Response received. Status code: {response.status_code}")
-                logging.debug(f"Response headers: {response.headers}")
-                logging.debug(f"Response content: {response.text[:1000]}...")  # Log first 1000 characters
+                logging.info(f"Full response content: {response.text}")  # Log the full response content
 
                 if response.status_code == 201:
                     if response.content and response.content != b'null':
                         json_response = response.json()
-                        logging.info(f"Upload successful. CID: {json_response.get('cid', 'N/A')}")
-                        return {"model_cid": json_response.get("ipfs_cid"), "size": json_response.get("size")}
+                        logging.info(f"JSON response: {json_response}")  # Log the parsed JSON response
+                        logging.info(f"Upload successful. CID: {json_response.get('ipfsCid', 'N/A')}")
+                        result = {"model_cid": json_response.get("ipfsCid"), "size": json_response.get("size")}
                     else:
                         logging.warning("Empty or null response content received. Assuming upload was successful.")
-                        return {"model_cid": None, "size": None}
+                        result = {"model_cid": None, "size": None}
                 elif response.status_code == 500:
                     error_message = "Internal server error occurred. Please try again later or contact support."
                     logging.error(error_message)
@@ -267,6 +267,8 @@ class Client:
                     logging.error(f"Upload failed with status code {response.status_code}: {error_message}")
                     raise OpenGradientError(f"Upload failed: {error_message}", status_code=response.status_code)
 
+                return result
+
         except requests.RequestException as e:
             logging.error(f"Request exception during upload: {str(e)}")
             if hasattr(e, 'response') and e.response is not None:

{opengradient-0.2.5.dist-info → opengradient-0.2.6.dist-info}/METADATA

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

opengradient-0.2.6.dist-info/RECORD

@@ -0,0 +1,13 @@
+opengradient/__init__.py,sha256=sImAqtmJU03jhmPXCye37bLu_KGILDCmCRLpS8S2HcI,1783
+opengradient/cli.py,sha256=U0hEQEab5-D-Abkxs1WmV4Qbalth80MBwBDnOCK4hWc,7354
+opengradient/client.py,sha256=tPHTi4rZiZ5Kf_XKdfD3uJZFP2j4AO4ld2lJS38nj24,18026
+opengradient/exceptions.py,sha256=v4VmUGTvvtjhCZAhR24Ga42z3q-DzR1Y5zSqP_yn2Xk,3366
+opengradient/types.py,sha256=EoJN-DkQrJ2WTUv8OenlrlWJWFY2jPGTl-T8C_OVjp8,1849
+opengradient/utils.py,sha256=95i5RVn-32MRsn00M21io8QHLtmEAoRbgueMhDh0TVk,5079
+opengradient/abi/inference.abi,sha256=HH2SmCJ_D4O0I-CFsln0vFHd2PU-A-fxgCnUtHg0ZQg,2373
+opengradient-0.2.6.dist-info/LICENSE,sha256=xEcvQ3AxZOtDkrqkys2Mm6Y9diEnaSeQRKvxi-JGnNA,1069
+opengradient-0.2.6.dist-info/METADATA,sha256=au_A8g75nxOpqCoo6zuHBQmKVNwxu_9sUGBMlVI1ybY,3000
+opengradient-0.2.6.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+opengradient-0.2.6.dist-info/entry_points.txt,sha256=yUKTaJx8RXnybkob0J62wVBiCp_1agVbgw9uzsmaeJc,54
+opengradient-0.2.6.dist-info/top_level.txt,sha256=oC1zimVLa2Yi1LQz8c7x-0IQm92milb5ax8gHBHwDqU,13
+opengradient-0.2.6.dist-info/RECORD,,

opengradient-0.2.5.dist-info/RECORD

@@ -1,13 +0,0 @@
-opengradient/__init__.py,sha256=KEs28Cnki2uQk3XMb2z2U0wPI7q5JXn0HToJMUmQDZI,1783
-opengradient/cli.py,sha256=U0hEQEab5-D-Abkxs1WmV4Qbalth80MBwBDnOCK4hWc,7354
-opengradient/client.py,sha256=hmBIb_UXMfcAXIfDAd9Baxwd9SDsLzPLBFaH8efS1IY,17978
-opengradient/exceptions.py,sha256=v4VmUGTvvtjhCZAhR24Ga42z3q-DzR1Y5zSqP_yn2Xk,3366
-opengradient/types.py,sha256=EoJN-DkQrJ2WTUv8OenlrlWJWFY2jPGTl-T8C_OVjp8,1849
-opengradient/utils.py,sha256=95i5RVn-32MRsn00M21io8QHLtmEAoRbgueMhDh0TVk,5079
-opengradient/abi/inference.abi,sha256=HH2SmCJ_D4O0I-CFsln0vFHd2PU-A-fxgCnUtHg0ZQg,2373
-opengradient-0.2.5.dist-info/LICENSE,sha256=xEcvQ3AxZOtDkrqkys2Mm6Y9diEnaSeQRKvxi-JGnNA,1069
-opengradient-0.2.5.dist-info/METADATA,sha256=VaOmJ1SXmpooKgm1zJKTgXjkSOI4r5YBG4-qs3qxAKs,3000
-opengradient-0.2.5.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-opengradient-0.2.5.dist-info/entry_points.txt,sha256=yUKTaJx8RXnybkob0J62wVBiCp_1agVbgw9uzsmaeJc,54
-opengradient-0.2.5.dist-info/top_level.txt,sha256=oC1zimVLa2Yi1LQz8c7x-0IQm92milb5ax8gHBHwDqU,13
-opengradient-0.2.5.dist-info/RECORD,,