be2e2387d4
* feat: add Anthropic API support with custom version header * refactor: use switch statement for API type header handling * refactor: add OpenAI & AzureAD types to be exhaustive * Update client.go need explicit fallthrough in empty case statements * constant for APIVersion; addtl tests
110 lines
2.5 KiB
Go
110 lines
2.5 KiB
Go
package openai
|
|
|
|
import (
|
|
"net/http"
|
|
"regexp"
|
|
)
|
|
|
|
const (
|
|
openaiAPIURLv1 = "https://api.openai.com/v1"
|
|
defaultEmptyMessagesLimit uint = 300
|
|
|
|
azureAPIPrefix = "openai"
|
|
azureDeploymentsPrefix = "deployments"
|
|
|
|
AnthropicAPIVersion = "2023-06-01"
|
|
)
|
|
|
|
type APIType string
|
|
|
|
const (
|
|
APITypeOpenAI APIType = "OPEN_AI"
|
|
APITypeAzure APIType = "AZURE"
|
|
APITypeAzureAD APIType = "AZURE_AD"
|
|
APITypeCloudflareAzure APIType = "CLOUDFLARE_AZURE"
|
|
APITypeAnthropic APIType = "ANTHROPIC"
|
|
)
|
|
|
|
const AzureAPIKeyHeader = "api-key"
|
|
|
|
const defaultAssistantVersion = "v2" // upgrade to v2 to support vector store
|
|
|
|
type HTTPDoer interface {
|
|
Do(req *http.Request) (*http.Response, error)
|
|
}
|
|
|
|
// ClientConfig is a configuration of a client.
|
|
type ClientConfig struct {
|
|
authToken string
|
|
|
|
BaseURL string
|
|
OrgID string
|
|
APIType APIType
|
|
APIVersion string // required when APIType is APITypeAzure or APITypeAzureAD or APITypeAnthropic
|
|
AssistantVersion string
|
|
AzureModelMapperFunc func(model string) string // replace model to azure deployment name func
|
|
HTTPClient HTTPDoer
|
|
|
|
EmptyMessagesLimit uint
|
|
}
|
|
|
|
func DefaultConfig(authToken string) ClientConfig {
|
|
return ClientConfig{
|
|
authToken: authToken,
|
|
BaseURL: openaiAPIURLv1,
|
|
APIType: APITypeOpenAI,
|
|
AssistantVersion: defaultAssistantVersion,
|
|
OrgID: "",
|
|
|
|
HTTPClient: &http.Client{},
|
|
|
|
EmptyMessagesLimit: defaultEmptyMessagesLimit,
|
|
}
|
|
}
|
|
|
|
func DefaultAzureConfig(apiKey, baseURL string) ClientConfig {
|
|
return ClientConfig{
|
|
authToken: apiKey,
|
|
BaseURL: baseURL,
|
|
OrgID: "",
|
|
APIType: APITypeAzure,
|
|
APIVersion: "2023-05-15",
|
|
AzureModelMapperFunc: func(model string) string {
|
|
return regexp.MustCompile(`[.:]`).ReplaceAllString(model, "")
|
|
},
|
|
|
|
HTTPClient: &http.Client{},
|
|
|
|
EmptyMessagesLimit: defaultEmptyMessagesLimit,
|
|
}
|
|
}
|
|
|
|
func DefaultAnthropicConfig(apiKey, baseURL string) ClientConfig {
|
|
if baseURL == "" {
|
|
baseURL = "https://api.anthropic.com/v1"
|
|
}
|
|
return ClientConfig{
|
|
authToken: apiKey,
|
|
BaseURL: baseURL,
|
|
OrgID: "",
|
|
APIType: APITypeAnthropic,
|
|
APIVersion: AnthropicAPIVersion,
|
|
|
|
HTTPClient: &http.Client{},
|
|
|
|
EmptyMessagesLimit: defaultEmptyMessagesLimit,
|
|
}
|
|
}
|
|
|
|
func (ClientConfig) String() string {
|
|
return "<OpenAI API ClientConfig>"
|
|
}
|
|
|
|
func (c ClientConfig) GetAzureDeploymentByModel(model string) string {
|
|
if c.AzureModelMapperFunc != nil {
|
|
return c.AzureModelMapperFunc(model)
|
|
}
|
|
|
|
return model
|
|
}
|