feat: support Elasticsearch datasource #108

Author: XiaJunjie2020

backend/apps/chat/task/llm.py

@@ -1040,6 +1040,7 @@ def run_task(self, in_chat: bool = True):
                                                                           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)

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: