diff --git a/backend/apps/system/crud/assistant.py b/backend/apps/system/crud/assistant.py index 696e2154..5d731847 100644 --- a/backend/apps/system/crud/assistant.py +++ b/backend/apps/system/crud/assistant.py @@ -170,7 +170,7 @@ def get_db_schema(self, ds_id: int, question: str, embedding: bool = True) -> st for table in ds.tables: i += 1 schema_table = '' - schema_table += f"# Table: {db_name}.{table.name}" if ds.type != "mysql" else f"# Table: {table.name}" + schema_table += f"# Table: {db_name}.{table.name}" if ds.type != "mysql" and ds.type != "es" else f"# Table: {table.name}" table_comment = table.comment if table_comment == '': schema_table += '\n[\n'