31 lines
960 B
Go
31 lines
960 B
Go
package router
|
||
|
||
import (
|
||
"github.com/flipped-aurora/gin-vue-admin/server/middleware"
|
||
"github.com/flipped-aurora/gin-vue-admin/server/plugin/volcengine/api"
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
type LLMRouter struct{}
|
||
|
||
// InitLLMRouter 初始化LLM路由
|
||
func (l *LLMRouter) InitLLMRouter(Router *gin.RouterGroup) {
|
||
// LLM相关路由组，使用操作记录中间件
|
||
llmRouter := Router.Group("llm").Use(middleware.OperationRecord())
|
||
// LLM查询路由组，不使用操作记录中间件（用于GET请求）
|
||
llmRouterWithoutRecord := Router.Group("llm")
|
||
|
||
// 获取API实例
|
||
llmApi := api.ApiGroupApp.LLMApi
|
||
|
||
{
|
||
// 需要记录操作的路由（POST请求）
|
||
llmRouter.POST("chat", llmApi.ChatCompletion) // LLM聊天完成
|
||
llmRouter.POST("stop", llmApi.StopGeneration) // 停止LLM生成
|
||
}
|
||
{
|
||
// 不需要记录操作的路由（GET请求）
|
||
llmRouterWithoutRecord.GET("sessions", llmApi.GetActiveSessionsCount) // 获取活跃会话数量
|
||
}
|
||
}
|