Update crew.py
Browse files
crew.py
CHANGED
@@ -1,3 +1,7 @@
|
|
|
|
|
|
|
|
|
|
1 |
import os
|
2 |
from crewai import Agent, Crew, Process, Task
|
3 |
from crewai.tools import tool
|
@@ -177,7 +181,7 @@ def run_crew(question, file_path):
|
|
177 |
@tool("Document Analysis Tool")
|
178 |
def document_analysis_tool(question: str, file_path: str) -> str:
|
179 |
"""Answer a question about a document file. Supported document types include:
|
180 |
-
.txt, .csv, .xml, .rtf, .pdf, .md, .html, .css, .js, .py
|
181 |
|
182 |
Args:
|
183 |
question (str): Question about a document file
|
@@ -324,7 +328,7 @@ def run_crew(question, file_path):
|
|
324 |
|
325 |
document_analysis_agent = Agent(
|
326 |
role="Document Analysis Agent",
|
327 |
-
goal="Analyze document of type .txt, .csv, .xml, .rtf, .pdf, .md, .html, .css, .js, .py to help answer question \"{question}\"",
|
328 |
backstory="As an expert document analysis assistant, you analyze the document to help answer the question.",
|
329 |
allow_delegation=False,
|
330 |
llm=AGENT_MODEL,
|
@@ -383,7 +387,7 @@ def run_crew(question, file_path):
|
|
383 |
audio_analysis_agent,
|
384 |
video_analysis_agent,
|
385 |
youtube_analysis_agent,
|
386 |
-
|
387 |
code_generation_agent,
|
388 |
code_execution_agent],
|
389 |
manager_agent=manager_agent,
|
|
|
1 |
+
# References:
|
2 |
+
# https://docs.crewai.com/introduction
|
3 |
+
# https://ai.google.dev/gemini-api/docs
|
4 |
+
|
5 |
import os
|
6 |
from crewai import Agent, Crew, Process, Task
|
7 |
from crewai.tools import tool
|
|
|
181 |
@tool("Document Analysis Tool")
|
182 |
def document_analysis_tool(question: str, file_path: str) -> str:
|
183 |
"""Answer a question about a document file. Supported document types include:
|
184 |
+
.xlxs, .txt, .csv, .xml, .rtf, .pdf, .md, .html, .css, .js, .py
|
185 |
|
186 |
Args:
|
187 |
question (str): Question about a document file
|
|
|
328 |
|
329 |
document_analysis_agent = Agent(
|
330 |
role="Document Analysis Agent",
|
331 |
+
goal="Analyze document of type .xlxs, .txt, .csv, .xml, .rtf, .pdf, .md, .html, .css, .js, .py to help answer question \"{question}\"",
|
332 |
backstory="As an expert document analysis assistant, you analyze the document to help answer the question.",
|
333 |
allow_delegation=False,
|
334 |
llm=AGENT_MODEL,
|
|
|
387 |
audio_analysis_agent,
|
388 |
video_analysis_agent,
|
389 |
youtube_analysis_agent,
|
390 |
+
document_analysis_agent,
|
391 |
code_generation_agent,
|
392 |
code_execution_agent],
|
393 |
manager_agent=manager_agent,
|