# Cài đặt thư viện: 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="You review Python code for bugs and optimization.",
    llm_config=config
)
coder = AssistantAgent(
    name="Coder",
    system_message="You write Python code following best practices.",
    llm_config=config
)
user_proxy = UserProxyAgent(
    name="User",
    code_execution_config={"work_dir": "python_scripts"}
)

init_msg_coder = "Write a Python function that checks if a string is a palindrome."

user_proxy.initiate_chat(
    agent=reviewer,
    messages=[
        ("User", init_msg_coder),
        ("Coder", "Here is the function implementation:\ndef 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