feat: data permission

backend/apps/chat/task/llm.py

@@ -497,7 +497,7 @@ class LLMService:
             filters.append({"table": table.table_name, "filter": wheres})
 
         filter = json.dumps(filters, ensure_ascii=False)
-        # filter = f"""[{{"table":"{tables[0]}","filter":"省份 = '广东省' or 销售额（万元） > 10000"}}]"""  # todo get filters
+
         self.chat_question.sql = sql
         self.chat_question.filter = filter
         msg: List[Union[BaseMessage, dict[str, Any]]] = []
@@ -746,7 +746,7 @@ def run_task(llm_service: LLMService, in_chat: bool = True):
         # todo row permission
         sql_json_str = extract_nested_json(full_sql_text)
         data = orjson.loads(sql_json_str)
-        sql_result = llm_service.generate_filter(data['sql'], data['tables'])
+        sql_result = llm_service.generate_filter(data['sql'], data['tables']) # maybe no sql and tables
         print(sql_result)
         sql = llm_service.check_save_sql(res=sql_result)
         # sql = llm_service.check_save_sql(res=full_sql_text)

frontend/src/views/ds/DatasourceForm.vue

@@ -34,7 +34,11 @@ const tableList = ref<any>([])
 const excelUploadSuccess = ref(false)
 const tableListLoading = ref(false)
 const token = wsCache.get('user.token')
-const headers = ref<any>({ 'X-SQLBOT-TOKEN': `Bearer ${token}` })
+// eslint-disable-next-line @typescript-eslint/ban-ts-comment
+// @ts-expect-error
+// eslint-disable-next-line no-undef
+const request_key = LicenseGenerator.generate()
+const headers = ref<any>({ 'X-SQLBOT-TOKEN': `Bearer ${token}`, 'X-SQLBOT-KEY': request_key })
 const dialogTitle = ref('')
 const getUploadURL = import.meta.env.VITE_API_BASE_URL + '/datasource/uploadExcel'
 const saveLoading = ref<boolean>(false)