15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 01:33:05 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

feat: add optional user_id parameter to tool queries and serializers

This commit is contained in:
CaptainB 2025-05-15 11:02:41 +08:00
parent 74fac81c7b
commit e24f211674
3 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apps/knowledge/serializers/knowledge_folder.py
View File

@ -15,8 +15,6 @@ class KnowledgeFolderTreeSerializer(serializers.ModelSerializer):
class KnowledgeFolderFlatSerializer(serializers.ModelSerializer):
children = serializers.SerializerMethodField()
class Meta:
model = KnowledgeFolder
fields = ['id', 'name', 'desc', 'user_id', 'workspace_id', 'parent_id']

7
apps/tools/api/tool.py
View File

@ -194,6 +194,13 @@ class ToolPageAPI(ToolReadAPI):
location='query',
required=True,
),
OpenApiParameter(
name="user_id",
description="创建者id",
type=OpenApiTypes.STR,
location='query',
required=False,
),
OpenApiParameter(
name="tool_type",
description="工具类型",

3
apps/tools/serializers/tool.py
View File

@ -392,6 +392,7 @@ class ToolTreeSerializer(serializers.Serializer):
workspace_id = serializers.CharField(required=True, label=_('workspace id'))
folder_id = serializers.CharField(required=True, label=_('folder id'))
name = serializers.CharField(required=False, allow_null=True, allow_blank=True, label=_('tool name'))
user_id = serializers.CharField(required=False, allow_null=True, allow_blank=True, label=_('user id'))
tool_type = serializers.CharField(required=True, label=_('tool type'))
def page_tool(self, current_page: int, page_size: int):
@ -408,10 +409,12 @@ class ToolTreeSerializer(serializers.Serializer):
tools = QuerySet(Tool).filter(Q(workspace_id=self.data.get('workspace_id')) &
Q(folder_id__in=all_folders) &
Q(tool_type=self.data.get('tool_type')) &
Q(user_id=self.data.get('user_id')) &
Q(name__contains=self.data.get('name')))
else:
tools = QuerySet(Tool).filter(Q(workspace_id=self.data.get('workspace_id')) &
Q(folder_id__in=all_folders) &
Q(user_id=self.data.get('user_id')) &
Q(tool_type=self.data.get('tool_type')))
return page_search(current_page, page_size, tools, lambda record: ToolModelSerializer(record).data)