openai-llm-rag

Build error

bstraehle commited on Dec 21, 2023

Commit

8573a63

1 Parent(s): 511bcdc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -141,7 +141,7 @@ def wandb_trace(rag_option, prompt, completion, result, chain, err_msg, start_ti
         },
         model_dict = {"llm_client": str(chain.llm.client) if (rag_option == RAG_OFF) else str(chain.combine_documents_chain.llm_chain.llm.client),
                       "chain_prompt": str(chain.prompt) if (rag_option == RAG_OFF) else str(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

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