pygeai 0.4.0b7__py3-none-any.whl → 0.4.0b9__py3-none-any.whl

pygeai/cli/commands/lab/options.py

@@ -0,0 +1,8 @@
+from pygeai.cli.commands import Option
+
+PROJECT_ID_OPTION = Option(
+    "project_id",
+    ["--project-id", "--pid"],
+    "GUID of the project (optional)",
+    True
+)

pygeai/core/common/exceptions.py

@@ -55,5 +55,6 @@ class InvalidAgentException(GEAIException):
     pass
 
 
-
+class APIResponseError(GEAIException):
+    pass
 

pygeai/lab/clients.py

@@ -6,9 +6,9 @@ from pygeai.core.common.exceptions import APIError
 
 class AILabClient(BaseClient):
 
-    def __init__(self, api_key: str = None, base_url: str = None, alias: str = None):
+    def __init__(self, api_key: str = None, base_url: str = None, alias: str = None, project_id: str = None):
         super().__init__(api_key, base_url, alias)
-        self.project_id = self.__get_project_id()
+        self.project_id = project_id if project_id else self.__get_project_id()
 
     def __get_project_id(self):
         response = None

pygeai/lab/managers.py

@@ -22,11 +22,11 @@ from pygeai.lab.tools.mappers import ToolMapper
 
 class AILabManager:
 
-    def __init__(self, api_key: str = None, base_url: str = None, alias: str = "default"):
-        self.__agent_client = AgentClient(api_key=api_key, base_url=base_url, alias=alias)
-        self.__tool_client = ToolClient(api_key=api_key, base_url=base_url, alias=alias)
-        self.__reasoning_strategy_client = ReasoningStrategyClient(api_key=api_key, base_url=base_url, alias=alias)
-        self.__process_client = AgenticProcessClient(api_key=api_key, base_url=base_url, alias=alias)
+    def __init__(self, api_key: str = None, base_url: str = None, alias: str = "default", project_id: str = None):
+        self.__agent_client = AgentClient(api_key=api_key, base_url=base_url, alias=alias, project_id=project_id)
+        self.__tool_client = ToolClient(api_key=api_key, base_url=base_url, alias=alias, project_id=project_id)
+        self.__reasoning_strategy_client = ReasoningStrategyClient(api_key=api_key, base_url=base_url, alias=alias, project_id=project_id)
+        self.__process_client = AgenticProcessClient(api_key=api_key, base_url=base_url, alias=alias, project_id=project_id)
 
     def get_agent_list(
             self,

pygeai/lab/tools/clients.py

@@ -2,7 +2,7 @@ import json
 from json import JSONDecodeError
 
 from pygeai import logger
-from pygeai.core.common.exceptions import InvalidAPIResponseException, MissingRequirementException
+from pygeai.core.common.exceptions import InvalidAPIResponseException, MissingRequirementException, APIResponseError
 from pygeai.lab.constants import VALID_SCOPES, VALID_ACCESS_SCOPES, VALID_REPORT_EVENTS
 from pygeai.lab.tools.endpoints import CREATE_TOOL_V2, LIST_TOOLS_V2, GET_TOOL_V2, UPDATE_TOOL_V2, UPSERT_TOOL_V2, \
     PUBLISH_TOOL_REVISION_V2, GET_PARAMETER_V2, SET_PARAMETER_V2, DELETE_TOOL_V2, EXPORT_TOOL_V2
@@ -114,7 +114,9 @@ class ToolClient(AILabClient):
             headers=headers,
             data=data
         )
-        response.raise_for_status()
+        if response.status_code != 200:
+            logger.error(f"Invalid status code returned from the API endpoint: {response.text}")
+            raise APIResponseError(f"API returned an error: {response.text}")
         try:
             result = response.json()
         except JSONDecodeError as e:
@@ -166,7 +168,9 @@ class ToolClient(AILabClient):
                 "allowExternal": allow_external
             }
         )
