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

pygeai/lab/processes/clients.py

@@ -85,6 +85,7 @@ class AgenticProcessClient(AILabClient):
         logger.debug(f"Creating agentic process with data: {data}")
 
         response = self.api_service.post(endpoint=endpoint, headers=headers, data=data)
+        validate_status_code(response)
         return parse_json_response(response, f"create process for project {self.project_id}")
 
 
@@ -190,6 +191,7 @@ class AgenticProcessClient(AILabClient):
             headers=headers,
             data=data
         )
+        validate_status_code(response)
         return parse_json_response(response, f"update process {process_id or name} in project {self.project_id}")
 
 
@@ -239,6 +241,7 @@ class AgenticProcessClient(AILabClient):
             headers=headers,
             params=params
         )
+        validate_status_code(response)
         return parse_json_response(response, f"retrieve process {process_id or process_name} for project {self.project_id}")
 
 
@@ -287,6 +290,7 @@ class AgenticProcessClient(AILabClient):
             headers=headers,
             params=params
         )
+        validate_status_code(response)
         return parse_json_response(response, f"list processes for project {self.project_id}")
 
 
@@ -329,6 +333,7 @@ class AgenticProcessClient(AILabClient):
             headers=headers,
             params=params
         )
+        validate_status_code(response)
         return parse_json_response(response, f"list process instances for process {process_id} in project {self.project_id}")
 
 
@@ -411,6 +416,7 @@ class AgenticProcessClient(AILabClient):
                 "revision": revision
             }
         )
+        validate_status_code(response)
         return parse_json_response(response, f"publish revision {revision} for process {process_id or process_name} in project {self.project_id}")
 
 
@@ -466,6 +472,7 @@ class AgenticProcessClient(AILabClient):
         logger.debug(f"Creating task with data: {data}")
 
         response = self.api_service.post(endpoint=endpoint, headers=headers, data=data)
