mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-07 22:36:02 +00:00
refactor: move entities
This commit is contained in:
27
pkg/api/http/controller/groups/models.py
Normal file
27
pkg/api/http/controller/groups/models.py
Normal 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
|
||||
@@ -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')
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
|
||||
35
pkg/api/http/service/model.py
Normal file
35
pkg/api/http/service/model.py
Normal 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
|
||||
@@ -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
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user