Khachatur Mirijanyan
commited on
Commit
·
b5dc7c4
1
Parent(s):
13fcfb3
Add query checker to db chain
Browse files
app.py
CHANGED
@@ -78,7 +78,8 @@ def answer_question(query):
|
|
78 |
db_chain = SQLDatabaseChain.from_llm(llm, db,
|
79 |
prompt=PROMPT,
|
80 |
verbose=True,
|
81 |
-
return_intermediate_steps=True
|
|
|
82 |
)
|
83 |
result = db_chain(q)
|
84 |
return result['result']
|
|
|
78 |
db_chain = SQLDatabaseChain.from_llm(llm, db,
|
79 |
prompt=PROMPT,
|
80 |
verbose=True,
|
81 |
+
return_intermediate_steps=True,
|
82 |
+
use_query_checker=True
|
83 |
)
|
84 |
result = db_chain(q)
|
85 |
return result['result']
|