修改知识库权限问题

dbgpt/app/knowledge/api.py

@@ -86,6 +86,7 @@ async def space_list(request: KnowledgeSpaceRequest, auth: Auth = Depends(FullAd
     try:
         #超级管理员返回全部知识库
         responses = knowledge_space_service.get_knowledge_space(request)
+        print(f"---1--->responses:{responses}")
         role_id = 0
 
         # 角色的id
@@ -100,10 +101,11 @@ async def space_list(request: KnowledgeSpaceRequest, auth: Auth = Depends(FullAd
         if role_id != 1:
             # 根据角色绑定的知识库id, 过滤掉其他知识库
             role_knowledge_list = await crud.RoleDal(auth.db).get_role_knowledge_list(role_id)
-            for know_id in role_knowledge_list:
-                print(f"----->knowledge space know_id:{know_id}:")
-                responses = [element for element in responses if element.id == know_id]
+            print(f"Role knowledge list: {role_knowledge_list}")
+            # 只保留id在role_knowledge_list中的元素
+            responses = [element for element in responses if element.id in role_knowledge_list]
 
+        print(f"---2--->responses:{responses}")
         return Result.succ(responses)
     except Exception as e:
         return Result.failed(code="E000X", msg=f"space list error {e}")