refactor: move entities

This commit is contained in:
Junyan Qin
2025-03-10 22:34:45 +08:00
parent a0fd152d19
commit c1f4de425a
14 changed files with 105 additions and 8 deletions

View File

@@ -0,0 +1,27 @@
import quart
from .. import group
from .....entity.persistence import model
@group.group_class('models/llm', '/api/v1/provider/models/llm')
class LLMModelsRouterGroup(group.RouterGroup):
async def initialize(self) -> None:
@self.route('', methods=['GET', 'POST'])
async def _() -> str:
if quart.request.method == 'GET':
return self.success(data={
'models': await self.ap.model_service.get_llm_models()
})
elif quart.request.method == 'POST':
pass
@self.route('/<model_uuid>', methods=['GET', 'PUT', 'DELETE'])
async def _(model_uuid: str) -> str:
if quart.request.method == 'GET':
pass
elif quart.request.method == 'PUT':
pass
elif quart.request.method == 'DELETE':
pass

View File

@@ -3,7 +3,7 @@ import sqlalchemy
import argon2
from .. import group
from .....persistence.entities import user
from .....entity.persistence import user
@group.group_class('user', '/api/v1/user')

View File

@@ -7,7 +7,7 @@ import quart
import quart_cors
from ....core import app, entities as core_entities
from .groups import logs, system, settings, plugins, stats, user
from .groups import logs, system, settings, plugins, stats, user, models
from . import group

View File

@@ -0,0 +1,35 @@
from __future__ import annotations
import sqlalchemy
from ....core import app
from ....entity.persistence import model
class ModelsService:
ap: app.Application
def __init__(self, ap: app.Application) -> None:
self.ap = ap
async def get_llm_models(self) -> list[model.LLMModel]:
result = await self.ap.persistence_mgr.execute_async(
sqlalchemy.select(model.LLMModel)
)
result_list = result.all()
return result_list
async def create_llm_model(self, model: model.LLMModel) -> None:
pass
async def get_llm_model(self, model_uuid: str) -> model.LLMModel:
pass
async def update_llm_model(self, model: model.LLMModel) -> None:
pass
async def delete_llm_model(self, model_uuid: str) -> None:
pass

View File

@@ -6,7 +6,7 @@ import jwt
import datetime
from ....core import app
from ....persistence.entities import user
from ....entity.persistence import user
from ....utils import constants