Update app.py

app.py

@@ -113,14 +113,8 @@ def rag_chain(llm, prompt, db):
 
 def evaluate(prompt, completion, rag_option):
     wandb.login(key = wandb_api_key)
-    wandb.config["model"] = config.model
-    wandb.config["temperature"] = config.temperature
-    wandb.config["rag_option"] = rag_option
-    wandb.config["prompt"] = prompt
-    wandb.config["completion"] = completion
     wandb.init(project = "openai-llm-rag", config = config)
-    #config = wandb.config
-    wandb.log({"prompt": prompt, "completion": completion})
+    wandb.log({"prompt": prompt, "completion": completion, "rag_option": rag_option})
     wandb.logout()
 
 def invoke(openai_api_key, rag_option, prompt):
@@ -131,9 +125,9 @@ def invoke(openai_api_key, rag_option, prompt):
     if (prompt == ""):
         raise gr.Error("Prompt is required.")
     try:
-        llm = ChatOpenAI(model_name = config.model, 
+        llm = ChatOpenAI(model_name = config["model"], 
                          openai_api_key = openai_api_key, 
-                         temperature = config.temperature)
+                         temperature = config["temperature"])
         if (rag_option == "Chroma"):
             #splits = document_loading_splitting()
             #document_storage_chroma(splits)