mirror of
https://hubproxy.babadafafafafa.cn/https://github.com/1Panel-dev/1Panel.git
synced 2026-09-21 00:24:12 +08:00
113 lines
3.3 KiB
Go
113 lines
3.3 KiB
Go
package v2
|
|
|
|
import (
|
|
"github.com/1Panel-dev/1Panel/agent/app/api/v2/helper"
|
|
"github.com/1Panel-dev/1Panel/agent/app/dto/request"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// @Tags TensorRT LLM
|
|
// @Summary Page TensorRT LLMs
|
|
// @Accept json
|
|
// @Param request body request.TensorRTLLMSearch true "request"
|
|
// @Success 200
|
|
// @Security ApiKeyAuth
|
|
// @Security Timestamp
|
|
// @Router /ai/tensorrt/search [post]
|
|
func (b *BaseApi) PageTensorRTLLMs(c *gin.Context) {
|
|
var req request.TensorRTLLMSearch
|
|
if err := helper.CheckBindAndValidate(&req, c); err != nil {
|
|
return
|
|
}
|
|
list := tensorrtLLMService.Page(req, helper.IsDemoRequest(c))
|
|
helper.SuccessWithData(c, list)
|
|
}
|
|
|
|
// @Tags TensorRT LLM
|
|
// @Summary Create TensorRT LLM
|
|
// @Accept json
|
|
// @Param request body request.TensorRTLLMCreate true "request"
|
|
// @Success 200
|
|
// @Security ApiKeyAuth
|
|
// @Security Timestamp
|
|
// @Router /ai/tensorrt/create [post]
|
|
// @x-panel-log {"bodyKeys":["name"],"paramKeys":[],"BeforeFunctions":[],"formatZH":"创建 TensorRT LLM [name]","formatEN":"create TensorRT LLM [name]"}
|
|
func (b *BaseApi) CreateTensorRTLLM(c *gin.Context) {
|
|
var req request.TensorRTLLMCreate
|
|
if err := helper.CheckBindAndValidate(&req, c); err != nil {
|
|
return
|
|
}
|
|
err := tensorrtLLMService.Create(req)
|
|
if err != nil {
|
|
helper.InternalServer(c, err)
|
|
return
|
|
}
|
|
helper.Success(c)
|
|
}
|
|
|
|
// @Tags TensorRT LLM
|
|
// @Summary Update TensorRT LLM
|
|
// @Accept json
|
|
// @Param request body request.TensorRTLLMUpdate true "request"
|
|
// @Success 200
|
|
// @Security ApiKeyAuth
|
|
// @Security Timestamp
|
|
// @Router /ai/tensorrt/update [post]
|
|
// @x-panel-log {"bodyKeys":["id","name"],"paramKeys":[],"BeforeFunctions":[],"formatZH":"更新 TensorRT LLM [id][name]","formatEN":"update TensorRT LLM [id][name]"}
|
|
func (b *BaseApi) UpdateTensorRTLLM(c *gin.Context) {
|
|
var req request.TensorRTLLMUpdate
|
|
if err := helper.CheckBindAndValidate(&req, c); err != nil {
|
|
return
|
|
}
|
|
err := tensorrtLLMService.Update(req)
|
|
if err != nil {
|
|
helper.InternalServer(c, err)
|
|
return
|
|
}
|
|
helper.Success(c)
|
|
}
|
|
|
|
// @Tags TensorRT LLM
|
|
// @Summary Delete TensorRT LLM
|
|
// @Accept json
|
|
// @Param request body request.TensorRTLLMDelete true "request"
|
|
// @Success 200
|
|
// @Security ApiKeyAuth
|
|
// @Security Timestamp
|
|
// @Router /ai/tensorrt/delete [post]
|
|
// @x-panel-log {"bodyKeys":["id"],"paramKeys":[],"BeforeFunctions":[],"formatZH":"删除 TensorRT LLM [id]","formatEN":"delete TensorRT LLM [id]"}
|
|
func (b *BaseApi) DeleteTensorRTLLM(c *gin.Context) {
|
|
var req request.TensorRTLLMDelete
|
|
if err := helper.CheckBindAndValidate(&req, c); err != nil {
|
|
return
|
|
}
|
|
err := tensorrtLLMService.Delete(req.ID)
|
|
if err != nil {
|
|
helper.InternalServer(c, err)
|
|
return
|
|
}
|
|
helper.Success(c)
|
|
}
|
|
|
|
// @Tags TensorRT LLM
|
|
// @Summary Operate TensorRT LLM
|
|
// @Accept json
|
|
// @Param request body request.TensorRTLLMOperate true "request"
|
|
// @Success 200
|
|
// @Security ApiKeyAuth
|
|
// @Security Timestamp
|
|
// @Router /ai/tensorrt/operate [post]
|
|
// @x-panel-log {"bodyKeys":["id","operate"],"paramKeys":[],"BeforeFunctions":[],"formatZH":"操作 TensorRT LLM [id][operate]","formatEN":"operate TensorRT LLM [id][operate]"}
|
|
func (b *BaseApi) OperateTensorRTLLM(c *gin.Context) {
|
|
var req request.TensorRTLLMOperate
|
|
if err := helper.CheckBindAndValidate(&req, c); err != nil {
|
|
return
|
|
}
|
|
err := tensorrtLLMService.Operate(req)
|
|
if err != nil {
|
|
helper.InternalServer(c, err)
|
|
return
|
|
}
|
|
helper.Success(c)
|
|
}
|