From bd9331ce62f8aea92dbe63757b8c38696464ff87 Mon Sep 17 00:00:00 2001
From: Junyan Qin <rockchinq@gmail.com>
Date: Fri, 11 Jul 2025 20:57:09 +0800
Subject: [PATCH] fix: kb get api format

---
 .../http/controller/groups/knowledge/base.py  | 24 +++++++++++--------
 1 file changed, 14 insertions(+), 10 deletions(-)

diff --git a/pkg/api/http/controller/groups/knowledge/base.py b/pkg/api/http/controller/groups/knowledge/base.py
index b5a48d29..70cf2b0c 100644
--- a/pkg/api/http/controller/groups/knowledge/base.py
+++ b/pkg/api/http/controller/groups/knowledge/base.py
@@ -24,7 +24,9 @@ class KnowledgeBaseRouterGroup(group.RouterGroup):
             elif quart.request.method == 'POST':
                 json_data = await quart.request.json
                 knowledge_base_uuid = await self.ap.knowledge_base_service.create_knowledge_base(
-                    json_data.get('name'), json_data.get('description'), json_data.get('embedding_model_uuid')
+                    json_data.get('name'),
+                    json_data.get('description'),
+                    json_data.get('embedding_model_uuid'),
                 )
                 return self.success(data={'uuid': knowledge_base_uuid})
 
@@ -35,20 +37,22 @@ class KnowledgeBaseRouterGroup(group.RouterGroup):
         )
         async def handle_specific_knowledge_base(knowledge_base_uuid: str) -> str:
             if quart.request.method == 'GET':
-                knowledge_base = await self.ap.knowledge_base_service.get_knowledge_base_by_id(int(knowledge_base_uuid))
+                knowledge_base = await self.ap.knowledge_base_service.get_knowledge_base_by_id(knowledge_base_uuid)
 
                 if knowledge_base is None:
                     return self.http_status(404, -1, 'knowledge base not found')
 
                 return self.success(
                     data={
-                        'name': knowledge_base.name,
-                        'description': knowledge_base.description,
-                        'uuid': knowledge_base.id,
-                    },
+                        'base': {
+                            'name': knowledge_base.name,
+                            'description': knowledge_base.description,
+                            'uuid': knowledge_base.id,
+                        },
+                    }
                 )
             elif quart.request.method == 'DELETE':
-                await self.ap.knowledge_base_service.delete_kb_by_id(int(knowledge_base_uuid))
+                await self.ap.knowledge_base_service.delete_kb_by_id(knowledge_base_uuid)
                 return self.success({})
 
         @self.route(
@@ -57,7 +61,7 @@ class KnowledgeBaseRouterGroup(group.RouterGroup):
             endpoint='get_knowledge_base_files',
         )
         async def get_knowledge_base_files(knowledge_base_uuid: str) -> str:
-            files = await self.ap.knowledge_base_service.get_files_by_knowledge_base(int(knowledge_base_uuid))
+            files = await self.ap.knowledge_base_service.get_files_by_knowledge_base(knowledge_base_uuid)
             return self.success(
                 data={
                     'files': [
@@ -77,7 +81,7 @@ class KnowledgeBaseRouterGroup(group.RouterGroup):
             endpoint='delete_specific_file_in_kb',
         )
         async def delete_specific_file_in_kb(file_id: str) -> str:
-            await self.ap.knowledge_base_service.delete_data_by_file_id(int(file_id))
+            await self.ap.knowledge_base_service.delete_data_by_file_id(file_id)
             return self.success({})
 
         @self.route(
@@ -95,5 +99,5 @@ class KnowledgeBaseRouterGroup(group.RouterGroup):
                 return self.http_status(400, -1, 'File ID is required')
 
             # 调用服务层方法将文件与知识库关联
-            await self.ap.knowledge_base_service.relate_file_id_with_kb(int(knowledge_base_uuid), int(file_id))
+            await self.ap.knowledge_base_service.relate_file_id_with_kb(knowledge_base_uuid, file_id)
             return self.success({})