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

pygeai/tests/lab/processes/test_mappers.py

@@ -160,7 +160,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
                 process_id=self.process_id,
                 name="Updated Process"
             )
-        except InvalidAPIResponseException as e:
+        except InvalidAPIResponseException:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")
@@ -186,7 +186,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
             result = self.client.get_process(
                 process_id=self.process_id
             )
-        except InvalidAPIResponseException as e:
+        except InvalidAPIResponseException:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")
@@ -197,7 +197,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
         try:
             result = self.client.list_processes()
-        except InvalidAPIResponseException as e:
+        except InvalidAPIResponseException:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")
@@ -223,7 +223,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
             result = self.client.list_process_instances(
                 process_id=self.process_id
             )
-        except InvalidAPIResponseException as e:
+        except InvalidAPIResponseException:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")
@@ -247,7 +247,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
             result = self.client.delete_process(
                 process_id=self.process_id
             )
-        except InvalidAPIResponseException as e:
+        except InvalidAPIResponseException:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")
@@ -280,7 +280,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
                 process_id=self.process_id,
                 revision="2"
             )
-        except InvalidAPIResponseException as e:
+        except InvalidAPIResponseException:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")
@@ -293,7 +293,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
             result = self.client.create_task(
                 name="Test Task"
             )
-        except InvalidAPIResponseException as e:
+        except InvalidAPIResponseException:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")
@@ -333,7 +333,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
             result = self.client.get_task(
                 task_id=self.task_id
             )
-        except InvalidAPIResponseException as e:
+        except InvalidAPIResponseException:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")
@@ -357,7 +357,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
         try:
             result = self.client.list_tasks()
-        except InvalidAPIResponseException as e:
+        except InvalidAPIResponseException:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")
@@ -407,7 +407,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
                 task_id=self.task_id,
                 name="Updated Task"
             )
-        except InvalidAPIResponseException as e:
+        except InvalidAPIResponseException:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")
@@ -432,7 +432,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
             result = self.client.delete_task(
                 task_id=self.task_id
             )
-        except InvalidAPIResponseException as e:
+        except InvalidAPIResponseException:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")
@@ -485,7 +485,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
                 task_id=self.task_id,
                 revision="3"
             )
-        except InvalidAPIResponseException as e:
+        except InvalidAPIResponseException:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")
@@ -498,7 +498,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
             result = self.client.start_instance(
                 process_name="Test Process"
             )
-        except InvalidAPIResponseException as e:
+        except InvalidAPIResponseException:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")
@@ -535,7 +535,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
             result = self.client.abort_instance(
                 instance_id=self.instance_id
             )
-        except InvalidAPIResponseException as e:
+        except InvalidAPIResponseException:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")
@@ -558,7 +558,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
             result = self.client.get_instance(
                 instance_id=self.instance_id
             )
-        except InvalidAPIResponseException as e:
+        except InvalidAPIResponseException:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")
@@ -589,7 +589,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
             result = self.client.get_instance_history(
                 instance_id=self.instance_id
             )
-        except InvalidAPIResponseException as e:
+        except InvalidAPIResponseException:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")
@@ -620,7 +620,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
             result = self.client.get_thread_information(
                 thread_id=self.thread_id
             )
-        except InvalidAPIResponseException as e:
+        except InvalidAPIResponseException:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")
@@ -653,7 +653,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
                 instance_id=self.instance_id,
                 signal_name="approval"
             )
-        except InvalidAPIResponseException as e:
+        except InvalidAPIResponseException:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")
@@ -675,7 +675,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
             result = self.client.create_kb(
                 name="Test KB"
             )
-        except InvalidAPIResponseException as e:
+        except InvalidAPIResponseException:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")
@@ -714,7 +714,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
             result = self.client.get_kb(
                 kb_id=self.kb_id
             )
-        except InvalidAPIResponseException as e:
+        except InvalidAPIResponseException:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")
@@ -746,7 +746,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
         try:
             result = self.client.list_kbs()
