Update crew.py

crew.py

@@ -208,12 +208,12 @@ def run_crew(question, file_path):
             raise RuntimeError(f"Processing failed: {str(e)}")
     
     @tool("Code Generation Tool")
-    def code_generation_tool(question: str, json_data: str) -> str:
-        """Given a question and JSON data, generate and execute code to answer the question.
+    def code_generation_tool(question: str, file_path: str) -> str:
+        """Given a question and data file, generate and execute code to answer the question.
 
            Args:
                question (str): Question to answer
-               json_data (str): The JSON data
+                file_path (str): The data file path
 
            Returns:
                str: Answer to the question
@@ -223,9 +223,11 @@ def run_crew(question, file_path):
         try:
             client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
 
+            file_data = read_file(file_path)
+            
             response = client.models.generate_content(
                 model=CODE_GENERATION_MODEL,
-                contents=[f"{question}\n{json_data}"],
+                contents=[f"{question}\n{file_data}"],
                 config=types.GenerateContentConfig(
                     tools=[types.Tool(code_execution=types.ToolCodeExecution)]
                 ),
@@ -339,7 +341,7 @@ def run_crew(question, file_path):
     
     code_generation_agent = Agent(
         role="Code Generation Agent",
-        goal="Given a question and JSON data, generate and execute code to answer the question: {question}",
+        goal="Given a question and data file, generate and execute code to answer the question: {question}",
         backstory="As an expert Python code generation assistant, you generate and execute code to answer the question.",
         allow_delegation=False,
         llm=AGENT_MODEL,
@@ -373,15 +375,15 @@ def run_crew(question, file_path):
 
     manager_task = Task(
         agent=manager_agent,
-        description="Answer the following question. If needed, delegate to one of your coworkers: "
-                    "- Web Search Agent requires a question only. "
-                    "- Image Analysis Agent requires a question and **.png, .jpeg, .webp, .heic, or .heif image file**. "
-                    "- Audio Analysis Agent requires a question and **.wav, .mp3, .aiff, .aac, .ogg, or .flac audio file**. "
-                    "- Video Analysis Agent requires a question and **.mp4, .mpeg, .mov, .avi, .x-flv, .mpg, .webm, .wmv, or .3gpp video file**. "
-                    "- Document Analysis Agent requires a question and **.pdf, .txt, .html, css, .js, .md, .xml, or .rtf document file**. "
-                    "- YouTube Analysis Agent requires a question and **YouTube URL**. "
-                    "- Code Generation Agent requires a question and **JSON data**. "
-                    "- Code Execution Agent requires a question and **.py Python file**. "
+        description="Answer the following question. If needed, delegate to one of your coworkers:\n"
+                    "- Web Search Agent requires a question only.\n"
+                    "- Image Analysis Agent requires a question and **.png, .jpeg, .webp, .heic, or .heif image file**.\n"
+                    "- Audio Analysis Agent requires a question and **.wav, .mp3, .aiff, .aac, .ogg, or .flac audio file**.\n"
+                    "- Video Analysis Agent requires a question and **.mp4, .mpeg, .mov, .avi, .x-flv, .mpg, .webm, .wmv, or .3gpp video file**.\n"
+                    "- Document Analysis Agent requires a question and **.pdf, .txt, .html, css, .js, .md, .xml, or .rtf document file**.\n"
+                    "- YouTube Analysis Agent requires a question and **YouTube URL**.\n"
+                    "- Code Generation Agent requires a question and **.csv, .xls, .xlsx, .json, or .jsonl data file**.\n"
+                    "- Code Execution Agent requires a question and **.py Python file**.\n"
                     "Question: {question}",
         expected_output="The answer to the question."
     )
@@ -405,12 +407,7 @@ def run_crew(question, file_path):
     # Process
 
     if file_path:
-        file_data = read_file(file_path)
-
-        if file_data:
-            question = f"{question} JSON data: {file_data}" # sandbox contraints
-        else:
-            question = f"{question} File path: {file_path}."
+        question = f"{question} File path: {file_path}."
 
     initial_answer = crew.kickoff(inputs={"question": question})
     final_answer = get_final_answer(FINAL_ANSWER_MODEL, question, str(initial_answer))