快速开始#
注意
See here for installation instructions.
在深入了解核心API之前，我们先从一个简单的例子开始，两个代理从10倒数到1。
我们首先定义代理类及其各自的处理消息的流程。
我们创建了两个代理类：Modifier 和 Checker。 Modifier 代理修改给定的数字，而 Check 代理根据条件检查该值。
我们还创建了一个 Message 数据类，它定义了在代理之间传递的消息。
from dataclasses import dataclass
from typing import Callable
from autogen_core import DefaultTopicId, MessageContext, RoutedAgent, default_subscription, message_handler
@dataclass
class Message:
content: int
@default_subscription
class Modifier(RoutedAgent):
def __init__(self, modify_val: Callable[[int], int]) -> None:
super().__init__("A modifier agent.")
self._modify_val = modify_val
@message_handler
async def handle_message(self, message: Message, ctx: MessageContext) -> None:
val = self._modify_val(message.content)
print(f"{'-'*80}\nModifier:\nModified {message.content} to {val}")
await self.publish_message(Message(content=val), DefaultTopicId()) # type: ignore
@default_subscription
class Checker(RoutedAgent):
def __init__(self, run_until: Callable[[int], bool]) -> None:
super().__init__("A checker agent.")
self._run_until = run_until
@message_handler
async def handle_message(self, message: Message, ctx: MessageContext) -> None:
if not self._run_until(message.content):
print(f"{'-'*80}\nChecker:\n{message.content} passed the check, continue.")
await self.publish_message(Message(content=message.content), DefaultTopicId())
else:
print(f"{'-'*80}\nChecker:\n{message.content} failed the check, stopping.")
你可能已经注意到，无论是使用模型还是代码执行器，代理的逻辑与消息的传递方式是完全解耦的。这是核心思想：框架提供了通信基础设施，而代理则负责自己的逻辑。我们将这种通信基础设施称为Agent Runtime。
代理运行时是这个框架的一个关键概念。除了传递消息外，它还管理代理的生命周期。因此，代理的创建由运行时处理。
以下代码展示了如何使用 SingleThreadedAgentRuntime 来注册和运行代理，这是一个本地嵌入式代理运行时实现。
注意
如果您正在使用VSCode或其他编辑器，请记得导入asyncio并将代码包裹在async def main() -> None:中，并使用asyncio.run(main())函数来运行代码。
from autogen_core import AgentId, SingleThreadedAgentRuntime
# Create an local embedded runtime.
runtime = SingleThreadedAgentRuntime()
# Register the modifier and checker agents by providing
# their agent types, the factory functions for creating instance and subscriptions.
await Modifier.register(
runtime,
"modifier",
# Modify the value by subtracting 1
lambda: Modifier(modify_val=lambda x: x - 1),
)
await Checker.register(
runtime,
"checker",
# Run until the value is less than or equal to 1
lambda: Checker(run_until=lambda x: x <= 1),
)
# Start the runtime and send a direct message to the checker.
runtime.start()
await runtime.send_message(Message(10), AgentId("checker", "default"))
await runtime.stop_when_idle()
--------------------------------------------------------------------------------
Checker:
10 passed the check, continue.
--------------------------------------------------------------------------------
Modifier:
Modified 10 to 9
--------------------------------------------------------------------------------
Checker:
9 passed the check, continue.
--------------------------------------------------------------------------------
Modifier:
Modified 9 to 8
--------------------------------------------------------------------------------
Checker:
8 passed the check, continue.
--------------------------------------------------------------------------------
Modifier:
Modified 8 to 7
--------------------------------------------------------------------------------
Checker:
7 passed the check, continue.
--------------------------------------------------------------------------------
Modifier:
Modified 7 to 6
--------------------------------------------------------------------------------
Checker:
6 passed the check, continue.
--------------------------------------------------------------------------------
Modifier:
Modified 6 to 5
--------------------------------------------------------------------------------
Checker:
5 passed the check, continue.
--------------------------------------------------------------------------------
Modifier:
Modified 5 to 4
--------------------------------------------------------------------------------
Checker:
4 passed the check, continue.
--------------------------------------------------------------------------------
Modifier:
Modified 4 to 3
--------------------------------------------------------------------------------
Checker:
3 passed the check, continue.
--------------------------------------------------------------------------------
Modifier:
Modified 3 to 2
--------------------------------------------------------------------------------
Checker:
2 passed the check, continue.
--------------------------------------------------------------------------------
Modifier:
Modified 2 to 1
--------------------------------------------------------------------------------
Checker:
1 failed the check, stopping.
从代理的输出中，我们可以看到值根据修改器和检查器的条件成功地从10递减到了1。
AutoGen 还支持分布式代理运行时，可以在不同的进程或机器上托管代理，这些代理具有不同的身份、语言和依赖关系。
要学习如何使用代理运行时、通信、消息处理和订阅，请继续阅读快速入门之后的部分。