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

pygeai/tests/core/feedback/test_clients.py

@@ -18,6 +18,7 @@ class TestFeedbackClient(unittest.TestCase):
     def test_send_feedback_without_comments(self, mock_post):
         mock_response = Mock()
         mock_response.json.return_value = {}
+        mock_response.status_code = 200
         mock_post.return_value = mock_response
 
         result = self.feedback_client.send_feedback(
@@ -40,6 +41,7 @@ class TestFeedbackClient(unittest.TestCase):
     def test_send_feedback_with_comments(self, mock_post):
         mock_response = Mock()
         mock_response.json.return_value = {}
+        mock_response.status_code = 200
         mock_post.return_value = mock_response
 
         result = self.feedback_client.send_feedback(

pygeai/tests/core/files/test_clients.py

@@ -18,6 +18,7 @@ class TestFileClient(unittest.TestCase):
         self.file_path = Path("test_file.txt")
         self.mock_response = MagicMock()
         self.mock_response.json.return_value = {"id": self.file_id, "status": "success"}
+        self.mock_response.status_code = 200
         self.mock_response.text = json.dumps({"id": self.file_id, "status": "success"})
         self.mock_response.content = b"file content"
         self.mock_file = MagicMock()

pygeai/tests/core/llm/test_clients.py

@@ -2,7 +2,7 @@ import unittest
 from unittest.mock import patch, MagicMock
 from json import JSONDecodeError
 from pygeai.core.llm.clients import LlmClient
-from pygeai.core.common.exceptions import InvalidAPIResponseException
+from pygeai.core.common.exceptions import InvalidAPIResponseException, APIResponseError, APIResponseError
 
 
 class TestLlmClient(unittest.TestCase):
@@ -21,6 +21,7 @@ class TestLlmClient(unittest.TestCase):
     def test_get_provider_list_success(self):
         expected_response = {"providers": [{"name": "provider1"}, {"name": "provider2"}]}
         self.mock_response.json.return_value = expected_response
+        self.mock_response.status_code = 200
         self.client.api_service.get.return_value = self.mock_response
 
         result = self.client.get_provider_list()
@@ -34,15 +35,16 @@ class TestLlmClient(unittest.TestCase):
         self.mock_response.status_code = 500
         self.client.api_service.get.return_value = self.mock_response
 
-        with self.assertRaises(InvalidAPIResponseException) as context:
+        with self.assertRaises(APIResponseError) as context:
             self.client.get_provider_list()
 
         self.client.api_service.get.assert_called_once()
-        self.assertIn("Unable to obtain provider list", str(context.exception))
+        self.assertIn("API returned an error", str(context.exception))  # "Unable to obtain provider list", str(context.exception))
 
     def test_get_provider_data_success(self):
         expected_response = {"name": self.provider_name, "details": "test details"}
         self.mock_response.json.return_value = expected_response
+        self.mock_response.status_code = 200
         self.client.api_service.get.return_value = self.mock_response
 
         result = self.client.get_provider_data(provider_name=self.provider_name)
@@ -58,15 +60,16 @@ class TestLlmClient(unittest.TestCase):
         self.mock_response.status_code = 500
         self.client.api_service.get.return_value = self.mock_response
 
-        with self.assertRaises(InvalidAPIResponseException) as context:
+        with self.assertRaises(APIResponseError) as context:
             self.client.get_provider_data(provider_name=self.provider_name)
 
         self.client.api_service.get.assert_called_once()
-        self.assertIn("Unable to obtain provider data", str(context.exception))
+        self.assertIn("API returned an error", str(context.exception))  # "Unable to obtain provider data", str(context.exception))
 
     def test_get_provider_models_success(self):
         expected_response = {"models": [{"name": "model1"}, {"name": "model2"}]}
         self.mock_response.json.return_value = expected_response
+        self.mock_response.status_code = 200
         self.client.api_service.get.return_value = self.mock_response
 
         result = self.client.get_provider_models(provider_name=self.provider_name)
@@ -82,15 +85,16 @@ class TestLlmClient(unittest.TestCase):
         self.mock_response.status_code = 500
         self.client.api_service.get.return_value = self.mock_response
 
-        with self.assertRaises(InvalidAPIResponseException) as context:
+        with self.assertRaises(APIResponseError) as context:
             self.client.get_provider_models(provider_name=self.provider_name)
 
         self.client.api_service.get.assert_called_once()
-        self.assertIn("Unable to obtain provider models", str(context.exception))
+        self.assertIn("API returned an error", str(context.exception))  # "Unable to obtain provider models", str(context.exception))
 
     def test_get_model_data_with_model_name_success(self):
         expected_response = {"name": self.model_name, "provider": self.provider_name}
         self.mock_response.json.return_value = expected_response
+        self.mock_response.status_code = 200
         self.client.api_service.get.return_value = self.mock_response
 
         result = self.client.get_model_data(
@@ -107,6 +111,7 @@ class TestLlmClient(unittest.TestCase):
     def test_get_model_data_with_model_id_success(self):
         expected_response = {"id": self.model_id, "provider": self.provider_name}
         self.mock_response.json.return_value = expected_response
+        self.mock_response.status_code = 200
         self.client.api_service.get.return_value = self.mock_response
 
         result = self.client.get_model_data(
@@ -126,12 +131,12 @@ class TestLlmClient(unittest.TestCase):
         self.mock_response.status_code = 500
         self.client.api_service.get.return_value = self.mock_response
 
-        with self.assertRaises(InvalidAPIResponseException) as context:
+        with self.assertRaises(APIResponseError) as context:
             self.client.get_model_data(
                 provider_name=self.provider_name,
                 model_name=self.model_name
             )
 
         self.client.api_service.get.assert_called_once()
-        self.assertIn("Unable to obtain model data", str(context.exception))
+        self.assertIn("API returned an error", str(context.exception))  # "Unable to obtain model data", str(context.exception))
 

pygeai/tests/core/plugins/test_clients.py

@@ -3,7 +3,7 @@ from unittest.mock import patch, MagicMock
 from json import JSONDecodeError
 
 from pygeai.core.plugins.clients import PluginClient
-from pygeai.core.common.exceptions import InvalidAPIResponseException
+from pygeai.core.common.exceptions import InvalidAPIResponseException, APIResponseError, APIResponseError
 
 
 class TestPluginClient(unittest.TestCase):
@@ -23,6 +23,7 @@ class TestPluginClient(unittest.TestCase):
                 {"id": "assistant-2", "name": "Assistant 2"}
             ]
         }
+        self.mock_response.status_code = 200
         mock_get.return_value = self.mock_response
         
         result = self.client.list_assistants(
@@ -39,6 +40,7 @@ class TestPluginClient(unittest.TestCase):
     @patch('pygeai.core.services.rest.ApiService.get')
     def test_list_assistants_empty(self, mock_get):
         self.mock_response.json.return_value = {"assistants": []}
+        self.mock_response.status_code = 200
         mock_get.return_value = self.mock_response
         
         result = self.client.list_assistants(
@@ -55,9 +57,9 @@ class TestPluginClient(unittest.TestCase):
         self.mock_response.text = "Internal server error"
         mock_get.return_value = self.mock_response
         
-        with self.assertRaises(InvalidAPIResponseException) as context:
+        with self.assertRaises(APIResponseError) as context:
             self.client.list_assistants("org-123", "proj-456")
-        self.assertIn("Unable to list assistants for organization org-123 and project proj-456", str(context.exception))
+        self.assertIn("API returned an error", str(context.exception))  # "Unable to list assistants for organization org-123 and project proj-456", str(context.exception))
 
 
 if __name__ == '__main__':

pygeai/tests/core/rerank/test_clients.py

@@ -13,6 +13,7 @@ class TestRerankClient(unittest.TestCase):
         self.client.api_service = MagicMock()
         self.mock_response = MagicMock()
         self.mock_response.json.return_value = {"results": [{"index": 0, "relevance_score": 0.95}]}
+        self.mock_response.status_code = 200
 
     def test_rerank_chunks_success(self):
         self.client.api_service.post.return_value = self.mock_response

pygeai/tests/core/secrets/test_clients.py

@@ -3,7 +3,7 @@ from unittest.mock import patch, MagicMock
 import json
 from json import JSONDecodeError
 from pygeai.core.secrets.clients import SecretClient
-from pygeai.core.common.exceptions import InvalidAPIResponseException
+from pygeai.core.common.exceptions import InvalidAPIResponseException, APIResponseError, APIResponseError
 
 
 class TestSecretClient(unittest.TestCase):
@@ -19,6 +19,7 @@ class TestSecretClient(unittest.TestCase):
         expected_response = {"secrets": [{"id": "1", "name": "secret1"}]}
         mock_response = MagicMock()
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
         self.mock_api_service.get.return_value = mock_response
 
         result = self.secret_client.list_secrets(name="secret1", start=0, count=5)
@@ -38,10 +39,10 @@ class TestSecretClient(unittest.TestCase):
         mock_response.status_code = 500
         self.mock_api_service.get.return_value = mock_response
 
-        with self.assertRaises(InvalidAPIResponseException) as context:
+        with self.assertRaises(APIResponseError) as context:
             self.secret_client.list_secrets()
 
-        self.assertIn("Unable to list secrets with params", str(context.exception))
+        self.assertIn("API returned an error", str(context.exception))  # "Unable to list secrets with params", str(context.exception))
         self.mock_api_service.get.assert_called_once()
 
     def test_get_secret_success(self):
@@ -49,6 +50,7 @@ class TestSecretClient(unittest.TestCase):
         expected_response = {"id": "secret-123", "name": "test-secret"}
         mock_response = MagicMock()
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
         self.mock_api_service.get.return_value = mock_response
 
         result = self.secret_client.get_secret(secret_id)
@@ -65,10 +67,10 @@ class TestSecretClient(unittest.TestCase):
         mock_response.status_code = 500
         self.mock_api_service.get.return_value = mock_response
 
-        with self.assertRaises(InvalidAPIResponseException) as context:
+        with self.assertRaises(APIResponseError) as context:
             self.secret_client.get_secret(secret_id)
 
-        self.assertIn(f"Unable to get secret with ID '{secret_id}'", str(context.exception))
+        self.assertIn("API returned an error", str(context.exception))  # f"Unable to get secret with ID '{secret_id}'", str(context.exception))
         self.mock_api_service.get.assert_called_once()
 
     def test_get_secret_invalid_id(self):
@@ -83,6 +85,7 @@ class TestSecretClient(unittest.TestCase):
         expected_response = {"id": "secret-123", "name": name}
         mock_response = MagicMock()
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
         self.mock_api_service.post.return_value = mock_response
 
         result = self.secret_client.create_secret(name, secret_string, description)
@@ -105,10 +108,10 @@ class TestSecretClient(unittest.TestCase):
         mock_response.status_code = 500
         self.mock_api_service.post.return_value = mock_response
 
-        with self.assertRaises(InvalidAPIResponseException) as context:
+        with self.assertRaises(APIResponseError) as context:
             self.secret_client.create_secret(name, secret_string)
 
-        self.assertIn(f"Unable to create secret with name '{name}'", str(context.exception))
+        self.assertIn("API returned an error", str(context.exception))  # f"Unable to create secret with name '{name}'", str(context.exception))
         self.mock_api_service.post.assert_called_once()
 
     def test_create_secret_invalid_input(self):
@@ -128,6 +131,7 @@ class TestSecretClient(unittest.TestCase):
         expected_response = {"id": secret_id, "name": name}
         mock_response = MagicMock()
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
         self.mock_api_service.put.return_value = mock_response
 
         result = self.secret_client.update_secret(secret_id, name, secret_string, description)
@@ -151,10 +155,10 @@ class TestSecretClient(unittest.TestCase):
         mock_response.status_code = 500
         self.mock_api_service.put.return_value = mock_response
 
-        with self.assertRaises(InvalidAPIResponseException) as context:
+        with self.assertRaises(APIResponseError) as context:
             self.secret_client.update_secret(secret_id, name, secret_string)
 
-        self.assertIn(f"Unable to update secret with ID '{secret_id}'", str(context.exception))
+        self.assertIn("API returned an error", str(context.exception))  # f"Unable to update secret with ID '{secret_id}'", str(context.exception))
         self.mock_api_service.put.assert_called_once()
 
     def test_update_secret_invalid_input(self):
@@ -179,6 +183,7 @@ class TestSecretClient(unittest.TestCase):
         expected_response = {"status": "success"}
         mock_response = MagicMock()
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
         self.mock_api_service.post.return_value = mock_response
 
         result = self.secret_client.set_secret_accesses(secret_id, access_list)
@@ -201,10 +206,10 @@ class TestSecretClient(unittest.TestCase):
         mock_response.status_code = 500
         self.mock_api_service.post.return_value = mock_response
 
-        with self.assertRaises(InvalidAPIResponseException) as context:
+        with self.assertRaises(APIResponseError) as context:
             self.secret_client.set_secret_accesses(secret_id, access_list)
 
-        self.assertIn(f"Unable to set accesses for secret with ID '{secret_id}'", str(context.exception))
+        self.assertIn("API returned an error", str(context.exception))  # f"Unable to set accesses for secret with ID '{secret_id}'", str(context.exception))
         self.mock_api_service.post.assert_called_once()
 
     def test_set_secret_accesses_invalid_input(self):
@@ -229,6 +234,7 @@ class TestSecretClient(unittest.TestCase):
         expected_response = {"accesses": [{"accessLevel": "write", "principalType": "service"}]}
         mock_response = MagicMock()
         mock_response.json.return_value = expected_response
+        mock_response.status_code = 200
         self.mock_api_service.get.return_value = mock_response
 
         result = self.secret_client.get_secret_accesses(secret_id)
@@ -245,10 +251,10 @@ class TestSecretClient(unittest.TestCase):
         mock_response.status_code = 500
         self.mock_api_service.get.return_value = mock_response
 
-        with self.assertRaises(InvalidAPIResponseException) as context:
+        with self.assertRaises(APIResponseError) as context:
             self.secret_client.get_secret_accesses(secret_id)
 
-        self.assertIn(f"Unable to get accesses for secret with ID '{secret_id}'", str(context.exception))
+        self.assertIn("API returned an error", str(context.exception))  # f"Unable to get accesses for secret with ID '{secret_id}'", str(context.exception))
         self.mock_api_service.get.assert_called_once()
 
     def test_get_secret_accesses_invalid_id(self):