You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
reflexion-human-eval/hotpotqa_runs/tests.py

14 lines
378 B
Python

import joblib
from react_cls import ReactReflectAgent
from mocks import DocStoreExplorerMock, LLMMock
test_q = "What is the elevation range for the area that the eastern sector of the Colorado orogeny extends into?"
test_a = "1,800 to 7,000 ft"
agent = ReactReflectAgent(test_q, test_a)
agent.run()
print(agent._build_agent_prompt())
print(agent._build_reflection_prompt())