-        except InvalidAPIResponseException as e:
+        except InvalidAPIResponseException:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")
@@ -771,7 +771,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
             result = self.client.delete_kb(
                 kb_id=self.kb_id
             )
-        except InvalidAPIResponseException as e:
+        except InvalidAPIResponseException:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")
@@ -791,7 +791,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
         try:
             result = self.client.list_jobs()
-        except InvalidAPIResponseException as e:
+        except InvalidAPIResponseException:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")

pygeai/tests/lab/spec/test_loader.py

@@ -1,8 +1,6 @@
-import unittest
 from unittest import TestCase
 import tempfile
 import os
-from pathlib import Path
 from pygeai.lab.spec.loader import JSONLoader
 from pygeai.core.common.exceptions import InvalidPathException, InvalidJSONException
 

pygeai/tests/lab/spec/test_parsers.py

@@ -1,8 +1,7 @@
-import unittest
 from unittest import TestCase
 from pydantic import ValidationError
 from pygeai.lab.spec.parsers import AgentParser, ToolParser, TaskParser, AgenticProcessParser
-from pygeai.lab.models import Agent, Tool, Task, AgenticProcess, Prompt, LlmConfig, Sampling, ModelList, Model, AgentData
+from pygeai.lab.models import Agent, Tool, Task, AgenticProcess
 
 
 class TestParser(TestCase):

pygeai/tests/lab/strategies/test_clients.py

@@ -2,7 +2,7 @@ import unittest
 from unittest.mock import patch
 from json import JSONDecodeError
 from pygeai.lab.strategies.clients import ReasoningStrategyClient
-from pygeai.core.common.exceptions import InvalidAPIResponseException, APIResponseError, APIResponseError
+from pygeai.core.common.exceptions import APIResponseError
 
 
 class TestReasoningStrategyClient(unittest.TestCase):
@@ -15,7 +15,7 @@ class TestReasoningStrategyClient(unittest.TestCase):
         self.reasoning_strategy_id = "strat-123"
         self.reasoning_strategy_name = "TestStrategy"
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_list_reasoning_strategies_success(self, mock_get):
         expected_response = {"strategies": [{"id": "strat-1", "name": "Strategy1"}]}
         mock_response = mock_get.return_value
