bstraehle commited on
Commit
af90c08
·
verified ·
1 Parent(s): 890784c

Update agents.py

Browse files
Files changed (1) hide show
  1. agents.py +6 -6
agents.py CHANGED
@@ -2,26 +2,26 @@ 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, temperature=0.7),
12
  tools = [search_tool(), scrape_tool()],
13
  allow_delegation=False,
14
- verbose=False
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 a 2000-word 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, temperature=0.7),
24
  tools = [today_tool()],
25
  allow_delegation=False,
26
- verbose=False
27
  )
 
2
  from langchain_openai import ChatOpenAI
3
  from tools import scrape_tool, search_tool, today_tool
4
 
5
+ def get_researcher_agent(model, temperature, 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, temperature=temperature),
12
  tools = [search_tool(), scrape_tool()],
13
  allow_delegation=False,
14
+ verbose=verbose
15
  )
16
 
17
+ def get_writer_agent(model, temperature, verbose):
18
  return Agent(
19
  role="Writer",
20
  goal="Write an article on topic: {topic}.",
21
  backstory="You're working on writing a 2000-word 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, temperature=temperature),
24
  tools = [today_tool()],
25
  allow_delegation=False,
26
+ verbose=verbose
27
  )