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:
@@ -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_
|
||||
|
||||
Reference in New Issue
Block a user