Spaces:
Paused
Paused
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,
|