Spaces:
Build error
Build error
Update app.py
Browse files
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,
|