pygeai 0.4.0b12__py3-none-any.whl → 0.6.0b3__py3-none-any.whl

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

@@ -0,0 +1,70 @@
+from unittest import TestCase
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import ReasoningStrategy
+from pygeai.core.common.exceptions import APIError
+
+ai_lab_manager: AILabManager
+
+class TestAILabGetReasoningStrategyIntegration(TestCase):   
+
+    def setUp(self):
+        self.ai_lab_manager = AILabManager()
+        self.strategy_id = "0a3b039e-25bd-4cf9-ad67-7af2b654874b"
+        self.strategy_name = "Chain of Thought"
+
+    def __get_reasoning_strategy(self, strategy_id = None, strategy_name = None):
+        return self.ai_lab_manager.get_reasoning_strategy(
+            reasoning_strategy_id= strategy_id if strategy_id is not None else self.strategy_id,
+            reasoning_strategy_name=strategy_name if strategy_name is not None else self.strategy_name
+        )
+
+    def test_get_reasoning_strategy_by_id(self):
+        strategy = self.__get_reasoning_strategy(strategy_name="")
+
+        self.assertIsInstance(strategy, ReasoningStrategy, "Expected a reasoning strategy")
+        self.assertEqual(strategy.id, self.strategy_id, "Expected reasoning strategy ID to match")
+
+
+    def test_get_reasoning_strategy_by_name(self):
+        strategy = self.__get_reasoning_strategy(strategy_id="")
+        
+        self.assertIsInstance(strategy, ReasoningStrategy, "Expected a reasoning strategy")
+        self.assertEqual(
+            strategy.name, 
+            self.strategy_name,
+            "Expected reasoning strategy name to match"
+        )
+
+
+    def test_get_reasoning_strategy_no_id_or_name(self):
+        with self.assertRaises(Exception) as context:
+            self.__get_reasoning_strategy(strategy_id="", strategy_name="")
+
+        self.assertIn(
+            "Either reasoning_strategy_id or reasoning_strategy_name must be provided.",
+            str(context.exception),
+            "Expected error message for missing id and name"
+        )
+
+
+    def test_get_reasoning_strategy_invalid_id(self):
+        invalid_id = "invalid-id-1234"
+        with self.assertRaises(APIError) as context:
+            self.__get_reasoning_strategy(strategy_id=invalid_id)
+        self.assertIn(
+            f"Reasoning-Strategy not found (idOrName={invalid_id}",
+            str(context.exception),
+            "Expected an error for invalid reasoning strategy id"
+        )
+
+
+    def test_get_reasoning_strategy_invalid_name(self):
+        invalid_name = "Nonexistent Strategy"
+        with self.assertRaises(APIError) as context:
+            self.__get_reasoning_strategy(strategy_id="", strategy_name=invalid_name)
+
+        self.assertIn(
+            f"Reasoning-Strategy not found (idOrName={invalid_name})",
+            str(context.exception),
+            "Expected an error for invalid reasoning strategy name"
+        )

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

@@ -0,0 +1,93 @@
+from unittest import TestCase
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import FilterSettings, ReasoningStrategyList
+import copy
+
+ai_lab_manager: AILabManager
+
+class TestAILabListReasoningStrategiesIntegration(TestCase):    
+
+    def setUp(self):
+        self.ai_lab_manager = AILabManager()
+        self.filter_settings = FilterSettings(
+            name="",
+            start=0,
+            count=100,
+            allow_external=True,
+            access_scope="public"
+        )
+
+    
+    def __list_reasoning_strategies(self, filter_settings: FilterSettings = None):
+        filter_settings = filter_settings if filter_settings is not None else self.filter_settings
+        return self.ai_lab_manager.list_reasoning_strategies(filter_settings=filter_settings)
+    
+    
+    def test_public_list_reasoning_strategies(self):
+        result = self.__list_reasoning_strategies()
+
+        self.assertIsInstance(result, ReasoningStrategyList, "Expected a list of reasoning strategies")
+        for strategy in result.strategies:
+            self.assertTrue(
+                strategy.access_scope == "public",
+                "Expected all reasoning strategies to be public"
+            )
+
+
+    def test_private_list_reasoning_strategies(self):
+        filter_settings = copy.deepcopy(self.filter_settings)
+        filter_settings.access_scope = "private"
+        result = self.__list_reasoning_strategies(filter_settings=filter_settings)
+
+        self.assertIsInstance(result, ReasoningStrategyList, "Expected a list of reasoning strategies")
+        for strategy in result.strategies:
+            self.assertTrue(
+                strategy.access_scope == "private",
+                "Expected all reasoning strategies to be private"
+            )
+
+
+    def test_list_reasoning_strategies_small_count(self):
+        filter_settings = copy.deepcopy(self.filter_settings)
+        filter_settings.count = 2
+        result = self.__list_reasoning_strategies(filter_settings=filter_settings)
+        self.assertEqual(
+            len(result), 2,
+            "Expected list of reasoning strategies returned to be 2"
+        )
+
+
+    def test_list_reasoning_strategies_big_count(self):
+        filter_settings = copy.deepcopy(self.filter_settings)
+        filter_settings.count = 500
+        result = self.__list_reasoning_strategies(filter_settings=filter_settings)
+        self.assertLessEqual(
+            len(result), 500,
+            "Expected list of reasoning strategies returned to be 500 or less"
+        )
+
+
+    def test_list_reasoning_strategies_name_filter(self):
+        filter_settings = copy.deepcopy(self.filter_settings)
+        filter_settings.name = "Chain of Thought"
+        result = self.__list_reasoning_strategies(filter_settings=filter_settings)
+
+        for strategy in result.strategies:
+            self.assertIn(
+                "Chain of Thought",
+                strategy.name,
+                "Expected reasoning strategy name to contain filter value"
+            )
+
+   
+    def test_list_reasoning_strategies_invalid_scope(self):
+        filter_settings = copy.deepcopy(self.filter_settings)
+        filter_settings.access_scope = "project"
+
+        with self.assertRaises(ValueError) as exception:
+            self.__list_reasoning_strategies(filter_settings=filter_settings)
+        self.assertIn(
+            "Access scope must be either 'public' or 'private'.",
+            str(exception.exception),
+            "The expected error about invalid scope was not returned"
+        )

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

