46 lines
1.0 KiB
Go
46 lines
1.0 KiB
Go
package gemini
|
|
|
|
import (
|
|
"fmt"
|
|
"one-api/providers/base"
|
|
"strings"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type GeminiProviderFactory struct{}
|
|
|
|
// 创建 ClaudeProvider
|
|
func (f GeminiProviderFactory) Create(c *gin.Context) base.ProviderInterface {
|
|
return &GeminiProvider{
|
|
BaseProvider: base.BaseProvider{
|
|
BaseURL: "https://generativelanguage.googleapis.com",
|
|
ChatCompletions: "/",
|
|
Context: c,
|
|
},
|
|
}
|
|
}
|
|
|
|
type GeminiProvider struct {
|
|
base.BaseProvider
|
|
}
|
|
|
|
func (p *GeminiProvider) GetFullRequestURL(requestURL string, modelName string) string {
|
|
baseURL := strings.TrimSuffix(p.GetBaseURL(), "/")
|
|
version := "v1"
|
|
if p.Context.GetString("api_version") != "" {
|
|
version = p.Context.GetString("api_version")
|
|
}
|
|
|
|
return fmt.Sprintf("%s/%s/models/%s:%s?key=%s", baseURL, version, modelName, requestURL, p.Context.GetString("api_key"))
|
|
|
|
}
|
|
|
|
// 获取请求头
|
|
func (p *GeminiProvider) GetRequestHeaders() (headers map[string]string) {
|
|
headers = make(map[string]string)
|
|
p.CommonRequestHeaders(headers)
|
|
|
|
return headers
|
|
}
|