gmicloud 0.1.2__py3-none-any.whl → 0.1.3__py3-none-any.whl

gmicloud/_internal/_client/_artifact_client.py

@@ -5,7 +5,6 @@ from ._iam_client import IAMClient
 from ._decorator import handle_refresh_token
 from .._models import *
 from .._config import ARTIFACT_SERVICE_BASE_URL
-from .._constants import ACCESS_TOKEN_HEADER, CLIENT_ID_HEADER
 
 
 class ArtifactClient:
@@ -33,11 +32,7 @@ class ArtifactClient:
         :return: The Artifact object.
         :rtype: Artifact
         """
-        custom_headers = {
-            ACCESS_TOKEN_HEADER: self.iam_client.get_access_token(),
-            CLIENT_ID_HEADER: self.iam_client.get_client_id()
-        }
-        result = self.client.get(f"/get_artifact", custom_headers, {"artifact_id": artifact_id})
+        result = self.client.get(f"/get_artifact", self.iam_client.get_custom_headers(), {"artifact_id": artifact_id})
 
         return Artifact.model_validate(result)
 
@@ -49,11 +44,7 @@ class ArtifactClient:
         :return: A list of Artifact objects.
         :rtype: List[Artifact]
         """
-        custom_headers = {
-            ACCESS_TOKEN_HEADER: self.iam_client.get_access_token(),
-            CLIENT_ID_HEADER: self.iam_client.get_client_id()
-        }
-        result = self.client.get("/get_all_artifacts", custom_headers)
+        result = self.client.get("/get_all_artifacts", self.iam_client.get_custom_headers())
         if not result:
             return []
         return [Artifact.model_validate(item) for item in result]
@@ -67,11 +58,7 @@ class ArtifactClient:
         :return: The response object containing the created artifact details.
         :rtype: CreateArtifactResponse
         """
-        custom_headers = {
-            ACCESS_TOKEN_HEADER: self.iam_client.get_access_token(),
-            CLIENT_ID_HEADER: self.iam_client.get_client_id()
-        }
-        result = self.client.post("/create_artifact", custom_headers, request.model_dump())
+        result = self.client.post("/create_artifact", self.iam_client.get_custom_headers(), request.model_dump())
 
         return CreateArtifactResponse.model_validate(result)
 
@@ -84,11 +71,7 @@ class ArtifactClient:
         :return: The response object containing the created artifact details.
         :rtype: CreateArtifactFromTemplateResponse
         """
