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

pygeai/tests/evaluation/result/test_clients.py

@@ -2,6 +2,7 @@ import unittest
 from unittest.mock import patch, MagicMock
 
 from pygeai.evaluation.result.clients import EvaluationResultClient
+from pygeai.core.base.session import Session
 
 
 class TestEvaluationResultClient(unittest.TestCase):
@@ -10,12 +11,13 @@ class TestEvaluationResultClient(unittest.TestCase):
     """
 
     def setUp(self):
-        self.client = EvaluationResultClient()
+        self.client = EvaluationResultClient(api_key="test-key", base_url="http://test.com", eval_url="http://eval.com")
         self.mock_response = MagicMock()
+        self.mock_response.status_code = 200
 
     @patch('pygeai.core.services.rest.ApiService.get')
     def test_list_evaluation_results_success(self, mock_get):
-        self.mock_response.content = b'[{"id": "result-1", "status": "completed"}, {"id": "result-2", "status": "pending"}]'
+        self.mock_response.json.return_value = [{"id": "result-1", "status": "completed"}, {"id": "result-2", "status": "pending"}]
         mock_get.return_value = self.mock_response
         
         result = self.client.list_evaluation_results("plan-123")
@@ -27,7 +29,7 @@ class TestEvaluationResultClient(unittest.TestCase):
 
     @patch('pygeai.core.services.rest.ApiService.get')
     def test_list_evaluation_results_empty(self, mock_get):
-        self.mock_response.content = b'[]'
+        self.mock_response.json.return_value = []
         mock_get.return_value = self.mock_response
         
         result = self.client.list_evaluation_results("plan-456")
@@ -37,7 +39,7 @@ class TestEvaluationResultClient(unittest.TestCase):
 
     @patch('pygeai.core.services.rest.ApiService.get')
     def test_get_evaluation_result_success(self, mock_get):
-        self.mock_response.content = b'{"id": "result-123", "status": "completed", "score": 0.95}'
+        self.mock_response.json.return_value = {"id": "result-123", "status": "completed", "score": 0.95}
         mock_get.return_value = self.mock_response
         
         result = self.client.get_evaluation_result("result-123")
@@ -49,7 +51,7 @@ class TestEvaluationResultClient(unittest.TestCase):
 
     @patch('pygeai.core.services.rest.ApiService.get')
     def test_get_evaluation_result_with_details(self, mock_get):
-        self.mock_response.content = b'{"id": "result-789", "status": "failed", "error": "Test error"}'
+        self.mock_response.json.return_value = {"id": "result-789", "status": "failed", "error": "Test error"}
         mock_get.return_value = self.mock_response
         
         result = self.client.get_evaluation_result("result-789")

pygeai/tests/gam/test_clients.py

@@ -15,7 +15,7 @@ class TestGAMClient(unittest.TestCase):
         self.mock_response = MagicMock()
         self.mock_response.json.return_value = {"status": "success"}
         self.mock_response.text = "success text"
-        self.mock_response.status_code = 500
+        self.mock_response.status_code = 200
         self.client.api_service.base_url = "https://api.example.com"
 
     def test_generate_signing_url_success(self):

pygeai/tests/health/test_clients.py

@@ -20,6 +20,7 @@ class TestHealthClient(unittest.TestCase):
         expected_response = {"status": "healthy", "version": "1.0.0"}
         mock_response = mock_get.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.check_api_status()
 

pygeai/tests/lab/agents/test_clients.py

@@ -29,6 +29,7 @@ class TestAgentClient(unittest.TestCase):
         expected_response = {"agents": [{"id": "agent-1", "name": "Agent1"}]}
         mock_response = mock_get.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.list_agents(
             status="active",
@@ -73,6 +74,7 @@ class TestAgentClient(unittest.TestCase):
         expected_response = {"id": "agent-123", "name": "Test Agent"}
         mock_response = mock_post.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.create_agent(
             name="Test Agent",
@@ -114,6 +116,7 @@ class TestAgentClient(unittest.TestCase):
         expected_response = {"id": "agent-123", "name": "Test Agent"}
         mock_response = mock_post.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.create_agent(
             name="Test Agent",
@@ -179,6 +182,7 @@ class TestAgentClient(unittest.TestCase):
         expected_response = {"id": self.agent_id, "name": "Test Agent"}
         mock_response = mock_get.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.get_agent(
             agent_id=self.agent_id,
@@ -223,6 +227,7 @@ class TestAgentClient(unittest.TestCase):
         expected_response = {"link": "http://example.com/share"}
         mock_response = mock_get.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.create_sharing_link(
             agent_id=self.agent_id
@@ -263,6 +268,7 @@ class TestAgentClient(unittest.TestCase):
         expected_response = {"status": "published"}
         mock_response = mock_post.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.publish_agent_revision(
             agent_id=self.agent_id,
@@ -334,6 +340,7 @@ class TestAgentClient(unittest.TestCase):
         expected_response = {"id": self.agent_id, "name": "Updated Agent"}
         mock_response = mock_put.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.update_agent(
             agent_id=self.agent_id,
@@ -377,6 +384,7 @@ class TestAgentClient(unittest.TestCase):
         expected_response = {"id": self.agent_id, "name": "Upserted Agent"}
         mock_response = mock_put.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.update_agent(
             agent_id=self.agent_id,
@@ -406,6 +414,7 @@ class TestAgentClient(unittest.TestCase):
         expected_response = {"id": self.agent_id, "name": "Updated Agent No Pools"}
         mock_response = mock_put.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.update_agent(
             agent_id=self.agent_id,

pygeai/tests/lab/processes/test_clients.py

@@ -35,6 +35,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"id": "process-123", "name": "Test Process"}
         mock_response = mock_post.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.create_process(
             key="test-key",
@@ -94,6 +95,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"id": self.process_id, "name": "Updated Process"}
         mock_response_put = mock_put.return_value
         mock_response_put.json.return_value = expected_response
+        mock_response_put.status_code = 200
 
         mock_response_get = mock_get.return_value
         mock_response_get.json.return_value = {
@@ -102,6 +104,7 @@ class TestAgenticProcessClient(unittest.TestCase):
                 "agenticActivities": [{"name": "current-activity"}]
             }
         }
+        mock_response_get.status_code = 200
 
         result = self.client.update_process(
             process_id=self.process_id,
@@ -130,6 +133,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"id": "process-123", "name": "Updated Process"}
         mock_response_put = mock_put.return_value
         mock_response_put.json.return_value = expected_response
+        mock_response_put.status_code = 200
 
         mock_response_get = mock_get.return_value
         mock_response_get.json.return_value = {
@@ -138,6 +142,7 @@ class TestAgenticProcessClient(unittest.TestCase):
                 "agenticActivities": [{"name": "current-activity"}]
             }
         }
+        mock_response_get.status_code = 200
 
         result = self.client.update_process(
             name=self.process_name,
@@ -160,6 +165,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"id": self.process_id, "name": "Upserted Process"}
         mock_response = mock_put.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.update_process(
             process_id=self.process_id,
@@ -194,6 +200,7 @@ class TestAgenticProcessClient(unittest.TestCase):
                 "agenticActivities": [{"name": "current-activity"}]
             }
         }
+        mock_response_get.status_code = 200
 
         with self.assertRaises(InvalidAPIResponseException) as context:
             self.client.update_process(
@@ -211,6 +218,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"id": self.process_id, "name": "Test Process"}
         mock_response = mock_get.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.get_process(
             process_id=self.process_id,
@@ -237,6 +245,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"id": "process-123", "name": self.process_name}
         mock_response = mock_get.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.get_process(
             process_name=self.process_name
@@ -274,6 +283,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"processes": [{"id": "process-1", "name": "Process1"}]}
         mock_response = mock_get.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.list_processes(
             id="process-1",
@@ -318,6 +328,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"instances": [{"id": "instance-1"}]}
         mock_response = mock_get.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.list_process_instances(
             process_id=self.process_id,
@@ -418,6 +429,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"status": "published"}
         mock_response = mock_post.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.publish_process_revision(
             process_id=self.process_id,
@@ -438,6 +450,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"status": "published"}
         mock_response = mock_post.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.publish_process_revision(
             process_name=self.process_name,
@@ -486,6 +499,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"id": "task-123", "name": "Test Task"}
         mock_response = mock_post.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.create_task(
             name="Test Task",
@@ -533,6 +547,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"id": self.task_id, "name": "Test Task"}
         mock_response = mock_get.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.get_task(
             task_id=self.task_id
@@ -551,6 +566,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"id": "task-123", "name": self.task_name}
         mock_response = mock_get.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.get_task(
             task_id=self.task_id,
@@ -588,6 +604,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"tasks": [{"id": "task-1", "name": "Task1"}]}
         mock_response = mock_get.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.list_tasks(
             id="task-1",
@@ -628,6 +645,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"id": self.task_id, "name": "Updated Task"}
         mock_response = mock_put.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.update_task(
             task_id=self.task_id,
@@ -660,6 +678,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"id": self.task_id, "name": "Upserted Task"}
         mock_response = mock_put.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.update_task(
             task_id=self.task_id,
@@ -755,6 +774,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"status": "published"}
         mock_response = mock_post.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.publish_task_revision(
             task_id=self.task_id,
@@ -775,6 +795,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"status": "published"}
         mock_response = mock_post.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.publish_task_revision(
             task_id=self.task_id,
@@ -826,6 +847,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"id": "instance-123", "status": "started"}
         mock_response = mock_post.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.start_instance(
             process_name=self.process_name,
@@ -866,6 +888,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"status": "aborted"}
         mock_response = mock_post.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.abort_instance(
             instance_id=self.instance_id
@@ -926,6 +949,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"history": [{"event": "start", "time": "2023-01-01"}]}
         mock_response = mock_get.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.get_instance_history(
             instance_id=self.instance_id
@@ -965,6 +989,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"thread": {"id": self.thread_id, "status": "active"}}
         mock_response = mock_get.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.get_thread_information(
             thread_id=self.thread_id
@@ -1004,6 +1029,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"status": "signal sent"}
         mock_response = mock_post.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.send_user_signal(
             instance_id=self.instance_id,
@@ -1057,6 +1083,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"id": "kb-123", "name": "Test KB"}
         mock_response = mock_post.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.create_kb(
             name="Test KB",
@@ -1098,6 +1125,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"id": self.kb_id, "name": "Test KB"}
         mock_response = mock_get.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.get_kb(
             kb_id=self.kb_id
@@ -1116,6 +1144,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"id": "kb-123", "name": self.kb_name}
         mock_response = mock_get.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.get_kb(
             kb_name=self.kb_name
@@ -1153,6 +1182,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"kbs": [{"id": "kb-1", "name": "KB1"}]}
         mock_response = mock_get.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.list_kbs(
             name="KB1",
@@ -1247,6 +1277,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"jobs": [{"id": "job-1", "name": "Job1"}]}
         mock_response = mock_get.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.list_jobs(
             start="0",
@@ -1291,6 +1322,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"id": self.process_id, "name": "Updated Process"}
         mock_response_put = mock_put.return_value
         mock_response_put.json.return_value = expected_response
+        mock_response_put.status_code = 200
 
         mock_response_get = mock_get.return_value
         mock_response_get.json.return_value = {
@@ -1299,6 +1331,7 @@ class TestAgenticProcessClient(unittest.TestCase):
                 "agenticActivities": [{"name": "current-activity"}]
             }
         }
+        mock_response_get.status_code = 200
 
         result = self.client.update_process(
             process_id=self.process_id,
@@ -1327,6 +1360,7 @@ class TestAgenticProcessClient(unittest.TestCase):
         expected_response = {"id": "process-123", "name": "Updated Process"}
         mock_response_put = mock_put.return_value
         mock_response_put.json.return_value = expected_response
+        mock_response_put.status_code = 200
 
         mock_response_get = mock_get.return_value
         mock_response_get.json.return_value = {
@@ -1335,6 +1369,7 @@ class TestAgenticProcessClient(unittest.TestCase):
                 "agenticActivities": [{"name": "current-activity"}]
             }
         }
+        mock_response_get.status_code = 200
 
         result = self.client.update_process(
             name=self.process_name,
@@ -1367,6 +1402,7 @@ class TestAgenticProcessClient(unittest.TestCase):
                 "agenticActivities": [{"name": "current-activity"}]
             }
         }
+        mock_response_get.status_code = 200
 
         with self.assertRaises(InvalidAPIResponseException) as context:
             self.client.update_process(

pygeai/tests/lab/processes/test_mappers.py

@@ -26,6 +26,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.revision = "1"
         self.mock_response = MagicMock()
         self.mock_response.json.return_value = {"status": "success"}
+        self.mock_response.status_code = 200
         self.mock_response.text = "success text"
 
     def test_map_to_agentic_process_full_data(self):
@@ -122,6 +123,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
     def test_update_process_success_with_name(self):
         self.client.api_service.put.return_value = self.mock_response
+        self.client.api_service.get.return_value = self.mock_response
 
         result = self.client.update_process(
             name="Test Process",
@@ -151,6 +153,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
     def test_update_process_json_decode_error(self):
         self.mock_response.json.side_effect = JSONDecodeError("Invalid JSON", "", 0)
         self.client.api_service.put.return_value = self.mock_response
+        self.client.api_service.get.return_value = self.mock_response
 
         try:
             result = self.client.update_process(

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
+from pygeai.core.common.exceptions import InvalidAPIResponseException, APIResponseError, APIResponseError
 
 
 class TestReasoningStrategyClient(unittest.TestCase):
@@ -20,6 +20,7 @@ class TestReasoningStrategyClient(unittest.TestCase):
         expected_response = {"strategies": [{"id": "strat-1", "name": "Strategy1"}]}
         mock_response = mock_get.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.list_reasoning_strategies(
             name="Strategy1",
@@ -46,11 +47,11 @@ class TestReasoningStrategyClient(unittest.TestCase):
         mock_response.text = "Raw response text"
         mock_response.status_code = 500
 
-        with self.assertRaises(InvalidAPIResponseException) as context:
+        with self.assertRaises(APIResponseError) as context:
             self.client.list_reasoning_strategies()
 
         mock_get.assert_called_once()
-        self.assertIn("Unable to list reasoning strategies", str(context.exception))
+        self.assertIn("API returned an error", str(context.exception))  # "Unable to list reasoning strategies", str(context.exception))
 
     def test_list_reasoning_strategies_invalid_access_scope(self):
         with self.assertRaises(ValueError) as context:
@@ -69,6 +70,7 @@ class TestReasoningStrategyClient(unittest.TestCase):
         expected_response = {"id": "strat-123", "name": name}
         mock_response = mock_post.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.create_reasoning_strategy(
             name=name,
@@ -101,14 +103,14 @@ class TestReasoningStrategyClient(unittest.TestCase):
         mock_response.text = "Raw response text"
         mock_response.status_code = 500
 
-        with self.assertRaises(InvalidAPIResponseException) as context:
+        with self.assertRaises(APIResponseError) as context:
             self.client.create_reasoning_strategy(
                 name=name,
                 system_prompt=system_prompt
             )
 
         mock_post.assert_called_once()
-        self.assertIn("Unable to create reasoning strategy", str(context.exception))
+        self.assertIn("API returned an error", str(context.exception))  # "Unable to create reasoning strategy", str(context.exception))
 
     @patch("pygeai.core.services.rest.ApiService.put")
     def test_update_reasoning_strategy_success(self, mock_put):
@@ -122,6 +124,7 @@ class TestReasoningStrategyClient(unittest.TestCase):
         expected_response = {"id": self.reasoning_strategy_id, "name": name}
         mock_response = mock_put.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.update_reasoning_strategy(
             reasoning_strategy_id=self.reasoning_strategy_id,
@@ -155,14 +158,14 @@ class TestReasoningStrategyClient(unittest.TestCase):
         mock_response.text = "Raw response text"
         mock_response.status_code = 500
 
-        with self.assertRaises(InvalidAPIResponseException) as context:
+        with self.assertRaises(APIResponseError) as context:
             self.client.update_reasoning_strategy(
                 reasoning_strategy_id=self.reasoning_strategy_id,
                 name=name
             )
 
         mock_put.assert_called_once()
-        self.assertIn("Unable to update reasoning strategy", str(context.exception))
+        self.assertIn("API returned an error", str(context.exception))  # "Unable to update reasoning strategy", str(context.exception))
 
     def test_update_reasoning_strategy_invalid_access_scope(self):
         with self.assertRaises(ValueError) as context:
@@ -187,6 +190,7 @@ class TestReasoningStrategyClient(unittest.TestCase):
         expected_response = {"id": self.reasoning_strategy_id, "name": "TestStrategy"}
         mock_response = mock_get.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.get_reasoning_strategy(
             reasoning_strategy_id=self.reasoning_strategy_id
@@ -202,6 +206,7 @@ class TestReasoningStrategyClient(unittest.TestCase):
         expected_response = {"name": self.reasoning_strategy_name}
         mock_response = mock_get.return_value
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
 
         result = self.client.get_reasoning_strategy(
             reasoning_strategy_name=self.reasoning_strategy_name
@@ -217,13 +222,13 @@ class TestReasoningStrategyClient(unittest.TestCase):
         mock_response.text = "Raw response text"
         mock_response.status_code = 500
 
-        with self.assertRaises(InvalidAPIResponseException) as context:
+        with self.assertRaises(APIResponseError) as context:
             self.client.get_reasoning_strategy(
                 reasoning_strategy_id=self.reasoning_strategy_id
             )
 
         mock_get.assert_called_once()
-        self.assertIn("Unable to retrieve reasoning strategy", str(context.exception))
+        self.assertIn("API returned an error", str(context.exception))  # "Unable to retrieve reasoning strategy", str(context.exception))
 
     def test_get_reasoning_strategy_invalid_input(self):
         with self.assertRaises(ValueError) as context: