Ismael Sampaio
2aad4b8bf0
- 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
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
from math import e
|
|
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 = PDFAgent()
|
|
return super().setUp()
|
|
|
|
|
|
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_
|