组件配置#
AutoGen 组件能够以通用方式进行声明式配置。这是为了支持基于配置的体验,例如 AutoGen studio,但它对于许多其他场景也很有用。
提供此功能的系统称为“组件配置”。在 AutoGen 中,组件只是可以从配置对象创建并自身可以转储到配置对象中的事物。通过这种方式,您可以在代码中定义一个组件,然后从中获取配置对象。
该系统是通用的,允许以相同的方式配置在 AutoGen 本身之外定义的组件(例如扩展)。
这与状态有何不同?#
这是一个非常重要的澄清点。当我们谈论序列化一个对象时,我们必须包含构成该对象本身的所有数据。包括消息历史等。从序列化状态反序列化时,您必须得到完全相同的对象。组件配置并非如此。
组件配置应被视为对象的蓝图,可以多次“盖章”以创建同一已配置对象的多个实例。
用法#
如果您在 Python 中有一个组件并希望获取其配置,只需在其上调用 dump_component()。生成的结果对象可以传回 load_component() 以获取回组件。
从配置加载组件#
要从配置对象加载组件,可以使用 load_component() 方法。此方法将接收一个配置对象并返回一个组件对象。最好在您想要的接口上调用此方法。例如,加载模型客户端
from autogen_core.models import ChatCompletionClient
config = {
"provider": "openai_chat_completion_client",
"config": {"model": "gpt-4o"},
}
client = ChatCompletionClient.load_component(config)
创建组件类#
要向给定类添加组件功能
在类继承列表中添加对
Component()的调用。实现
_to_config()和_from_config()方法。
例如
from autogen_core import Component, ComponentBase
from pydantic import BaseModel
class Config(BaseModel):
value: str
class MyComponent(ComponentBase[Config], Component[Config]):
component_type = "custom"
component_config_schema = Config
def __init__(self, value: str):
self.value = value
def _to_config(self) -> Config:
return Config(value=self.value)
@classmethod
def _from_config(cls, config: Config) -> "MyComponent":
return cls(value=config.value)
秘密#
如果配置对象的字段是秘密值,则应使用 SecretStr 进行标记,这将确保该值不会转储到配置对象中。
例如
from pydantic import BaseModel, SecretStr
class ClientConfig(BaseModel):
endpoint: str
api_key: SecretStr