Update app.py

app.py

@@ -22,7 +22,10 @@ from dotenv import load_dotenv, find_dotenv
 _ = load_dotenv(find_dotenv())
 
 #openai.api_key = os.environ["OPENAI_API_KEY"]
-WANDB_API_KEY = os.environ["WANDB_API_KEY"]
+wandb_api_key = os.environ["WANDB_API_KEY"]
+
+os.environ["LANGCHAIN_WANDB_TRACING"] = "true"
+os.environ["WANDB_PROJECT"] = "openai-llm-rag"
 
 MONGODB_URI = os.environ["MONGODB_ATLAS_CLUSTER_URI"]
 client = MongoClient(MONGODB_URI)
@@ -132,7 +135,7 @@ def invoke(openai_api_key, rag_option, prompt):
     completion = ""
     #wandb.login(key = wandb_api_key)
     #wandb.init(project = "openai-llm-rag", config = config)
-    autolog({"project": "openai-llm-rag"})
+    autolog()
     try:
         llm = ChatOpenAI(model_name = config["model"], 
                          openai_api_key = openai_api_key,