config.go

package openai

import (
"net/http"
"regexp"
)

const (
openaiAPIURLv1="https://api.openai.com/v1"
defaultEmptyMessagesLimituint=300

azureAPIPrefix="openai"
azureDeploymentsPrefix="deployments"

AnthropicAPIVersion="2023-06-01"
)

typeAPITypestring

const (
APITypeOpenAIAPIType="OPEN_AI"
APITypeAzureAPIType="AZURE"
APITypeAzureADAPIType="AZURE_AD"
APITypeCloudflareAzureAPIType="CLOUDFLARE_AZURE"
APITypeAnthropicAPIType="ANTHROPIC"
)

constAzureAPIKeyHeader="api-key"

constdefaultAssistantVersion="v2"// upgrade to v2 to support vector store

typeHTTPDoerinterface {
Do(req*http.Request) (*http.Response, error)
}

// ClientConfig is a configuration of a client.
typeClientConfigstruct {
authTokenstring

BaseURLstring
OrgIDstring
APITypeAPIType
APIVersionstring// required when APIType is APITypeAzure or APITypeAzureAD or APITypeAnthropic
AssistantVersionstring
AzureModelMapperFuncfunc(modelstring) string// replace model to azure deployment name func
HTTPClientHTTPDoer

EmptyMessagesLimituint
}

funcDefaultConfig(authTokenstring) ClientConfig {
returnClientConfig{
authToken: authToken,
BaseURL: openaiAPIURLv1,
APIType: APITypeOpenAI,
AssistantVersion: defaultAssistantVersion,
OrgID: "",

HTTPClient: &http.Client{},

EmptyMessagesLimit: defaultEmptyMessagesLimit,
 }
}

funcDefaultAzureConfig(apiKey, baseURLstring) ClientConfig {
returnClientConfig{
authToken: apiKey,
BaseURL: baseURL,
OrgID: "",
APIType: APITypeAzure,
APIVersion: "2023-05-15",
AzureModelMapperFunc: func(modelstring) string {
returnregexp.MustCompile(`[.:]`).ReplaceAllString(model, "")
 },

HTTPClient: &http.Client{},

EmptyMessagesLimit: defaultEmptyMessagesLimit,
 }
}

funcDefaultAnthropicConfig(apiKey, baseURLstring) ClientConfig {
ifbaseURL=="" {
baseURL="https://api.anthropic.com/v1"
 }
returnClientConfig{
authToken: apiKey,
BaseURL: baseURL,
OrgID: "",
APIType: APITypeAnthropic,
APIVersion: AnthropicAPIVersion,

HTTPClient: &http.Client{},

EmptyMessagesLimit: defaultEmptyMessagesLimit,
 }
}

func (ClientConfig) String() string {
return"<OpenAI API ClientConfig>"
}

func (cClientConfig) GetAzureDeploymentByModel(modelstring) string {
ifc.AzureModelMapperFunc!=nil {
returnc.AzureModelMapperFunc(model)
 }

returnmodel
}