forked from pranjalkecilin01/crewai_development
-
Notifications
You must be signed in to change notification settings - Fork 0
/
pdf_task.py
30 lines (26 loc) · 870 Bytes
/
pdf_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
from crewai import Task, Crew
from Crew_agent.pdf_research_agent import pdf_reasercher
from Crew_agent.writer_agent import writer
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