Spaces:
Sleeping
Sleeping
| 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 | |
| ) |