Update trace.py

trace.py

@@ -15,9 +15,9 @@ def trace_wandb(config,
                 end_time_ms):
     wandb.init(project = "openai-llm-rag")
 
-    if (rag_option = RAG_LANGCHAIN):
+    if (rag_option == RAG_LANGCHAIN):
         prompt_template = os.environ["LANGCHAIN_TEMPLATE"]
-    elif (rag_option = RAG_LLAMAINDEX):
+    elif (rag_option == RAG_LLAMAINDEX):
         prompt_template = os.environ["LLAMAINDEX_TEMPLATE"]
     else:
         prompt_template = os.environ["TEMPLATE"]