mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-10 16:48:30 +00:00
22 lines
542 B
Python
22 lines
542 B
Python
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"
|
|
)
|