From 7b384cb933c4d395a6ac01df94c479fd75a006d9 Mon Sep 17 00:00:00 2001 From: "1808837298@qq.com" <1808837298@qq.com> Date: Mon, 17 Feb 2025 18:15:01 +0800 Subject: [PATCH] feat: Add support for DeepSeek completions endpoint --- relay/channel/deepseek/adaptor.go | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/relay/channel/deepseek/adaptor.go b/relay/channel/deepseek/adaptor.go index 14dd74f0..d779ee65 100644 --- a/relay/channel/deepseek/adaptor.go +++ b/relay/channel/deepseek/adaptor.go @@ -10,6 +10,7 @@ import ( "one-api/relay/channel" "one-api/relay/channel/openai" relaycommon "one-api/relay/common" + "one-api/relay/constant" ) type Adaptor struct { @@ -29,7 +30,12 @@ func (a *Adaptor) Init(info *relaycommon.RelayInfo) { } func (a *Adaptor) GetRequestURL(info *relaycommon.RelayInfo) (string, error) { - return fmt.Sprintf("%s/v1/chat/completions", info.BaseUrl), nil + switch info.RelayMode { + case constant.RelayModeCompletions: + return fmt.Sprintf("%s/beta/completions", info.BaseUrl), nil + default: + return fmt.Sprintf("%s/v1/chat/completions", info.BaseUrl), nil + } } func (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Header, info *relaycommon.RelayInfo) error {