Quick tools reorganization

This commit is contained in:
João Moura
2024-02-15 14:02:42 -03:00
parent a232bfbe60
commit db5d371769
8 changed files with 75 additions and 44 deletions

View File

@@ -0,0 +1,21 @@
from crewai_tools.tools.rag.rag_tool import Adapter, RagTool
class MockAdapter(Adapter):
answer: str
def query(self, question: str) -> str:
return self.answer
def test_rag_tool():
adapter = MockAdapter(answer="42")
rag_tool = RagTool(adapter=adapter)
assert rag_tool.name == "Knowledge base"
assert (
rag_tool.description == "A knowledge base that can be used to answer questions."
)
assert (
rag_tool.run("What is the answer to life, the universe and everything?") == "42"
)