pygeai 0.6.0b6__py3-none-any.whl → 0.6.0b10__py3-none-any.whl

pygeai/lab/tools/clients.py

@@ -1,11 +1,12 @@
 import json
-from json import JSONDecodeError
 
 from pygeai import logger
 from pygeai.core.common.exceptions import InvalidAPIResponseException, MissingRequirementException, APIResponseError
+from pygeai.core.utils.validators import validate_status_code
+from pygeai.core.utils.parsers import parse_json_response
 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
+    PUBLISH_TOOL_REVISION_V2, GET_PARAMETER_V2, SET_PARAMETER_V2, DELETE_TOOL_V2, EXPORT_TOOL_V4
 from pygeai.lab.clients import AILabClient
 
 
@@ -115,16 +116,9 @@ class ToolClient(AILabClient):
             data=data
         )
         
-        if response.status_code >= 300:
-            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:
-            logger.error(f"Unable to create tool for project {self.project_id}: JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to create tool for project {self.project_id}: {response.text}")
+        validate_status_code(response)
+        return parse_json_response(response, f"create tool for project {self.project_id}")
 
-        return result
 
     def list_tools(
             self,
@@ -169,16 +163,9 @@ class ToolClient(AILabClient):
                 "allowExternal": allow_external
             }
         )
-        if response.status_code >= 300:
-            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:
-            logger.error(f"Unable to list tools for project {self.project_id}: JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to list tools for project {self.project_id}: {response.text}")
+        validate_status_code(response)
+        return parse_json_response(response, f"list tools for project {self.project_id}")
 
-        return result
 
     def get_tool(
             self,
@@ -213,16 +200,9 @@ class ToolClient(AILabClient):
                 "allowDrafts": allow_drafts,
             }
         )
-        if response.status_code >= 300:
-            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:
-            logger.error(f"Unable to retrieve tool {tool_id} for project {self.project_id}: JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to retrieve tool {tool_id} for project {self.project_id}: {response.text}")
+        validate_status_code(response)
+        return parse_json_response(response, f"retrieve tool {tool_id} for project {self.project_id}")
 
-        return result
 
     def delete_tool(
             self,
@@ -256,9 +236,7 @@ class ToolClient(AILabClient):
             endpoint=endpoint,
             headers=headers
         )
-        if response.status_code >= 300:
-            logger.error(f"Invalid status code returned from the API endpoint: {response.text}")
-            raise APIResponseError(f"API returned an error: {response.text}")
+        validate_status_code(response)
 
         if response.status_code != 204:
             logger.error(f"Unable to delete tool {tool_id or tool_name} in project {self.project_id}: JSON parsing error (status {response.status_code}). Response: {response.text}")
@@ -376,16 +354,9 @@ class ToolClient(AILabClient):
             headers=headers,
             data=data
         )
-        if response.status_code >= 300:
-            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:
-            logger.error(f"Unable to update tool {tool_id} in project {self.project_id}: JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to update tool {tool_id} in project {self.project_id}: {response.text}")
+        validate_status_code(response)
+        return parse_json_response(response, f"update tool {tool_id} in project {self.project_id}")
 
-        return result
 
     def publish_tool_revision(
             self,
@@ -414,13 +385,9 @@ class ToolClient(AILabClient):
                 "revision": revision,
             }
         )
-        try:
-            result = response.json()
-        except JSONDecodeError as e:
-            logger.error(f"Unable to publish revision {revision} for tool {tool_id} in project {self.project_id}: JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to publish revision {revision} for tool {tool_id} in project {self.project_id}: {response.text}")
+        validate_status_code(response)
+        return parse_json_response(response, f"publish revision {revision} for tool {tool_id} in project {self.project_id}")
 
-        return result
 
     def get_parameter(
             self,
@@ -464,16 +431,10 @@ class ToolClient(AILabClient):
                 "allowDrafts": allow_drafts,
             }
         )
-        if response.status_code >= 300:
-            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:
-            logger.error(f"Unable to retrieve parameters for tool {tool_id or tool_public_name} in project {self.project_id}: JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to retrieve parameters for tool {tool_id or tool_public_name} in project {self.project_id}: {response.text}")
+        validate_status_code(response)
+        tool_identifier = tool_id or tool_public_name
+        return parse_json_response(response, f"retrieve parameters for tool {tool_identifier} in project {self.project_id}")
 
-        return result
 
     def set_parameter(
             self,
@@ -527,6 +488,7 @@ class ToolClient(AILabClient):
         else:
             return {}
 
+    '''
     def export_tool(
             self,
             tool_id: str,
@@ -543,7 +505,7 @@ class ToolClient(AILabClient):
         if not tool_id:
             raise MissingRequirementException("tool_id must be specified in order to retrieve the tool")
 
-        endpoint = EXPORT_TOOL_V2.format(toolId=tool_id)
+        endpoint = EXPORT_TOOL_V4.format(toolId=tool_id)
         headers = {
             "Authorization": self.api_service.token,
             "ProjectId": self.project_id
@@ -555,13 +517,7 @@ class ToolClient(AILabClient):
             endpoint=endpoint,
             headers=headers,
         )
-        if response.status_code >= 300:
-            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:
-            logger.error(f"Unable to export tool {tool_id} for project {self.project_id}: JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to export tool {tool_id} for project {self.project_id}: {response.text}")
-
-        return result
+        validate_status_code(response)
+        return parse_json_response(response, f"export tool {tool_id} for project {self.project_id}")
+
+    '''

pygeai/lab/tools/endpoints.py

@@ -8,3 +8,6 @@ PUBLISH_TOOL_REVISION_V2 = "v2/tools/{toolId}/publish-revision"  # POST -> Publi
 GET_PARAMETER_V2 = "v2/tools/{toolPublicName}/config"  # GET -> Get tool parameter
 SET_PARAMETER_V2 = "v2/tools/{toolPublicName}/config"  # POST -> Set tool parameter
 EXPORT_TOOL_V2 = "v2/tools/{toolId}/export"  # GET -> Export tool
+IMPORT_TOOL_V2 = "v2/tools/import"  # POST -> Import tool
+EXPORT_TOOL_V4 = "v4/tools/{toolId}/export"  # GET -> Export tool
+IMPORT_TOOL_V4 = "v4/tools/import"  # POST -> Import tool

pygeai/migration/__init__.py

@@ -0,0 +1,31 @@
+from pygeai.migration.strategies import (
+    MigrationStrategy,
+    ProjectMigrationStrategy,
+    AgentMigrationStrategy,
+    ToolMigrationStrategy,
+    AgenticProcessMigrationStrategy,
+    TaskMigrationStrategy,
+    UsageLimitMigrationStrategy,
+    RAGAssistantMigrationStrategy,
+    FileMigrationStrategy
+)
+from pygeai.migration.tools import (
+    MigrationTool,
+    MigrationPlan,
+    MigrationOrchestrator
+)
+
+__all__ = [
+    "MigrationStrategy",
+    "ProjectMigrationStrategy",
+    "AgentMigrationStrategy",
+    "ToolMigrationStrategy",
+    "AgenticProcessMigrationStrategy",
+    "TaskMigrationStrategy",
+    "UsageLimitMigrationStrategy",
+    "RAGAssistantMigrationStrategy",
+    "FileMigrationStrategy",
+    "MigrationTool",
+    "MigrationPlan",
+    "MigrationOrchestrator"
+]