bstraehle commited on
Commit
ec1b07b
·
verified ·
1 Parent(s): b2c1039

Update crew.py

Browse files
Files changed (1) hide show
  1. crew.py +7 -3
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
- #document_analysis_agent,
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,