forked from pranjalkecilin01/crewai_development
-
Notifications
You must be signed in to change notification settings - Fork 0
/
task.py
61 lines (50 loc) · 1.58 KB
/
task.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
from crewai import Agent, Task, Crew, Process
from Crew_agent.research_agent import researcher
from Crew_agent.pdf_research_agent import pdf_reasercher
from Crew_agent.writer_agent import writer
def crewai_response(research_task, write_task):
print(research_task)
print(write_task)
task1 = Task(
max_iter=5,
description="""Conduct a comprehensive analysis""",
expected_output= "Full analysis report in bullet points",
agent=researcher(research_task)
)
task2 = Task(
max_iter=5,
description="""Using the insights provided, develop an engaging answer""",
expected_output= "Comprehensive answer",
agent=writer(write_task)
)
crew = Crew(
agents=[researcher(research_task), writer(write_task)],
tasks=[task1, task2],
verbose=2,
)
result = crew.kickoff()
print(result)
return result
def crewai_pdf_response(research_task, write_task):
print(research_task)
print(write_task)
task1 = Task(
max_iter=2,
description="""Conduct a comprehensive analysis. Identify the exact and accuracte answers""",
expected_output= "Full analysis report in bullet points",
agent=pdf_reasercher(research_task)
)
task2 = Task(
max_iter=2,
description="""Using the insights provided, develop an engaging answer.""",
expected_output= "An exact comprehensive answer",
agent=writer(write_task)
)
crew = Crew(
agents=[pdf_reasercher(research_task), writer(write_task)],
tasks=[task1, task2],
verbose=2,
)
result = crew.kickoff()
print(result)
return result