Skip to content

Commit ea63d70

Browse files
committed
feat: check sql None
1 parent 57f4ded commit ea63d70

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

backend/apps/chat/task/llm.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -672,6 +672,8 @@ def generate_chart(self):
672672
self.chart_message]).decode())
673673
def check_sql(self, res: str) -> tuple[any]:
674674
json_str = extract_nested_json(res)
675+
if json_str is None:
676+
raise Exception("Cannot parse sql from answer")
675677
data: dict = orjson.loads(json_str)
676678
sql = ''
677679
message = ''

0 commit comments

Comments
 (0)