pygeai 0.6.0b14__py3-none-any.whl → 0.7.0b1__py3-none-any.whl

pygeai/tests/integration/lab/processes/test_create_task.py

@@ -1,9 +1,8 @@
 from unittest import TestCase
-import unittest
 import uuid
 from pygeai.lab.managers import AILabManager
-from pygeai.lab.models import Task, Prompt, ArtifactTypeList, ArtifactType
-from pygeai.core.common.exceptions import APIResponseError, APIError
+from pygeai.lab.models import Task, Prompt
+from pygeai.core.common.exceptions import APIError
 
 
 class TestAILabCreateTaskIntegration(TestCase): 

pygeai/tests/integration/lab/processes/test_delete_process.py

@@ -3,7 +3,6 @@ import uuid
 from pygeai.lab.managers import AILabManager
 from pygeai.lab.models import AgenticProcess, KnowledgeBase, AgenticActivity, ArtifactSignal, UserSignal, Event, SequenceFlow
 from pygeai.core.common.exceptions import InvalidAPIResponseException, MissingRequirementException
-from pygeai.core.base.responses import EmptyResponse
 
 ai_lab_manager: AILabManager
 

pygeai/tests/integration/lab/processes/test_get_process.py

@@ -1,10 +1,8 @@
 from unittest import TestCase
-import unittest
 from pygeai.lab.managers import AILabManager
-from pygeai.lab.models import AgenticProcess, FilterSettings, KnowledgeBase, AgenticActivity, ArtifactSignal, UserSignal, Event, SequenceFlow
-from pygeai.core.common.exceptions import APIResponseError, APIError, MissingRequirementException
+from pygeai.lab.models import AgenticProcess, FilterSettings
+from pygeai.core.common.exceptions import APIError, MissingRequirementException
 import copy
-import uuid
 
 
 class TestAILabGetProcessIntegration(TestCase):   

pygeai/tests/integration/lab/processes/test_list_process_instances.py

@@ -1,8 +1,6 @@
 from unittest import TestCase
-import uuid
 from pygeai.lab.managers import AILabManager
-from pygeai.lab.models import FilterSettings, ProcessInstanceList, AgenticProcess, KnowledgeBase, AgenticActivity, ArtifactSignal, UserSignal, Event, SequenceFlow
-from pygeai.core.common.exceptions import APIError, MissingRequirementException
+from pygeai.lab.models import FilterSettings, ProcessInstanceList
 import copy
 
 ai_lab_manager: AILabManager

pygeai/tests/integration/lab/processes/test_update_process.py

@@ -1,12 +1,11 @@
 from unittest import TestCase
 import uuid
 
-from pydantic import ValidationError
-from pygeai.core.common.exceptions import APIResponseError, APIError
+from pygeai.core.common.exceptions import APIError
 from pygeai.lab.managers import AILabManager
 from pygeai.lab.models import (
     AgenticProcess, KnowledgeBase, AgenticActivity, ArtifactSignal, Task, 
-    UserSignal, Event, SequenceFlow, Variable
+    UserSignal, Event, SequenceFlow
 )
 
 
@@ -20,7 +19,6 @@ class TestAILabUpdateProcessIntegration(TestCase):
         self.created_process: AgenticProcess = None
         self.updated_process: AgenticProcess = None
 
-
     def tearDown(self):
         """
         Clean up after each test if necessary.
@@ -29,7 +27,6 @@ class TestAILabUpdateProcessIntegration(TestCase):
         if isinstance(self.created_process, AgenticProcess):
             self.ai_lab_manager.delete_process(self.created_process.id)
 
-    
     def __load_process(self):
         """
         Helper to load a complete process configuration for testing.
@@ -137,12 +134,11 @@ class TestAILabUpdateProcessIntegration(TestCase):
                         self.__update_process(automatic_publish=auto_publish)
                     
                     self.assertIn(
-                        f"Invalid character in name ({':' if ":" in invalid_name else "/"} is not allowed).",
+                        f"Invalid character in name ({':' if ':' in invalid_name else '/'} is not allowed).",
                         str(exception.exception),                    
                         f"Expected an error about invalid character in process name with autopublish {'enabled' if auto_publish else 'disabled'}"
                     )
 
-
     def test_update_process_duplicated_name(self):
         """
         Test updating a process with a name that already exists.
@@ -161,7 +157,6 @@ class TestAILabUpdateProcessIntegration(TestCase):
                     str(exception.exception),                    
                     f"Expected an error about duplicated process name with autopublish {'enabled' if auto_publish else 'disabled'}"
                 )
-    
 
     def test_update_process_no_name(self):
         """
@@ -182,7 +177,6 @@ class TestAILabUpdateProcessIntegration(TestCase):
                     f"Expected an error when process name is empty with autopublish {'enabled' if auto_publish else 'disabled'}"
                 )
 
