Update app.py

app.py

@@ -42,9 +42,6 @@ config = {
 
 langchain.verbose = True
 
-os.environ["LANGCHAIN_WANDB_TRACING"] = "true"
-os.environ["WANDB_PROJECT"] = "openai-llm-rag"
-
 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: "
@@ -186,8 +183,7 @@ def invoke(openai_api_key, rag_option, prompt):
         raise gr.Error(e)
     finally:
         end_time_ms = round(time.time() * 1000)
-        #wandb_trace(rag_option, prompt, prompt_template, result, completion, chain_name, status_msg, start_time_ms, end_time_ms)
-        wandb.finish()
+        wandb_trace(rag_option, prompt, prompt_template, result, completion, chain_name, status_msg, start_time_ms, end_time_ms)
     return result
 
 gr.close_all()