Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -33,9 +33,9 @@ if not GOOGLE_API_KEY:
|
|
| 33 |
st.stop()
|
| 34 |
|
| 35 |
|
| 36 |
-
def generateResponse(prompt):
|
| 37 |
llm = GoogleGemini(api_key=gemini_api_key)
|
| 38 |
-
pandas_agent = SmartDatalake(
|
| 39 |
answer = pandas_agent.chat(prompt)
|
| 40 |
return answer
|
| 41 |
|
|
|
|
| 33 |
st.stop()
|
| 34 |
|
| 35 |
|
| 36 |
+
def generateResponse(prompt, dfs):
|
| 37 |
llm = GoogleGemini(api_key=gemini_api_key)
|
| 38 |
+
pandas_agent = SmartDatalake(dfs,config={"llm":llm, "response_parser":StreamLitResponse})
|
| 39 |
answer = pandas_agent.chat(prompt)
|
| 40 |
return answer
|
| 41 |
|