openai-llm-rag

Build error

bstraehle commited on Dec 21, 2023

Commit

5faffaa

1 Parent(s): 1affff8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -142,7 +142,7 @@ def wandb_trace(rag_option, prompt, completion, result, chain, err_msg, start_ti
                                      str(chain.combine_documents_chain.llm_chain.llm.client)) if (str(err_msg) == "") else "",
                       "chain_prompt": (str(chain.prompt) if (rag_option == RAG_OFF) else
                                        str(chain.combine_documents_chain.llm_chain.prompt)) if (str(err_msg) == "") else "",
-                      "retriever": "" (if (rag_option == RAG_OFF) else str(chain.retriever)) if (str(err_msg) == "") else "",
         },
         start_time_ms = start_time_ms,
         end_time_ms = end_time_ms

                                      str(chain.combine_documents_chain.llm_chain.llm.client)) if (str(err_msg) == "") else "",
                       "chain_prompt": (str(chain.prompt) if (rag_option == RAG_OFF) else
                                        str(chain.combine_documents_chain.llm_chain.prompt)) if (str(err_msg) == "") else "",
+                      "retriever": ("" if (rag_option == RAG_OFF) else str(chain.retriever)) if (str(err_msg) == "") else "",
         },
         start_time_ms = start_time_ms,
         end_time_ms = end_time_ms