Update app.py

app.py

@@ -117,7 +117,7 @@ def rag_chain(llm, prompt, db):
 def wandb_log(prompt, completion, rag_option):
     wandb.login(key = wandb_api_key)
     wandb.init(project = "openai-llm-rag", config = config)
-    wandb.log({"prompt": prompt, "completion": completion, "rag_option": rag_option})
+    wandb.log({"prompt": prompt, "completion": str(completion), "rag_option": rag_option})
     wandb.finish()
 
 def invoke(openai_api_key, rag_option, prompt):
@@ -137,20 +137,17 @@ def invoke(openai_api_key, rag_option, prompt):
             db = document_retrieval_chroma(llm, prompt)
             completion = rag_chain(llm, prompt, db)
             result = completion["result"]
-            completion = json.dumps(completion)
         elif (rag_option == "MongoDB"):
             #splits = document_loading_splitting()
             #document_storage_mongodb(splits)
             db = document_retrieval_mongodb(llm, prompt)
             completion = rag_chain(llm, prompt, db)
             result = completion["result"]
-            completion = json.dumps(completion)
         else:
             result = llm_chain(llm, prompt)
             completion = result
     except Exception as e:
         raise gr.Error(e)
-    print(completion)
     wandb_log(prompt, completion, rag_option)
     return result