Update app.py

app.py

@@ -544,8 +544,10 @@ try:
         fullPrompt += "<|eot_id|><|start_header_id|>user<|end_header_id|>\n\n"
         fullPrompt += userPrompt
         fullPrompt += "<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n"
+
+        st.session_state.userpTA = userPrompt
           
-        logger.debug(f"### userPrompt: {fullPrompt}")
+        logger.debug(f"### fullPrompt: {fullPrompt}")
         logger.info("setPrompt exited.")
         return fullPrompt
 
@@ -561,10 +563,12 @@ try:
             with st.spinner('Generating Completion...'):
                 st.session_state.sysTAtext = st.session_state.sysTA
                 logger.debug(f"sysTAtext: {st.session_state.sysTAtext}") 
-                wrklist = setPrompt(st.session_state.userpTA,st.selectRag)
-                st.session_state.userpTA = wrklist[1]["content"]
+                #wrklist = setPrompt(st.session_state.userpTA,st.selectRag)
+                fullPrompt = setPrompt(st.session_state.userpTA,st.selectRag)
+                #st.session_state.userpTA = wrklist[1]["content"]
                 logger.debug(f"userpTAtext: {st.session_state.userpTA}")
-                rsp = runLLM(wrklist)
+                #rsp = runLLM(wrklist)
+                rsp = runLLM(fullPrompt)
                 st.session_state.rspTA = rsp
                 logger.debug(f"rspTAtext: {st.session_state.rspTA}")
                 #if "sysTA" not in st.session_state: