mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-11-19 22:53:41 +08:00
008ffe4662
- 添加Anthropic适配器实现 | Add Anthropic adaptor implementation - 支持Anthropic消息格式转换 | Support Anthropic message format conversion - 添加Vertex AI Claude适配器支持 | Add Vertex AI Claude adapter support - 更新Anthropic的中继模式定义 | Update relay mode definitions for Anthropic - 添加Anthropic控制器和路由 | Add Anthropic controller and routing 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: zgdmemail@gmail.com
87 lines
4.7 KiB
Go
87 lines
4.7 KiB
Go
package router
|
|
|
|
import (
|
|
"github.com/songquanpeng/one-api/controller"
|
|
"github.com/songquanpeng/one-api/middleware"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func SetRelayRouter(router *gin.Engine) {
|
|
router.Use(middleware.CORS())
|
|
router.Use(middleware.GzipDecodeMiddleware())
|
|
// https://platform.openai.com/docs/api-reference/introduction
|
|
modelsRouter := router.Group("/v1/models")
|
|
modelsRouter.Use(middleware.TokenAuth())
|
|
{
|
|
modelsRouter.GET("", controller.ListModels)
|
|
modelsRouter.GET("/:model", controller.RetrieveModel)
|
|
}
|
|
relayV1Router := router.Group("/v1")
|
|
relayV1Router.Use(middleware.RelayPanicRecover(), middleware.TokenAuth(), middleware.Distribute())
|
|
{
|
|
relayV1Router.Any("/oneapi/proxy/:channelid/*target", controller.Relay)
|
|
relayV1Router.POST("/completions", controller.Relay)
|
|
relayV1Router.POST("/chat/completions", controller.Relay)
|
|
relayV1Router.POST("/edits", controller.Relay)
|
|
relayV1Router.POST("/images/generations", controller.Relay)
|
|
relayV1Router.POST("/images/edits", controller.RelayNotImplemented)
|
|
relayV1Router.POST("/images/variations", controller.RelayNotImplemented)
|
|
relayV1Router.POST("/embeddings", controller.Relay)
|
|
relayV1Router.POST("/engines/:model/embeddings", controller.Relay)
|
|
relayV1Router.POST("/audio/transcriptions", controller.Relay)
|
|
relayV1Router.POST("/audio/translations", controller.Relay)
|
|
relayV1Router.POST("/audio/speech", controller.Relay)
|
|
relayV1Router.GET("/files", controller.RelayNotImplemented)
|
|
relayV1Router.POST("/files", controller.RelayNotImplemented)
|
|
relayV1Router.DELETE("/files/:id", controller.RelayNotImplemented)
|
|
relayV1Router.GET("/files/:id", controller.RelayNotImplemented)
|
|
relayV1Router.GET("/files/:id/content", controller.RelayNotImplemented)
|
|
relayV1Router.POST("/fine_tuning/jobs", controller.RelayNotImplemented)
|
|
relayV1Router.GET("/fine_tuning/jobs", controller.RelayNotImplemented)
|
|
relayV1Router.GET("/fine_tuning/jobs/:id", controller.RelayNotImplemented)
|
|
relayV1Router.POST("/fine_tuning/jobs/:id/cancel", controller.RelayNotImplemented)
|
|
relayV1Router.GET("/fine_tuning/jobs/:id/events", controller.RelayNotImplemented)
|
|
relayV1Router.DELETE("/models/:model", controller.RelayNotImplemented)
|
|
relayV1Router.POST("/moderations", controller.Relay)
|
|
relayV1Router.POST("/assistants", controller.RelayNotImplemented)
|
|
relayV1Router.GET("/assistants/:id", controller.RelayNotImplemented)
|
|
relayV1Router.POST("/assistants/:id", controller.RelayNotImplemented)
|
|
relayV1Router.DELETE("/assistants/:id", controller.RelayNotImplemented)
|
|
relayV1Router.GET("/assistants", controller.RelayNotImplemented)
|
|
relayV1Router.POST("/assistants/:id/files", controller.RelayNotImplemented)
|
|
relayV1Router.GET("/assistants/:id/files/:fileId", controller.RelayNotImplemented)
|
|
relayV1Router.DELETE("/assistants/:id/files/:fileId", controller.RelayNotImplemented)
|
|
relayV1Router.GET("/assistants/:id/files", controller.RelayNotImplemented)
|
|
relayV1Router.POST("/threads", controller.RelayNotImplemented)
|
|
relayV1Router.GET("/threads/:id", controller.RelayNotImplemented)
|
|
relayV1Router.POST("/threads/:id", controller.RelayNotImplemented)
|
|
relayV1Router.DELETE("/threads/:id", controller.RelayNotImplemented)
|
|
relayV1Router.POST("/threads/:id/messages", controller.RelayNotImplemented)
|
|
relayV1Router.GET("/threads/:id/messages/:messageId", controller.RelayNotImplemented)
|
|
relayV1Router.POST("/threads/:id/messages/:messageId", controller.RelayNotImplemented)
|
|
relayV1Router.GET("/threads/:id/messages/:messageId/files/:filesId", controller.RelayNotImplemented)
|
|
relayV1Router.GET("/threads/:id/messages/:messageId/files", controller.RelayNotImplemented)
|
|
relayV1Router.POST("/threads/:id/runs", controller.RelayNotImplemented)
|
|
relayV1Router.GET("/threads/:id/runs/:runsId", controller.RelayNotImplemented)
|
|
relayV1Router.POST("/threads/:id/runs/:runsId", controller.RelayNotImplemented)
|
|
relayV1Router.GET("/threads/:id/runs", controller.RelayNotImplemented)
|
|
relayV1Router.POST("/threads/:id/runs/:runsId/submit_tool_outputs", controller.RelayNotImplemented)
|
|
relayV1Router.POST("/threads/:id/runs/:runsId/cancel", controller.RelayNotImplemented)
|
|
relayV1Router.GET("/threads/:id/runs/:runsId/steps/:stepId", controller.RelayNotImplemented)
|
|
relayV1Router.GET("/threads/:id/runs/:runsId/steps", controller.RelayNotImplemented)
|
|
}
|
|
|
|
// Anthropic API compatibility - https://docs.anthropic.com/claude/reference/
|
|
anthropicRouter := router.Group("/anthropic")
|
|
anthropicRouter.Use(middleware.RelayPanicRecover(), middleware.TokenAuth(), middleware.Distribute())
|
|
{
|
|
// Models API
|
|
anthropicRouter.GET("/v1/models", controller.ListModels)
|
|
anthropicRouter.GET("/v1/models/:model", controller.RetrieveModel)
|
|
|
|
// Messages API - main endpoint for chat completions
|
|
anthropicRouter.POST("/v1/messages", controller.Relay)
|
|
}
|
|
}
|