feat(rag): make embedding and retrieving available

pkg/api/http/controller/groups/files.py

@@ -34,8 +34,9 @@ class FilesRouterGroup(group.RouterGroup):
 
             file_bytes = await asyncio.to_thread(file.stream.read)
             extension = file.filename.split('.')[-1]
+            file_name = file.filename.split('.')[0]
 
-            file_key = str(uuid.uuid4()) + '.' + extension
+            file_key = file_name + '_' + str(uuid.uuid4())[:8] + '.' + extension
             # save file to storage
             await self.ap.storage_mgr.storage_provider.save(file_key, file_bytes)
             return self.success(

pkg/api/http/controller/groups/knowledge/base.py

@@ -72,3 +72,13 @@ class KnowledgeBaseRouterGroup(group.RouterGroup):
         async def delete_specific_file_in_kb(file_id: str, knowledge_base_uuid: str) -> str:
             await self.ap.knowledge_service.delete_file(knowledge_base_uuid, file_id)
             return self.success({})
+
+        @self.route(
+            '/<knowledge_base_uuid>/retrieve',
+            methods=['POST'],
+        )
+        async def retrieve_knowledge_base(knowledge_base_uuid: str) -> str:
+            json_data = await quart.request.json
+            query = json_data.get('query')
+            results = await self.ap.knowledge_service.retrieve_knowledge_base(knowledge_base_uuid, query)
+            return self.success(data={'results': results})