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()