Update app.py
Browse files
app.py
CHANGED
@@ -107,7 +107,8 @@ async def chat(request: ChatRequest):
|
|
107 |
),
|
108 |
)
|
109 |
|
110 |
-
# put back the unformatted user message
|
|
|
111 |
request.chat_history.append({"role": "user", "parts": [{"text": request.message}]})
|
112 |
|
113 |
return {"response": response.text, "dataset_str": concise_text_string, "history": request.chat_history, "RAG_prompt": rag_prompt, "chunks": chunks, "results": results}
|
|
|
107 |
),
|
108 |
)
|
109 |
|
110 |
+
# delete the prompt and put back the unformatted user message
|
111 |
+
del request.chat_history[-1]
|
112 |
request.chat_history.append({"role": "user", "parts": [{"text": request.message}]})
|
113 |
|
114 |
return {"response": response.text, "dataset_str": concise_text_string, "history": request.chat_history, "RAG_prompt": rag_prompt, "chunks": chunks, "results": results}
|