import os
from crewai import Agent, Task, Crew, Process

# 에이전트 정의
researcher = Agent(
    role="Researcher",
    goal="AI와 AI 에이전트에 대한 철저한 연구와 분석 수행",
    backstory="기술, 소프트웨어 엔지니어링, AI, 스타트업 분야의 전문 연구원으로, 프리랜서로 활동하며 새로운 클라이언트를 위해 연구 중이다.",
    allow_delegation=False,
)

writer = Agent(
    role="Senior Writer",
    goal="AI와 AI 에이전트에 관한 매력적인 콘텐츠 작성",
    backstory="기술, 소프트웨어 엔지니어링, AI, 스타트업 분야의 시니어 작가로, 프리랜서로 활동하며 새로운 클라이언트를 위해 콘텐츠를 작성 중이다.",
    allow_delegation=False,
)

# 작업 정의
task = Task(
    description="기사에 대한 5가지 흥미로운 아이디어를 생성하고, 각 아이디어에 대해 전체 기사의 잠재력을 보여줄 수 있는 매력적인 단락을 작성한다. 아이디어 목록과 각각의 단락, 그리고 노트를 반환한다.",
    expected_output="5개의 불릿 포인트, 각각 단락과 함께 노트가 포함된 형태.",
)

# 매니저 에이전트 정의
manager = Agent(
    role="Project Manager",
    goal="크루를 효율적으로 관리하고 고품질 작업 완료 보장",
    backstory="복잡한 프로젝트를 감독하고 팀을 성공으로 이끄는 데 능숙한 경험 많은 프로젝트 매니저다. 크루 멤버들의 노력을 조율하여 각 작업이 시간 내에 최고의 수준으로 완료되도록 한다.",
    allow_delegation=True,
)

# 커스텀 매니저와 함께 크루 인스턴스 생성
crew = Crew(
    agents=[researcher, writer],
    tasks=[task],
    manager_agent=manager,
    process=Process.hierarchical,
)

# 크루 작업 시작
result = crew.kickoff()

from crewai import LLM

manager_llm = LLM(model="gpt-4o")

crew = Crew(
    agents=[researcher, writer],
    tasks=[task],
    process=Process.hierarchical,
    manager_llm=manager_llm
)