-        custom_headers = {
-            ACCESS_TOKEN_HEADER: self.iam_client.get_access_token(),
-            CLIENT_ID_HEADER: self.iam_client.get_client_id()
-        }
-        result = self.client.post("/create_artifact_from_template", custom_headers,
+        result = self.client.post("/create_artifact_from_template", self.iam_client.get_custom_headers(),
                                   {"artifact_template_id": artifact_template_id})
 
         return CreateArtifactFromTemplateResponse.model_validate(result)
@@ -102,11 +85,8 @@ class ArtifactClient:
         :return: The response object containing the rebuilt artifact details.
         :rtype: RebuildArtifactResponse
         """
-        custom_headers = {
-            ACCESS_TOKEN_HEADER: self.iam_client.get_access_token(),
-            CLIENT_ID_HEADER: self.iam_client.get_client_id()
-        }
-        result = self.client.post("/rebuild_artifact", custom_headers, {"artifact_id": artifact_id})
+        result = self.client.post("/rebuild_artifact", self.iam_client.get_custom_headers(),
+                                  {"artifact_id": artifact_id})
 
         return CreateArtifactResponse.model_validate(result)
 
@@ -119,11 +99,8 @@ class ArtifactClient:
         :return: The response object containing the deleted artifact details.
         :rtype: DeleteArtifactResponse
         """
-        custom_headers = {
-            ACCESS_TOKEN_HEADER: self.iam_client.get_access_token(),
-            CLIENT_ID_HEADER: self.iam_client.get_client_id()
-        }
-        result = self.client.delete("/delete_artifact", custom_headers, {"artifact_id": artifact_id})
+        result = self.client.delete("/delete_artifact", self.iam_client.get_custom_headers(),
+                                    {"artifact_id": artifact_id})
 
         return DeleteArtifactResponse.model_validate(result)
 
@@ -136,11 +113,7 @@ class ArtifactClient:
         :return: The response object containing the pre-signed URL and upload details.
         :rtype: GetBigFileUploadUrlResponse
         """
-        custom_headers = {
-            ACCESS_TOKEN_HEADER: self.iam_client.get_access_token(),
-            CLIENT_ID_HEADER: self.iam_client.get_client_id()
-        }
-        result = self.client.post("/get_bigfile_upload_url", custom_headers, request.model_dump())
+        result = self.client.post("/get_bigfile_upload_url", self.iam_client.get_custom_headers(), request.model_dump())
 
         return GetBigFileUploadUrlResponse.model_validate(result)
 
@@ -153,11 +126,7 @@ class ArtifactClient:
         :return: The response object containing the deletion status.
         :rtype: DeleteBigfileResponse
         """
-        custom_headers = {
-            ACCESS_TOKEN_HEADER: self.iam_client.get_access_token(),
-            CLIENT_ID_HEADER: self.iam_client.get_client_id()
-        }
-        result = self.client.delete("/delete_bigfile", custom_headers, request.dict())
+        result = self.client.delete("/delete_bigfile", self.iam_client.get_custom_headers(), request.dict())
 
         return DeleteBigfileResponse.model_validate(result)
 
@@ -169,9 +138,5 @@ class ArtifactClient:
         :return: A list of ArtifactTemplate objects.
         :rtype: List[ArtifactTemplate]
         """
-        custom_headers = {
-            ACCESS_TOKEN_HEADER: self.iam_client.get_access_token(),
-            CLIENT_ID_HEADER: self.iam_client.get_client_id()
-        }
-        result = self.client.get("/get_artifact_templates", custom_headers)
+        result = self.client.get("/get_artifact_templates", self.iam_client.get_custom_headers())
         return GetArtifactTemplatesResponse.model_validate(result)

gmicloud/_internal/_client/_http_client.py

@@ -62,6 +62,7 @@ class HTTPClient:
                 raise APIError(f"HTTP Request failed: {error_message}")
             # Raise for HTTP errors
             response.raise_for_status()
+            print(response.text)
 
         except requests.exceptions.RequestException as e:
             raise APIError(f"HTTP Request failed: {str(e)}")

gmicloud/_internal/_client/_iam_client.py

@@ -3,7 +3,7 @@ import jwt
 from ._http_client import HTTPClient
 from .._config import IAM_SERVICE_BASE_URL
 from .._models import *
-from .._constants import CLIENT_ID_HEADER
+from .._constants import CLIENT_ID_HEADER, AUTHORIZATION_HEADER
 
 
 class IAMClient:
@@ -49,7 +49,7 @@ class IAMClient:
         custom_headers = {
             CLIENT_ID_HEADER: self._client_id
         }
-        result = self.client.patch("/me/sessions", custom_headers, {"refreshToken": self.refresh_token})
+        result = self.client.patch("/me/sessions", custom_headers, {"refreshToken": self._refresh_token})
 
         resp = LoginResponse.model_validate(result)
         self._access_token = resp.accessToken
@@ -90,3 +90,12 @@ class IAMClient:
         Gets the current client ID.
         """
         return self._client_id
+
+    def get_custom_headers(self) -> dict:
+        """
+        Gets the custom headers for the IAM client.
+        """
+        return {
+            AUTHORIZATION_HEADER: f'Bearer {self._access_token}',
+            CLIENT_ID_HEADER: self._client_id
+        }

gmicloud/_internal/_client/_task_client.py

@@ -3,7 +3,6 @@ from ._decorator import handle_refresh_token
 from ._iam_client import IAMClient
 from .._config import TASK_SERVICE_BASE_URL
 from .._models import *
-from .._constants import ACCESS_TOKEN_HEADER, CLIENT_ID_HEADER
 
 
 class TaskClient:
@@ -30,11 +29,7 @@ class TaskClient:
         :return: An instance of Task containing the details of the retrieved task.
         :rtype: Task
         """
-        custom_headers = {
-            ACCESS_TOKEN_HEADER: self.iam_client.get_access_token(),
-            CLIENT_ID_HEADER: self.iam_client.get_client_id()
-        }
-        result = self.client.get("/get_task", custom_headers, {"task_id": task_id})
+        result = self.client.get("/get_task", self.iam_client.get_custom_headers(), {"task_id": task_id})
 
         return Task.model_validate(result)
 
@@ -46,11 +41,7 @@ class TaskClient:
         :return: An instance of GetAllTasksResponse containing the retrieved tasks.
         :rtype: GetAllTasksResponse
         """
-        custom_headers = {
-            ACCESS_TOKEN_HEADER: self.iam_client.get_access_token(),
-            CLIENT_ID_HEADER: self.iam_client.get_client_id()
-        }
-        result = self.client.get("/get_tasks", custom_headers)
+        result = self.client.get("/get_tasks", self.iam_client.get_custom_headers())
         if not result:
             return GetAllTasksResponse(tasks=[])
 
@@ -63,12 +54,7 @@ class TaskClient:
 
         :param task: The Task object containing the details of the task to be created.
         """
-        custom_headers = {
-            ACCESS_TOKEN_HEADER: self.iam_client.get_access_token(),
-            CLIENT_ID_HEADER: self.iam_client.get_client_id()
-        }
-
-        result = self.client.post("/create_task", custom_headers, task.model_dump())
+        result = self.client.post("/create_task", self.iam_client.get_custom_headers(), task.model_dump())
 
         return CreateTaskResponse.model_validate(result)
 
@@ -79,11 +65,7 @@ class TaskClient:
 
         :param task: The Task object containing the updated task details.
         """
-        custom_headers = {
-            ACCESS_TOKEN_HEADER: self.iam_client.get_access_token(),
-            CLIENT_ID_HEADER: self.iam_client.get_client_id()
-        }
-        self.client.put("/update_schedule", custom_headers, task.model_dump())
+        self.client.put("/update_schedule", self.iam_client.get_custom_headers(), task.model_dump())
 
     @handle_refresh_token
     def start_task(self, task_id: str):
@@ -92,11 +74,7 @@ class TaskClient:
 
         :param task_id: The ID of the task to be started.
         """
-        custom_headers = {
-            ACCESS_TOKEN_HEADER: self.iam_client.get_access_token(),
-            CLIENT_ID_HEADER: self.iam_client.get_client_id()
-        }
-        self.client.post("/start_task", custom_headers, {"task_id": task_id})
+        self.client.post("/start_task", self.iam_client.get_custom_headers(), {"task_id": task_id})
 
     @handle_refresh_token
     def stop_task(self, task_id: str):
@@ -105,11 +83,7 @@ class TaskClient:
 
         :param task_id: The ID of the task to be stopped.
         """
-        custom_headers = {
-            ACCESS_TOKEN_HEADER: self.iam_client.get_access_token(),
-            CLIENT_ID_HEADER: self.iam_client.get_client_id()
-        }
-        self.client.post("/stop_task", custom_headers, {"task_id": task_id})
+        self.client.post("/stop_task", self.iam_client.get_custom_headers(), {"task_id": task_id})
 
     @handle_refresh_token
     def get_usage_data(self, start_timestamp: str, end_timestamp: str) -> GetUsageDataResponse:
@@ -119,11 +93,7 @@ class TaskClient:
         :param start_timestamp: The start timestamp of the usage data.
         :param end_timestamp: The end timestamp of the usage data.
         """
-        custom_headers = {
-            ACCESS_TOKEN_HEADER: self.iam_client.get_access_token(),
-            CLIENT_ID_HEADER: self.iam_client.get_client_id()
-        }
-        result = self.client.get("/get_usage_data", custom_headers,
+        result = self.client.get("/get_usage_data", self.iam_client.get_custom_headers(),
                                  {"start_timestamp": start_timestamp, "end_timestamp": end_timestamp})
 
         return result
@@ -135,8 +105,4 @@ class TaskClient:
 
         :param task_id: The ID of the task to be archived.
         """
-        custom_headers = {
-            ACCESS_TOKEN_HEADER: self.iam_client.get_access_token(),
-            CLIENT_ID_HEADER: self.iam_client.get_client_id()
-        }
-        self.client.post("/archive_task", custom_headers, {"task_id": task_id})
+        self.client.post("/archive_task", self.iam_client.get_custom_headers(), {"task_id": task_id})

gmicloud/_internal/_constants.py

@@ -1,4 +1,5 @@
 ACCEPT_HEADER = 'Accept'
+AUTHORIZATION_HEADER = 'Authorization'
 CONTENT_TYPE_HEADER = 'Content-Type'
 CLIENT_ID_HEADER = 'CE-ClientId'
 ACCESS_TOKEN_HEADER = 'CE-AccessToken'

{gmicloud-0.1.2.dist-info → gmicloud-0.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: gmicloud
-Version: 0.1.2
+Version: 0.1.3
 Summary: GMI Cloud Python SDK
 Author-email: GMI <gmi@gmitec.net>
 License: MIT
@@ -10,10 +10,12 @@ Classifier: Operating System :: OS Independent
 Requires-Python: >=3.6
 Description-Content-Type: text/markdown
 
-# GMICloud SDK
+# GMICloud SDK (Beta)
 
 ## Overview
 
+Before you start: Our service and GPU resource is currenly invite-only so please contact our team (getstarted@gmicloud.ai) to get invited if you don't have one yet.
+
 The GMI Inference Engine SDK provides a Python interface for deploying and managing machine learning models in
 production environments. It allows users to create model artifacts, schedule tasks for serving models, and call
 inference APIs easily.

{gmicloud-0.1.2.dist-info → gmicloud-0.1.3.dist-info}/RECORD

@@ -2,17 +2,17 @@ gmicloud/__init__.py,sha256=AJP9Z3Ba-AClp5P49YL32qH5-XzDTbEXhWZFF1iLP8Q,750
 gmicloud/client.py,sha256=G3sgH7zzODhdW_Ad56xgomDWXKAhnjzZuj1_mEBGYCI,4733
 gmicloud/_internal/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 gmicloud/_internal/_config.py,sha256=qIH76TSyS3MQWe62LHI46RJhDnklNFisdajY75oUAqE,218
-gmicloud/_internal/_constants.py,sha256=EyhjJp_mEIsAuopFyfnRzGRVjQH9jOhU5AQvtBF_IeU,339
+gmicloud/_internal/_constants.py,sha256=Y085dwFlqdFkCf39iBfxz39QiiB7lX59ayNJjB86_m4,378
 gmicloud/_internal/_enums.py,sha256=C4oy6Ps7OGdYd3tK-Wcpi6uXMyAKbOZW8_KykYAKsgw,510
 gmicloud/_internal/_exceptions.py,sha256=hScBq7n2fOit4_umlkabZJchY8zVbWSRfWM2Y0rLCbw,306
 gmicloud/_internal/_models.py,sha256=1yUk1yIyfXybIeR8guwYLbXc8gAKzXRpIZ7_6SOId28,13212
 gmicloud/_internal/_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-gmicloud/_internal/_client/_artifact_client.py,sha256=V8q78L1wys0d2GYJ87m2EFsLUOJG7nsJvAkrHBeXVU8,6895
+gmicloud/_internal/_client/_artifact_client.py,sha256=py6SQ1tVoK365hIIxnla433mpCapActZ6KE6Yakds-8,5588
 gmicloud/_internal/_client/_decorator.py,sha256=sy4gxzsUB6ORXHw5pqmMf7TTlK41Nmu1fhIhK2AIsbY,670
 gmicloud/_internal/_client/_file_upload_client.py,sha256=1JRs4X57S3EScPIP9w2DC1Uo6_Wbcjumcw3nVM7uIGM,4667
-gmicloud/_internal/_client/_http_client.py,sha256=y5u7jn71exFWYRvh81vs_-8WWmueW7WnEkRjPPyPctI,5671
-gmicloud/_internal/_client/_iam_client.py,sha256=57KHyGg_0Vj5AzwhIto0kmbqQYxnixKIOilNO8hCwr0,2698
-gmicloud/_internal/_client/_task_client.py,sha256=G0MqsNDHhdL885jo-isuu9H_Pv_6DLimN7lT-gz2Uv4,5074
+gmicloud/_internal/_client/_http_client.py,sha256=F7qpq6WLp6PJkqusWojdi483LijWBsSIY43NjJcyXNU,5704
+gmicloud/_internal/_client/_iam_client.py,sha256=D7PhQerSzjGGerXsd5Si-e-Wa5f-fJpOACkfWOnkPJg,2979
+gmicloud/_internal/_client/_task_client.py,sha256=JgYewfDy2iblRVrc9sdA8RbcooXyAho79ESuCK8PRWI,3842
 gmicloud/_internal/_manager/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 gmicloud/_internal/_manager/_artifact_manager.py,sha256=cc367Kd-W9Zn2d_wUcG6tT06544HRItIZ5IsFfA5_RQ,12201
 gmicloud/_internal/_manager/_task_manager.py,sha256=QQfpYXFKAAI_FSI--Nxvjlgf_jeVZuVnTuRGTQzrZso,8034
@@ -20,7 +20,7 @@ gmicloud/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 gmicloud/tests/test_artifacts.py,sha256=YiC1HBMS7g491Ra4acTLI9AdwyjXZfnY9f-fNKn2azQ,17108
 gmicloud/tests/test_tasks.py,sha256=AY90zTJdsXk1cxn6Jxhi4TDdwXRiGxz_r_aRk_Jkl8Y,10956
 gmicloud/utils/uninstall_packages.py,sha256=zzuuaJPf39oTXWZ_7tUAGseoxocuCbbkoglJSD5yDrE,1127
-gmicloud-0.1.2.dist-info/METADATA,sha256=Y50szRi4ci9WFSCSn7l0S0IoKX-Tv5nHG0Xb0Vzl76c,6635
-gmicloud-0.1.2.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-gmicloud-0.1.2.dist-info/top_level.txt,sha256=AZimLw3y0WPpLiSiOidZ1gD0dxALh-jQNk4fxC05hYE,9
-gmicloud-0.1.2.dist-info/RECORD,,
+gmicloud-0.1.3.dist-info/METADATA,sha256=OuNXnrFTE_tTDuArqevbW81fKwB--E23YUv6-uj91u0,6808
+gmicloud-0.1.3.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+gmicloud-0.1.3.dist-info/top_level.txt,sha256=AZimLw3y0WPpLiSiOidZ1gD0dxALh-jQNk4fxC05hYE,9
+gmicloud-0.1.3.dist-info/RECORD,,