Files
gpt-load/internal/protocol/protocol.go
T
tbphp aee2111290 feat(gateway): 支持 OpenAI Embeddings API (#496)
新增原生 OpenAI Embeddings 端点,覆盖首期 API Key 渠道,并复用现有路由、健康、用量、日志与管理流程。
2026-08-30 00:45:19 +08:00

47 lines
1.0 KiB
Go

// Package protocol defines protocol identifiers shared by runtime domains.
package protocol
type Protocol string
const (
OpenAICompletions Protocol = "openai-completions"
OpenAIResponses Protocol = "openai-responses"
OpenAIImages Protocol = "openai-images"
OpenAIEmbeddings Protocol = "openai-embeddings"
Anthropic Protocol = "anthropic"
Gemini Protocol = "gemini"
)
func (p Protocol) Valid() bool {
switch p {
case OpenAICompletions, OpenAIResponses, OpenAIImages, OpenAIEmbeddings, Anthropic, Gemini:
return true
default:
return false
}
}
func (p Protocol) DataPlaneEnabled() bool {
switch p {
case OpenAICompletions, OpenAIResponses, OpenAIImages, OpenAIEmbeddings, Anthropic, Gemini:
return true
default:
return false
}
}
func (p Protocol) SupportsModelOptionalRequests() bool {
return p == OpenAIResponses
}
func DataPlaneProtocols() []Protocol {
return []Protocol{
OpenAICompletions,
OpenAIResponses,
OpenAIImages,
OpenAIEmbeddings,
Anthropic,
Gemini,
}
}