feat: add PPIO as model provider

This commit is contained in:
saikidev
2025-02-13 17:00:04 +08:00
parent a82be00fc1
commit 37e55d014c
8 changed files with 57 additions and 1 deletions

View File

@@ -3,6 +3,7 @@ package openai
import (
"errors"
"fmt"
"github.com/songquanpeng/one-api/relay/adaptor/ppio"
"io"
"net/http"
"strings"
@@ -59,6 +60,8 @@ func (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {
return baiduv2.GetRequestURL(meta)
case channeltype.AliBailian:
return alibailian.GetRequestURL(meta)
case channeltype.PPIO:
return ppio.GetRequestURL(meta)
default:
return GetFullRequestURL(meta.BaseURL, meta.RequestURLPath, meta.ChannelType), nil
}

View File

@@ -14,6 +14,7 @@ import (
"github.com/songquanpeng/one-api/relay/adaptor/moonshot"
"github.com/songquanpeng/one-api/relay/adaptor/novita"
"github.com/songquanpeng/one-api/relay/adaptor/openrouter"
"github.com/songquanpeng/one-api/relay/adaptor/ppio"
"github.com/songquanpeng/one-api/relay/adaptor/siliconflow"
"github.com/songquanpeng/one-api/relay/adaptor/stepfun"
"github.com/songquanpeng/one-api/relay/adaptor/togetherai"
@@ -40,6 +41,7 @@ var CompatibleChannels = []int{
channeltype.XAI,
channeltype.BaiduV2,
channeltype.XunfeiV2,
channeltype.PPIO,
}
func GetCompatibleChannelMeta(channelType int) (string, []string) {
@@ -82,6 +84,8 @@ func GetCompatibleChannelMeta(channelType int) (string, []string) {
return "openrouter", openrouter.ModelList
case channeltype.AliBailian:
return "alibailian", alibailian.ModelList
case channeltype.PPIO:
return "ppio", ppio.ModelList
default:
return "openai", ModelList
}

View File

@@ -0,0 +1,25 @@
package ppio
// https://api.ppinfra.com/v3/openai/v1/models
var ModelList = []string{
"deepseek/deepseek-r1/community",
"deepseek/deepseek-v3/community",
"deepseek/deepseek-r1",
"deepseek/deepseek-v3",
"deepseek/deepseek-r1-distill-llama-70b",
"deepseek/deepseek-r1-distill-qwen-32b",
"deepseek/deepseek-r1-distill-qwen-14b",
"deepseek/deepseek-r1-distill-llama-8b",
"qwen/qwen-2.5-72b-instruct",
"qwen/qwen-2-vl-72b-instruct",
"meta-llama/llama-3.2-3b-instruct",
"qwen/qwen2.5-32b-instruct",
"baichuan/baichuan2-13b-chat",
"meta-llama/llama-3.1-70b-instruct",
"meta-llama/llama-3.1-8b-instruct",
"01-ai/yi-1.5-34b-chat",
"01-ai/yi-1.5-9b-chat",
"thudm/glm-4-9b-chat",
"qwen/qwen-2-7b-instruct",
}

View File

@@ -0,0 +1,15 @@
package ppio
import (
"fmt"
"github.com/songquanpeng/one-api/relay/meta"
"github.com/songquanpeng/one-api/relay/relaymode"
)
func GetRequestURL(meta *meta.Meta) (string, error) {
if meta.Mode == relaymode.ChatCompletions {
return fmt.Sprintf("%s/chat/completions", meta.BaseURL), nil
}
return "", fmt.Errorf("unsupported relay mode %d for ppio", meta.Mode)
}