Usar um agente AG2

Além das instruções gerais para usar um agente, esta página descreve recursos específicos do AG2Agent.

Antes de começar

Este tutorial pressupõe que você leu e seguiu as instruções em:

Operações suportadas

As seguintes operações são compatíveis com AG2Agent:

  • query: para receber uma resposta a uma consulta de forma síncrona.

O método query aceita os seguintes argumentos:

  • input: a mensagem a ser enviada ao agente.
  • max_turns: o número máximo de turnos de conversa permitidos. Ao usar ferramentas, é necessário um mínimo de max_turns=2: uma vez para gerar argumentos de ferramenta e uma segunda para executar a ferramenta.

Consultar o agente

O método query() oferece uma maneira simplificada de interagir com o agente. Uma chamada típica é assim:

response = agent.query(input="What is the exchange rate from US dollars to Swedish currency?", max_turns=2)

Esse método processa a comunicação subjacente com o agente e retorna a resposta final dele como um dicionário. É equivalente ao seguinte (na forma completa):

from autogen import ConversableAgent
import dataclasses
import json

input_message: str = "What is the exchange rate from US dollars to Swedish currency?"
max_turns: int = 2

with agent._runnable._create_or_get_executor(
    tools=agent._ag2_tool_objects,            # Use the agent's existing tools
    agent_name="user",                        # Default
    agent_human_input_mode="NEVER",           # query() enforces this
) as executor:
    chat_result = executor.initiate_chat(
        agent._runnable,
        message=input_message,
        max_turns=max_turns,
        clear_history=False,                  # Default
        summary_method="last_msg"             # Default
    )

response = json.loads(
  json.dumps(dataclasses.asdict(chat_result)) # query() does this conversion
)

É possível personalizar o comportamento do agente além de input e max_turns transmitindo outros argumentos de palavra-chave para query().

response = agent.query(
    input="What is the exchange rate from US dollars to Swedish currency?",
    max_turns=2,
    msg_to="user"  # Start the conversation with the "user" agent
)
print(response)

Consulte a documentação do ConversableAgent.run para ver uma lista completa dos parâmetros disponíveis. No entanto, lembre-se de que user_input sempre será substituído por False pelo modelo AG2Agent.

A seguir