mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-02 12:05:54 +00:00
feat: requesters api
This commit is contained in:
0
pkg/api/http/controller/groups/provider/__init__.py
Normal file
0
pkg/api/http/controller/groups/provider/__init__.py
Normal file
@@ -1,8 +1,8 @@
|
||||
import quart
|
||||
import uuid
|
||||
|
||||
from .. import group
|
||||
from .....entity.persistence import model
|
||||
from ... import group
|
||||
from ......entity.persistence import model
|
||||
|
||||
|
||||
@group.group_class('models/llm', '/api/v1/provider/models/llm')
|
||||
@@ -22,7 +22,7 @@ class LLMModelsRouterGroup(group.RouterGroup):
|
||||
|
||||
return self.success()
|
||||
|
||||
@self.route('/<model_uuid>', methods=['GET', 'PUT', 'DELETE'])
|
||||
@self.route('/<model_uuid>', methods=['GET', 'DELETE'])
|
||||
async def _(model_uuid: str) -> str:
|
||||
if quart.request.method == 'GET':
|
||||
model = await self.ap.model_service.get_llm_model(model_uuid)
|
||||
@@ -30,13 +30,15 @@ class LLMModelsRouterGroup(group.RouterGroup):
|
||||
if model is None:
|
||||
return self.http_status(404, -1, 'model not found')
|
||||
|
||||
return self.success(data=model)
|
||||
elif quart.request.method == 'PUT':
|
||||
json_data = await quart.request.json
|
||||
return self.success(data={
|
||||
'model': model
|
||||
})
|
||||
# elif quart.request.method == 'PUT':
|
||||
# json_data = await quart.request.json
|
||||
|
||||
await self.ap.model_service.update_llm_model(model_uuid, json_data)
|
||||
# await self.ap.model_service.update_llm_model(model_uuid, json_data)
|
||||
|
||||
return self.success()
|
||||
# return self.success()
|
||||
elif quart.request.method == 'DELETE':
|
||||
await self.ap.model_service.delete_llm_model(model_uuid)
|
||||
|
||||
26
pkg/api/http/controller/groups/provider/requesters.py
Normal file
26
pkg/api/http/controller/groups/provider/requesters.py
Normal file
@@ -0,0 +1,26 @@
|
||||
import quart
|
||||
|
||||
from ... import group
|
||||
|
||||
|
||||
@group.group_class('provider/requesters', '/api/v1/provider/requesters')
|
||||
class RequestersRouterGroup(group.RouterGroup):
|
||||
|
||||
async def initialize(self) -> None:
|
||||
@self.route('', methods=['GET'])
|
||||
async def _() -> quart.Response:
|
||||
return self.success(data={
|
||||
'requesters': self.ap.model_mgr.get_available_requesters_info()
|
||||
})
|
||||
|
||||
@self.route('/<requester_name>', methods=['GET'])
|
||||
async def _(requester_name: str) -> quart.Response:
|
||||
|
||||
requester_info = self.ap.model_mgr.get_available_requester_info_by_name(requester_name)
|
||||
|
||||
if requester_info is None:
|
||||
return self.http_status(404, -1, 'requester not found')
|
||||
|
||||
return self.success(data={
|
||||
'requester': requester_info
|
||||
})
|
||||
Reference in New Issue
Block a user