package apiserver import ( "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) { // 主页路由不需要认证，所以要放在认证中间件之前 e.GET("/", handleHome) // 添加Bearer Token认证中间件，只对API路由生效 apiGroup := e.Group("/hf") apiGroup.Use(middleware.BearerAuth()) // API路由 apiGroup.POST("/v1/chat/completions", handleChatCompletion) apiGroup.GET("/v1/models", handleListModels) } // handleHome 处理主页请求 func handleHome(c echo.Context) error { html := ` Monica Proxy API

Monica Proxy API

This is a proxy service that converts Monica's service to ChatGPT-compatible API format.

API Endpoints

Example Usage

curl -X POST https://xxx-xxx.hf.space/hf/v1/chat/completions \
  -H "Authorization: Bearer YOUR_BEARER_TOKEN" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "gpt-4o-mini",
    "messages": [
      {
        "role": "user",
        "content": "Hello!"
      }
    ],
    "stream": true
  }'
` return c.HTML(http.StatusOK, html) } // 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", }) } // 将 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) }