Spaces:
Running
Running
Update agents.py
Browse files
agents.py
CHANGED
@@ -2,25 +2,25 @@ from crewai import Agent
|
|
2 |
from langchain_openai import ChatOpenAI
|
3 |
from tools import scrape_tool, search_tool, today_tool
|
4 |
|
5 |
-
def get_researcher_agent(model,
|
6 |
return Agent(
|
7 |
role="Researcher",
|
8 |
goal="Research content on topic: {topic}.",
|
9 |
backstory="You're working on researching content on topic: {topic}. "
|
10 |
"Your work is the basis for the Writer to write on this topic.",
|
11 |
-
llm=ChatOpenAI(model=model
|
12 |
tools = [search_tool(), scrape_tool()],
|
13 |
allow_delegation=False,
|
14 |
verbose=verbose
|
15 |
)
|
16 |
|
17 |
-
def get_writer_agent(model,
|
18 |
return Agent(
|
19 |
role="Writer",
|
20 |
goal="Write an article on topic: {topic}.",
|
21 |
backstory="You're working on writing an article on topic: {topic}. "
|
22 |
"You base your writing on the work of the Researcher, who provides context on this topic.",
|
23 |
-
llm=ChatOpenAI(model=model
|
24 |
tools = [today_tool()],
|
25 |
allow_delegation=False,
|
26 |
verbose=verbose
|
|
|
2 |
from langchain_openai import ChatOpenAI
|
3 |
from tools import scrape_tool, search_tool, today_tool
|
4 |
|
5 |
+
def get_researcher_agent(model, verbose):
|
6 |
return Agent(
|
7 |
role="Researcher",
|
8 |
goal="Research content on topic: {topic}.",
|
9 |
backstory="You're working on researching content on topic: {topic}. "
|
10 |
"Your work is the basis for the Writer to write on this topic.",
|
11 |
+
llm=ChatOpenAI(model=model),
|
12 |
tools = [search_tool(), scrape_tool()],
|
13 |
allow_delegation=False,
|
14 |
verbose=verbose
|
15 |
)
|
16 |
|
17 |
+
def get_writer_agent(model, verbose):
|
18 |
return Agent(
|
19 |
role="Writer",
|
20 |
goal="Write an article on topic: {topic}.",
|
21 |
backstory="You're working on writing an article on topic: {topic}. "
|
22 |
"You base your writing on the work of the Researcher, who provides context on this topic.",
|
23 |
+
llm=ChatOpenAI(model=model),
|
24 |
tools = [today_tool()],
|
25 |
allow_delegation=False,
|
26 |
verbose=verbose
|