Create internal/apiserver/router.go

internal/apiserver/router.go

@@ -0,0 +1,83 @@
+package apiserver
+
+import (
+	"encoding/json"
+	"log"
+	"monica-proxy/internal/middleware"
+	"monica-proxy/internal/monica"
+	"monica-proxy/internal/types"
+	"net/http"
+
+	"github.com/labstack/echo/v4"
+	"github.com/sashabaranov/go-openai"
+)
+
+// RegisterRoutes 注册 Echo 路由
+func RegisterRoutes(e *echo.Echo) {
+	// 添加Bearer Token认证中间件
+	e.Use(middleware.BearerAuth())
+
+	// ChatGPT 风格的请求转发到 /v1/chat/completions
+	e.POST("/hf/v1/chat/completions", handleChatCompletion)
+	// 获取支持的模型列表
+	e.GET("/hf/v1/models", handleListModels)
+}
+
+// handleChatCompletion 接收 ChatGPT 形式的对话请求并转发给 Monica
+func handleChatCompletion(c echo.Context) error {
+	var req openai.ChatCompletionRequest
+	if err := c.Bind(&req); err != nil {
+		return c.JSON(http.StatusBadRequest, map[string]interface{}{
+			"error": "Invalid request payload",
+		})
+	}
+
+	// 检查请求是否包含消息
+	if len(req.Messages) == 0 {
+		return c.JSON(http.StatusBadRequest, map[string]interface{}{
+			"error": "No messages found",
+		})
+	}
+
+	marshalIndent, err := json.MarshalIndent(req, "", "  ")
+	if err != nil {
+		return err
+	}
+	log.Printf("Received completion request: \n%s\n", marshalIndent)
+	// 将 ChatGPTRequest 转换为 MonicaRequest
+	monicaReq, err := types.ChatGPTToMonica(req)
+	if err != nil {
+		return c.JSON(http.StatusInternalServerError, map[string]interface{}{
+			"error": err.Error(),
+		})
+	}
+
+	// 调用 Monica 并获取 SSE Stream
+	stream, err := monica.SendMonicaRequest(c.Request().Context(), monicaReq)
+	if err != nil {
+		return c.JSON(http.StatusInternalServerError, map[string]interface{}{
+			"error": err.Error(),
+		})
+	}
+	// Resty 不会自动关闭 Body，需要我们自己来处理
+	defer stream.RawBody().Close()
+
+	// 这里直接用流式方式把 SSE 数据返回
+	c.Response().Header().Set(echo.HeaderContentType, "text/event-stream")
+	c.Response().Header().Set("Cache-Control", "no-cache")
+	c.Response().Header().Set("Transfer-Encoding", "chunked")
+	c.Response().WriteHeader(http.StatusOK)
+
+	// 将 Monica 的 SSE 数据逐行读出，再以 SSE 格式返回给调用方
+	if err := monica.StreamMonicaSSEToClient(req.Model, c.Response().Writer, stream.RawBody()); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+// handleListModels 返回支持的模型列表
+func handleListModels(c echo.Context) error {
+	models := types.GetSupportedModels()
+	return c.JSON(http.StatusOK, models)
+}