Skip to main content

agentchat.contrib.agent_optimizer

execute_func

def execute_func(name, packages, code, **args)

The wrapper for generated functions.

AgentOptimizer

class AgentOptimizer()

Base class for optimizing AutoGen agents. Specifically, it is used to optimize the functions used in the agent. More information could be found in the following paper: https://arxiv.org/abs/2402.11359.

__init__

def __init__(max_actions_per_step: int,
llm_config: dict,
optimizer_model: Optional[str] = "gpt-4-1106-preview")

(These APIs are experimental and may change in the future.)

Arguments:

  • max_actions_per_step int - the maximum number of actions that the optimizer can take in one step.
  • llm_config dict - llm inference configuration. Please refer to OpenAIWrapper.create for available options. When using OpenAI or Azure OpenAI endpoints, please specify a non-empty 'model' either in llm_config or in each config of 'config_list' in llm_config.
  • optimizer_model - the model used for the optimizer.

record_one_conversation

def record_one_conversation(conversation_history: List[Dict],
is_satisfied: bool = None)

record one conversation history.

Arguments:

  • conversation_history List[Dict] - the chat messages of the conversation.
  • is_satisfied bool - whether the user is satisfied with the solution. If it is none, the user will be asked to input the satisfaction.

step

def step()

One step of training. It will return register_for_llm and register_for_executor at each iteration, which are subsequently utilized to update the assistant and executor agents, respectively. See example: https://github.com/autogenhub/autogen/blob/main/notebook/agentchat_agentoptimizer.ipynb

reset_optimizer

def reset_optimizer()

reset the optimizer.