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

feat: model sync between api and manager layer

Browse Source
This commit is contained in:
Junyan Qin
2025-03-19 23:58:14 +08:00
parent 81481c9050
commit 4275459d45
2 changed files with 47 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -35,7 +35,7 @@ class ModelsService:
**model_data
)
)
# TODO: add to runtime
await self.ap.model_mgr.load_model(model_data)
async def get_llm_model(self, model_uuid: str) -> dict | None:
result = await self.ap.persistence_mgr.execute_async(
@@ -54,8 +54,12 @@ class ModelsService:
sqlalchemy.update(persistence_model.LLMModel).where(persistence_model.LLMModel.uuid == model_uuid).values(**model_data)
)
await self.ap.model_mgr.remove_model(model_uuid)
await self.ap.model_mgr.load_model(model_data)
async def delete_llm_model(self, model_uuid: str) -> None:
await self.ap.persistence_mgr.execute_async(
sqlalchemy.delete(persistence_model.LLMModel).where(persistence_model.LLMModel.uuid == model_uuid)
)
# TODO: delete from runtime
await self.ap.model_mgr.remove_model(model_uuid)