diff --git a/python/packages/lab/gaia/agent_framework_lab_gaia/gaia.py b/python/packages/lab/gaia/agent_framework_lab_gaia/gaia.py index 1ee2aae167..d5894d086a 100644 --- a/python/packages/lab/gaia/agent_framework_lab_gaia/gaia.py +++ b/python/packages/lab/gaia/agent_framework_lab_gaia/gaia.py @@ -298,7 +298,6 @@ class GAIA: repo_type="dataset", token=token, local_dir=str(self.data_dir), - local_dir_use_symlinks=False, force_download=False, ) return Path(local_dir) diff --git a/python/pyproject.toml b/python/pyproject.toml index dba87561cd..f24f726003 100644 --- a/python/pyproject.toml +++ b/python/pyproject.toml @@ -53,6 +53,9 @@ dev = [ "rich", "tomli", "tomli-w", + # AutoGen migration samples + "autogen-agentchat", + "autogen-ext[openai]", ] docs = [ # Documentation @@ -175,15 +178,11 @@ omit = [ [tool.pyright] include = ["agent_framework", "samples"] -exclude = ["**/tests/**", "docs", "**/.venv/**", "samples/semantic-kernel-migration/**", "packages/devui/frontend/**"] +exclude = ["**/tests/**", "docs", "**/.venv/**", "samples/semantic-kernel-migration/**", "samples/autogen-migration/**", "packages/devui/frontend/**"] typeCheckingMode = "strict" reportUnnecessaryIsInstance = false reportMissingTypeStubs = false -[[tool.pyright.executionEnvironments]] -root = "samples/semantic-kernel-migration" -typeCheckingMode = "off" - [tool.mypy] plugins = ['pydantic.mypy'] strict = true diff --git a/python/samples/autogen-migration/.gitignore b/python/samples/autogen-migration/.gitignore new file mode 100644 index 0000000000..8df4fd2601 --- /dev/null +++ b/python/samples/autogen-migration/.gitignore @@ -0,0 +1,2 @@ +# Ignore autogen source files +autogen diff --git a/python/samples/autogen-migration/README.md b/python/samples/autogen-migration/README.md new file mode 100644 index 0000000000..c2ddacee70 --- /dev/null +++ b/python/samples/autogen-migration/README.md @@ -0,0 +1,61 @@ +# AutoGen → Microsoft Agent Framework Migration Samples + +This gallery helps AutoGen developers move to the Microsoft Agent Framework (AF) with minimal guesswork. Each script pairs AutoGen code with its AF equivalent so you can compare primitives, tooling, and orchestration patterns side by side while you migrate production workloads. + +## What's Included + +### Single-Agent Parity + +- [01_basic_assistant_agent.py](single_agent/01_basic_assistant_agent.py) — Minimal AutoGen `AssistantAgent` and AF `ChatAgent` comparison. +- [02_assistant_agent_with_tool.py](single_agent/02_assistant_agent_with_tool.py) — Function tool integration in both SDKs. +- [03_assistant_agent_thread_and_stream.py](single_agent/03_assistant_agent_thread_and_stream.py) — Thread management and streaming responses. +- [04_agent_as_tool.py](single_agent/04_agent_as_tool.py) — Using agents as tools (hierarchical agent pattern) and streaming with tools. + +### Multi-Agent Orchestration + +- [01_round_robin_group_chat.py](orchestrations/01_round_robin_group_chat.py) — AutoGen `RoundRobinGroupChat` → AF `GroupChatBuilder`/`SequentialBuilder`. +- [02_selector_group_chat.py](orchestrations/02_selector_group_chat.py) — AutoGen `SelectorGroupChat` → AF `GroupChatBuilder`. +- [03_swarm.py](orchestrations/03_swarm.py) — AutoGen Swarm pattern → AF `HandoffBuilder`. +- [04_magentic_one.py](orchestrations/04_magentic_one.py) — AutoGen `MagenticOneGroupChat` → AF `MagenticBuilder`. + +Each script is fully async and the `main()` routine runs both implementations back to back so you can observe their outputs in a single execution. + +## Prerequisites + +- Python 3.10 or later. +- Access to the necessary model endpoints (Azure OpenAI, OpenAI, etc.). +- Installed SDKs: Install AutoGen and the Microsoft Agent Framework with: + ```bash + pip install "autogen-agentchat autogen-ext[openai] agent-framework" + ``` +- Service credentials exposed through environment variables (e.g., `OPENAI_API_KEY`). + +## Running Single-Agent Samples + +From the repository root: + +```bash +python samples/autogen-migration/single_agent/01_basic_assistant_agent.py +``` + +Every script accepts no CLI arguments and will first call the AutoGen implementation, followed by the AF version. Adjust the prompt or credentials inside the file as necessary before running. + +## Running Orchestration Samples + +Advanced comparisons are in `autogen-migration/orchestrations` (RoundRobin, Selector, Swarm, Magentic). You can run them directly: + +```bash +python samples/autogen-migration/orchestrations/01_round_robin_group_chat.py +python samples/autogen-migration/orchestrations/04_magentic_one.py +``` + +## Tips for Migration + +- **Default behavior differences**: AutoGen's `AssistantAgent` is single-turn by default (`max_tool_iterations=1`), while AF's `ChatAgent` is multi-turn and continues tool execution automatically. +- **Thread management**: AF agents are stateless by default. Use `agent.get_new_thread()` and pass it to `run()`/`run_stream()` to maintain conversation state, similar to AutoGen's conversation context. +- **Tools**: AutoGen uses `FunctionTool` wrappers; AF uses `@ai_function` decorators with automatic schema inference. +- **Orchestration patterns**: + - `RoundRobinGroupChat` → `SequentialBuilder` or `WorkflowBuilder` + - `SelectorGroupChat` → `GroupChatBuilder` with LLM-based speaker selection + - `Swarm` → `HandoffBuilder` for agent handoff coordination + - `MagenticOneGroupChat` → `MagenticBuilder` for orchestrated multi-agent workflows diff --git a/python/samples/autogen-migration/orchestrations/01_round_robin_group_chat.py b/python/samples/autogen-migration/orchestrations/01_round_robin_group_chat.py new file mode 100644 index 0000000000..bd48e34861 --- /dev/null +++ b/python/samples/autogen-migration/orchestrations/01_round_robin_group_chat.py @@ -0,0 +1,185 @@ +# Copyright (c) Microsoft. All rights reserved. +"""AutoGen RoundRobinGroupChat vs Agent Framework GroupChatBuilder/SequentialBuilder. + +Demonstrates sequential agent orchestration where agents take turns processing +the task in a round-robin fashion. +""" + +import asyncio + + +async def run_autogen() -> None: + """AutoGen's RoundRobinGroupChat for sequential agent orchestration.""" + from autogen_agentchat.agents import AssistantAgent + from autogen_agentchat.conditions import TextMentionTermination + from autogen_agentchat.teams import RoundRobinGroupChat + from autogen_agentchat.ui import Console + from autogen_ext.models.openai import OpenAIChatCompletionClient + + client = OpenAIChatCompletionClient(model="gpt-4.1-mini") + + # Create specialized agents + researcher = AssistantAgent( + name="researcher", + model_client=client, + system_message="You are a researcher. Provide facts and data about the topic.", + model_client_stream=True, + ) + + writer = AssistantAgent( + name="writer", + model_client=client, + system_message="You are a writer. Turn research into engaging content.", + model_client_stream=True, + ) + + editor = AssistantAgent( + name="editor", + model_client=client, + system_message="You are an editor. Review and finalize the content. End with APPROVED if satisfied.", + model_client_stream=True, + ) + + # Create round-robin team + team = RoundRobinGroupChat( + participants=[researcher, writer, editor], + termination_condition=TextMentionTermination("APPROVED"), + ) + + # Run the team and display the conversation. + print("[AutoGen] Round-robin conversation:") + await Console(team.run_stream(task="Create a brief summary about electric vehicles")) + + +async def run_agent_framework() -> None: + """Agent Framework's SequentialBuilder for sequential agent orchestration.""" + from agent_framework import AgentRunUpdateEvent, SequentialBuilder + from agent_framework.openai import OpenAIChatClient + + client = OpenAIChatClient(model_id="gpt-4.1-mini") + + # Create specialized agents + researcher = client.create_agent( + name="researcher", + instructions="You are a researcher. Provide facts and data about the topic.", + ) + + writer = client.create_agent( + name="writer", + instructions="You are a writer. Turn research into engaging content.", + ) + + editor = client.create_agent( + name="editor", + instructions="You are an editor. Review and finalize the content.", + ) + + # Create sequential workflow + workflow = SequentialBuilder().participants([researcher, writer, editor]).build() + + # Run the workflow + print("[Agent Framework] Sequential conversation:") + current_executor = None + async for event in workflow.run_stream("Create a brief summary about electric vehicles"): + if isinstance(event, AgentRunUpdateEvent): + # Print executor name header when switching to a new agent + if current_executor != event.executor_id: + if current_executor is not None: + print() # Newline after previous agent's message + print(f"---------- {event.executor_id} ----------") + current_executor = event.executor_id + if event.data: + print(event.data.text, end="", flush=True) + print() # Final newline after conversation + + +async def run_agent_framework_with_cycle() -> None: + """Agent Framework's WorkflowBuilder with cyclic edges and conditional exit.""" + from agent_framework import ( + AgentExecutorRequest, + AgentExecutorResponse, + AgentRunUpdateEvent, + WorkflowBuilder, + WorkflowContext, + WorkflowOutputEvent, + executor, + ) + from agent_framework.openai import OpenAIChatClient + + client = OpenAIChatClient(model_id="gpt-4.1-mini") + + # Create specialized agents + researcher = client.create_agent( + name="researcher", + instructions="You are a researcher. Provide facts and data about the topic.", + ) + + writer = client.create_agent( + name="writer", + instructions="You are a writer. Turn research into engaging content.", + ) + + editor = client.create_agent( + name="editor", + instructions="You are an editor. Review and finalize the content. End with APPROVED if satisfied.", + ) + + # Create custom executor for checking approval + @executor + async def check_approval( + response: AgentExecutorResponse, context: WorkflowContext[AgentExecutorRequest, str] + ) -> None: + assert response.full_conversation is not None + last_message = response.full_conversation[-1] + if last_message and "APPROVED" in last_message.text: + await context.yield_output("Content approved.") + else: + await context.send_message(AgentExecutorRequest(messages=response.full_conversation, should_respond=True)) + + workflow = ( + WorkflowBuilder() + .add_edge(researcher, writer) + .add_edge(writer, editor) + .add_edge( + editor, + check_approval, + ) + .add_edge(check_approval, researcher) + .set_start_executor(researcher) + .build() + ) + + # Run the workflow + print("[Agent Framework with Cycle] Cyclic conversation:") + current_executor = None + async for event in workflow.run_stream("Create a brief summary about electric vehicles"): + if isinstance(event, WorkflowOutputEvent): + print("\n---------- Workflow Output ----------") + print(event.data) + elif isinstance(event, AgentRunUpdateEvent): + # Print executor name header when switching to a new agent + if current_executor != event.executor_id: + if current_executor is not None: + print() # Newline after previous agent's message + print(f"---------- {event.executor_id} ----------") + current_executor = event.executor_id + if event.data: + print(event.data.text, end="", flush=True) + print() # Final newline after conversation + + +async def main() -> None: + print("=" * 60) + print("Round-Robin / Sequential Orchestration Comparison") + print("=" * 60) + print("AutoGen: RoundRobinGroupChat") + print("Agent Framework: SequentialBuilder + WorkflowBuilder with cycles\n") + await run_autogen() + print() + await run_agent_framework() + print() + await run_agent_framework_with_cycle() + + +if __name__ == "__main__": + asyncio.run(main()) diff --git a/python/samples/autogen-migration/orchestrations/02_selector_group_chat.py b/python/samples/autogen-migration/orchestrations/02_selector_group_chat.py new file mode 100644 index 0000000000..097680a3b7 --- /dev/null +++ b/python/samples/autogen-migration/orchestrations/02_selector_group_chat.py @@ -0,0 +1,125 @@ +# Copyright (c) Microsoft. All rights reserved. +"""AutoGen SelectorGroupChat vs Agent Framework GroupChatBuilder. + +Demonstrates LLM-based speaker selection where an orchestrator decides +which agent should speak next based on the conversation context. +""" + +import asyncio + + +async def run_autogen() -> None: + """AutoGen's SelectorGroupChat with LLM-based speaker selection.""" + from autogen_agentchat.agents import AssistantAgent + from autogen_agentchat.conditions import MaxMessageTermination + from autogen_agentchat.teams import SelectorGroupChat + from autogen_agentchat.ui import Console + from autogen_ext.models.openai import OpenAIChatCompletionClient + + client = OpenAIChatCompletionClient(model="gpt-4.1-mini") + + # Create specialized agents + python_expert = AssistantAgent( + name="python_expert", + model_client=client, + system_message="You are a Python programming expert. Answer Python-related questions.", + description="Expert in Python programming", + model_client_stream=True, + ) + + javascript_expert = AssistantAgent( + name="javascript_expert", + model_client=client, + system_message="You are a JavaScript programming expert. Answer JavaScript-related questions.", + description="Expert in JavaScript programming", + model_client_stream=True, + ) + + database_expert = AssistantAgent( + name="database_expert", + model_client=client, + system_message="You are a database expert. Answer SQL and database-related questions.", + description="Expert in databases and SQL", + model_client_stream=True, + ) + + # Create selector group chat - LLM selects appropriate expert + team = SelectorGroupChat( + participants=[python_expert, javascript_expert, database_expert], + model_client=client, + termination_condition=MaxMessageTermination(2), + selector_prompt="Based on the conversation so far:\n{history}\n, " + "select the most appropriate expert from {roles} to respond next.", + ) + + # Run with a question that requires expert selection + print("[AutoGen] Selector group chat conversation:") + await Console(team.run_stream(task="How do I connect to a PostgreSQL database using Python?")) + + +async def run_agent_framework() -> None: + """Agent Framework's GroupChatBuilder with LLM-based speaker selection.""" + from agent_framework import AgentRunUpdateEvent, GroupChatBuilder + from agent_framework.openai import OpenAIChatClient + + client = OpenAIChatClient(model_id="gpt-4.1-mini") + + # Create specialized agents + python_expert = client.create_agent( + name="python_expert", + instructions="You are a Python programming expert. Answer Python-related questions.", + description="Expert in Python programming", + ) + + javascript_expert = client.create_agent( + name="javascript_expert", + instructions="You are a JavaScript programming expert. Answer JavaScript-related questions.", + description="Expert in JavaScript programming", + ) + + database_expert = client.create_agent( + name="database_expert", + instructions="You are a database expert. Answer SQL and database-related questions.", + description="Expert in databases and SQL", + ) + + workflow = ( + GroupChatBuilder() + .participants([python_expert, javascript_expert, database_expert]) + .set_prompt_based_manager( + chat_client=client, + instructions="Based on the conversation, select the most appropriate expert to respond next.", + ) + .with_max_rounds(1) + .build() + ) + + # Run with a question that requires expert selection + print("[Agent Framework] Group chat conversation:") + current_executor = None + async for event in workflow.run_stream("How do I connect to a PostgreSQL database using Python?"): + if isinstance(event, AgentRunUpdateEvent): + # Print executor name header when switching to a new agent + if current_executor != event.executor_id: + if current_executor is not None: + print() # Newline after previous agent's message + print(f"---------- {event.executor_id} ----------") + current_executor = event.executor_id + if event.data: + print(event.data.text, end="", flush=True) + print() # Final newline after conversation + + +async def main() -> None: + print("=" * 60) + print("Selector Group Chat Comparison") + print("=" * 60) + print("AutoGen: SelectorGroupChat") + print("Agent Framework: GroupChatBuilder with standard_manager\n") + await run_autogen() + print() + await run_agent_framework() + + +if __name__ == "__main__": + asyncio.run(main()) diff --git a/python/samples/autogen-migration/orchestrations/03_swarm.py b/python/samples/autogen-migration/orchestrations/03_swarm.py new file mode 100644 index 0000000000..a70169c531 --- /dev/null +++ b/python/samples/autogen-migration/orchestrations/03_swarm.py @@ -0,0 +1,238 @@ +# Copyright (c) Microsoft. All rights reserved. +"""AutoGen Swarm pattern vs Agent Framework HandoffBuilder. + +Demonstrates agent handoff coordination where agents can transfer control +to other specialized agents based on the task requirements. +""" + +import asyncio + + +async def run_autogen() -> None: + """AutoGen's Swarm pattern with human-in-the-loop handoffs.""" + from autogen_agentchat.agents import AssistantAgent + from autogen_agentchat.conditions import HandoffTermination, TextMentionTermination + from autogen_agentchat.messages import HandoffMessage + from autogen_agentchat.teams import Swarm + from autogen_agentchat.ui import Console + from autogen_ext.models.openai import OpenAIChatCompletionClient + + client = OpenAIChatCompletionClient(model="gpt-4.1-mini") + + # Create triage agent that routes to specialists + triage_agent = AssistantAgent( + name="triage", + model_client=client, + system_message=( + "You are a triage agent. Analyze the user's request and hand off to the appropriate specialist.\n" + "If you need information from the user, first send your message, then handoff to user.\n" + "Use TERMINATE when the issue is fully resolved." + ), + handoffs=["billing_agent", "technical_support", "user"], + model_client_stream=True, + ) + + # Create billing specialist + billing_agent = AssistantAgent( + name="billing_agent", + model_client=client, + system_message=( + "You are a billing specialist. Help with payment and billing questions.\n" + "If you need information from the user, first send your message, then handoff to user.\n" + "When the issue is resolved, handoff to triage to finalize." + ), + handoffs=["triage", "user"], + model_client_stream=True, + ) + + # Create technical support specialist + tech_support = AssistantAgent( + name="technical_support", + model_client=client, + system_message=( + "You are technical support. Help with technical issues.\n" + "If you need information from the user, first send your message, then handoff to user.\n" + "When the issue is resolved, handoff to triage to finalize." + ), + handoffs=["triage", "user"], + model_client_stream=True, + ) + + # Create swarm team with human-in-the-loop termination + termination = HandoffTermination(target="user") | TextMentionTermination("TERMINATE") + team = Swarm( + participants=[triage_agent, billing_agent, tech_support], + termination_condition=termination, + ) + + # Scripted user responses for demonstration + scripted_responses = [ + "I was charged twice for my subscription", + "Yes, the charge of $49.99 appears twice on my credit card statement.", + "Thank you for your help!", + ] + response_index = 0 + + # Run with human-in-the-loop pattern + print("[AutoGen] Swarm handoff conversation:") + task_result = await Console(team.run_stream(task=scripted_responses[response_index])) + last_message = task_result.messages[-1] + response_index += 1 + + # Continue conversation when agents handoff to user + while ( + isinstance(last_message, HandoffMessage) + and last_message.target == "user" + and response_index < len(scripted_responses) + ): + user_message = scripted_responses[response_index] + task_result = await Console( + team.run_stream(task=HandoffMessage(source="user", target=last_message.source, content=user_message)) + ) + last_message = task_result.messages[-1] + response_index += 1 + + +async def run_agent_framework() -> None: + """Agent Framework's HandoffBuilder for agent coordination.""" + from agent_framework import ( + AgentRunUpdateEvent, + HandoffBuilder, + HandoffUserInputRequest, + RequestInfoEvent, + WorkflowRunState, + WorkflowStatusEvent, + ) + from agent_framework.openai import OpenAIChatClient + + client = OpenAIChatClient(model_id="gpt-4.1-mini") + + # Create triage agent + triage_agent = client.create_agent( + name="triage", + instructions=( + "You are a triage agent. Analyze the user's request and route to the appropriate specialist:\n" + "- For billing issues: call handoff_to_billing_agent\n" + "- For technical issues: call handoff_to_technical_support" + ), + description="Routes requests to appropriate specialists", + ) + + # Create billing specialist + billing_agent = client.create_agent( + name="billing_agent", + instructions="You are a billing specialist. Help with payment and billing questions. Provide clear assistance.", + description="Handles billing and payment questions", + ) + + # Create technical support specialist + tech_support = client.create_agent( + name="technical_support", + instructions="You are technical support. Help with technical issues. Provide clear assistance.", + description="Handles technical support questions", + ) + + # Create handoff workflow - simpler configuration + # After specialists respond, control returns to user (via triage as coordinator) + workflow = ( + HandoffBuilder( + name="support_handoff", + participants=[triage_agent, billing_agent, tech_support], + ) + .set_coordinator(triage_agent) + .add_handoff(triage_agent, [billing_agent, tech_support]) + .with_termination_condition(lambda conv: sum(1 for msg in conv if msg.role.value == "user") > 3) + .build() + ) + + # Scripted user responses + scripted_responses = [ + "I was charged twice for my subscription", + "Yes, the charge of $49.99 appears twice on my credit card statement.", + "Thank you for your help!", + ] + + # Run with initial message + print("[Agent Framework] Handoff conversation:") + print("---------- user ----------") + print(scripted_responses[0]) + + current_executor = None + stream_line_open = False + pending_requests: list[RequestInfoEvent] = [] + + async for event in workflow.run_stream(scripted_responses[0]): + if isinstance(event, AgentRunUpdateEvent): + # Print executor name header when switching to a new agent + if current_executor != event.executor_id: + if stream_line_open: + print() + stream_line_open = False + print(f"---------- {event.executor_id} ----------") + current_executor = event.executor_id + stream_line_open = True + if event.data: + print(event.data.text, end="", flush=True) + elif isinstance(event, RequestInfoEvent): + if isinstance(event.data, HandoffUserInputRequest): + pending_requests.append(event) + elif isinstance(event, WorkflowStatusEvent): + if event.state in {WorkflowRunState.IDLE_WITH_PENDING_REQUESTS} and stream_line_open: + print() + stream_line_open = False + + # Process scripted responses + response_index = 1 + while pending_requests and response_index < len(scripted_responses): + user_response = scripted_responses[response_index] + print("---------- user ----------") + print(user_response) + + responses = {req.request_id: user_response for req in pending_requests} + pending_requests = [] + current_executor = None + stream_line_open = False + + async for event in workflow.send_responses_streaming(responses): + if isinstance(event, AgentRunUpdateEvent): + # Print executor name header when switching to a new agent + if current_executor != event.executor_id: + if stream_line_open: + print() + stream_line_open = False + print(f"---------- {event.executor_id} ----------") + current_executor = event.executor_id + stream_line_open = True + if event.data: + print(event.data.text, end="", flush=True) + elif isinstance(event, RequestInfoEvent): + if isinstance(event.data, HandoffUserInputRequest): + pending_requests.append(event) + elif isinstance(event, WorkflowStatusEvent): + if ( + event.state in {WorkflowRunState.IDLE_WITH_PENDING_REQUESTS, WorkflowRunState.IDLE} + and stream_line_open + ): + print() + stream_line_open = False + + response_index += 1 + + if stream_line_open: + print() + print() # Final newline after conversation + + +async def main() -> None: + print("=" * 60) + print("Swarm / Handoff Pattern Comparison") + print("=" * 60) + print("AutoGen: Swarm with handoffs") + print("Agent Framework: HandoffBuilder\n") + await run_autogen() + print() + await run_agent_framework() + + +if __name__ == "__main__": + asyncio.run(main()) diff --git a/python/samples/autogen-migration/orchestrations/04_magentic_one.py b/python/samples/autogen-migration/orchestrations/04_magentic_one.py new file mode 100644 index 0000000000..ca81f0faf9 --- /dev/null +++ b/python/samples/autogen-migration/orchestrations/04_magentic_one.py @@ -0,0 +1,153 @@ +# Copyright (c) Microsoft. All rights reserved. +"""AutoGen MagenticOneGroupChat vs Agent Framework MagenticBuilder. + +Demonstrates orchestrated multi-agent workflows with a central coordinator +managing specialized agents for complex tasks. +""" + +import asyncio + + +async def run_autogen() -> None: + """AutoGen's MagenticOneGroupChat for orchestrated collaboration.""" + from autogen_agentchat.agents import AssistantAgent + from autogen_agentchat.teams import MagenticOneGroupChat + from autogen_agentchat.ui import Console + from autogen_ext.models.openai import OpenAIChatCompletionClient + + client = OpenAIChatCompletionClient(model="gpt-4.1-mini") + + # Create specialized agents + researcher = AssistantAgent( + name="researcher", + model_client=client, + system_message="You are a research analyst. Gather and analyze information.", + description="Research analyst for data gathering", + model_client_stream=True, + ) + + coder = AssistantAgent( + name="coder", + model_client=client, + system_message="You are a programmer. Write code based on requirements.", + description="Software developer for implementation", + model_client_stream=True, + ) + + reviewer = AssistantAgent( + name="reviewer", + model_client=client, + system_message="You are a code reviewer. Review code for quality and correctness.", + description="Code reviewer for quality assurance", + model_client_stream=True, + ) + + # Create MagenticOne team with coordinator + team = MagenticOneGroupChat( + participants=[researcher, coder, reviewer], + model_client=client, # Coordinator uses this client + max_turns=20, + max_stalls=3, + ) + + # Run complex task and display the conversation + print("[AutoGen] Magentic One conversation:") + await Console(team.run_stream(task="Research Python async patterns and write a simple example")) + + +async def run_agent_framework() -> None: + """Agent Framework's MagenticBuilder for orchestrated collaboration.""" + from agent_framework import ( + MagenticAgentDeltaEvent, + MagenticAgentMessageEvent, + MagenticBuilder, + MagenticFinalResultEvent, + MagenticOrchestratorMessageEvent, + ) + from agent_framework.openai import OpenAIChatClient + + client = OpenAIChatClient(model_id="gpt-4.1-mini") + + # Create specialized agents + researcher = client.create_agent( + name="researcher", + instructions="You are a research analyst. Gather and analyze information.", + description="Research analyst for data gathering", + ) + + coder = client.create_agent( + name="coder", + instructions="You are a programmer. Write code based on requirements.", + description="Software developer for implementation", + ) + + reviewer = client.create_agent( + name="reviewer", + instructions="You are a code reviewer. Review code for quality and correctness.", + description="Code reviewer for quality assurance", + ) + + # Create Magentic workflow + workflow = ( + MagenticBuilder() + .participants(researcher=researcher, coder=coder, reviewer=reviewer) + .with_standard_manager( + chat_client=client, + max_round_count=20, + max_stall_count=3, + max_reset_count=1, + ) + .build() + ) + + # Run complex task + print("[Agent Framework] Magentic conversation:") + last_stream_agent_id: str | None = None + stream_line_open: bool = False + + async for event in workflow.run_stream("Research Python async patterns and write a simple example"): + if isinstance(event, MagenticOrchestratorMessageEvent): + if stream_line_open: + print() + stream_line_open = False + print(f"---------- Orchestrator:{event.kind} ----------") + print(getattr(event.message, "text", "")) + elif isinstance(event, MagenticAgentDeltaEvent): + if last_stream_agent_id != event.agent_id or not stream_line_open: + if stream_line_open: + print() + print(f"---------- {event.agent_id} ----------") + last_stream_agent_id = event.agent_id + stream_line_open = True + if event.text: + print(event.text, end="", flush=True) + elif isinstance(event, MagenticAgentMessageEvent): + if stream_line_open: + print() + stream_line_open = False + elif isinstance(event, MagenticFinalResultEvent): + if stream_line_open: + print() + stream_line_open = False + print("---------- Final Result ----------") + if event.message is not None: + print(event.message.text) + + if stream_line_open: + print() + print() # Final newline after conversation + + +async def main() -> None: + print("=" * 60) + print("Magentic One Orchestration Comparison") + print("=" * 60) + print("AutoGen: MagenticOneGroupChat") + print("Agent Framework: MagenticBuilder\n") + await run_autogen() + print() + await run_agent_framework() + + +if __name__ == "__main__": + asyncio.run(main()) diff --git a/python/samples/autogen-migration/pyrightconfig.json b/python/samples/autogen-migration/pyrightconfig.json new file mode 100644 index 0000000000..0f332d6342 --- /dev/null +++ b/python/samples/autogen-migration/pyrightconfig.json @@ -0,0 +1,5 @@ +{ + "exclude": [ + "autogen" + ] +} \ No newline at end of file diff --git a/python/samples/autogen-migration/single_agent/01_basic_assistant_agent.py b/python/samples/autogen-migration/single_agent/01_basic_assistant_agent.py new file mode 100644 index 0000000000..febc6d9ffb --- /dev/null +++ b/python/samples/autogen-migration/single_agent/01_basic_assistant_agent.py @@ -0,0 +1,56 @@ +# Copyright (c) Microsoft. All rights reserved. +"""Basic AutoGen AssistantAgent vs Agent Framework ChatAgent. + +Both samples expect OpenAI-compatible environment variables (OPENAI_API_KEY or +Azure OpenAI configuration). Update the prompts or client wiring to match your +model of choice before running. +""" + +import asyncio + + +async def run_autogen() -> None: + """Call AutoGen's AssistantAgent for a simple question.""" + from autogen_agentchat.agents import AssistantAgent + from autogen_ext.models.openai import OpenAIChatCompletionClient + + # AutoGen agent with OpenAI model client + client = OpenAIChatCompletionClient(model="gpt-4.1-mini") + agent = AssistantAgent( + name="assistant", + model_client=client, + system_message="You are a helpful assistant. Answer in one sentence.", + ) + + # Run the agent (AutoGen maintains conversation state internally) + result = await agent.run(task="What is the capital of France?") + print("[AutoGen]", result.messages[-1].to_text()) + + +async def run_agent_framework() -> None: + """Call Agent Framework's ChatAgent created from OpenAIChatClient.""" + from agent_framework.openai import OpenAIChatClient + + # AF constructs a lightweight ChatAgent backed by OpenAIChatClient + client = OpenAIChatClient(model_id="gpt-4.1-mini") + agent = client.create_agent( + name="assistant", + instructions="You are a helpful assistant. Answer in one sentence.", + ) + + # Run the agent (AF agents are stateless by default) + result = await agent.run("What is the capital of France?") + print("[Agent Framework]", result.text) + + +async def main() -> None: + print("=" * 60) + print("Basic Assistant Agent Comparison") + print("=" * 60) + await run_autogen() + print() + await run_agent_framework() + + +if __name__ == "__main__": + asyncio.run(main()) diff --git a/python/samples/autogen-migration/single_agent/02_assistant_agent_with_tool.py b/python/samples/autogen-migration/single_agent/02_assistant_agent_with_tool.py new file mode 100644 index 0000000000..6f9cd2303a --- /dev/null +++ b/python/samples/autogen-migration/single_agent/02_assistant_agent_with_tool.py @@ -0,0 +1,89 @@ +# Copyright (c) Microsoft. All rights reserved. +"""AutoGen AssistantAgent vs Agent Framework ChatAgent with function tools. + +Demonstrates how to create and attach tools to agents in both frameworks. +""" + +import asyncio + + +async def run_autogen() -> None: + """AutoGen agent with a FunctionTool.""" + from autogen_agentchat.agents import AssistantAgent + from autogen_core.tools import FunctionTool + from autogen_ext.models.openai import OpenAIChatCompletionClient + + # Define a simple tool function + def get_weather(location: str) -> str: + """Get the weather for a location. + + Args: + location: The city name or location. + + Returns: + A weather description. + """ + return f"The weather in {location} is sunny and 72°F." + + # Wrap function in FunctionTool + weather_tool = FunctionTool( + func=get_weather, + description="Get weather information for a location", + ) + + # Create agent with tool + client = OpenAIChatCompletionClient(model="gpt-4.1-mini") + agent = AssistantAgent( + name="assistant", + model_client=client, + tools=[weather_tool], + system_message="You are a helpful assistant. Use available tools to answer questions.", + ) + + # Run with tool usage + result = await agent.run(task="What's the weather in Seattle?") + print("[AutoGen]", result.messages[-1].to_text()) + + +async def run_agent_framework() -> None: + """Agent Framework agent with @ai_function decorator.""" + from agent_framework import ai_function + from agent_framework.openai import OpenAIChatClient + + # Define tool with @ai_function decorator (automatic schema inference) + @ai_function + def get_weather(location: str) -> str: + """Get the weather for a location. + + Args: + location: The city name or location. + + Returns: + A weather description. + """ + return f"The weather in {location} is sunny and 72°F." + + # Create agent with tool + client = OpenAIChatClient(model_id="gpt-4.1-mini") + agent = client.create_agent( + name="assistant", + instructions="You are a helpful assistant. Use available tools to answer questions.", + tools=[get_weather], + ) + + # Run with tool usage + result = await agent.run("What's the weather in Seattle?") + print("[Agent Framework]", result.text) + + +async def main() -> None: + print("=" * 60) + print("Assistant Agent with Tools Comparison") + print("=" * 60) + await run_autogen() + print() + await run_agent_framework() + + +if __name__ == "__main__": + asyncio.run(main()) diff --git a/python/samples/autogen-migration/single_agent/03_assistant_agent_thread_and_stream.py b/python/samples/autogen-migration/single_agent/03_assistant_agent_thread_and_stream.py new file mode 100644 index 0000000000..9949b3c0ef --- /dev/null +++ b/python/samples/autogen-migration/single_agent/03_assistant_agent_thread_and_stream.py @@ -0,0 +1,79 @@ +# Copyright (c) Microsoft. All rights reserved. +"""AutoGen vs Agent Framework: Thread management and streaming responses. + +Demonstrates conversation state management and streaming in both frameworks. +""" + +import asyncio + + +async def run_autogen() -> None: + """AutoGen agent with conversation history and streaming.""" + from autogen_agentchat.agents import AssistantAgent + from autogen_agentchat.ui import Console + from autogen_ext.models.openai import OpenAIChatCompletionClient + + client = OpenAIChatCompletionClient(model="gpt-4.1-mini") + agent = AssistantAgent( + name="assistant", + model_client=client, + system_message="You are a helpful math tutor.", + model_client_stream=True, + ) + + print("[AutoGen] Conversation with history:") + # First turn - AutoGen maintains state internally with Console for streaming + result = await agent.run(task="What is 15 + 27?") + print(f" Q1: {result.messages[-1].to_text()}") + + # Second turn - agent remembers context + result = await agent.run(task="What about that number times 2?") + print(f" Q2: {result.messages[-1].to_text()}") + + print("\n[AutoGen] Streaming response:") + # Stream response with Console for token streaming + await Console(agent.run_stream(task="Count from 1 to 5")) + + +async def run_agent_framework() -> None: + """Agent Framework agent with explicit thread and streaming.""" + from agent_framework.openai import OpenAIChatClient + + client = OpenAIChatClient(model_id="gpt-4.1-mini") + agent = client.create_agent( + name="assistant", + instructions="You are a helpful math tutor.", + ) + + print("[Agent Framework] Conversation with thread:") + # Create a thread to maintain state + thread = agent.get_new_thread() + + # First turn - pass thread to maintain history + result1 = await agent.run("What is 15 + 27?", thread=thread) + print(f" Q1: {result1.text}") + + # Second turn - agent remembers context via thread + result2 = await agent.run("What about that number times 2?", thread=thread) + print(f" Q2: {result2.text}") + + print("\n[Agent Framework] Streaming response:") + # Stream response + print(" ", end="") + async for chunk in agent.run_stream("Count from 1 to 5"): + if chunk.text: + print(chunk.text, end="", flush=True) + print() + + +async def main() -> None: + print("=" * 60) + print("Thread Management and Streaming Comparison") + print("=" * 60) + await run_autogen() + print() + await run_agent_framework() + + +if __name__ == "__main__": + asyncio.run(main()) diff --git a/python/samples/autogen-migration/single_agent/04_agent_as_tool.py b/python/samples/autogen-migration/single_agent/04_agent_as_tool.py new file mode 100644 index 0000000000..2f19fc7c8a --- /dev/null +++ b/python/samples/autogen-migration/single_agent/04_agent_as_tool.py @@ -0,0 +1,130 @@ +# Copyright (c) Microsoft. All rights reserved. +"""AutoGen vs Agent Framework: Agent-as-a-Tool pattern. + +Demonstrates hierarchical agent architectures where one agent delegates +work to specialized sub-agents wrapped as tools. +""" + +import asyncio + + +async def run_autogen() -> None: + """AutoGen's AgentTool for hierarchical agents with streaming.""" + from autogen_agentchat.agents import AssistantAgent + from autogen_agentchat.tools import AgentTool + from autogen_agentchat.ui import Console + from autogen_ext.models.openai import OpenAIChatCompletionClient + + # Create a specialized writer agent + writer_client = OpenAIChatCompletionClient(model="gpt-4.1-mini") + writer = AssistantAgent( + name="writer", + model_client=writer_client, + system_message="You are a creative writer. Write short, engaging content.", + model_client_stream=True, + ) + + # Wrap writer agent as a tool (description is taken from agent.description) + writer_tool = AgentTool(agent=writer) + + # Create coordinator agent with writer as a tool + # IMPORTANT: Disable parallel_tool_calls when using AgentTool + coordinator_client = OpenAIChatCompletionClient( + model="gpt-4.1-mini", + parallel_tool_calls=False, + ) + coordinator = AssistantAgent( + name="coordinator", + model_client=coordinator_client, + tools=[writer_tool], + system_message="You coordinate with specialized agents. Delegate writing tasks to the writer agent.", + model_client_stream=True, + ) + + # Run coordinator with streaming - it will delegate to writer + print("[AutoGen]") + await Console(coordinator.run_stream(task="Create a tagline for a coffee shop")) + + +async def run_agent_framework() -> None: + """Agent Framework's as_tool() for hierarchical agents with streaming.""" + from agent_framework import FunctionCallContent, FunctionResultContent + from agent_framework.openai import OpenAIChatClient + + client = OpenAIChatClient(model_id="gpt-4.1-mini") + + # Create specialized writer agent + writer = client.create_agent( + name="writer", + instructions="You are a creative writer. Write short, engaging content.", + ) + + # Convert writer to a tool using as_tool() + writer_tool = writer.as_tool( + name="creative_writer", + description="Generate creative content", + arg_name="request", + arg_description="What to write", + ) + + # Create coordinator agent with writer tool + coordinator = client.create_agent( + name="coordinator", + instructions="You coordinate with specialized agents. Delegate writing tasks to the writer agent.", + tools=[writer_tool], + ) + + # Run coordinator with streaming - it will delegate to writer + print("[Agent Framework]") + + # Track accumulated function calls (they stream in incrementally) + accumulated_calls: dict[str, FunctionCallContent] = {} + + async for chunk in coordinator.run_stream("Create a tagline for a coffee shop"): + # Stream text tokens + if chunk.text: + print(chunk.text, end="", flush=True) + + # Process streaming function calls and results + if chunk.contents: + for content in chunk.contents: + if isinstance(content, FunctionCallContent): + # Accumulate function call content as it streams in + call_id = content.call_id + if call_id in accumulated_calls: + # Add to existing call (arguments stream in gradually) + accumulated_calls[call_id] = accumulated_calls[call_id] + content + else: + # First chunk of this function call + accumulated_calls[call_id] = content + print("\n[Function Call - streaming]", flush=True) + print(f" Call ID: {call_id}", flush=True) + print(f" Name: {content.name}", flush=True) + + # Show accumulated arguments so far + current_args = accumulated_calls[call_id].arguments + print(f" Arguments: {current_args}", flush=True) + + elif isinstance(content, FunctionResultContent): + # Tool result - shows writer's response + result_text = content.result if isinstance(content.result, str) else str(content.result) + if result_text.strip(): + print("\n[Function Result]", flush=True) + print(f" Call ID: {content.call_id}", flush=True) + print(f" Result: {result_text[:150]}{'...' if len(result_text) > 150 else ''}", flush=True) + print() + + +async def main() -> None: + print("=" * 60) + print("Agent-as-Tool Pattern Comparison") + print("=" * 60) + print("Note: AutoGen requires parallel_tool_calls=False for AgentTool") + print(" Agent Framework handles this automatically\n") + await run_autogen() + print() + await run_agent_framework() + + +if __name__ == "__main__": + asyncio.run(main()) diff --git a/python/uv.lock b/python/uv.lock index d722d1e183..6209c65dce 100644 --- a/python/uv.lock +++ b/python/uv.lock @@ -88,6 +88,8 @@ dependencies = [ [package.dev-dependencies] dev = [ + { name = "autogen-agentchat", marker = "sys_platform == 'darwin' or sys_platform == 'linux' or sys_platform == 'win32'" }, + { name = "autogen-ext", extra = ["openai"], marker = "sys_platform == 'darwin' or sys_platform == 'linux' or sys_platform == 'win32'" }, { name = "flit", marker = "sys_platform == 'darwin' or sys_platform == 'linux' or sys_platform == 'win32'" }, { name = "mypy", marker = "sys_platform == 'darwin' or sys_platform == 'linux' or sys_platform == 'win32'" }, { name = "poethepoet", marker = "sys_platform == 'darwin' or sys_platform == 'linux' or sys_platform == 'win32'" }, @@ -127,6 +129,8 @@ requires-dist = [ [package.metadata.requires-dev] dev = [ + { name = "autogen-agentchat" }, + { name = "autogen-ext", extras = ["openai"] }, { name = "flit", specifier = ">=3.12.0" }, { name = "mypy", specifier = ">=1.16.1" }, { name = "poethepoet", specifier = ">=0.36.0" }, @@ -722,6 +726,54 @@ wheels = [ { url = "https://files.pythonhosted.org/packages/3a/2a/7cc015f5b9f5db42b7d48157e23356022889fc354a2813c15934b7cb5c0e/attrs-25.4.0-py3-none-any.whl", hash = "sha256:adcf7e2a1fb3b36ac48d97835bb6d8ade15b8dcce26aba8bf1d14847b57a3373", size = 67615, upload-time = "2025-10-06T13:54:43.17Z" }, ] +[[package]] +name = "autogen-agentchat" +version = "0.7.5" +source = { registry = "https://pypi.org/simple" } +dependencies = [ + { name = "autogen-core", marker = "sys_platform == 'darwin' or sys_platform == 'linux' or sys_platform == 'win32'" }, +] +sdist = { url = "https://files.pythonhosted.org/packages/e5/b6/df2f835ce3aaaa2716a3dfbbd4ab8855839184f08b35ce0baa23b26a1885/autogen_agentchat-0.7.5.tar.gz", hash = "sha256:8d9c718db52ef24a518806b3a0ef848f0e4c1902877675dc0abed73a8e6e7755", size = 147716, upload-time = "2025-09-30T06:16:14.413Z" } +wheels = [ + { url = "https://files.pythonhosted.org/packages/9e/82/23490a70837d77d691948863d393cef71a06d36903249f635b28f579292b/autogen_agentchat-0.7.5-py3-none-any.whl", hash = "sha256:d19ca8ec26cb15e071a56c4269140aea2bf3c718bdc7e06f6677af9a905815ba", size = 119302, upload-time = "2025-09-30T06:16:12.895Z" }, +] + +[[package]] +name = "autogen-core" +version = "0.7.5" +source = { registry = "https://pypi.org/simple" } +dependencies = [ + { name = "jsonref", marker = "sys_platform == 'darwin' or sys_platform == 'linux' or sys_platform == 'win32'" }, + { name = "opentelemetry-api", marker = "sys_platform == 'darwin' or sys_platform == 'linux' or sys_platform == 'win32'" }, + { name = "pillow", marker = "sys_platform == 'darwin' or sys_platform == 'linux' or sys_platform == 'win32'" }, + { name = "protobuf", marker = "sys_platform == 'darwin' or sys_platform == 'linux' or sys_platform == 'win32'" }, + { name = "pydantic", marker = "sys_platform == 'darwin' or sys_platform == 'linux' or sys_platform == 'win32'" }, + { name = "typing-extensions", marker = "sys_platform == 'darwin' or sys_platform == 'linux' or sys_platform == 'win32'" }, +] +sdist = { url = "https://files.pythonhosted.org/packages/b1/11/fea52bf3541c5308bed1ee9b9b3596fa510b2c5db893d32b649d22f02b87/autogen_core-0.7.5.tar.gz", hash = "sha256:70c2871389f1d0a7f6db8ef78717a51b7ce877ff4a08a836b7758d604dece203", size = 101980, upload-time = "2025-09-30T06:16:25.957Z" } +wheels = [ + { url = "https://files.pythonhosted.org/packages/33/83/8ad899fca9dd2d2b3e5e37be13dd9e6aee3e53a621041b0624d74b07e1ee/autogen_core-0.7.5-py3-none-any.whl", hash = "sha256:4f4a0d3b88a36da75b2ef0d40be2d5e3a207cae7f7d951511e498ad1d68f8ef4", size = 101874, upload-time = "2025-09-30T06:16:24.306Z" }, +] + +[[package]] +name = "autogen-ext" +version = "0.7.5" +source = { registry = "https://pypi.org/simple" } +dependencies = [ + { name = "autogen-core", marker = "sys_platform == 'darwin' or sys_platform == 'linux' or sys_platform == 'win32'" }, +] +sdist = { url = "https://files.pythonhosted.org/packages/e5/c8/f0651372f814c48eb64ffe921166995b7734bec0df7f0ba663383e831f58/autogen_ext-0.7.5.tar.gz", hash = "sha256:711ab9238ea66ff2abef163c331e538092bdea661620727a4a9b2ebce1c22df9", size = 417568, upload-time = "2025-09-30T06:16:24.278Z" } +wheels = [ + { url = "https://files.pythonhosted.org/packages/5c/10/9333ba6c532086cce7ec7fb39e36b9a08afdbc39e2d3519f00af712e403a/autogen_ext-0.7.5-py3-none-any.whl", hash = "sha256:18cecc8aab37c7c4861fbad038a1017f0ef25e35e273aa158066ccf9d93fea4f", size = 331380, upload-time = "2025-09-30T06:16:22.832Z" }, +] + +[package.optional-dependencies] +openai = [ + { name = "aiofiles", marker = "sys_platform == 'darwin' or sys_platform == 'linux' or sys_platform == 'win32'" }, + { name = "openai", marker = "sys_platform == 'darwin' or sys_platform == 'linux' or sys_platform == 'win32'" }, + { name = "tiktoken", marker = "sys_platform == 'darwin' or sys_platform == 'linux' or sys_platform == 'win32'" }, +] + [[package]] name = "azure-ai-agents" version = "1.2.0b5" @@ -1599,7 +1651,7 @@ name = "exceptiongroup" version = "1.3.0" source = { registry = "https://pypi.org/simple" } dependencies = [ - { name = "typing-extensions", marker = "(python_full_version < '3.13' and sys_platform == 'darwin') or (python_full_version < '3.13' and sys_platform == 'linux') or (python_full_version < '3.13' and sys_platform == 'win32')" }, + { name = "typing-extensions", marker = "(python_full_version < '3.11' and sys_platform == 'darwin') or (python_full_version < '3.11' and sys_platform == 'linux') or (python_full_version < '3.11' and sys_platform == 'win32')" }, ] sdist = { url = "https://files.pythonhosted.org/packages/0b/9f/a65090624ecf468cdca03533906e7c69ed7588582240cfe7cc9e770b50eb/exceptiongroup-1.3.0.tar.gz", hash = "sha256:b241f5885f560bc56a59ee63ca4c6a8bfa46ae4ad651af316d4e81817bb9fd88", size = 29749, upload-time = "2025-05-10T17:42:51.123Z" } wheels = [ @@ -2487,6 +2539,15 @@ wheels = [ { url = "https://files.pythonhosted.org/packages/35/5a/73ecb3d82f8615f32ccdadeb9356726d6cae3a4bbc840b437ceb95708063/jsonpath_ng-1.7.0-py3-none-any.whl", hash = "sha256:f3d7f9e848cba1b6da28c55b1c26ff915dc9e0b1ba7e752a53d6da8d5cbd00b6", size = 30105, upload-time = "2024-11-20T17:58:30.418Z" }, ] +[[package]] +name = "jsonref" +version = "1.1.0" +source = { registry = "https://pypi.org/simple" } +sdist = { url = "https://files.pythonhosted.org/packages/aa/0d/c1f3277e90ccdb50d33ed5ba1ec5b3f0a242ed8c1b1a85d3afeb68464dca/jsonref-1.1.0.tar.gz", hash = "sha256:32fe8e1d85af0fdefbebce950af85590b22b60f9e95443176adbde4e1ecea552", size = 8814, upload-time = "2023-01-16T16:10:04.455Z" } +wheels = [ + { url = "https://files.pythonhosted.org/packages/0c/ec/e1db9922bceb168197a558a2b8c03a7963f1afe93517ddd3cf99f202f996/jsonref-1.1.0-py3-none-any.whl", hash = "sha256:590dc7773df6c21cbf948b5dac07a72a251db28b0238ceecce0a2abfa8ec30a9", size = 9425, upload-time = "2023-01-16T16:10:02.255Z" }, +] + [[package]] name = "jsonschema" version = "4.25.1"