Spaces:
Sleeping
Sleeping
File size: 522 Bytes
d26c7f3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
# OUTDATED !
from agents_stage_2 import MathAgent, MainAgent
import asyncio
def test_math_agent():
math_agent = MathAgent(temperature=0.7, max_tokens=100)
print(math_agent.get_system_prompt())
asyncio.run(math_agent.query("What is 345 times 281?"))
def test_main_agent():
main_agent = MainAgent(temperature=0.7, max_tokens=100)
print(main_agent.get_system_prompt())
asyncio.run(main_agent.query("What is 345 times 281?"))
if __name__ == "__main__":
# test_math_agent()
test_main_agent()
|