Spaces:

bstraehle
/

rag

Running

bstraehle commited on Dec 20, 2023

Commit

dc7f3cd

•

1 Parent(s): 8700249

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -126,7 +126,7 @@ def wandb_trace(rag_option, prompt, completion, status_msg, start_time_ms, end_t
         document_1 = completion["source_documents"][1]
         document_2 = completion["source_documents"][2]
     trace = Trace(
-        kind = "chain",
         name = "LLMChain" if (rag_option == "Off") else "RetrievalQA",
         status_code = "SUCCESS" if (str(status_msg) == "") else "ERROR",
         status_message = str(status_msg),

         document_1 = completion["source_documents"][1]
         document_2 = completion["source_documents"][2]
     trace = Trace(
+        kind = "llm",
         name = "LLMChain" if (rag_option == "Off") else "RetrievalQA",
         status_code = "SUCCESS" if (str(status_msg) == "") else "ERROR",
         status_message = str(status_msg),