pygeai 0.2.7b38__tar.gz → 0.2.7b39__tar.gz

{pygeai-0.2.7b38/pygeai.egg-info → pygeai-0.2.7b39}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pygeai
-Version: 0.2.7b38
+Version: 0.2.7b39
 Summary: Software Development Kit to interact with Globant Enterprise AI.
 Author-email: Globant <geai-sdk@globant.com>
 License-Expression: MIT

{pygeai-0.2.7b38 → pygeai-0.2.7b39}/pygeai/core/files/managers.py

@@ -1,4 +1,5 @@
 from pygeai import logger
+from pygeai.admin.clients import AdminClient
 from pygeai.core.base.mappers import ErrorMapper, ResponseMapper
 from pygeai.core.base.responses import EmptyResponse
 from pygeai.core.files.clients import FileClient
@@ -28,8 +29,26 @@ class FileManager:
             base_url,
             alias
         )
-        self.__organization_id = organization_id
-        self.__project_id = project_id
+        self.organization_id = self.__get_organization_id() if not organization_id else organization_id
+        self.project_id = self.__get_project_id() if not project_id else project_id
+
+    def __get_organization_id(self):
+        response = None
+        try:
+            response = AdminClient().validate_api_token()
+            return response.get("organizationId")
+        except Exception as e:
+            logger.error(f"Error retrieving organization_id from GEAI. Response: {response}: {e}")
+            raise APIError(f"Error retrieving organization_id from GEAI: {e}")
+
+    def __get_project_id(self):
+        response = None
+        try:
+            response = AdminClient().validate_api_token()
+            return response.get("projectId")
+        except Exception as e:
+            logger.error(f"Error retrieving project_id from GEAI. Response: {response}: {e}")
+            raise APIError(f"Error retrieving project_id from GEAI: {e}")
 
     def upload_file(
             self,
@@ -47,8 +66,8 @@ class FileManager:
         """
         response_data = self.__client.upload_file(
             file_path=file.path,
-            organization_id=self.__organization_id,
-            project_id=self.__project_id,
+            organization_id=self.organization_id,
+            project_id=self.project_id,
             folder=file.folder,
             file_name=file.name,
         )
@@ -75,8 +94,8 @@ class FileManager:
         :raises APIError: If the API returns errors.
         """
         response_data = self.__client.get_file(
-            organization=self.__organization_id,
-            project=self.__project_id,
+            organization=self.organization_id,
+            project=self.project_id,
             file_id=file_id
         )
         if ErrorHandler.has_errors(response_data):
@@ -101,8 +120,8 @@ class FileManager:
         :raises APIError: If the API returns errors.
         """
         response_data = self.__client.delete_file(
-            organization=self.__organization_id,
-            project=self.__project_id,
+            organization=self.organization_id,
+            project=self.project_id,
             file_id=file_id
         )
         if ErrorHandler.has_errors(response_data):
@@ -128,8 +147,8 @@ class FileManager:
         :raises APIError: If the API returns errors.
         """
         response_data = self.__client.get_file_content(
-            organization=self.__organization_id,
-            project=self.__project_id,
+            organization=self.organization_id,
+            project=self.project_id,
             file_id=file_id
         )
         if isinstance(response_data, dict) and "errors" in response_data:
@@ -151,8 +170,8 @@ class FileManager:
         :raises APIError: If the API returns errors.
         """
         response_data = self.__client.get_file_list(
-            organization=self.__organization_id,
-            project=self.__project_id
+            organization=self.organization_id,
+            project=self.project_id
         )
         if ErrorHandler.has_errors(response_data):
             error = ErrorHandler.extract_error(response_data)

pygeai-0.2.7b39/pygeai/tests/snippets/files/delete_file.py

@@ -0,0 +1,9 @@
+from pygeai.core.files.managers import FileManager
+from pygeai.core.files.models import File
+
+file = File(id="7fe2393d-8f86-4020-a51f-bc14ab957e1e")
+
+file_manager = FileManager()
+
+response = file_manager.delete_file(file_id=file.id)
+print(response)

pygeai-0.2.7b39/pygeai/tests/snippets/files/get_file_content.py

@@ -0,0 +1,10 @@
+from pygeai.core.files.managers import FileManager
+from pygeai.core.files.models import File
+
+
+file = File(id="7fe2393d-8f86-4020-a51f-bc14ab957e1e")
+
+file_manager = FileManager()
+
+response = file_manager.get_file_content(file_id=file.id)
+print(response)

pygeai-0.2.7b39/pygeai/tests/snippets/files/get_file_data.py

@@ -0,0 +1,9 @@
+from pygeai.core.files.managers import FileManager
+from pygeai.core.files.models import File
+
+file = File(id="9984b837-fe88-4014-ad14-91e1596c8ead")
+
+file_manager = FileManager()
+
+response = file_manager.get_file_data(file_id=file.id)
+print(response)

pygeai-0.2.7b39/pygeai/tests/snippets/files/get_file_list.py

@@ -0,0 +1,6 @@
+from pygeai.core.files.managers import FileManager
+
+file_manager = FileManager()
+
+response = file_manager.get_file_list()
+print(response)

pygeai-0.2.7b39/pygeai/tests/snippets/files/upload_file.py

@@ -0,0 +1,13 @@
+from pygeai.core.files.managers import FileManager
+from pygeai.core.files.models import UploadFile
+
+file = UploadFile(
+    path="test.txt",
+    name="TestyFile",
+    folder="TestyTestTemp"
+)
+
+file_manager = FileManager()
+
+response = file_manager.upload_file(file)
+print(response)

{pygeai-0.2.7b38 → pygeai-0.2.7b39/pygeai.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pygeai
-Version: 0.2.7b38
+Version: 0.2.7b39
 Summary: Software Development Kit to interact with Globant Enterprise AI.
 Author-email: Globant <geai-sdk@globant.com>
 License-Expression: MIT

{pygeai-0.2.7b38 → pygeai-0.2.7b39}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pygeai"
-version = "0.2.7b38"
+version = "0.2.7b39"
 description = "Software Development Kit to interact with Globant Enterprise AI."
 license = "MIT"
 requires-python = ">=3.10"

pygeai-0.2.7b38/pygeai/tests/snippets/files/delete_file.py

@@ -1,12 +0,0 @@
-from pygeai.core.models import Organization, Project
-from pygeai.core.files.managers import FileManager
-from pygeai.core.files.models import File
-
-organization = Organization(id="4aa15b61-d3c7-4a5c-99b8-052d18a04ff2")
-project = Project(id="1956c032-3c66-4435-acb8-6a06e52f819f")
-file = File(id="7fe2393d-8f86-4020-a51f-bc14ab957e1e")
-
-file_manager = FileManager(organization_id=organization.id, project_id=project.id)
-
-response = file_manager.delete_file(file_id=file.id)
-print(response)

pygeai-0.2.7b38/pygeai/tests/snippets/files/get_file_content.py

@@ -1,12 +0,0 @@
-from pygeai.core.models import Organization, Project
-from pygeai.core.files.managers import FileManager
-from pygeai.core.files.models import File
-
-organization = Organization(id="4aa15b61-d3c7-4a5c-99b8-052d18a04ff2")
-project = Project(id="1956c032-3c66-4435-acb8-6a06e52f819f")
-file = File(id="7fe2393d-8f86-4020-a51f-bc14ab957e1e")
-
-file_manager = FileManager(organization_id=organization.id, project_id=project.id)
-
-response = file_manager.get_file_content(file_id=file.id)
-print(response)

pygeai-0.2.7b38/pygeai/tests/snippets/files/get_file_data.py

@@ -1,12 +0,0 @@
-from pygeai.core.models import Organization, Project
-from pygeai.core.files.managers import FileManager
-from pygeai.core.files.models import File
-
-organization = Organization(id="4aa15b61-d3c7-4a5c-99b8-052d18a04ff2")
-project = Project(id="1956c032-3c66-4435-acb8-6a06e52f819f")
-file = File(id="9984b837-fe88-4014-ad14-91e1596c8ead")
-
-file_manager = FileManager(organization_id=organization.id, project_id=project.id)
-
-response = file_manager.get_file_data(file_id=file.id)
-print(response)

pygeai-0.2.7b38/pygeai/tests/snippets/files/get_file_list.py

@@ -1,11 +0,0 @@
-from pygeai.core.models import Organization, Project
-from pygeai.core.files.managers import FileManager
-
-organization = Organization(id="4aa15b61-d3c7-4a5c-99b8-052d18a04ff2")
-project = Project(id="1956c032-3c66-4435-acb8-6a06e52f819f")
-
-
-file_manager = FileManager(organization_id=organization.id, project_id=project.id)
-
-response = file_manager.get_file_list()
-print(response)

pygeai-0.2.7b38/pygeai/tests/snippets/files/upload_file.py

@@ -1,17 +0,0 @@
-from pygeai.core.models import Organization, Project
-from pygeai.core.files.managers import FileManager
-from pygeai.core.files.models import UploadFile
-
-organization = Organization(id="4aa15b61-d3c7-4a5c-99b8-052d18a04ff2")
-project = Project(id="1956c032-3c66-4435-acb8-6a06e52f819f")
-
-file = UploadFile(
-    path="test.txt",
-    name="TestyFile",
-    folder="TestyTestTemp"
-)
-
-file_manager = FileManager(organization_id=organization.id, project_id=project.id)
-
-response = file_manager.upload_file(file)
-print(response)