bstraehle's picture
Update agents.py
d4ebe88 verified
from crewai import Agent
from langchain_openai import ChatOpenAI
from tools import scrape_tool, search_tool, today_tool
def get_researcher_agent(model, verbose):
return Agent(
role="Researcher",
goal="Research content on topic: {topic}.",
backstory="You're working on researching content on topic: {topic}. "
"Your work is the basis for the Writer to write on this topic.",
llm=ChatOpenAI(model=model),
tools = [search_tool(), scrape_tool()],
allow_delegation=False,
verbose=verbose
)
def get_author_agent(model, verbose):
return Agent(
role="Author",
goal="Write an article on topic: {topic}.",
backstory="You're working on writing an article on topic: {topic}. "
"You base your writing on the work of the Researcher, who provides content on this topic. "
"Your work is the basis for the Editor to edit the article on this topic.",
llm=ChatOpenAI(model=model, max_tokens=32768),
tools = [],
allow_delegation=False,
verbose=verbose
)
def get_editor_agent(model, verbose):
return Agent(
role="Editor",
goal="Edit an article on topic: {topic}.",
backstory="You're working on editing an article on topic: {topic}. "
"You base your editing on the work of the Author, who provides an article on this topic.",
llm=ChatOpenAI(model=model, max_tokens=32768),
tools = [],
allow_delegation=False,
verbose=verbose
)