Update app.py

app.py

@@ -44,8 +44,8 @@ langchain.verbose = True
 
 template = """If you don't know the answer, just say that you don't know, don't try to make up an answer. Keep the answer as concise as possible. Always say "Thanks for using the 🧠 app - Bernd" at the end of the answer. """
 
-llm_template = "Answer the question at the end. " + template + "Question: {question} Helpful Answer: "
-rag_template = "Use the following pieces of context to answer the question at the end. " + template + "{context} Question: {question} Helpful Answer: "
+llm_template = "Answer the question at the end. " + template + "Question: {question}. Helpful Answer: "
+rag_template = "Use the following pieces of context to answer the question at the end. " + template + "{context} Question: {question}. Helpful Answer: "
 
 LLM_CHAIN_PROMPT = PromptTemplate(input_variables = ["question"], template = llm_template)
 RAG_CHAIN_PROMPT = PromptTemplate(input_variables = ["context", "question"], template = rag_template)
@@ -130,13 +130,12 @@ def wandb_trace(rag_option, prompt, completion, status_msg, start_time_ms, end_t
             "model": config["model"],
             "temperature": config["temperature"],
         },
-        inputs = {"rag_option": rag_option,
+        inputs = {"rag_option": rag_option if (str(status_msg) == "") else "",
                   "prompt": str(prompt if (str(status_msg) == "") else ""), 
                   "prompt_template": str((llm_template if (rag_option == "Off") else rag_template) if (str(status_msg) == "") else "")},
         outputs = {"completion": str(completion)},
         start_time_ms = start_time_ms,
-        end_time_ms = end_time_ms,
-        #model_dict = {"llm": str(llm)}
+        end_time_ms = end_time_ms}
     )
     trace.log("test")
     wandb.finish()