mirror of
https://github.com/microsoft/autogen.git
synced 2025-06-26 22:30:10 +00:00
f113c9a959
* Move core samples to /python/samples * Fix proto check * Add sample code check workflow * Update pyright settings; fix types
39 lines
1.5 KiB
Python
39 lines
1.5 KiB
Python
import chainlit as cl
|
|
from autogen_agentchat.agents import AssistantAgent
|
|
from autogen_agentchat.base import TaskResult
|
|
from autogen_agentchat.conditions import MaxMessageTermination, TextMentionTermination
|
|
from autogen_agentchat.teams import RoundRobinGroupChat
|
|
from autogen_ext.models.openai import OpenAIChatCompletionClient
|
|
|
|
|
|
async def get_weather(city: str) -> str:
|
|
return f"The weather in {city} is 73 degrees and Sunny."
|
|
|
|
|
|
@cl.on_chat_start # type: ignore
|
|
async def start_chat():
|
|
cl.user_session.set("prompt_history", "") # type: ignore
|
|
|
|
|
|
async def run_team(query: str):
|
|
assistant_agent = AssistantAgent(
|
|
name="assistant_agent", tools=[get_weather], model_client=OpenAIChatCompletionClient(model="gpt-4o-2024-08-06")
|
|
)
|
|
|
|
termination = TextMentionTermination("TERMINATE") | MaxMessageTermination(10)
|
|
team = RoundRobinGroupChat(participants=[assistant_agent], termination_condition=termination)
|
|
|
|
response_stream = team.run_stream(task=query)
|
|
async for msg in response_stream:
|
|
if hasattr(msg, "content"):
|
|
cl_msg = cl.Message(content=msg.content, author="Agent Team") # type: ignore
|
|
await cl_msg.send()
|
|
if isinstance(msg, TaskResult):
|
|
cl_msg = cl.Message(content="Termination condition met. Team and Agents are reset.", author="Agent Team")
|
|
await cl_msg.send()
|
|
|
|
|
|
@cl.on_message # type: ignore
|
|
async def chat(message: cl.Message):
|
|
await run_team(message.content) # type: ignore
|