Update crew.py

crew.py

@@ -220,12 +220,12 @@ def run_crew(question, file_path):
             raise RuntimeError(f"Processing failed: {str(e)}")
 
     @tool("Code Generation Tool")
-    def code_generation_tool(question: str, file_path: str) -> str:
-        """Given a question and data file, generate and execute code to answer the question.
+    def code_generation_tool(question: str, file_data: str) -> str:
+        """Given a question and file data, generate and execute code to answer the question.
 
            Args:
                question (str): Question to answer
-                file_path (str): The data file path
+                file_path (str): The file data
 
            Returns:
                str: Answer to the question
@@ -234,10 +234,6 @@ def run_crew(question, file_path):
                RuntimeError: If processing fails"""
         try:
             client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
-
-            file_data = read_file_json(file_path)
-            
-            print(f"=> File data:\n{file_data}")
                     
             response = client.models.generate_content(
                 model=CODE_GENERATION_MODEL,
@@ -396,7 +392,7 @@ def run_crew(question, file_path):
                     "- 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 **.docx, .pptx, .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 Generation Agent requires a question and **file data**.\n"
                     "- Code Execution Agent requires a question and **.py Python file**.\n"
                     "Question: {question}",
         expected_output="The answer to the question."
@@ -421,7 +417,15 @@ def run_crew(question, file_path):
     # Process
 
     if file_path:
-        question = f"{question} File path: {file_path}."
+        if is_ext(file_path, ".csv") or
+           is_ext(file_path, ".xls") or
+           is_ext(file_path, ".xlsx") or
+           is_ext(file_path, ".json") or
+           is_ext(file_path, ".jsonl"):
+            file_data = read_file_json(file_path)
+            question = f"{question} File data: {file_data}."
+        else:
+            question = f"{question} File path: {file_path}."
 
     answer = crew.kickoff(inputs={"question": question})
     final_answer = get_final_answer(FINAL_ANSWER_MODEL, question, str(answer))