Coverage for docs/docs_src/user_guide/runtimes/ag2/websurfer_tool.py: 100%

1import os 5 ctx1bacde

2from typing import Any 5 ctx1bacde

3 

4from autogen import UserProxyAgent, ConversableAgent, LLMConfig 5 ctx1bacde

5 

6from fastagency import UI, FastAgency 5 ctx1bacde

7from fastagency.runtimes.ag2 import Workflow 5 ctx1bacde

8from fastagency.runtimes.ag2.tools import WebSurferTool 5 ctx1bacde

9from fastagency.ui.console import ConsoleUI 5 ctx1bacde

10 

11llm_config = LLMConfig( 5 ctx1bacde

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

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

14 temperature=0.8, 

15) 

16 

17wf = Workflow() 5 ctx1bacde

18 

19 

20@wf.register(name="simple_websurfer", description="WebSurfer chat") # type: ignore[type-var] 5 ctx1bacde

21def websurfer_workflow( 5 ctx1bacde

22 ui: UI, params: dict[str, Any] 

23) -> str: 

24 initial_message = ui.text_input( 1 ctx1a

25 sender="Workflow", 

26 recipient="User", 

27 prompt="I can help you with your web search. What would you like to know?", 

28 ) 

29 

30 with llm_config: 1 ctx1a

31 user_agent = UserProxyAgent( 1 ctx1a

32 name="User_Agent", 

33 system_message="You are a user agent", 

34 human_input_mode="NEVER", 

35 ) 

36 assistant_agent = ConversableAgent( 1 ctx1a

37 name="Assistant_Agent", 

38 system_message="You are a useful assistant", 

39 human_input_mode="NEVER", 

40 ) 

41 

42 web_surfer = WebSurferTool( 1 ctx1a

43 name_prefix="Web_Surfer", 

44 llm_config=llm_config, 

45 summarizer_llm_config=llm_config, 

46 bing_api_key=os.getenv("BING_API_KEY"), 

47 ) 

48 

49 web_surfer.register( 1 ctx1a

50 caller=assistant_agent, 

51 executor=user_agent, 

52 ) 

53 

54 response = user_agent.run( 1 ctx1a

55 assistant_agent, 

56 message=initial_message, 

57 summary_method="reflection_with_llm", 

58 max_turns=3, 

59 ) 

60 

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

62 

63 

64app = FastAgency(provider=wf, ui=ConsoleUI()) 5 ctx1bacde