openai-llm-rag

Build error

bstraehle commited on Dec 24, 2023

Commit

b4f203e

1 Parent(s): c6bc22c

Update trace.py

Files changed (1) hide show

trace.py CHANGED Viewed

@@ -33,7 +33,7 @@ def wandb_trace(config,
                   "source_documents": "" if (is_rag_off) else str([doc.metadata["source"] for doc in completion["source_documents"]]),
                  } if (str(err_msg) == "") else {},
         outputs = {"result": result,
-                   "cb": cb,
                    "generation_info": str(generation_info),
                    "llm_output": str(llm_output),
                    "completion": str(completion),

                   "source_documents": "" if (is_rag_off) else str([doc.metadata["source"] for doc in completion["source_documents"]]),
                  } if (str(err_msg) == "") else {},
         outputs = {"result": result,
+                   "cb": str(cb),
                    "generation_info": str(generation_info),
                    "llm_output": str(llm_output),
                    "completion": str(completion),