Update main.py
Browse files
main.py
CHANGED
@@ -55,7 +55,7 @@ def create_agent(filename: str):
|
|
55 |
df = pd.read_csv(filename)
|
56 |
|
57 |
# Create a Pandas DataFrame agent.
|
58 |
-
return create_csv_agent(llm, filename, verbose=False, agent_type=AgentType.OPENAI_FUNCTIONS)
|
59 |
|
60 |
def query_agent(agent, query):
|
61 |
"""
|
|
|
55 |
df = pd.read_csv(filename)
|
56 |
|
57 |
# Create a Pandas DataFrame agent.
|
58 |
+
return create_csv_agent(llm, filename, verbose=False, allow_dangerous_code=True, agent_type=AgentType.OPENAI_FUNCTIONS)
|
59 |
|
60 |
def query_agent(agent, query):
|
61 |
"""
|