bstraehle commited on
Commit
99cc5cd
·
verified ·
1 Parent(s): 8395e8e

Update crew.py

Browse files
Files changed (1) hide show
  1. crew.py +5 -4
crew.py CHANGED
@@ -181,14 +181,15 @@ def run_crew(question, file_path):
181
  raise RuntimeError(f"Processing failed: {str(e)}")
182
 
183
  @tool("Code Execution Tool")
184
- def code_execution_tool(file_path: str) -> str:
185
- """Execute a Python code file and return the result.
186
 
187
  Args:
 
188
  file_path (str): The Python code file path
189
 
190
  Returns:
191
- str: The code execution result
192
 
193
  Raises:
194
  RuntimeError: If processing fails"""
@@ -199,7 +200,7 @@ def run_crew(question, file_path):
199
 
200
  response = client.models.generate_content(
201
  model=CODE_MODEL,
202
- contents=[file],
203
  config=types.GenerateContentConfig(
204
  tools=[types.Tool(code_execution=types.ToolCodeExecution)]
205
  ),
 
181
  raise RuntimeError(f"Processing failed: {str(e)}")
182
 
183
  @tool("Code Execution Tool")
184
+ def code_execution_tool(question: str, file_path: str) -> str:
185
+ """Execute a Python code file and answer a question about it.
186
 
187
  Args:
188
+ question (str): Question about a Pyton code file
189
  file_path (str): The Python code file path
190
 
191
  Returns:
192
+ str: Answer to the question about the Python code file
193
 
194
  Raises:
195
  RuntimeError: If processing fails"""
 
200
 
201
  response = client.models.generate_content(
202
  model=CODE_MODEL,
203
+ contents=[file, question],
204
  config=types.GenerateContentConfig(
205
  tools=[types.Tool(code_execution=types.ToolCodeExecution)]
206
  ),