@@ -0,0 +1,149 @@
+from unittest import TestCase
+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
+
+class TestAILabCreateReasoningStrategyIntegration(TestCase):    
+    def setUp(self):
+        """
+        Set up the test environment.
+        """
+        self.ai_lab_manager = AILabManager()
+        self.strategy_to_update = self.__load_strategy()
+
+    
+    def __load_strategy(self):
+        self.random_str = str(uuid.uuid4())
+        return ReasoningStrategy(
+            id="323f5f94-5a3d-4717-89f3-3554a33c093f",
+            name=f"UpdatedStrategy_{self.random_str}",
+            system_prompt=f"Let's think step by step. {self.random_str}",
+            access_scope="private",
+            type="addendum",
+            localized_descriptions=[
+                LocalizedDescription(language="spanish", description=f"RSName spanish description {self.random_str}"),
+                LocalizedDescription(language="english", description=f"RSName english description {self.random_str}"),
+                LocalizedDescription(language="japanese", description=f"RSName japanese description {self.random_str}")
+            ]
+        )
+    
+
+    def __update_strategy(self, strategy=None, upsert = False):
+        """
+        Helper to create a reasoning strategy using ai_lab_manager.
+        """
+        return self.ai_lab_manager.update_reasoning_strategy(
+            strategy=self.strategy_to_update if strategy is None else strategy,
+            upsert=upsert
+        )
+    
+
+    def test_update_strategy_full_data(self):
+        self.updated_strategy = self.__update_strategy()
+
+        self.assertEqual(self.updated_strategy.name, self.strategy_to_update.name)
+        self.assertEqual(self.updated_strategy.system_prompt, self.strategy_to_update.system_prompt)
+
+        for locale in self.updated_strategy.localized_descriptions:
+            self.assertIn(
+                self.random_str, 
+                locale.description,
+                "Expected the localized description to be updated correctly"
+            )
+
+
+    def test_update_strategy_no_name(self):  
+        self.strategy_to_update.name = None
+
+        with self.assertRaises(APIError) as exception:
+            self.__update_strategy()
+        self.assertIn(
+            "ReasoningStrategy name cannot be empty",
+            str(exception.exception),
+            "The expected error about empty name was not returned"
+        )
+    
+
+    def test_update_strategy_no_system_prompt(self):
+        self.strategy_to_update.system_prompt = None
+        
+        with self.assertRaises(APIError) as exception:
+            self.__update_strategy()
+        self.assertIn(
+            "ReasoningStrategy template or systemPrompt are required.",
+            str(exception.exception),
+            "The expected error about empty system_prompt was not returned"
+        )
+
+
+    def test_update_strategy_no_public_name(self):
+        self.strategy_to_update.access_scope = "public"
+        with self.assertRaises(APIError) as exception:
+            self.__update_strategy()
+        self.assertIn(
+            "ReasoningStrategy publicName is required for public strategies",
+            str(exception.exception),
+            "The expected error about missing public name was not returned"
+        )
+
+
+    def test_update_strategy_no_access_scope(self):
+        self.strategy_to_update.access_scope = ""
+        with self.assertRaises(ValueError) as exception:
+            self.__update_strategy()
+        self.assertIn(
+            "Access scope must be either 'public' or 'private'",
+            str(exception.exception),
+            "The expected error about missing access scope was not returned"
+        )
+
+    
+    def test_update_strategy_no_type(self):
+        self.strategy_to_update.type = ""
+        with self.assertRaises(ValueError) as exception:
+            self.__update_strategy()
+        self.assertIn(
+            "Type must be 'addendum'",
+            str(exception.exception),
+            "The expected error about missing type was not returned"
+        )
+
+
+    def test_update_strategy_invalid_type(self):
+        self.strategy_to_update.type = "strategy_type"
+        
+        with self.assertRaises(ValueError) as exception:
+            self.__update_strategy()
+        self.assertIn(
+            "Type must be 'addendum'",
+            str(exception.exception),
+            "The expected error about missing type was not returned"
+        )
+
+    
+    def test_update_strategy_no_localized_descriptions(self):
+        self.strategy_to_update.localized_descriptions = []    
+        updated_strategy = self.__update_strategy()
+        self.assertIsNone(
+            updated_strategy.localized_descriptions,
+            "Expected no localized descriptions after update"
+        )    
+
+    @unittest.skip("Skipping upsert test to avoid creating new strategies during routine testing before having a cleanup mechanism.")
+    def test_update_strategy_upsert(self):			
+        new_id = str(uuid.uuid4())
+        new_strategy = ReasoningStrategy(
+            id=new_id,
+            name=f"UpsertStrategy_{self.random_str}",
+            system_prompt=f"Upsert system prompt {self.random_str}",
+            access_scope="private",
+            type="addendum",
+            localized_descriptions=[
+                LocalizedDescription(language="english", description=f"Upsert description {self.random_str}")
+            ]
+        )
+        upserted_strategy = self.__update_strategy(strategy=new_strategy, upsert=True)
+        self.assertEqual(upserted_strategy.id, new_id, "Expected the reasoning strategy to be created via upsert")

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

@@ -2,8 +2,7 @@ from unittest import TestCase
 import uuid
 from pygeai.lab.managers import AILabManager
 from pygeai.lab.models import Tool, ToolParameter
-from pydantic import ValidationError
-from pygeai.core.common.exceptions import APIError 
+from pygeai.core.common.exceptions import APIResponseError
 
 
 class TestAILabCreateToolIntegration(TestCase):    
@@ -11,7 +10,7 @@ class TestAILabCreateToolIntegration(TestCase):
         """
         Set up the test environment.
         """
-        self.ai_lab_manager = AILabManager(alias="beta")
+        self.ai_lab_manager = AILabManager()
         self.new_tool = self.__load_tool()
         self.created_tool: Tool = None
 
@@ -94,15 +93,12 @@ class TestAILabCreateToolIntegration(TestCase):
 
         for auto_publish in test_params:
             
