fix: 【知识库】- 知识库能关联非自己的应用

This commit is contained in:
shaohuzhang1 2024-01-31 17:01:56 +08:00
parent d31c6399f6
commit 057decf867
2 changed files with 3 additions and 2 deletions

View File

@ -377,7 +377,7 @@ class ApplicationSerializer(serializers.Serializer):
application = QuerySet(Application).get(id=self.data.get("application_id"))
return select_list(get_file_content(
os.path.join(PROJECT_DIR, "apps", "application", 'sql', 'list_application_dataset.sql')),
[self.data.get('user_id'), application.user_id, self.data.get('user_id')])
[self.data.get('user_id') if self.data.get('user_id')==str(application.user_id) else None, application.user_id, self.data.get('user_id')])
class ApplicationKeySerializerModel(serializers.ModelSerializer):
class Meta:

View File

@ -585,7 +585,8 @@ class DataSetSerializers(serializers.ModelSerializer):
dataset = QuerySet(DataSet).get(id=self.data.get("id"))
return select_list(get_file_content(
os.path.join(PROJECT_DIR, "apps", "dataset", 'sql', 'list_dataset_application.sql')),
[self.data.get('user_id'), dataset.user_id, self.data.get('user_id')])
[self.data.get('user_id') if self.data.get('user_id') == str(dataset.user_id) else None,
dataset.user_id, self.data.get('user_id')])
def one(self, user_id, with_valid=True):
if with_valid: