test_agent_creation.py 681 B

1234567891011121314151617
  1. # make sure we can create agents
  2. from pyopenagi.agents.agent_process import AgentProcess
  3. from pyopenagi.utils.chat_template import Query
  4. def test_agent_creation():
  5. agent_process = AgentProcess(
  6. agent_name="example/academic_agent",
  7. query=Query(
  8. messages = [
  9. {"role": "user", "content": "Summarize researches of quantum computing in recent five years."}
  10. ]
  11. )
  12. )
  13. # Use plain assert statements for testing conditions
  14. assert agent_process.agent_name == "example/academic_agent", "Agent name does not match"
  15. # Add more assertions here as necessary to validate the properties of the agent_process object