-            with self.subTest(input=auto_publish):     
-                with self.assertRaises(ValidationError) as context:
-                    self.new_tool = Tool(
+            with self.subTest(input=auto_publish): 
+                self.new_tool = Tool(
                         name=str(uuid.uuid4())
                     )
-                    self.__create_tool(automatic_publish=auto_publish)
-
-                self.assertIn("description", str(context.exception))
-                self.assertIn("Field required", str(context.exception))
+                created_tool = self.__create_tool(automatic_publish=auto_publish)
+                self.assertTrue(isinstance(created_tool, Tool), "Expected a created tool")    
 
 
     def test_create_tool_no_name(self):
@@ -111,7 +107,7 @@ class TestAILabCreateToolIntegration(TestCase):
         for auto_publish in test_params:
             with self.subTest(input=auto_publish):
                 self.new_tool.name = ""
-                with self.assertRaises(APIError) as exception:
+                with self.assertRaises(APIResponseError) as exception:
                     self.__create_tool(automatic_publish=auto_publish)
 
                 self.assertIn(
@@ -128,7 +124,7 @@ class TestAILabCreateToolIntegration(TestCase):
 
             with self.subTest(input=auto_publish):
                 self.new_tool.name = "sdk_project_gemini_tool"
-                with self.assertRaises(APIError) as exception:
+                with self.assertRaises(APIResponseError) as exception:
                     self.__create_tool(automatic_publish=auto_publish)
                 self.assertIn(
                     "Tool already exists [name=sdk_project_gemini_tool]..",
@@ -145,7 +141,7 @@ class TestAILabCreateToolIntegration(TestCase):
                 new_tool = self.__load_tool()
                 new_tool2 = self.__load_tool()
 
-                with self.assertRaises(APIError) as exception:
+                with self.assertRaises(APIResponseError) as exception:
                     new_tool.name = f"{new_tool.name}:invalid"
                     self.__create_tool(tool=new_tool, automatic_publish=auto_publish)
                 self.assertIn(
@@ -154,7 +150,7 @@ class TestAILabCreateToolIntegration(TestCase):
                     f"Expected an error about invalid character (:) in tool name with autopublish {'enabled' if auto_publish else 'disabled'}"
                 )
 
-                with self.assertRaises(APIError) as exception:
+                with self.assertRaises(APIResponseError) as exception:
                     new_tool2.name = f"{new_tool2.name}/invalid"
                     self.__create_tool(tool=new_tool2, automatic_publish=auto_publish)
                 self.assertIn(
@@ -190,7 +186,7 @@ class TestAILabCreateToolIntegration(TestCase):
             with self.subTest(input=auto_publish):
                 self.new_tool.access_scope = "public"
                 self.new_tool.public_name = None
-                with self.assertRaises(APIError) as exception:
+                with self.assertRaises(APIResponseError) as exception:
                     self.__create_tool(automatic_publish=auto_publish)
                 self.assertIn(
                     "Tool publicName is required for tools with accessScope=public.",
@@ -206,7 +202,7 @@ class TestAILabCreateToolIntegration(TestCase):
             with self.subTest(input=auto_publish): 
                 self.new_tool.access_scope = "public"
                 self.new_tool.public_name = "com.sdk.testing#"  # Add invalid character to public name
-                with self.assertRaises(APIError) as exception:
+                with self.assertRaises(APIResponseError) as exception:
                     self.__create_tool(automatic_publish=auto_publish)
 
                 self.assertIn(
@@ -232,7 +228,7 @@ class TestAILabCreateToolIntegration(TestCase):
                 duplicated_pn_tool.access_scope = "public"
                 duplicated_pn_tool.public_name = self.created_tool.public_name
                 
-                with self.assertRaises(APIError) as exception:
+                with self.assertRaises(APIResponseError) as exception:
                     self.__create_tool(tool=duplicated_pn_tool, automatic_publish=auto_publish)
                 self.assertIn(
                     f"Tool already exists [publicName={self.created_tool.public_name}].",

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

@@ -1,15 +1,15 @@
 from unittest import TestCase
 import uuid
 from pygeai.lab.managers import AILabManager
-from pygeai.lab.models import Tool, AgentData, Prompt, LlmConfig, Model
-from pygeai.core.common.exceptions import MissingRequirementException, InvalidAPIResponseException
+from pygeai.lab.models import Tool
+from pygeai.core.common.exceptions import APIResponseError, MissingRequirementException, InvalidAPIResponseException
 
 ai_lab_manager: AILabManager
 
 class TestAILabDeleteToolIntegration(TestCase):  
 
     def setUp(self):
-        self.ai_lab_manager = AILabManager(alias="beta")
+        self.ai_lab_manager = AILabManager()
         
 
     def __create_tool(self):
@@ -66,7 +66,7 @@ class TestAILabDeleteToolIntegration(TestCase):
     def test_delete_tool_invalid_id_valid_name(self):
         invalid_id = "0026e53d-ea78-4cac-af9f-12650invalid"
         created_tool = self.__create_tool() 
-        with self.assertRaises(InvalidAPIResponseException) as exception:
+        with self.assertRaises(APIResponseError) as exception:
             self.__delete_tool(tool_name=created_tool.name, tool_id=invalid_id)
 
         self.assertIn(

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

@@ -9,7 +9,7 @@ ai_lab_manager: AILabManager
 class TestAILabGetParameterIntegration(TestCase):   
 
     def setUp(self):
-        self.ai_lab_manager = AILabManager(alias="beta")
+        self.ai_lab_manager = AILabManager()
         self.tool_id = "42610b61-cb47-49bf-9475-36ee652f31f3"
         self.filter_settings = FilterSettings(
             revision="0",

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

@@ -2,7 +2,7 @@ from unittest import TestCase
 import unittest
 from pygeai.lab.managers import AILabManager
 from pygeai.lab.models import Tool, FilterSettings
-from pygeai.core.common.exceptions import APIError
+from pygeai.core.common.exceptions import APIResponseError
 import copy
 
 ai_lab_manager: AILabManager
@@ -10,7 +10,7 @@ ai_lab_manager: AILabManager
 class TestAILabGetToolIntegration(TestCase):   
 
     def setUp(self):
-        self.ai_lab_manager = AILabManager(alias="beta")
+        self.ai_lab_manager = AILabManager()
         self.tool_id = "e3e4d64f-ce52-467e-90a9-aa4d08425e82"
         self.filter_settings = FilterSettings(
             revision="0",
@@ -36,7 +36,7 @@ class TestAILabGetToolIntegration(TestCase):
     
     def test_get_tool_invalid_tool_id(self):
         invalid_id = "0026e53d-ea78-4cac-af9f-12650invalid"
-        with self.assertRaises(APIError) as context:
+        with self.assertRaises(APIResponseError) as context:
             self.__get_tool(tool_id=invalid_id)
         self.assertIn(
             f"Tool not found [IdOrName= {invalid_id}].",
@@ -65,7 +65,7 @@ class TestAILabGetToolIntegration(TestCase):
     def test_get_tool_by_earlier_revision(self):
         filter_settings = copy.deepcopy(self.filter_settings)
         filter_settings.revision = "2"
-        with self.assertRaises(APIError) as context:
+        with self.assertRaises(APIResponseError) as context:
             self.__get_tool(filter_settings=filter_settings)
         self.assertIn(
             f"Requested revision not found [revision={filter_settings.revision}].",

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

@@ -8,7 +8,7 @@ ai_lab_manager: AILabManager
 class TestAILabListToolsIntegration(TestCase):    
 
     def setUp(self):
-        self.ai_lab_manager = AILabManager(alias="beta")
+        self.ai_lab_manager = AILabManager()
         self.filter_settings = FilterSettings(
             allow_external=False,
             allow_drafts=True,

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

@@ -9,7 +9,7 @@ ai_lab_manager: AILabManager
 class TestAILabPublishToolRevisionIntegration(TestCase):   
 
     def setUp(self):
-        self.ai_lab_manager = AILabManager(alias="beta")
+        self.ai_lab_manager = AILabManager()
         self.tool_id = "c77e1f2e-0322-4dd0-b6ec-aff217f1cb32"
 
 

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

@@ -8,7 +8,7 @@ ai_lab_manager: AILabManager
 
 class TestAILabSetParameterIntegration(TestCase):
     def setUp(self):
-        self.ai_lab_manager = AILabManager(alias="beta")
+        self.ai_lab_manager = AILabManager()
         self.tool_id = "42610b61-cb47-49bf-9475-36ee652f31f3"
         self.tool_parameters = self.__load_parameters()
 

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

@@ -1,6 +1,6 @@
 from unittest import TestCase
 import uuid
-from pygeai.core.common.exceptions import APIError
+from pygeai.core.common.exceptions import APIResponseError
 from pygeai.lab.managers import AILabManager
 from pygeai.lab.models import Tool, ToolParameter
 
@@ -10,7 +10,7 @@ class TestAILabUpdateToolIntegration(TestCase):
         """
         Set up the test environment.
         """
-        self.ai_lab_manager = AILabManager(alias="beta")
+        self.ai_lab_manager = AILabManager()
         self.tool_to_update = self.__load_tool()
 
     
@@ -65,7 +65,7 @@ class TestAILabUpdateToolIntegration(TestCase):
                 tool = self.__load_tool()
                 tool2 = self.__load_tool()
 
-                with self.assertRaises(APIError) as exception:
+                with self.assertRaises(APIResponseError) as exception:
                     tool.name = f"{tool.name}:invalid"
                     self.__update_tool(tool=tool, automatic_publish=auto_publish)
                 self.assertIn(
@@ -74,7 +74,7 @@ class TestAILabUpdateToolIntegration(TestCase):
                     f"Expected an error about invalid character (:) in tool name with autopublish {'enabled' if auto_publish else 'disabled'}"
                 )
 
-                with self.assertRaises(APIError) as exception:
+                with self.assertRaises(APIResponseError) as exception:
                     tool2.name = f"{tool2.name}/invalid"
                     self.__update_tool(tool=tool2, automatic_publish=auto_publish)
                 self.assertIn(
@@ -91,7 +91,7 @@ class TestAILabUpdateToolIntegration(TestCase):
 
             with self.subTest(input=auto_publish):
                 self.tool_to_update.name = "sdk_project_gemini_tool"
-                with self.assertRaises(APIError) as exception:
+                with self.assertRaises(APIResponseError) as exception:
                     self.__update_tool(automatic_publish=auto_publish)
                 self.assertIn(
                     "Tool already exists",
@@ -107,7 +107,7 @@ class TestAILabUpdateToolIntegration(TestCase):
 
             with self.subTest(input=auto_publish):
                 self.tool_to_update.name = ""
-                with self.assertRaises(APIError) as exception:
+                with self.assertRaises(APIResponseError) as exception:
                     self.__update_tool(automatic_publish=auto_publish)
                 self.assertIn(
                     "Tool name cannot be empty",
@@ -123,7 +123,7 @@ class TestAILabUpdateToolIntegration(TestCase):
             with self.subTest(input=auto_publish):
                 invalid_id = "0026e53d-ea78-4cac-af9f-12650invalid"
                 self.tool_to_update.id = invalid_id
-                with self.assertRaises(APIError) as exception:
+                with self.assertRaises(APIResponseError) as exception:
                     self.__update_tool(automatic_publish=auto_publish)
                 self.assertIn(
                     f"Tool not found [IdOrName= {invalid_id}",
@@ -172,7 +172,7 @@ class TestAILabUpdateToolIntegration(TestCase):
         self.tool_to_update.public_name = "invalid#name"
         for auto_publish in test_params:
             with self.subTest(input=auto_publish):                
-                with self.assertRaises(APIError) as exception:
+                with self.assertRaises(APIResponseError) as exception:
                     self.__update_tool(automatic_publish=auto_publish)
                 self.assertIn(
                     "Invalid public name, it can only contain lowercase letters, numbers, periods (.), dashes (-), and underscores (_). Please remove any other characters.",
@@ -219,7 +219,7 @@ class TestAILabUpdateToolIntegration(TestCase):
 
         for auto_publish in test_params:
             with self.subTest(input=auto_publish):  
-                with self.assertRaises(APIError) as exc:
+                with self.assertRaises(APIResponseError) as exc:
                     self.__update_tool(automatic_publish=auto_publish)
                 self.assertIn(
                     "Either openApi (URL with the OpenAPI definition) or the openApiJson (with the conent) are required for api-tools.",
@@ -245,7 +245,6 @@ class TestAILabUpdateToolIntegration(TestCase):
 
    
     def test_update_tool_upsert(self):
-        print(self.tool_to_update)
         self.tool_to_update.id = str(uuid.uuid4())
         self.tool_to_update.name = str(uuid.uuid4())
         result = self.__update_tool(upsert=True)