Spaces:

bstraehle
/

rag

Running

bstraehle commited on Dec 17, 2023

Commit

96115b9

1 Parent(s): 84e076a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -145,6 +145,7 @@ def invoke(openai_api_key, rag_option, prompt):
             result = completion["result"]
         else:
             result = llm_chain(llm, prompt)
     except Exception as e:
         raise gr.Error(e)
     wandb_log(prompt, completion, rag_option)

             result = completion["result"]
         else:
             result = llm_chain(llm, prompt)
+            completion = result
     except Exception as e:
         raise gr.Error(e)
     wandb_log(prompt, completion, rag_option)