Komponenten serialisieren#
AutoGen stellt die Konfigurationsklasse Component bereit, die Verhaltensweisen zum Serialisieren/Deserialisieren von Komponenten in deklarative Spezifikationen definiert. Dies können wir durch Aufruf von .dump_component() bzw. .load_component() erreichen. Dies ist nützlich für Debugging, Visualisierung und sogar zum Teilen Ihrer Arbeit mit anderen. In diesem Notebook demonstrieren wir, wie mehrere Komponenten in eine deklarative Spezifikation wie eine JSON-Datei serialisiert werden.
Warnung
LADEN SIE KOMPONENTEN NUR AUS VERTRAUENSWÜRDIGEN QUELLEN.
Bei serialisierten Komponenten implementiert jede Komponente die Logik, wie sie serialisiert und deserialisiert wird – d. h. wie die deklarative Spezifikation generiert und wie sie zurück in ein Objekt konvertiert wird.
In einigen Fällen kann das Erstellen eines Objekts die Ausführung von Code beinhalten (z. B. eine serialisierte Funktion). LADEN SIE KOMPONENTEN NUR AUS VERTRAUENSWÜRDIGEN QUELLEN.
Hinweis
selector_func ist nicht serialisierbar und wird während des Serialisierungs- und Deserialisierungsprozesses ignoriert.
Beispiel für Abbruchbedingungen#
Im folgenden Beispiel definieren wir Abbruchbedingungen (ein Teil eines Agententeams) in Python, exportieren diese in ein Dictionary/JSON und demonstrieren auch, wie das Objekt der Abbruchbedingung aus dem Dictionary/JSON geladen werden kann.
from autogen_agentchat.conditions import MaxMessageTermination, StopMessageTermination
max_termination = MaxMessageTermination(5)
stop_termination = StopMessageTermination()
or_termination = max_termination | stop_termination
or_term_config = or_termination.dump_component()
print("Config: ", or_term_config.model_dump_json())
new_or_termination = or_termination.load_component(or_term_config)
Config: {"provider":"autogen_agentchat.base.OrTerminationCondition","component_type":"termination","version":1,"component_version":1,"description":null,"config":{"conditions":[{"provider":"autogen_agentchat.conditions.MaxMessageTermination","component_type":"termination","version":1,"component_version":1,"config":{"max_messages":5}},{"provider":"autogen_agentchat.conditions.StopMessageTermination","component_type":"termination","version":1,"component_version":1,"config":{}}]}}
Beispiel für Agenten#
Im folgenden Beispiel definieren wir einen Agenten in Python, exportieren diesen in ein Dictionary/JSON und demonstrieren auch, wie das Agentenobjekt aus dem Dictionary/JSON geladen werden kann.
from autogen_agentchat.agents import AssistantAgent, UserProxyAgent
from autogen_ext.models.openai import OpenAIChatCompletionClient
# Create an agent that uses the OpenAI GPT-4o model.
model_client = OpenAIChatCompletionClient(
model="gpt-4o",
# api_key="YOUR_API_KEY",
)
agent = AssistantAgent(
name="assistant",
model_client=model_client,
handoffs=["flights_refunder", "user"],
# tools=[], # serializing tools is not yet supported
system_message="Use tools to solve tasks.",
)
user_proxy = UserProxyAgent(name="user")
user_proxy_config = user_proxy.dump_component() # dump component
print(user_proxy_config.model_dump_json())
up_new = user_proxy.load_component(user_proxy_config) # load component
{"provider":"autogen_agentchat.agents.UserProxyAgent","component_type":"agent","version":1,"component_version":1,"description":null,"config":{"name":"user","description":"A human user"}}
agent_config = agent.dump_component() # dump component
print(agent_config.model_dump_json())
agent_new = agent.load_component(agent_config) # load component
{"provider":"autogen_agentchat.agents.AssistantAgent","component_type":"agent","version":1,"component_version":1,"description":null,"config":{"name":"assistant","model_client":{"provider":"autogen_ext.models.openai.OpenAIChatCompletionClient","component_type":"model","version":1,"component_version":1,"config":{"model":"gpt-4o"}},"handoffs":[{"target":"flights_refunder","description":"Handoff to flights_refunder.","name":"transfer_to_flights_refunder","message":"Transferred to flights_refunder, adopting the role of flights_refunder immediately."},{"target":"user","description":"Handoff to user.","name":"transfer_to_user","message":"Transferred to user, adopting the role of user immediately."}],"model_context":{"provider":"autogen_core.model_context.UnboundedChatCompletionContext","component_type":"chat_completion_context","version":1,"component_version":1,"config":{}},"description":"An agent that provides assistance with ability to use tools.","system_message":"Use tools to solve tasks.","reflect_on_tool_use":false,"tool_call_summary_format":"{result}"}}
Ein ähnlicher Ansatz kann verwendet werden, um den Agenten MultiModalWebSurfer zu serialisieren.
from autogen_ext.agents.web_surfer import MultimodalWebSurfer
agent = MultimodalWebSurfer(
name="web_surfer",
model_client=model_client,
headless=False,
)
web_surfer_config = agent.dump_component() # dump component
print(web_surfer_config.model_dump_json())
Beispiel für ein Team#
Im folgenden Beispiel definieren wir ein Team in Python, exportieren dieses in ein Dictionary/JSON und demonstrieren auch, wie das Teamobjekt aus dem Dictionary/JSON geladen werden kann.
from autogen_agentchat.agents import AssistantAgent, UserProxyAgent
from autogen_agentchat.conditions import MaxMessageTermination
from autogen_agentchat.teams import RoundRobinGroupChat
from autogen_ext.models.openai import OpenAIChatCompletionClient
# Create an agent that uses the OpenAI GPT-4o model.
model_client = OpenAIChatCompletionClient(
model="gpt-4o",
# api_key="YOUR_API_KEY",
)
agent = AssistantAgent(
name="assistant",
model_client=model_client,
handoffs=["flights_refunder", "user"],
# tools=[], # serializing tools is not yet supported
system_message="Use tools to solve tasks.",
)
team = RoundRobinGroupChat(participants=[agent], termination_condition=MaxMessageTermination(2))
team_config = team.dump_component() # dump component
print(team_config.model_dump_json())
await model_client.close()
{"provider":"autogen_agentchat.teams.RoundRobinGroupChat","component_type":"team","version":1,"component_version":1,"description":null,"config":{"participants":[{"provider":"autogen_agentchat.agents.AssistantAgent","component_type":"agent","version":1,"component_version":1,"config":{"name":"assistant","model_client":{"provider":"autogen_ext.models.openai.OpenAIChatCompletionClient","component_type":"model","version":1,"component_version":1,"config":{"model":"gpt-4o"}},"handoffs":[{"target":"flights_refunder","description":"Handoff to flights_refunder.","name":"transfer_to_flights_refunder","message":"Transferred to flights_refunder, adopting the role of flights_refunder immediately."},{"target":"user","description":"Handoff to user.","name":"transfer_to_user","message":"Transferred to user, adopting the role of user immediately."}],"model_context":{"provider":"autogen_core.model_context.UnboundedChatCompletionContext","component_type":"chat_completion_context","version":1,"component_version":1,"config":{}},"description":"An agent that provides assistance with ability to use tools.","system_message":"Use tools to solve tasks.","reflect_on_tool_use":false,"tool_call_summary_format":"{result}"}}],"termination_condition":{"provider":"autogen_agentchat.conditions.MaxMessageTermination","component_type":"termination","version":1,"component_version":1,"config":{"max_messages":2}}}}