openai-llm-rag

Build error

bstraehle commited on Dec 21, 2023

Commit

968a729

1 Parent(s): f45ad50

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -134,10 +134,13 @@ def wandb_trace(rag_option, prompt, completion, result, chain, err_msg, start_ti
             "temperature": config["temperature"],
         },
         inputs = {"rag_option": rag_option if (str(err_msg) == "") else "",
-                  "prompt": str(prompt if (str(err_msg) == "") else ""),
-                  "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)},
         start_time_ms = start_time_ms,
         end_time_ms = end_time_ms
     )

             "temperature": config["temperature"],
         },
         inputs = {"rag_option": rag_option if (str(err_msg) == "") else "",
+                  "prompt": str(prompt if (str(err_msg) == "") else "")},
+                  #"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
     )