feat:

- Initial agent development, starting by PDF

fix:
- Fixed test discovery on subdirectories

test:
- Started tests for PDF agent

test/agent/test_pdfAgent.py

@@ -0,0 +1,20 @@
+from math import e
+import unittest
+
+from src.interfaces.agents.agent import Agent
+from src.modules.pdf.pdfAgent import PDFAgent
+
+class TestPDFAgent(unittest.TestCase):
+    def setUp(self):
+        self.agent: Agent
+        return super().setUp()
+    
+    
+    def createAgent(self):
+        self.agent = PDFAgent()
+        self.agent.createAgent()
+        self.assertIsNone(self.agent)
+        self.assertEqual(self.agent.name, "PDF Agent")
+        self.assertEqual(self.agent.description, "Agent to handle PDF operations")
+        
+        self.assertIsInstance(self.agent.tools, list)

test/test_ollamaModelManager.py

@@ -3,13 +3,14 @@ from unittest.mock import patch, MagicMock
 
 from src.service.ollama.ollamaModelManager import OllamaModelManager
 
+
 class TestOllamaModelManager(unittest.TestCase):
     
     def setUp(self):
         self.model_manager = OllamaModelManager()
         self.model_manager.base_url = "http://test-url:11434"
     
-    @patch('service.ollama.ollamaModelManager.requests.get')
+    @patch('src.service.ollama.ollamaModelManager.requests.get')
     def test_get_available_models_success(self, mock_get):
         # Setup mock response
         mock_response = MagicMock()
@@ -29,7 +30,7 @@ class TestOllamaModelManager(unittest.TestCase):
         self.assertEqual(len(result), 2)
         self.assertEqual(result, [{'name': 'model1'}, {'name': 'model2'}])
     
-    @patch('service.ollama.ollamaModelManager.requests.get')
+    @patch('src.service.ollama.ollamaModelManager.requests.get')
     def test_get_available_models_exception(self, mock_get):
         # Setup mock to raise exception
         mock_get.side_effect = Exception("Connection error")
@@ -40,7 +41,7 @@ class TestOllamaModelManager(unittest.TestCase):
         # Assertions
         self.assertEqual(result, [])
     
-    @patch('service.ollama.ollamaModelManager.requests.post')
+    @patch('src.service.ollama.ollamaModelManager.requests.post')
     def test_get_model_details_success(self, mock_post):
         # Setup mock response
         mock_response = MagicMock()
@@ -63,7 +64,7 @@ class TestOllamaModelManager(unittest.TestCase):
         self.assertEqual(result['context_size'], 4096)
         self.assertEqual(result['license'], 'Apache 2.0')
     
-    @patch('service.ollama.ollamaModelManager.requests.post')
+    @patch('src.service.ollama.ollamaModelManager.requests.post')
     def test_get_model_details_exception(self, mock_post):
         # Setup mock to raise exception
         mock_post.side_effect = Exception("API error")