bstraehle commited on
Commit
09018fc
·
1 Parent(s): b4f203e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -7
app.py CHANGED
@@ -36,8 +36,6 @@ def invoke(openai_api_key, rag_option, prompt):
36
  chain = None
37
  completion = ""
38
  result = ""
39
- generation_info = ""
40
- llm_output = ""
41
  cb = ""
42
  err_msg = ""
43
 
@@ -49,11 +47,9 @@ def invoke(openai_api_key, rag_option, prompt):
49
 
50
  if (completion.generations[0] != None and completion.generations[0][0] != None):
51
  result = completion.generations[0][0].text
52
- generation_info = completion.generations[0][0].generation_info
53
-
54
- llm_output = completion.llm_output
55
  else:
56
  completion, chain, cb = rag_chain(config, openai_api_key, rag_option, prompt)
 
57
  result = completion["result"]
58
  except Exception as e:
59
  err_msg = e
@@ -67,8 +63,6 @@ def invoke(openai_api_key, rag_option, prompt):
67
  prompt,
68
  completion,
69
  result,
70
- generation_info,
71
- llm_output,
72
  chain,
73
  cb,
74
  err_msg,
 
36
  chain = None
37
  completion = ""
38
  result = ""
 
 
39
  cb = ""
40
  err_msg = ""
41
 
 
47
 
48
  if (completion.generations[0] != None and completion.generations[0][0] != None):
49
  result = completion.generations[0][0].text
 
 
 
50
  else:
51
  completion, chain, cb = rag_chain(config, openai_api_key, rag_option, prompt)
52
+
53
  result = completion["result"]
54
  except Exception as e:
55
  err_msg = e
 
63
  prompt,
64
  completion,
65
  result,
 
 
66
  chain,
67
  cb,
68
  err_msg,