@@ -40,7 +40,7 @@ class TestReasoningStrategyClient(unittest.TestCase):
         self.assertTrue(params['allowExternal'])
         self.assertEqual(params['accessScope'], "public")
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_list_reasoning_strategies_json_decode_error(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.side_effect = JSONDecodeError("Invalid JSON", "", 0)
@@ -59,7 +59,7 @@ class TestReasoningStrategyClient(unittest.TestCase):
 
         self.assertEqual(str(context.exception), "Access scope must be either 'public' or 'private'.")
 
-    @patch("pygeai.core.services.rest.ApiService.post")
+    @patch("pygeai.core.services.rest.GEAIApiService.post")
     def test_create_reasoning_strategy_success(self, mock_post):
         name = "TestStrategy"
         system_prompt = "Test system prompt"
@@ -94,7 +94,7 @@ class TestReasoningStrategyClient(unittest.TestCase):
         headers = 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_reasoning_strategy_json_decode_error(self, mock_post):
         name = "TestStrategy"
         system_prompt = "Test system prompt"
@@ -112,7 +112,7 @@ class TestReasoningStrategyClient(unittest.TestCase):
         mock_post.assert_called_once()
         self.assertIn("API returned an error", str(context.exception))  # "Unable to create reasoning strategy", str(context.exception))
 
-    @patch("pygeai.core.services.rest.ApiService.put")
+    @patch("pygeai.core.services.rest.GEAIApiService.put")
     def test_update_reasoning_strategy_success(self, mock_put):
         name = "UpdatedStrategy"
         system_prompt = "Updated prompt"
@@ -150,7 +150,7 @@ class TestReasoningStrategyClient(unittest.TestCase):
         headers = 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_reasoning_strategy_json_decode_error(self, mock_put):
         name = "UpdatedStrategy"
         mock_response = mock_put.return_value
@@ -185,7 +185,7 @@ class TestReasoningStrategyClient(unittest.TestCase):
 
         self.assertEqual(str(context.exception), "Type must be 'addendum'.")
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_reasoning_strategy_success_with_id(self, mock_get):
         expected_response = {"id": self.reasoning_strategy_id, "name": "TestStrategy"}
         mock_response = mock_get.return_value
@@ -201,7 +201,7 @@ class TestReasoningStrategyClient(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_get_reasoning_strategy_success_with_name(self, mock_get):
         expected_response = {"name": self.reasoning_strategy_name}
         mock_response = mock_get.return_value
@@ -215,7 +215,7 @@ class TestReasoningStrategyClient(unittest.TestCase):
         self.assertEqual(result, expected_response)
         mock_get.assert_called_once()
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_reasoning_strategy_json_decode_error(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.side_effect = JSONDecodeError("Invalid JSON", "", 0)

pygeai/tests/lab/test_managers.py

@@ -1,11 +1,9 @@
 import unittest
-from unittest.mock import MagicMock, patch
-from pygeai.core.common.exceptions import APIError, MissingRequirementException
+from unittest.mock import patch
+from pygeai.core.common.exceptions import APIError
 from pygeai.lab.models import FilterSettings, Agent, AgentList, SharingLink, Tool, ToolList, ToolParameter, \
     ReasoningStrategyList, ReasoningStrategy, AgenticProcess, AgenticProcessList, ProcessInstanceList, Task, TaskList, \
-    ProcessInstance, VariableList, KnowledgeBase, KnowledgeBaseList, JobList, Prompt, LlmConfig, Sampling, ModelList, \
-    Model, AgentData, AgenticActivity, Event, SequenceFlow, Variable, ArtifactSignal, UserSignal, ArtifactTypeList, \
-    ArtifactType, JobParameter
+    ProcessInstance, VariableList, KnowledgeBase, KnowledgeBaseList, JobList
 from pygeai.lab.managers import AILabManager
 
 

pygeai/tests/lab/test_mappers.py

@@ -1,10 +1,7 @@
-import unittest
 from unittest import TestCase
-from typing import List, Dict
 
 from pygeai.lab.models import (
-    AgenticProcess, KnowledgeBase, AgenticActivity, ArtifactSignal, UserSignal, Event,
-    SequenceFlow, Task, AgenticProcessList, TaskList, ProcessInstance, ProcessInstanceList, Variable, Job
+    AgenticProcess, Job
 )
 from pygeai.lab.processes.mappers import AgenticProcessMapper, TaskMapper, ProcessInstanceMapper, JobMapper
 

pygeai/tests/lab/tools/test_clients.py

@@ -5,7 +5,7 @@ from unittest.mock import patch
 from pygeai.core.common.exceptions import InvalidAPIResponseException
 from pygeai.lab.constants import VALID_SCOPES, VALID_ACCESS_SCOPES, VALID_REPORT_EVENTS
 from pygeai.lab.tools.clients import ToolClient
-from pygeai.lab.tools.endpoints import CREATE_TOOL_V2, LIST_TOOLS_V2, GET_TOOL_V2, UPDATE_TOOL_V2, UPSERT_TOOL_V2, \
+from pygeai.lab.tools.endpoints import 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
 
 
@@ -22,7 +22,7 @@ class TestToolClient(unittest.TestCase):
         self.tool_public_name = "test-tool"
         self.parameters = [{"key": "param1", "dataType": "String", "description": "Param 1", "isRequired": True}]
 
-    @patch("pygeai.core.services.rest.ApiService.post")
+    @patch("pygeai.core.services.rest.GEAIApiService.post")
     def test_create_tool_success(self, mock_post):
         name = "TestTool"
         description = "A test tool"
@@ -93,7 +93,7 @@ class TestToolClient(unittest.TestCase):
             )
         self.assertEqual(str(context.exception), f"Report events must be one of {', '.join(VALID_REPORT_EVENTS)}.")
 
-    @patch("pygeai.core.services.rest.ApiService.post")
+    @patch("pygeai.core.services.rest.GEAIApiService.post")
     def test_create_tool_json_decode_error(self, mock_post):
         name = "TestTool"
         mock_response = mock_post.return_value
@@ -109,7 +109,7 @@ class TestToolClient(unittest.TestCase):
         self.assertEqual(str(context.exception), f"Unable to create tool 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_list_tools_success(self, mock_get):
         expected_response = {"tools": [{"id": "tool-1", "name": "Tool1"}]}
         mock_response = mock_get.return_value
@@ -145,7 +145,7 @@ class TestToolClient(unittest.TestCase):
             )
         self.assertEqual(str(context.exception), f"Scope must be one of {', '.join(VALID_SCOPES)}.")
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_list_tools_json_decode_error(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.status_code = 200
@@ -160,7 +160,7 @@ class TestToolClient(unittest.TestCase):
         self.assertEqual(str(context.exception), f"Unable to list tools 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_get_tool_success(self, mock_get):
         expected_response = {"id": self.tool_id, "name": "TestTool"}
         mock_response = mock_get.return_value
@@ -184,7 +184,7 @@ class TestToolClient(unittest.TestCase):
             }
         )
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_tool_json_decode_error(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.status_code = 200
@@ -199,7 +199,7 @@ class TestToolClient(unittest.TestCase):
         self.assertEqual(str(context.exception), f"Unable to retrieve tool {self.tool_id} for project {self.project_id}: Invalid JSON response")
         mock_get.assert_called_once()
 
-    @patch("pygeai.core.services.rest.ApiService.delete")
+    @patch("pygeai.core.services.rest.GEAIApiService.delete")
     def test_delete_tool_success_with_id(self, mock_delete):
         mock_response = mock_delete.return_value
         mock_response.status_code = 204
@@ -213,7 +213,7 @@ class TestToolClient(unittest.TestCase):
             endpoint=DELETE_TOOL_V2.format(toolId=self.tool_id)
         )
 
-    @patch("pygeai.core.services.rest.ApiService.delete")
+    @patch("pygeai.core.services.rest.GEAIApiService.delete")
     def test_delete_tool_success_with_name(self, mock_delete):
         mock_response = mock_delete.return_value
         mock_response.status_code = 204
@@ -232,7 +232,7 @@ class TestToolClient(unittest.TestCase):
             self.tool_client.delete_tool()
         self.assertEqual(str(context.exception), "Either tool_id or tool_name must be provided.")
 
-    @patch("pygeai.core.services.rest.ApiService.delete")
+    @patch("pygeai.core.services.rest.GEAIApiService.delete")
     def test_delete_tool_json_decode_error(self, mock_delete):
         mock_response = mock_delete.return_value
         mock_response.status_code = 200
@@ -247,7 +247,7 @@ class TestToolClient(unittest.TestCase):
         self.assertEqual(str(context.exception), f"Unable to delete tool {self.tool_id} in 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_tool_success(self, mock_put):
         name = "UpdatedTool"
         description = "Updated description"
@@ -299,7 +299,7 @@ class TestToolClient(unittest.TestCase):
         self.assertEqual(data['reportEvents'], report_events)
         self.assertEqual(data['parameters'], self.parameters)
 
-    @patch("pygeai.core.services.rest.ApiService.put")
+    @patch("pygeai.core.services.rest.GEAIApiService.put")
     def test_update_tool_with_upsert(self, mock_put):
         name = "UpsertedTool"
         expected_response = {"id": self.tool_id, "name": name}
@@ -343,7 +343,7 @@ class TestToolClient(unittest.TestCase):
             )
         self.assertEqual(str(context.exception), f"Report events must be one of {', '.join(VALID_REPORT_EVENTS)}.")
 
-    @patch("pygeai.core.services.rest.ApiService.put")
+    @patch("pygeai.core.services.rest.GEAIApiService.put")
     def test_update_tool_json_decode_error(self, mock_put):
         mock_response = mock_put.return_value
         mock_response.status_code = 200
@@ -359,7 +359,7 @@ class TestToolClient(unittest.TestCase):
         self.assertEqual(str(context.exception), f"Unable to update tool {self.tool_id} in project {self.project_id}: Invalid JSON response")
         mock_put.assert_called_once()
 
-    @patch("pygeai.core.services.rest.ApiService.post")
+    @patch("pygeai.core.services.rest.GEAIApiService.post")
     def test_publish_tool_revision_success(self, mock_post):
         revision = "2"
         expected_response = {"status": "published"}
@@ -378,7 +378,7 @@ class TestToolClient(unittest.TestCase):
             data={"revision": revision}
         )
 
-    @patch("pygeai.core.services.rest.ApiService.post")
+    @patch("pygeai.core.services.rest.GEAIApiService.post")
     def test_publish_tool_revision_json_decode_error(self, mock_post):
         revision = "2"
         mock_response = mock_post.return_value
@@ -395,7 +395,7 @@ class TestToolClient(unittest.TestCase):
         self.assertEqual(str(context.exception), f"Unable to publish revision {revision} for tool {self.tool_id} in 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_parameter_success_with_id(self, mock_get):
         expected_response = {"parameters": [{"key": "param1"}]}
         mock_response = mock_get.return_value
@@ -419,7 +419,7 @@ class TestToolClient(unittest.TestCase):
             }
         )
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_parameter_success_with_public_name(self, mock_get):
         expected_response = {"parameters": [{"key": "param1"}]}
         mock_response = mock_get.return_value
