Update crew.py

crew.py

@@ -181,7 +181,7 @@ def run_crew(question, file_path):
 
     @tool("Code Generation Tool")
     def code_generation_tool(question: str, file_path: str) -> str:
-        """Generate and execute Python code, using an optional csv, xls, xlsx, json, or jsonl document as input, to answer a question.
+        """Generate and execute Python code, optionally reading a .csv, .xls, .xlsx, .json, .jsonl document, to answer a question.
 
            Args:
                question (str): Question to answer
@@ -193,16 +193,9 @@ def run_crew(question, file_path):
            Raises:
                RuntimeError: If processing fails"""
         try:
-            print("###")
-            print(question)
-            print(file_path)
-            print("###")
             if file_path:
                 df = read_file(file_path)
                 question = f"Question:\n{question}\nData:\n{df.to_string()}"
-                print("###")
-                print(question)
-                print("###")
             
             client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
 
@@ -224,7 +217,7 @@ def run_crew(question, file_path):
             
     @tool("Code Execution Tool")
     def code_execution_tool(question: str, file_path: str) -> str:
-        """Execute a Python code file to answer a question.
+        """Execute a .py Python code file to answer a question.
     
            Args:
                question (str): Question to answer
@@ -313,7 +306,7 @@ def run_crew(question, file_path):
 
     code_generation_agent = Agent(
         role="Code Generation Agent",
-        goal="Generate and execute Python code, using an optional csv, xls, xlsx, json, or jsonl document as input, to help answer question \"{question}\"",
+        goal="Generate and execute Python code, optionally reading a .csv, .xls, .xlsx, .json, .jsonl document, to help answer question \"{question}\"",
         backstory="As an expert Python code generation assistant, you generate and execute code to help answer the question.",
         allow_delegation=False,
         llm=AGENT_MODEL,