-
     def test_update_process_invalid_id(self):
         """
         Test updating a process with an invalid ID.

pygeai/tests/integration/lab/reasoning_strategies/test_update_reasoning_strategy.py

@@ -3,8 +3,7 @@ import unittest
 import uuid
 from pygeai.lab.managers import AILabManager
 from pygeai.lab.models import LocalizedDescription, ReasoningStrategy
-from pydantic import ValidationError
-from pygeai.core.common.exceptions import APIError, APIResponseError
+from pygeai.core.common.exceptions import APIError
 
 class TestAILabCreateReasoningStrategyIntegration(TestCase):    
     def setUp(self):

pygeai/tests/integration/lab/tools/test_delete_tool.py

@@ -2,7 +2,7 @@ from unittest import TestCase
 import uuid
 from pygeai.lab.managers import AILabManager
 from pygeai.lab.models import Tool
-from pygeai.core.common.exceptions import APIResponseError, MissingRequirementException, InvalidAPIResponseException
+from pygeai.core.common.exceptions import APIResponseError, MissingRequirementException
 
 ai_lab_manager: AILabManager
 

pygeai/tests/integration/lab/tools/test_list_tools.py

@@ -91,7 +91,7 @@ class TestAILabListToolsIntegration(TestCase):
         self.assertIn(
             "Scope must be one of builtin, external, api, proxied.",
             str(exception.exception),                    
-            f"The expected error about invalid scope was not returned"
+            "The expected error about invalid scope was not returned"
         )
 
 

pygeai/tests/integration/lab/tools/test_update_tool.py

@@ -148,7 +148,7 @@ class TestAILabUpdateToolIntegration(TestCase):
                 with self.assertRaises(ValueError) as exception:
                     self.__update_tool(automatic_publish=auto_publish)
                 self.assertIn(
-                    f"Scope must be one of builtin, external, api, proxied",
+                    "Scope must be one of builtin, external, api, proxied",
                     str(exception.exception),                    
                     f"Expected an error when tool scope is invalid and autopublish {'enabled' if auto_publish else 'disabled'}"
                 )

pygeai/tests/lab/agents/test_clients.py

@@ -24,7 +24,7 @@ class TestAgentClient(unittest.TestCase):
         self.agent_data_models = [{"name": "gpt-4o"}]
         self.agent_data_resource_pools = [{"name": "pool1", "tools": [{"name": "tool1"}]}]
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_list_agents_success(self, mock_get):
         expected_response = {"agents": [{"id": "agent-1", "name": "Agent1"}]}
         mock_response = mock_get.return_value
@@ -56,7 +56,7 @@ class TestAgentClient(unittest.TestCase):
         headers = mock_get.call_args[1]['headers']
         self.assertEqual(headers['ProjectId'], self.project_id)
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_list_agents_json_decode_error(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.status_code = 200
@@ -69,7 +69,7 @@ class TestAgentClient(unittest.TestCase):
         self.assertEqual(str(context.exception), f"Unable to list agents for project {self.project_id}: Invalid JSON response")
         mock_get.assert_called_once()
 
-    @patch("pygeai.core.services.rest.ApiService.post")
+    @patch("pygeai.core.services.rest.GEAIApiService.post")
     def test_create_agent_success(self, mock_post):
         expected_response = {"id": "agent-123", "name": "Test Agent"}
         mock_response = mock_post.return_value
@@ -111,7 +111,7 @@ class TestAgentClient(unittest.TestCase):
         headers = mock_post.call_args[1]['headers']
         self.assertEqual(headers['ProjectId'], self.project_id)
 
-    @patch("pygeai.core.services.rest.ApiService.post")
+    @patch("pygeai.core.services.rest.GEAIApiService.post")
     def test_create_agent_without_resource_pools(self, mock_post):
         expected_response = {"id": "agent-123", "name": "Test Agent"}
         mock_response = mock_post.return_value
@@ -153,7 +153,7 @@ class TestAgentClient(unittest.TestCase):
             )
         self.assertEqual(str(context.exception), f"Access scope must be one of {', '.join(VALID_ACCESS_SCOPES)}.")
 
-    @patch("pygeai.core.services.rest.ApiService.post")
+    @patch("pygeai.core.services.rest.GEAIApiService.post")
     def test_create_agent_json_decode_error(self, mock_post):
         mock_response = mock_post.return_value
         mock_response.status_code = 200
@@ -177,7 +177,7 @@ class TestAgentClient(unittest.TestCase):
         self.assertEqual(str(context.exception), f"Unable to create agent for project {self.project_id}: Invalid JSON response")
         mock_post.assert_called_once()
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_agent_success(self, mock_get):
         expected_response = {"id": self.agent_id, "name": "Test Agent"}
         mock_response = mock_get.return_value
@@ -209,7 +209,7 @@ class TestAgentClient(unittest.TestCase):
             self.client.get_agent(agent_id="")
         self.assertEqual(str(context.exception), "agent_id must be specified in order to retrieve the agent")
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_agent_json_decode_error(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.status_code = 200
@@ -222,7 +222,7 @@ class TestAgentClient(unittest.TestCase):
         self.assertEqual(str(context.exception), f"Unable to retrieve agent {self.agent_id} for project {self.project_id}: Invalid JSON response")
         mock_get.assert_called_once()
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_create_sharing_link_success(self, mock_get):
         expected_response = {"link": "http://example.com/share"}
         mock_response = mock_get.return_value
@@ -247,7 +247,7 @@ class TestAgentClient(unittest.TestCase):
             self.client.create_sharing_link(agent_id="")
         self.assertEqual(str(context.exception), "agent_id must be specified in order to create sharing link")
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_create_sharing_link_json_decode_error(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.status_code = 200
@@ -262,7 +262,7 @@ class TestAgentClient(unittest.TestCase):
         self.assertEqual(str(context.exception), f"Unable to create sharing link for agent {self.agent_id} in project {self.project_id}: Invalid JSON response")
         mock_get.assert_called_once()
 
-    @patch("pygeai.core.services.rest.ApiService.post")
+    @patch("pygeai.core.services.rest.GEAIApiService.post")
     def test_publish_agent_revision_success(self, mock_post):
         revision = "2"
         expected_response = {"status": "published"}
@@ -284,7 +284,7 @@ class TestAgentClient(unittest.TestCase):
         headers = mock_post.call_args[1]['headers']
         self.assertEqual(headers['ProjectId'], self.project_id)
 
-    @patch("pygeai.core.services.rest.ApiService.post")
+    @patch("pygeai.core.services.rest.GEAIApiService.post")
     def test_publish_agent_revision_json_decode_error(self, mock_post):
         revision = "2"
         mock_response = mock_post.return_value
@@ -301,7 +301,7 @@ class TestAgentClient(unittest.TestCase):
         self.assertEqual(str(context.exception), f"Unable to publish revision {revision} for agent {self.agent_id} in project {self.project_id}: Invalid JSON response")
         mock_post.assert_called_once()
 
-    @patch("pygeai.core.services.rest.ApiService.delete")
+    @patch("pygeai.core.services.rest.GEAIApiService.delete")
     def test_delete_agent_success(self, mock_delete):
         expected_response = {}
         mock_response = mock_delete.return_value
@@ -320,7 +320,7 @@ class TestAgentClient(unittest.TestCase):
         headers = mock_delete.call_args[1]['headers']
         self.assertEqual(headers['ProjectId'], self.project_id)
 
-    @patch("pygeai.core.services.rest.ApiService.delete")
+    @patch("pygeai.core.services.rest.GEAIApiService.delete")
     def test_delete_agent_json_decode_error(self, mock_delete):
         mock_response = mock_delete.return_value
         mock_response.status_code = 200
@@ -335,7 +335,7 @@ class TestAgentClient(unittest.TestCase):
         self.assertEqual(str(context.exception), f"Unable to delete agent {self.agent_id} from project {self.project_id}: Invalid JSON response")
         mock_delete.assert_called_once()
 
-    @patch("pygeai.core.services.rest.ApiService.put")
+    @patch("pygeai.core.services.rest.GEAIApiService.put")
     def test_update_agent_success(self, mock_put):
         expected_response = {"id": self.agent_id, "name": "Updated Agent"}
         mock_response = mock_put.return_value
@@ -379,7 +379,7 @@ class TestAgentClient(unittest.TestCase):
         headers = mock_put.call_args[1]['headers']
         self.assertEqual(headers['ProjectId'], self.project_id)
 
-    @patch("pygeai.core.services.rest.ApiService.put")
+    @patch("pygeai.core.services.rest.GEAIApiService.put")
     def test_update_agent_with_upsert(self, mock_put):
         expected_response = {"id": self.agent_id, "name": "Upserted Agent"}
         mock_response = mock_put.return_value
@@ -409,7 +409,7 @@ class TestAgentClient(unittest.TestCase):
             data=mock_put.call_args[1]['data']
         )
 
-    @patch("pygeai.core.services.rest.ApiService.put")
+    @patch("pygeai.core.services.rest.GEAIApiService.put")
     def test_update_agent_without_resource_pools(self, mock_put):
         expected_response = {"id": self.agent_id, "name": "Updated Agent No Pools"}
         mock_response = mock_put.return_value
@@ -454,7 +454,7 @@ class TestAgentClient(unittest.TestCase):
             )
         self.assertEqual(str(context.exception), f"Access scope must be one of {', '.join(VALID_ACCESS_SCOPES)}.")
 
-    @patch("pygeai.core.services.rest.ApiService.put")
+    @patch("pygeai.core.services.rest.GEAIApiService.put")
     def test_update_agent_json_decode_error(self, mock_put):
         mock_response = mock_put.return_value
         mock_response.status_code = 200