refactor:

- changed method get_agent_tools for only get_tools on agent interface
- changed OllamaModelManager to correc semantic OllamaService

test:
- added test for service initialization
- added tests to list tools and create tools into  PDF agent
This commit is contained in:
2025-03-23 21:52:11 -03:00
parent 8851ad888c
commit 2aad4b8bf0
20 changed files with 1071 additions and 37 deletions

View File

@@ -3,18 +3,34 @@ import unittest
from src.interfaces.agents.agent import Agent
from src.modules.pdf.pdfAgent import PDFAgent
from src.modules.pdf.tools.metadataPDFTool import PDFMetadataTool
class TestPDFAgent(unittest.TestCase):
def setUp(self):
self.agent: Agent
self.agent: Agent = PDFAgent()
return super().setUp()
def createAgent(self):
self.agent = PDFAgent()
self.agent.createAgent()
self.assertIsNone(self.agent)
def test_createAgent(self):
self.assertIsInstance(self.agent, Agent)
self.assertEqual(self.agent.name, "PDF Agent")
self.assertEqual(self.agent.description, "Agent to handle PDF operations")
self.assertIsInstance(self.agent.tools, list)
def test_list_tools(self):
self.assertEqual(self.agent.list_tools(), [])
def test_create_tools(self):
metadataTool = PDFMetadataTool()
self.assertIsInstance(metadataTool, PDFMetadataTool)
self.assertEqual(metadataTool.name, "PDF Metadata Tool")
self.agent.add_tool(metadataTool)
tools = self.agent.get_tools()
self.assertEqual(tools, [metadataTool])
self.assertEqual(len(self.agent.tools), 1)
self.assertEqual(self.agent.tools[0].name, "PDF Metadata Tool")
# def test_