Update app.py

app.py

@@ -117,13 +117,13 @@ def rag_chain(llm, prompt, db):
 
 def wandb_trace(rag_option, prompt, completion, chain, status_msg, start_time_ms, end_time_ms):
     if (chain != None):
-        if (type(chain).__name__ == "LLMChain")
-        print("1=" + chain.llm)
-        print("2=" + chain.prompt)
-        if (type(chain).__name__ == "RetrievalQA")
-        print("3=" + chain.llm)
-        print("4=" + chain.chain_type_kwargs)
-        print("5=" + chain.retriever)
+        if (type(chain).__name__ == "LLMChain"):
+            print("1=" + chain.llm)
+            print("2=" + chain.prompt)
+        if (type(chain).__name__ == "RetrievalQA"):
+            print("3=" + chain.llm)
+            print("4=" + chain.chain_type_kwargs)
+            print("5=" + chain.retriever)
     wandb.init(project = "openai-llm-rag")
     if (rag_option == "Off" or str(status_msg) != ""):
         result = completion