refactor: NewHTTPProvider - now accept more params, removed long names
NewHTTPProvider*
This commit is contained in:
parent
5005066fd1
commit
0068613bcb
2 changed files with 4 additions and 29 deletions
|
|
@ -85,7 +85,7 @@ func CreateProviderFromConfig(cfg *config.ModelConfig) (LLMProvider, string, err
|
||||||
apiBase = getDefaultAPIBase(protocol)
|
apiBase = getDefaultAPIBase(protocol)
|
||||||
}
|
}
|
||||||
|
|
||||||
return NewHTTPProviderWithMaxTokensFieldAndRequestTimeoutAndUsePromptCaching(
|
return NewHTTPProvider(
|
||||||
cfg.APIKey,
|
cfg.APIKey,
|
||||||
apiBase,
|
apiBase,
|
||||||
cfg.Proxy,
|
cfg.Proxy,
|
||||||
|
|
@ -105,7 +105,7 @@ func CreateProviderFromConfig(cfg *config.ModelConfig) (LLMProvider, string, err
|
||||||
if apiBase == "" {
|
if apiBase == "" {
|
||||||
apiBase = getDefaultAPIBase(protocol)
|
apiBase = getDefaultAPIBase(protocol)
|
||||||
}
|
}
|
||||||
return NewHTTPProviderWithMaxTokensFieldAndRequestTimeoutAndUsePromptCaching(
|
return NewHTTPProvider(
|
||||||
cfg.APIKey,
|
cfg.APIKey,
|
||||||
apiBase,
|
apiBase,
|
||||||
cfg.Proxy,
|
cfg.Proxy,
|
||||||
|
|
@ -131,7 +131,7 @@ func CreateProviderFromConfig(cfg *config.ModelConfig) (LLMProvider, string, err
|
||||||
if cfg.APIKey == "" {
|
if cfg.APIKey == "" {
|
||||||
return nil, "", fmt.Errorf("api_key is required for anthropic protocol (model: %s)", cfg.Model)
|
return nil, "", fmt.Errorf("api_key is required for anthropic protocol (model: %s)", cfg.Model)
|
||||||
}
|
}
|
||||||
return NewHTTPProviderWithMaxTokensFieldAndRequestTimeoutAndUsePromptCaching(
|
return NewHTTPProvider(
|
||||||
cfg.APIKey,
|
cfg.APIKey,
|
||||||
apiBase,
|
apiBase,
|
||||||
cfg.Proxy,
|
cfg.Proxy,
|
||||||
|
|
|
||||||
|
|
@ -17,32 +17,7 @@ type HTTPProvider struct {
|
||||||
delegate *openai_compat.Provider
|
delegate *openai_compat.Provider
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewHTTPProvider(apiKey, apiBase, proxy string) *HTTPProvider {
|
func NewHTTPProvider(
|
||||||
return &HTTPProvider{
|
|
||||||
delegate: openai_compat.NewProvider(apiKey, apiBase, proxy),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewHTTPProviderWithMaxTokensField(apiKey, apiBase, proxy, maxTokensField string) *HTTPProvider {
|
|
||||||
return NewHTTPProviderWithMaxTokensFieldAndRequestTimeout(apiKey, apiBase, proxy, maxTokensField, 0)
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewHTTPProviderWithMaxTokensFieldAndRequestTimeout(
|
|
||||||
apiKey, apiBase, proxy, maxTokensField string,
|
|
||||||
requestTimeoutSeconds int,
|
|
||||||
) *HTTPProvider {
|
|
||||||
return &HTTPProvider{
|
|
||||||
delegate: openai_compat.NewProvider(
|
|
||||||
apiKey,
|
|
||||||
apiBase,
|
|
||||||
proxy,
|
|
||||||
openai_compat.WithMaxTokensField(maxTokensField),
|
|
||||||
openai_compat.WithRequestTimeout(time.Duration(requestTimeoutSeconds)*time.Second),
|
|
||||||
),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewHTTPProviderWithMaxTokensFieldAndRequestTimeoutAndUsePromptCaching(
|
|
||||||
apiKey, apiBase, proxy, maxTokensField string,
|
apiKey, apiBase, proxy, maxTokensField string,
|
||||||
requestTimeoutSeconds int,
|
requestTimeoutSeconds int,
|
||||||
usePromptCaching bool,
|
usePromptCaching bool,
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue