openai-llm-rag

Build error

bstraehle commited on Dec 21, 2023

Commit

1affff8

1 Parent(s): 63fd66f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -139,10 +139,10 @@ def wandb_trace(rag_option, prompt, completion, result, chain, err_msg, start_ti
         model_dict = {"model_name": config["model_name"] if (str(err_msg) == "") else "",
                       "temperature": config["temperature"] if (str(err_msg) == "") else "",
                       "llm_client": (str(chain.llm.client) if (rag_option == RAG_OFF) else
-                                     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

         model_dict = {"model_name": config["model_name"] if (str(err_msg) == "") else "",
                       "temperature": config["temperature"] if (str(err_msg) == "") else "",
                       "llm_client": (str(chain.llm.client) if (rag_option == RAG_OFF) else
+                                     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