# Cài đặt dependencies: pip install pyautogen openai
import autogen
from autogen.agentchat.user_proxy_agent import UserProxyAgent
from autogen.agentchat.assistant_agent import AssistantAgent

config = {
    "llm": "openai",
    "config_list": [
        {"model": "gpt-3.5-turbo", "api_key": "YOUR_OPENAI_API_KEY"}
    ]
}

reviewer = AssistantAgent(
    name="Reviewer",
    system_message="Bạn kiểm tra code Python để tìm lỗi và tối ưu.",
    llm_config=config,
)

coder = AssistantAgent(
    name="Coder",
    system_message="Bạn viết code Python theo chuẩn tốt nhất.",
    llm_config=config,
)

user_proxy = UserProxyAgent(
    name="User",
    code_execution_config={"work_dir": "python_scripts"}
)

init_msg_coder = "Viết hàm Python kiểm tra xem một chuỗi có phải là palindrome không."

user_proxy.initiate_chat(
    agent=reviewer,
    messages=[
        ("User", init_msg_coder),
        ("Coder", "def is_palindrome(s):\n    return s == s[::-1]")
    ],
    n_results=2,
)

agents:
  - name: "DevOpsAgent"
    capabilities:
      - "build"
      - "deploy"
    endpoint: "https://devops.internal/api"
  - name: "TestAgent"
    capabilities:
      - "run_tests"
      - "report_coverage"
    endpoint: "https://ci.internal/api"
  - name: "DocAgent"
    capabilities:
      - "generate_docs"
      - "tag_codebase"
    endpoint: "https://docs.internal/api"

workflow:
  - step: "build"
    agent: "DevOpsAgent"
    next: "run_tests"
  - step: "run_tests"
    agent: "TestAgent"
    next: "generate_docs"
  - step: "generate_docs"
    agent: "DocAgent"
    next: "deploy"
  - step: "deploy"
    agent: "DevOpsAgent"
    end: true