Coverage for docs/docs_src/user_guide/adapters/fastapi/security/workflows.py: 82%

1import os 9 ctx1aefghibcd

2from typing import Any 9 ctx1aefghibcd

3 

4from autogen import ConversableAgent, LLMConfig 9 ctx1aefghibcd

5 

6from fastagency import UI 9 ctx1aefghibcd

7from fastagency.runtimes.ag2 import Workflow 9 ctx1aefghibcd

8 

9__init__ = ["wf"] 9 ctx1aefghibcd

10 

11llm_config = LLMConfig( 9 ctx1aefghibcd

12 model="gpt-4o-mini", 

13 api_key=os.getenv("OPENAI_API_KEY"), 

14 temperature=0.8, 

15) 

16 

17wf = Workflow() 9 ctx1aefghibcd

18 

19 

20@wf.register(name="simple_learning", description="Student and teacher learning chat") 9 ctx1aefghibcd

21def simple_workflow(ui: UI, params: dict[str, Any]) -> str: 9 ctx1aefghibcd

22 initial_message = ui.text_input( 4 ctx1abcd

23 sender="Workflow", 

24 recipient="User", 

25 prompt="I can help you learn about mathematics. What subject you would like to explore?", 

26 ) 

27 

28 with llm_config: 4 ctx1abcd

29 student_agent = ConversableAgent( 4 ctx1abcd

30 name="Student_Agent", 

31 system_message="You are a student willing to learn.", 

32 # human_input_mode="ALWAYS", 

33 ) 

34 teacher_agent = ConversableAgent( 1 ctx1a

35 name="Teacher_Agent", 

36 system_message="You are a math teacher.", 

37 # human_input_mode="ALWAYS", 

38 ) 

39 

40 response = student_agent.run( 

41 teacher_agent, 

42 message=initial_message, 

43 summary_method="reflection_with_llm", 

44 max_turns=5, 

45 ) 

46 

47 return ui.process(response) # type: ignore[no-any-return]