bstraehle commited on
Commit
b5e753e
·
verified ·
1 Parent(s): 7ad3b1e

Update crew.py

Browse files
Files changed (1) hide show
  1. crew.py +7 -7
crew.py CHANGED
@@ -181,11 +181,11 @@ def run_crew(question, file_path):
181
 
182
  @tool("Code Generation Tool")
183
  def code_generation_tool(question: str, file_path: str) -> str:
184
- """Given a question and optional .csv, .xls, .xlsx, .json, .jsonl file, generate and execute code to answer the question.
185
 
186
  Args:
187
  question (str): Question to answer
188
- file_path (str): The optional .csv, .xls, .xlsx, .json, .jsonl file path
189
 
190
  Returns:
191
  str: Answer to the question
@@ -219,11 +219,11 @@ def run_crew(question, file_path):
219
 
220
  @tool("Code Execution Tool")
221
  def code_execution_tool(question: str, file_path: str) -> str:
222
- """Given a question and .py file, execute the file to answer the question.
223
 
224
  Args:
225
  question (str): Question to answer
226
- file_path (str): The .py file path
227
 
228
  Returns:
229
  str: Answer to the question
@@ -308,8 +308,8 @@ def run_crew(question, file_path):
308
 
309
  code_generation_agent = Agent(
310
  role="Code Generation Agent",
311
- goal="Given a question and optional .csv, .xls, .xlsx, .json, .jsonl file, generate and execute code to help answer question \"{question}\"",
312
- backstory="As an expert Python code generation assistant, you generate and execute code to help answer the question.",
313
  allow_delegation=False,
314
  llm=AGENT_MODEL,
315
  max_iter=2,
@@ -319,7 +319,7 @@ def run_crew(question, file_path):
319
 
320
  code_execution_agent = Agent(
321
  role="Code Execution Agent",
322
- goal="Given a question and .py file, execute the file to help answer question \"{question}\"",
323
  backstory="As an expert Python code execution assistant, you execute the code file to help answer the question.",
324
  allow_delegation=False,
325
  llm=AGENT_MODEL,
 
181
 
182
  @tool("Code Generation Tool")
183
  def code_generation_tool(question: str, file_path: str) -> str:
184
+ """Given a question and optional .csv, .xls, .xlsx, .json, .jsonl document file, generate code to answer the question.
185
 
186
  Args:
187
  question (str): Question to answer
188
+ file_path (str): The optional .csv, .xls, .xlsx, .json, .jsonl document file path
189
 
190
  Returns:
191
  str: Answer to the question
 
219
 
220
  @tool("Code Execution Tool")
221
  def code_execution_tool(question: str, file_path: str) -> str:
222
+ """Given a question and .py Python file, execute the file to answer the question.
223
 
224
  Args:
225
  question (str): Question to answer
226
+ file_path (str): The .py Python file path
227
 
228
  Returns:
229
  str: Answer to the question
 
308
 
309
  code_generation_agent = Agent(
310
  role="Code Generation Agent",
311
+ goal="Given a question and optional .csv, .xls, .xlsx, .json, .jsonl document file, generate code to help answer the question: \"{question}\"",
312
+ backstory="As an expert Python code generation assistant, you generate code to help answer the question.",
313
  allow_delegation=False,
314
  llm=AGENT_MODEL,
315
  max_iter=2,
 
319
 
320
  code_execution_agent = Agent(
321
  role="Code Execution Agent",
322
+ goal="Given a question and .py Python file, execute the file to help answer the question: \"{question}\"",
323
  backstory="As an expert Python code execution assistant, you execute the code file to help answer the question.",
324
  allow_delegation=False,
325
  llm=AGENT_MODEL,