diff --git a/pilot/scene/base_chat.py b/pilot/scene/base_chat.py
index 726db8304..161f84953 100644
--- a/pilot/scene/base_chat.py
+++ b/pilot/scene/base_chat.py
@@ -231,7 +231,6 @@ class BaseChat(ABC):
except StopAsyncIteration:
break
-
def _blocking_nostream_call(self):
logger.warn(
"_blocking_nostream_call is only temporarily used in webserver and will be deleted soon, please use nostream_call to replace it for higher performance"
@@ -242,7 +241,6 @@ class BaseChat(ABC):
finally:
loop.close()
-
def call(self):
if self.prompt_template.stream_out:
yield self._blocking_stream_call()
diff --git a/pilot/scene/chat_data/chat_excel/excel_learning/verify_sql.py b/pilot/scene/chat_data/chat_excel/excel_learning/verify_sql.py
new file mode 100644
index 000000000..af5c566af
--- /dev/null
+++ b/pilot/scene/chat_data/chat_excel/excel_learning/verify_sql.py
@@ -0,0 +1,29 @@
+import re
+import sqlparse
+
+
+def add_quotes(sql, column_names=[]):
+ parsed = sqlparse.parse(sql)
+ for stmt in parsed:
+ for token in stmt.tokens:
+ deep_quotes(token, column_names)
+ return str(parsed[0])
+
+
+def deep_quotes(token, column_names=[]):
+ if hasattr(token, "tokens"):
+ for token_child in token.tokens:
+ deep_quotes(token_child, column_names)
+ else:
+ if token.ttype == sqlparse.tokens.Name:
+ if len(column_names) > 0:
+ if token.value in column_names:
+ token.value = f'"{token.value}"'
+ else:
+ token.value = f'"{token.value}"'
+
+
+if __name__ == "__main__":
+ sql = "SELECT SUM(预算) AS 总预算 FROM yhj-zx"
+ new_sql = add_quotes(sql, ["预算"])
+ print(new_sql)
diff --git a/pilot/server/static/404.html b/pilot/server/static/404.html
index b9b8281be..954920cda 100644
--- a/pilot/server/static/404.html
+++ b/pilot/server/static/404.html
@@ -1 +1 @@
-404: This page could not be found
404
This page could not be found.
\ No newline at end of file
+404: This page could not be found
404
This page could not be found.
\ No newline at end of file
diff --git a/pilot/server/static/404/index.html b/pilot/server/static/404/index.html
index b9b8281be..954920cda 100644
--- a/pilot/server/static/404/index.html
+++ b/pilot/server/static/404/index.html
@@ -1 +1 @@
-404: This page could not be found
404
This page could not be found.
\ No newline at end of file
+404: This page could not be found