openai-llm-rag

Build error

bstraehle commited on Dec 20, 2023

Commit

b220060

1 Parent(s): 64931b6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -116,8 +116,8 @@ def rag_chain(llm, prompt, db):
     return completion, rag_chain
 def wandb_trace(rag_option, prompt, completion, chain, status_msg, start_time_ms, end_time_ms):
-    print(chain.inputKey)
-    print(chain.outputKey)
     print(chain.retriever)
     wandb.init(project = "openai-llm-rag")
     if (rag_option == "Off" or str(status_msg) != ""):

     return completion, rag_chain
 def wandb_trace(rag_option, prompt, completion, chain, status_msg, start_time_ms, end_time_ms):
+    #print(chain.inputKey)
+    #print(chain.outputKey)
     print(chain.retriever)
     wandb.init(project = "openai-llm-rag")
     if (rag_option == "Off" or str(status_msg) != ""):