Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -93,7 +93,7 @@ llm = ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=g_api_key, tempe
|
|
93 |
|
94 |
def generateResponse(prompt):
|
95 |
llm = GoogleGemini(api_key=g_api_key)
|
96 |
-
pandas_agent = Agent(df,config={"llm":llm, "save_charts_path": user_defined_path,"save_charts":
|
97 |
answer = pandas_agent.chat(prompt)
|
98 |
return answer
|
99 |
|
|
|
93 |
|
94 |
def generateResponse(prompt):
|
95 |
llm = GoogleGemini(api_key=g_api_key)
|
96 |
+
pandas_agent = Agent(df,config={"llm":llm, "save_charts_path": user_defined_path,"save_charts": False,"response_parser":FlaskResponse, "enable_cache": False, "conversational":True})
|
97 |
answer = pandas_agent.chat(prompt)
|
98 |
return answer
|
99 |
|