diff --git a/backend/apps/chat/task/llm.py b/backend/apps/chat/task/llm.py index 8cc1f470..d8a422ea 100644 --- a/backend/apps/chat/task/llm.py +++ b/backend/apps/chat/task/llm.py @@ -1040,6 +1040,7 @@ class LLMService: subsql) real_execute_sql = assistant_dynamic_sql + print(tables) result = self.execute_sql(sql=real_execute_sql, tables=tables) print(result) self.save_sql_data(data_obj=result) diff --git a/backend/apps/db/db.py b/backend/apps/db/db.py index a1da1492..3a4dc94f 100644 --- a/backend/apps/db/db.py +++ b/backend/apps/db/db.py @@ -420,6 +420,7 @@ def exec_sql(ds: CoreDatasource | AssistantOutDsSchema, sql: str, origin_column= except Exception as ex: raise ParseSQLResultError(str(ex)) elif ds.type == 'es': + print(table_name) try: if table_name and table_name[0]: res, columns = get_es_data(conf, sql, table_name[0]) @@ -431,6 +432,7 @@ def exec_sql(ds: CoreDatasource | AssistantOutDsSchema, sql: str, origin_column= enumerate(tuple_item)} for tuple_item in res ] + print(result_list) return {"fields": columns, "data": result_list, "sql": bytes.decode(base64.b64encode(bytes(sql, 'utf-8')))} except Exception as ex: