From c5ab4a8461e1b995d5c97c6b02a6abce05403960 Mon Sep 17 00:00:00 2001 From: fit2cloud-chenyw Date: Wed, 30 Jul 2025 15:39:26 +0800 Subject: [PATCH] perf: Disable model validate --- backend/apps/chat/task/llm.py | 11 +++++++++-- backend/common/core/config.py | 1 - frontend/src/views/system/model/Model.vue | 3 +++ 3 files changed, 12 insertions(+), 3 deletions(-) diff --git a/backend/apps/chat/task/llm.py b/backend/apps/chat/task/llm.py index fcaf3bed..895aea5e 100644 --- a/backend/apps/chat/task/llm.py +++ b/backend/apps/chat/task/llm.py @@ -836,7 +836,8 @@ class LLMService: self.ds.id) if self.out_ds_instance else get_table_schema(session=self.session, current_user=self.current_user, ds=self.ds) - + else: + self.validate_history_ds() # generate sql sql_res = self.generate_sql() full_sql_text = '' @@ -1025,7 +1026,13 @@ class LLMService: finally: # end pass - + + def validate_history_ds(self): + _invalid_ds = False + _ds = self.ds + _assistant = self.current_assistant + if _invalid_ds: + yield orjson.dumps({'content': 'ds is invalid', 'type': 'error'}).decode() + '\n\n' def execute_sql_with_db(db: SQLDatabase, sql: str) -> str: """Execute SQL query using SQLDatabase diff --git a/backend/common/core/config.py b/backend/common/core/config.py index 5d16c61a..9de1279b 100644 --- a/backend/common/core/config.py +++ b/backend/common/core/config.py @@ -9,7 +9,6 @@ from pydantic import ( ) from pydantic_core import MultiHostUrl from pydantic_settings import BaseSettings, SettingsConfigDict -from typing_extensions import Self def parse_cors(v: Any) -> list[str] | str: diff --git a/frontend/src/views/system/model/Model.vue b/frontend/src/views/system/model/Model.vue index 32a9ed2e..8c991f1a 100644 --- a/frontend/src/views/system/model/Model.vue +++ b/frontend/src/views/system/model/Model.vue @@ -71,6 +71,8 @@ const defaultModelListWithSearch = computed(() => { }) const modelCheckHandler = async (item: any) => { + console.log(item) + /** const response = await modelApi.check(item) const reader = response.body.getReader() const decoder = new TextDecoder() @@ -110,6 +112,7 @@ const modelCheckHandler = async (item: any) => { currentRef?.showErrorMask(checkMsg) } }) + */ } const duplicateName = async (item: any) => { const res = await modelApi.queryAll()