pygeai 0.4.0b6__py3-none-any.whl → 0.4.0b8__py3-none-any.whl

pygeai/core/common/exceptions.py

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

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
         )
-
+        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,6 +168,9 @@ class ToolClient(AILabClient):
                 "allowExternal": allow_external
             }
         )
+        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:
@@ -207,6 +212,9 @@ class ToolClient(AILabClient):
                 "allowDrafts": allow_drafts,
             }
         )
+        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:
@@ -247,6 +255,9 @@ class ToolClient(AILabClient):
             endpoint=endpoint,
             headers=headers
         )
+        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}")
@@ -256,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,
@@ -313,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:
@@ -362,7 +375,9 @@ class ToolClient(AILabClient):
             headers=headers,
             data=data
         )
-
+        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:
@@ -448,6 +463,9 @@ class ToolClient(AILabClient):
                 "allowDrafts": allow_drafts,
             }
         )
+        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:
@@ -502,6 +520,9 @@ class ToolClient(AILabClient):
             headers=headers,
             data=data
         )
+        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:
@@ -538,6 +559,9 @@ class ToolClient(AILabClient):
             endpoint=endpoint,
             headers=headers,
         )
+        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.0b6.dist-info → pygeai-0.4.0b8.dist-info}/METADATA

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

{pygeai-0.4.0b6.dist-info → pygeai-0.4.0b8.dist-info}/RECORD

@@ -77,7 +77,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
@@ -164,7 +164,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=cjoNIdCTYqVIw4MpEnwtFt4HSGl1snYFSEEOo_jZWUs,26515
+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 +495,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.0b6.dist-info/licenses/LICENSE,sha256=eHfqo7-AWS8cMq0cg03lq7owsLeCmZA-xS5L0kuHnl8,1474
-pygeai-0.4.0b6.dist-info/METADATA,sha256=Vw_S0OHG34fpDSiQqF_VexSPvyelYRGqn-Pf5og7Ack,6940
-pygeai-0.4.0b6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pygeai-0.4.0b6.dist-info/entry_points.txt,sha256=OAmwuXVCQBTCE3HeVegVd37hbhCcp9TPahvdrCuMYWw,178
-pygeai-0.4.0b6.dist-info/top_level.txt,sha256=bJFwp2tURmCfB94yXDF7ylvdSJXFDDJsyUOb-7PJgwc,7
-pygeai-0.4.0b6.dist-info/RECORD,,
+pygeai-0.4.0b8.dist-info/licenses/LICENSE,sha256=eHfqo7-AWS8cMq0cg03lq7owsLeCmZA-xS5L0kuHnl8,1474
+pygeai-0.4.0b8.dist-info/METADATA,sha256=L9KOzdMh6Dkg_M1dcgADWiJUDlF9VGV21f_4iUwvx6U,6940
+pygeai-0.4.0b8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pygeai-0.4.0b8.dist-info/entry_points.txt,sha256=OAmwuXVCQBTCE3HeVegVd37hbhCcp9TPahvdrCuMYWw,178
+pygeai-0.4.0b8.dist-info/top_level.txt,sha256=bJFwp2tURmCfB94yXDF7ylvdSJXFDDJsyUOb-7PJgwc,7
+pygeai-0.4.0b8.dist-info/RECORD,,