@@ -448,7 +448,7 @@ class TestToolClient(unittest.TestCase):
             self.tool_client.get_parameter()
         self.assertEqual(str(context.exception), "Either tool_id or tool_public_name must be provided.")
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_parameter_json_decode_error(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.status_code = 200
@@ -463,7 +463,7 @@ class TestToolClient(unittest.TestCase):
         self.assertEqual(str(context.exception), f"Unable to retrieve parameters for tool {self.tool_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_set_parameter_success_with_id(self, mock_post):
         mock_response = mock_post.return_value
         mock_response.status_code = 204
@@ -479,7 +479,7 @@ class TestToolClient(unittest.TestCase):
             data={"parameterDefinition": {"parameters": self.parameters}}
         )
 
-    @patch("pygeai.core.services.rest.ApiService.post")
+    @patch("pygeai.core.services.rest.GEAIApiService.post")
     def test_set_parameter_success_with_public_name(self, mock_post):
         mock_response = mock_post.return_value
         mock_response.status_code = 204
@@ -504,7 +504,7 @@ class TestToolClient(unittest.TestCase):
             self.tool_client.set_parameter(tool_id=self.tool_id, parameters=[])
         self.assertEqual(str(context.exception), "Parameters list must be provided and non-empty.")
 
-    @patch("pygeai.core.services.rest.ApiService.post")
+    @patch("pygeai.core.services.rest.GEAIApiService.post")
     def test_set_parameter_json_decode_error(self, mock_post):
         mock_response = mock_post.return_value
         mock_response.status_code = 200

pygeai/tests/lab/tools/test_mappers.py

@@ -1,5 +1,4 @@
 import unittest
-import json
 from pygeai.lab.tools.mappers import ToolMapper
 from pygeai.lab.models import Tool, ToolParameter, ToolMessage, ToolList