Update app.py

app.py

@@ -43,7 +43,7 @@ def invoke(openai_api_key, prompt, rag_option):
         elif (rag_option == RAG_LLAMAINDEX):
             rag = LlamaIndexRAG()
             rag.ingestion(config)
-    
+
     completion = ""
     result = ""
     callback = ""
@@ -53,14 +53,17 @@ def invoke(openai_api_key, prompt, rag_option):
         start_time_ms = round(time.time() * 1000)
 
         if (rag_option == RAG_LANGCHAIN):
+            prompt_template = ""
             rag = LangChainRAG()
             completion, chain, callback = rag.rag_chain(config, prompt)
 
             result = completion["result"]
         elif (rag_option == RAG_LLAMAINDEX):
+            prompt_template = ""
             rag = LlamaIndexRAG()
             result = rag.retrieval(config, prompt)
         else:
+            prompt_template = ""
             rag = LangChainRAG()
             completion, chain, callback = rag.llm_chain(config, prompt)
             
@@ -76,7 +79,8 @@ def invoke(openai_api_key, prompt, rag_option):
         
         trace_wandb(
             config,
-            rag_option, 
+            rag_option,
+            prompt_template,
             prompt, 
             completion, 
             result,