+        validate_status_code(response)
         return parse_json_response(response, f"create task for project {self.project_id}")
     def get_task(
             self,
@@ -497,6 +504,7 @@ class AgenticProcessClient(AILabClient):
             logger.debug(f"Retrieving task detail with name {task_name}")
 
         response = self.api_service.get(endpoint=endpoint, headers=headers)
+        validate_status_code(response)
         return parse_json_response(response, f"retrieve task {task_id or task_name} for project {self.project_id}")
     def list_tasks(
             self,
@@ -531,6 +539,7 @@ class AgenticProcessClient(AILabClient):
         logger.debug(f"Listing tasks for project with ID {self.project_id}")
 
         response = self.api_service.get(endpoint=endpoint, headers=headers, params=params)
+        validate_status_code(response)
         return parse_json_response(response, f"list tasks for project {self.project_id}")
     def update_task(
             self,
@@ -594,6 +603,7 @@ class AgenticProcessClient(AILabClient):
         logger.debug(f"Updating task with ID {task_id} with data: {data}")
 
         response = self.api_service.put(endpoint=endpoint, headers=headers, data=data)
+        validate_status_code(response)
         return parse_json_response(response, f"update task {task_id} in project {self.project_id}")
     def delete_task(
             self,
@@ -671,6 +681,7 @@ class AgenticProcessClient(AILabClient):
             logger.debug(f"Publishing revision {revision} for task with name {task_name}")
 
         response = self.api_service.post(endpoint=endpoint, headers=headers, data=data)
+        validate_status_code(response)
         return parse_json_response(response, f"publish revision {revision} for task {task_id or task_name} in project {self.project_id}")
     def start_instance(
             self,
@@ -707,6 +718,7 @@ class AgenticProcessClient(AILabClient):
         logger.info(f"Starting instance for process with name '{process_name}'")
 
         response = self.api_service.post(endpoint=endpoint, headers=headers, data=data)
+        validate_status_code(response)
         return parse_json_response(response, f"start instance for process {process_name} in project {self.project_id}")
     def abort_instance(
             self,
@@ -734,6 +746,7 @@ class AgenticProcessClient(AILabClient):
         logger.info(f"Aborting instance with ID '{instance_id}'")
 
         response = self.api_service.post(endpoint=endpoint, headers=headers, data={})
+        validate_status_code(response)
         return parse_json_response(response, f"abort instance {instance_id} in project {self.project_id}")
     def get_instance(
             self,
@@ -759,6 +772,7 @@ class AgenticProcessClient(AILabClient):
         logger.info(f"Retrieving instance detail with ID '{instance_id}'")
 
         response = self.api_service.get(endpoint=endpoint, headers=headers)
+        validate_status_code(response)
         return parse_json_response(response, f"retrieve instance {instance_id} for project {self.project_id}")
     def get_instance_history(
             self,
@@ -784,6 +798,7 @@ class AgenticProcessClient(AILabClient):
         logger.info(f"Retrieving instance history with ID '{instance_id}'")
 
         response = self.api_service.get(endpoint=endpoint, headers=headers)
+        validate_status_code(response)
         return parse_json_response(response, f"retrieve history for instance {instance_id} in project {self.project_id}")
     def get_thread_information(
             self,
@@ -809,6 +824,7 @@ class AgenticProcessClient(AILabClient):
         logger.debug(f"Retrieving information about thread with ID {thread_id}")
 
         response = self.api_service.get(endpoint=endpoint, headers=headers)
+        validate_status_code(response)
         return parse_json_response(response, f"retrieve thread information for thread {thread_id} in project {self.project_id}")
     def send_user_signal(
             self,
@@ -843,6 +859,7 @@ class AgenticProcessClient(AILabClient):
         logger.debug(f"Sending user signal to process instance with ID {instance_id} with data: {data}")
 
         response = self.api_service.post(endpoint=endpoint, headers=headers, data=data)
+        validate_status_code(response)
         return parse_json_response(response, f"send user signal {signal_name} to instance {instance_id} in project {self.project_id}")
     def create_kb(
             self,
@@ -883,6 +900,7 @@ class AgenticProcessClient(AILabClient):
             headers=headers,
             data=data
         )
+        validate_status_code(response)
         return parse_json_response(response, f"create knowledge base for project {self.project_id}")
     def get_kb(
             self,
@@ -915,6 +933,7 @@ class AgenticProcessClient(AILabClient):
 
         response = self.api_service.get(endpoint=endpoint, headers=headers)
         kb_identifier = kb_id or kb_name
+        validate_status_code(response)
         return parse_json_response(response, f"retrieve knowledge base {kb_identifier} for project {self.project_id}")
     def list_kbs(
             self,
@@ -946,6 +965,7 @@ class AgenticProcessClient(AILabClient):
         logger.debug(f"Listing tasks in project with ID {self.project_id}")
 
         response = self.api_service.get(endpoint=endpoint, headers=headers, params=params)
+        validate_status_code(response)
         return parse_json_response(response, f"list knowledge bases for project {self.project_id}")
     def delete_kb(
             self,
@@ -1026,5 +1046,6 @@ class AgenticProcessClient(AILabClient):
             headers=headers,
             params=params
         )
+        validate_status_code(response)
         return parse_json_response(response, f"list jobs for project {self.project_id}")
 

pygeai/lab/strategies/clients.py

@@ -47,6 +47,7 @@ class ReasoningStrategyClient(AILabClient):
         logger.debug("Listing reasoning strategies")
 
         response = self.api_service.get(endpoint=endpoint, headers=headers, params=params)
+        validate_status_code(response)
         return parse_json_response(response, "list reasoning strategies")
 
 
@@ -94,6 +95,7 @@ class ReasoningStrategyClient(AILabClient):
         logger.debug(f"Creating reasoning strategy with data: {data}")
 
         response = self.api_service.post(endpoint=endpoint, headers=headers, data=data)
+        validate_status_code(response)
         return parse_json_response(response, f"create reasoning strategy for project {self.project_id}")
 
 
@@ -165,6 +167,7 @@ class ReasoningStrategyClient(AILabClient):
             headers=headers,
             data=data
         )
+        validate_status_code(response)
         return parse_json_response(response, f"update reasoning strategy {reasoning_strategy_id} in project {self.project_id}")
 
 
@@ -203,6 +206,7 @@ class ReasoningStrategyClient(AILabClient):
             headers=headers
         )
         strategy_identifier = reasoning_strategy_id or reasoning_strategy_name
+        validate_status_code(response)
         return parse_json_response(response, f"retrieve reasoning strategy {strategy_identifier} for project {self.project_id}")
 
     

pygeai/lab/tools/clients.py

@@ -385,6 +385,7 @@ class ToolClient(AILabClient):
                 "revision": revision,
             }
         )
+        validate_status_code(response)
         return parse_json_response(response, f"publish revision {revision} for tool {tool_id} in project {self.project_id}")
 
 

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"
+]