-        response.raise_for_status()
+        if response.status_code != 200:
+            logger.error(f"Invalid status code returned from the API endpoint: {response.text}")
+            raise APIResponseError(f"API returned an error: {response.text}")
         try:
             result = response.json()
         except JSONDecodeError as e:
@@ -208,7 +212,9 @@ class ToolClient(AILabClient):
                 "allowDrafts": allow_drafts,
             }
         )
-        response.raise_for_status()
+        if response.status_code != 200:
+            logger.error(f"Invalid status code returned from the API endpoint: {response.text}")
+            raise APIResponseError(f"API returned an error: {response.text}")
         try:
             result = response.json()
         except JSONDecodeError as e:
@@ -249,7 +255,9 @@ class ToolClient(AILabClient):
             endpoint=endpoint,
             headers=headers
         )
-        response.raise_for_status()
+        if response.status_code != 200:
+            logger.error(f"Invalid status code returned from the API endpoint: {response.text}")
+            raise APIResponseError(f"API returned an error: {response.text}")
 
         if response.status_code != 204:
             logger.error(f"Unable to delete tool {tool_id or tool_name} from project {self.project_id}: JSON parsing error (status {response.status_code}). Response: {response.text}")
@@ -259,7 +267,7 @@ class ToolClient(AILabClient):
 
     def update_tool(
             self,
-            tool_id: str,
+            tool_id: str = None,
             name: str = None,
             description: str = None,
             scope: str = None,
@@ -316,6 +324,8 @@ class ToolClient(AILabClient):
         :raises JSONDecodeError: Caught internally if the response cannot be parsed as JSON; returns raw response text.
         :raises Exception: May be raised by `api_service.put` for network issues, authentication errors, or server-side problems (not caught here).
         """
+        if not (tool_id or name):
+            raise ValueError(f"Either tool ID or tool Name must be defined in order to update tool.")
         if scope and scope not in VALID_SCOPES:
             raise ValueError(f"Scope must be one of {', '.join(VALID_SCOPES)}.")
         if access_scope and access_scope not in VALID_ACCESS_SCOPES:
@@ -365,7 +375,9 @@ class ToolClient(AILabClient):
             headers=headers,
             data=data
         )
-        response.raise_for_status()
+        if response.status_code != 200:
+            logger.error(f"Invalid status code returned from the API endpoint: {response.text}")
+            raise APIResponseError(f"API returned an error: {response.text}")
         try:
             result = response.json()
         except JSONDecodeError as e:
@@ -451,7 +463,9 @@ class ToolClient(AILabClient):
                 "allowDrafts": allow_drafts,
             }
         )
-        response.raise_for_status()
+        if response.status_code != 200:
+            logger.error(f"Invalid status code returned from the API endpoint: {response.text}")
+            raise APIResponseError(f"API returned an error: {response.text}")
         try:
             result = response.json()
         except JSONDecodeError as e:
@@ -506,7 +520,9 @@ class ToolClient(AILabClient):
             headers=headers,
             data=data
         )
-        response.raise_for_status()
+        if response.status_code != 200:
+            logger.error(f"Invalid status code returned from the API endpoint: {response.text}")
+            raise APIResponseError(f"API returned an error: {response.text}")
         try:
             result = response.json()
         except JSONDecodeError as e:
@@ -543,7 +559,9 @@ class ToolClient(AILabClient):
             endpoint=endpoint,
             headers=headers,
         )
-        response.raise_for_status()
+        if response.status_code != 200:
+            logger.error(f"Invalid status code returned from the API endpoint: {response.text}")
+            raise APIResponseError(f"API returned an error: {response.text}")
         try:
             result = response.json()
         except JSONDecodeError as e:

{pygeai-0.4.0b7.dist-info → pygeai-0.4.0b9.dist-info}/METADATA

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

{pygeai-0.4.0b7.dist-info → pygeai-0.4.0b9.dist-info}/RECORD

@@ -57,8 +57,9 @@ pygeai/cli/commands/validators.py,sha256=lNtYSJvKFrEeg_h0oYURMuoQbNG5r6QdjzDL-aT
 pygeai/cli/commands/version.py,sha256=vyJcnxwL_TfpOQI0yE2a1ZyA3VRAE7ssh9APNBXpmqk,1701
 pygeai/cli/commands/flows/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pygeai/cli/commands/lab/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pygeai/cli/commands/lab/ai_lab.py,sha256=mS6QtRuCjQvA71CTD657tUYYvtw_mtDWLsGvzARSi2U,122999
+pygeai/cli/commands/lab/ai_lab.py,sha256=Ecvp8E6tlEh56sPsmcAwYF0CjNn3zdJwggs-BmtdKqU,129292
 pygeai/cli/commands/lab/common.py,sha256=YBenPCVgK01Xaxgj1429bp_Ri1SN4beBxZk3dCLp7X0,6590
+pygeai/cli/commands/lab/options.py,sha256=T13Vi97zochr0cU4yjyvvwWRPENILFDYpvqpU4uEBis,165
 pygeai/cli/commands/lab/spec.py,sha256=EjNdEnljKpYPQyT4d4ViAPrM1g7oIitv6ddnWVkWXPk,8301
 pygeai/cli/commands/lab/utils.py,sha256=uxhgHvCRrV6WYRxR2qd3nED_hhXcxJ1tAU8MlzoshEg,456
 pygeai/cli/texts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -77,7 +78,7 @@ pygeai/core/base/session.py,sha256=WVb4MmptwdgK7paHOSvfEle_HPXRRXO8CHgi0qbgtOg,2
 pygeai/core/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pygeai/core/common/config.py,sha256=zrlNF-0AE54qZ-XpR0RgYzM_7v8LtS0jUpM7UXrbSqQ,4475
 pygeai/core/common/decorators.py,sha256=X7Tv5XBmsuS7oZHSmI95eX8UkuukKoiOiNRl5w9lgR4,1227
-pygeai/core/common/exceptions.py,sha256=hfyCj5iLuWJzg93rGrR9wpVESGnq6Dq0HJQBuuitNo0,1204
+pygeai/core/common/exceptions.py,sha256=-eF4V0B-27zfp0aHMlZWqWRIty6P7TCOrzMRW87ZnlE,1251
 pygeai/core/embeddings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pygeai/core/embeddings/clients.py,sha256=0r-BX4ptivIBNrsOAMgw0q5nNLrIU7UxJ3SD6MkfXX4,3543
 pygeai/core/embeddings/endpoints.py,sha256=b__cuKQjribog9PSUeDzwrQ0vBO4WyYahLhLjDiUpL0,98
@@ -143,9 +144,9 @@ pygeai/health/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pygeai/health/clients.py,sha256=U2eb1tkXt1Rf_KdV0ZFQS2k4wGnJTXHHd9-Er0eWQuw,1011
 pygeai/health/endpoints.py,sha256=UAzMcqSXZtMj4r8M8B7a_a5LT6X_jMFNsCTvcsjNTYA,71
 pygeai/lab/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pygeai/lab/clients.py,sha256=mhtJw_cEFC1Cd6KIrcX6KxcRseKN3N_GtjlyUlLaah4,762
+pygeai/lab/clients.py,sha256=5JebyNpKCVwAQpeOAkJPIHRf_6hoKA3uUm9xewm01UQ,816
 pygeai/lab/constants.py,sha256=ddgDnXP4GD0woi-FUJaJXzaWS3H6zmDN0B-v8utM95Q,170
-pygeai/lab/managers.py,sha256=eDnT31n-xPUcK_Hjfe9FXjz6dAwAqmrbaYZ6XEHonQE,70554
+pygeai/lab/managers.py,sha256=_i9tQACTGblKkw05WMu5RLpCYlyL2PtLvNI42F11u8Y,70670
 pygeai/lab/models.py,sha256=1m41gSqpXZVO9AcPVxzlsC-TgxZcCsgGUbpN5zoDMjU,71451
 pygeai/lab/runners.py,sha256=-uaCPHpFyiKtVOxlEjPjAc9h-onSdGAcYJ5IAZPqlb0,4147
 pygeai/lab/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -164,7 +165,7 @@ pygeai/lab/strategies/clients.py,sha256=_a1yc8kwd50Yv4g1jqfa0gRnMiROR7Dn0gx3xqFU
 pygeai/lab/strategies/endpoints.py,sha256=LgEvUgeeN-X6VMl-tpl9_N12GRppLPScQmiMRk7Ri28,541
 pygeai/lab/strategies/mappers.py,sha256=6C_jubAVXMKLGQy5NUD0OX7SlrU2mLe2QsgzeJ1-WKw,2437
 pygeai/lab/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pygeai/lab/tools/clients.py,sha256=KkXLvZ8PRUZHP4X-YrvezdC2B2lf4MZYzRhdOmEbGWw,26801
+pygeai/lab/tools/clients.py,sha256=Jnc4AU_1fnEqhHOsey3AoKpNl1Pfw01qq_hDOmmDvE8,28394
 pygeai/lab/tools/endpoints.py,sha256=HiGoMs4OVeCgH7EAERTtifFPl53NryA1Awh7D6AO8bA,699
 pygeai/lab/tools/mappers.py,sha256=bYi5k36h0k4mCvOnV-r8YOHKz0U9P0mH21GNs20w2eM,4998
 pygeai/man/__init__.py,sha256=gqGI92vUPt6RPweoWX3mTUYPWNDlm6aGUjQOnYXqthk,53
@@ -495,9 +496,9 @@ pygeai/vendor/a2a/utils/helpers.py,sha256=6Tbd8SVfXvdNEk6WYmLOjrAxkzFf1aIg8dkFfB
 pygeai/vendor/a2a/utils/message.py,sha256=gc_EKO69CJ4HkR76IFgsy-kENJz1dn7CfSgWJWvt-gs,2197
 pygeai/vendor/a2a/utils/task.py,sha256=BYRA_L1HpoUGJAVlyHML0lCM9Awhf2Ovjj7oPFXKbh0,1647
 pygeai/vendor/a2a/utils/telemetry.py,sha256=VvSp1Ztqaobkmq9-3sNhhPEilJS32-JTSfKzegkj6FU,10861
-pygeai-0.4.0b7.dist-info/licenses/LICENSE,sha256=eHfqo7-AWS8cMq0cg03lq7owsLeCmZA-xS5L0kuHnl8,1474
-pygeai-0.4.0b7.dist-info/METADATA,sha256=M_uGUknAyMR5q6rhNQZDL43yZWMqhQup34VJVMmhzgU,6940
-pygeai-0.4.0b7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pygeai-0.4.0b7.dist-info/entry_points.txt,sha256=OAmwuXVCQBTCE3HeVegVd37hbhCcp9TPahvdrCuMYWw,178
-pygeai-0.4.0b7.dist-info/top_level.txt,sha256=bJFwp2tURmCfB94yXDF7ylvdSJXFDDJsyUOb-7PJgwc,7
-pygeai-0.4.0b7.dist-info/RECORD,,
+pygeai-0.4.0b9.dist-info/licenses/LICENSE,sha256=eHfqo7-AWS8cMq0cg03lq7owsLeCmZA-xS5L0kuHnl8,1474
+pygeai-0.4.0b9.dist-info/METADATA,sha256=vYLI-sOMXnaeMkEDUFZlbXyx4cgYVgPFJNqsqHuVT90,6940
+pygeai-0.4.0b9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pygeai-0.4.0b9.dist-info/entry_points.txt,sha256=OAmwuXVCQBTCE3HeVegVd37hbhCcp9TPahvdrCuMYWw,178
+pygeai-0.4.0b9.dist-info/top_level.txt,sha256=bJFwp2tURmCfB94yXDF7ylvdSJXFDDJsyUOb-7PJgwc,7
+pygeai-0.4.0b9.dist-info/RECORD,,