Spaces:

bstraehle
/

rag

Running

bstraehle commited on Dec 21, 2023

Commit

315505c

1 Parent(s): 968a729

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -138,9 +138,9 @@ def wandb_trace(rag_option, prompt, completion, result, chain, err_msg, start_ti
                   #"prompt_template": str((llm_template if (rag_option == RAG_OFF) else rag_template) if (str(err_msg) == "") else "")},
         outputs = {"result": result if (str(err_msg) == "") else "",
                    "completion": str(completion)},
-        dict = {"llm_client": chain.llm.client if (rag_option == RAG_OFF) else chain.combine_documents_chain.llm_chain.llm.client,
-                "prompt_engineered": chain.prompt if (rag_option == RAG_OFF) else chain.combine_documents_chain.llm_chain.prompt,
-                "retriever": "" if (rag_option == RAG_OFF) else chain.retriever},
         start_time_ms = start_time_ms,
         end_time_ms = end_time_ms
     )

                   #"prompt_template": str((llm_template if (rag_option == RAG_OFF) else rag_template) if (str(err_msg) == "") else "")},
         outputs = {"result": result if (str(err_msg) == "") else "",
                    "completion": str(completion)},
+        model_dict = {"llm_client": chain.llm.client if (rag_option == RAG_OFF) else chain.combine_documents_chain.llm_chain.llm.client,
+                      "prompt_engineered": chain.prompt if (rag_option == RAG_OFF) else chain.combine_documents_chain.llm_chain.prompt,
+                      "retriever": "" if (rag_option == RAG_OFF) else chain.retriever},
         start_time_ms = start_time_ms,
         end_time_ms = end_time_ms
     )