xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-11 16:26:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: crud of platform/bots

Browse Source
This commit is contained in:
Junyan Qin
2025-03-27 01:20:00 +08:00
parent 6c1ee922de
commit 5be17c55d2
14 changed files with 188 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pkg/api/http/service/model.py
View File

@@ -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)
)