Update trace.py

trace.py

@@ -2,7 +2,16 @@ import wandb
 
 from wandb.sdk.data_types.trace_tree import Trace
 
-def wandb_trace(rag_option, prompt, completion, result, generation_info, llm_output, chain, err_msg, start_time_ms, end_time_ms):
+def wandb_trace(rag_option, 
+                prompt, 
+                completion, 
+                result, 
+                generation_info, 
+                llm_output, 
+                chain, 
+                err_msg, 
+                start_time_ms, 
+                end_time_ms):
     wandb.init(project = "openai-llm-rag")
     
     trace = Trace(
@@ -37,4 +46,5 @@ def wandb_trace(rag_option, prompt, completion, result, generation_info, llm_out
     )
     
     trace.log("evaluation")
+                    
     wandb.finish()