klinsmaya/SQLBot
1
0
Fork 0
mirror of https://gh-proxy.org/https://github.com/klinsmaya/SQLBot.git synced 2026-11-08 01:06:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: improve docs

Browse Source 
#1183
This commit is contained in:
ulleo
2026-06-10 11:45:35 +08:00
committed by ulleo
parent 2da3ae4111
commit b472bc0977
2 changed files with 19 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/chat/api/chat.py
+4 -3
View File
@@ -16,7 +16,7 @@ from apps.chat.curd.chat import delete_chat_with_user, get_chart_data_with_user,
format_json_data, format_json_list_data, get_chart_config, list_recent_questions, get_chat as get_chat_exec, \
rename_chat_with_user, get_chat_log_history, get_chart_data_with_user_live
from apps.chat.models.chat_model import CreateChat, ChatRecord, RenameChat, ChatQuestion, AxisObj, QuickCommand, \
ChatInfo, Chat, ChatFinishStep
ChatInfo, Chat, ChatFinishStep, ChatQuestionBase
from apps.chat.task.llm import LLMService
from apps.swagger.i18n import PLACEHOLDER_PREFIX
from apps.system.schemas.permission import SqlbotPermission, require_permissions
@@ -269,9 +269,10 @@ def find_base_question(record_id: int, session: SessionDep):
@router.post("/question", summary=f"{PLACEHOLDER_PREFIX}ask_question")
@require_permissions(permission=SqlbotPermission(type='chat', keyExpression="request_question.chat_id"))
async def question_answer(session: SessionDep, current_user: CurrentUser, request_question: ChatQuestion,
async def question_answer(session: SessionDep, current_user: CurrentUser, request_question: ChatQuestionBase,
current_assistant: CurrentAssistant):
return await question_answer_inner(session, current_user, request_question, current_assistant, embedding=True)
question = ChatQuestion(chat_id=request_question.chat_id, question=request_question.question)
return await question_answer_inner(session, current_user, question, current_assistant, embedding=True)
async def question_answer_inner(session: SessionDep, current_user: CurrentUser, request_question: ChatQuestion,
backend/apps/chat/models/chat_model.py
+15 -7
View File
@@ -250,16 +250,18 @@ class AiModelQuestion(BaseModel):
_example_answer_3 = _sql_template['example_answer_3_with_limit'] if enable_query_limit else _sql_template[
'example_answer_3']
templates['system'] = _base_template['system'].format(lang=self.lang, process_check=_process_check, sqlbot_name=self.sqlbot_name)
templates['system'] = _base_template['system'].format(lang=self.lang, process_check=_process_check,
sqlbot_name=self.sqlbot_name)
templates['rules'] = _base_template['generate_rules'].format(lang=self.lang,
sqlbot_name = self.sqlbot_name,
sqlbot_name=self.sqlbot_name,
base_sql_rules=_base_sql_rules,
basic_sql_examples=_sql_examples,
example_engine=_example_engine,
example_answer_1=_example_answer_1,
example_answer_2=_example_answer_2,
example_answer_3=_example_answer_3)
templates['schema'] = _base_template['generate_basic_info'].format(engine=self.engine, schema=self.db_schema, sample_data=self.sample_data)
templates['schema'] = _base_template['generate_basic_info'].format(engine=self.engine, schema=self.db_schema,
sample_data=self.sample_data)
if self.terminologies:
templates['terminologies'] = _base_template['generate_terminologies_info'].format(
@@ -303,7 +305,8 @@ class AiModelQuestion(BaseModel):
return get_analysis_template()['user'].format(fields=self.fields, data=self.data)
def predict_sys_question(self):
return get_predict_template()['system'].format(lang=self.lang, custom_prompt=self.custom_prompt, sqlbot_name=self.sqlbot_name)
return get_predict_template()['system'].format(lang=self.lang, custom_prompt=self.custom_prompt,
sqlbot_name=self.sqlbot_name)
def predict_user_question(self):
return get_predict_template()['user'].format(fields=self.fields, data=self.data)
@@ -315,14 +318,16 @@ class AiModelQuestion(BaseModel):
return get_datasource_template()['user'].format(lang=self.lang, question=self.question, data=datasource_list)
def guess_sys_question(self, articles_number: int = 4):
return get_guess_question_template()['system'].format(lang=self.lang, articles_number=articles_number, sqlbot_name=self.sqlbot_name)
return get_guess_question_template()['system'].format(lang=self.lang, articles_number=articles_number,
sqlbot_name=self.sqlbot_name)
def guess_user_question(self, old_questions: str = "[]"):
return get_guess_question_template()['user'].format(question=self.question, schema=self.db_schema,
old_questions=old_questions)
def filter_sys_question(self):
return get_permissions_template()['system'].format(lang=self.lang, engine=self.engine, sqlbot_name=self.sqlbot_name)
return get_permissions_template()['system'].format(lang=self.lang, engine=self.engine,
sqlbot_name=self.sqlbot_name)
def filter_user_question(self):
return get_permissions_template()['user'].format(sql=self.sql, filter=self.filter)
@@ -353,9 +358,12 @@ class ChatStart(BaseModel):
password: str = Body(description='密码')
class McpQuestion(BaseModel):
class ChatQuestionBase(BaseModel):
question: str = Body(description='用户提问')
chat_id: int = Body(description='会话ID')
class McpQuestion(ChatQuestionBase):
token: str = Body(description='token')
stream: Optional[bool] = Body(description='是否流式输出，默认为true开启, 关闭false则返回JSON对象', default=True)
lang: Optional[str] = Body(description='语言：zh-CN|zh-TW|en|ko-KR', default='zh-CN')