AutoGen - OpenAI Assistant

autogen_openai_assistant.py

import os
from autogen import UserProxyAgent
from autogen.agentchat.contrib.gpt_assistant_agent import GPTAssistantAgent

os.environ["AUTOGEN_USE_DOCKER"] = "False"
config_list = [
        {
            "model": "gpt-4o",
            "api_key": "sk-..."
        }
    ]


gpt_assistant = GPTAssistantAgent(
    name="assistant",
    llm_config={
        "config_list": config_list,
        "cache_seed": None,
        "assistant_id": None
    })


user_proxy = UserProxyAgent(name="user_proxy",
    code_execution_config={
        "work_dir": "coding"
    },
    is_termination_msg=lambda msg: "TERMINATE" in msg["content"],
    human_input_mode="NEVER")

user_proxy.initiate_chat(gpt_assistant, message="Print hello world")