feat: crud of platform/bots

pkg/api/http/service/bot.py

@@ -0,0 +1,70 @@
+from __future__ import annotations
+
+import uuid
+import datetime
+import sqlalchemy
+
+from ....core import app
+from ....entity.persistence import bot as persistence_bot
+
+
+class BotService:
+    """机器人服务"""
+
+    ap: app.Application
+
+    def __init__(self, ap: app.Application) -> None:
+        self.ap = ap
+
+    async def get_bots(self) -> list[dict]:
+        """获取所有机器人"""
+        result = await self.ap.persistence_mgr.execute_async(
+            sqlalchemy.select(persistence_bot.Bot)
+        )
+
+        bots = result.all()
+
+        return [
+            self.ap.persistence_mgr.serialize_model(persistence_bot.Bot, bot)
+            for bot in bots
+        ]
+    
+    async def get_bot(self, bot_uuid: str) -> dict | None:
+        """获取机器人"""
+        result = await self.ap.persistence_mgr.execute_async(
+            sqlalchemy.select(persistence_bot.Bot).where(persistence_bot.Bot.uuid == bot_uuid)
+        )
+        
+        bot = result.first()
+
+        if bot is None:
+            return None
+
+        return self.ap.persistence_mgr.serialize_model(persistence_bot.Bot, bot)
+
+    async def create_bot(self, bot_data: dict) -> str:
+        """创建机器人"""
+        bot_data['uuid'] = str(uuid.uuid4())
+        await self.ap.persistence_mgr.execute_async(
+            sqlalchemy.insert(persistence_bot.Bot).values(bot_data)
+        )
+        # TODO: 加载机器人到机器人管理器
+        return bot_data['uuid']
+
+    async def update_bot(self, bot_uuid: str, bot_data: dict) -> None:
+        """更新机器人"""
+        if 'uuid' in bot_data:
+            del bot_data['uuid']
+        await self.ap.persistence_mgr.execute_async(
+            sqlalchemy.update(persistence_bot.Bot).values(bot_data).where(persistence_bot.Bot.uuid == bot_uuid)
+        )
+        # TODO: 加载机器人到机器人管理器
+
+    async def delete_bot(self, bot_uuid: str) -> None:
+        """删除机器人"""
+        await self.ap.persistence_mgr.execute_async(
+            sqlalchemy.delete(persistence_bot.Bot).where(persistence_bot.Bot.uuid == bot_uuid)
+        )
+        # TODO: 从机器人管理器中删除机器人
+
+

pkg/api/http/service/model.py

@@ -26,7 +26,7 @@ class ModelsService:
             for model in models
         ]
     
-    async def create_llm_model(self, model_data: dict) -> None:
+    async def create_llm_model(self, model_data: dict) -> str:
 
         model_data['uuid'] = str(uuid.uuid4())
 
@@ -37,6 +37,8 @@ class ModelsService:
         )
         await self.ap.model_mgr.load_llm_model(model_data)
 
+        return model_data['uuid']
+
     async def get_llm_model(self, model_uuid: str) -> dict | None:
         result = await self.ap.persistence_mgr.execute_async(
             sqlalchemy.select(persistence_model.LLMModel).where(persistence_model.LLMModel.uuid == model_uuid)
@@ -50,6 +52,9 @@ class ModelsService:
         return self.ap.persistence_mgr.serialize_model(persistence_model.LLMModel, model)
 
     async def update_llm_model(self, model_uuid: str, model_data: dict) -> None:
+        if 'uuid' in model_data:
+            del model_data['uuid']
+        
         await self.ap.persistence_mgr.execute_async(
             sqlalchemy.update(persistence_model.LLMModel).where(persistence_model.LLMModel.uuid == model_uuid).values(**model_data)
         )

pkg/api/http/service/pipeline.py

@@ -46,7 +46,7 @@ class PipelineService:
 
         return self.ap.persistence_mgr.serialize_model(persistence_pipeline.LegacyPipeline, pipeline)
 
-    async def create_pipeline(self, pipeline_data: dict) -> None:
+    async def create_pipeline(self, pipeline_data: dict) -> str:
         pipeline_data['uuid'] = str(uuid.uuid4())
         pipeline_data['for_version'] = self.ap.ver_mgr.get_current_version()
         pipeline_data['stages'] = stagemgr.stage_order.copy()
@@ -58,6 +58,8 @@ class PipelineService:
         )
         # TODO: 更新到pipeline manager
 
+        return pipeline_data['uuid']
+
     async def update_pipeline(self, pipeline_uuid: str, pipeline_data: dict) -> None:
         del pipeline_data['uuid']
         del pipeline_data['for_version']