Update app.py

app.py

@@ -4,11 +4,11 @@ import os, time
 from dotenv import load_dotenv, find_dotenv
 
 from rag import run_llm_chain, run_rag_chain, run_rag_batch
-from trace import wandb_trace
+from trace import trace_wandb
 
 _ = load_dotenv(find_dotenv())
 
-RUN_RAG_BATCH = False # document loading, splitting, storage
+RUN_RAG_BATCH = False # load, split, embed, and store documents
 
 config = {
     "chunk_overlap": 150,       # split documents
@@ -58,7 +58,7 @@ def invoke(openai_api_key, rag_option, prompt):
     finally:
         end_time_ms = round(time.time() * 1000)
         
-        wandb_trace(config,
+        trace_wandb(config,
                     rag_option == RAG_OFF, 
                     prompt, 
                     completion,