bstraehle commited on
Commit
09ae5f0
·
1 Parent(s): a5827eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -124,7 +124,7 @@ def wandb_trace(rag_option, prompt, completion, chain, err_msg, start_time_ms, e
124
  if (str(err_msg) != ""):
125
  result = ""
126
  elif (rag_option == RAG_OFF):
127
- result = completion.generations[0][0]
128
  else:
129
  result = completion["result"]
130
  docs_meta = str([doc.metadata for doc in completion["source_documents"]])
@@ -186,7 +186,7 @@ def invoke(openai_api_key, rag_option, prompt):
186
  print(completion)
187
  print(completion.llm_output)
188
  print(completion.generations[0])
189
- result = completion.generations[0][0]
190
  except Exception as e:
191
  err_msg = e
192
  raise gr.Error(e)
 
124
  if (str(err_msg) != ""):
125
  result = ""
126
  elif (rag_option == RAG_OFF):
127
+ result = completion.generations[0][0].text
128
  else:
129
  result = completion["result"]
130
  docs_meta = str([doc.metadata for doc in completion["source_documents"]])
 
186
  print(completion)
187
  print(completion.llm_output)
188
  print(completion.generations[0])
189
+ result = completion.generations[0][0].text
190
  except Exception as e:
191
  err_msg = e
192
  raise gr.Error(e)