Archives/fabric
2
0
Fork 0
mirror of https://github.com/danielmiessler/fabric synced 2024-11-08 07:11:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: #979 add support for configurable base url for Anthropic

Browse Source
This commit is contained in:
Eugen Eisler 2024-09-20 13:16:49 +02:00
parent a31af9fa80
commit f4044cde7e
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
vendors/anthropic/anthropic.go vendored
View File

@ -10,6 +10,8 @@ import (
"github.com/liushuangls/go-anthropic/v2"
)
const baseUrl = "https://api.anthropic.com/v1"
func NewClient() (ret *Client) {
vendorName := "Anthropic"
ret = &Client{}
@ -20,6 +22,8 @@ func NewClient() (ret *Client) {
ConfigureCustom: ret.configure,
}
ret.ApiBaseURL = ret.AddSetupQuestion("API Base URL", false)
ret.ApiBaseURL.Value = baseUrl
ret.ApiKey = ret.Configurable.AddSetupQuestion("API key", true)
// we could provide a setup question for the following settings
@ -36,7 +40,8 @@ func NewClient() (ret *Client) {
type Client struct {
*common.Configurable
ApiKey *common.SetupQuestion
ApiBaseURL *common.SetupQuestion
ApiKey *common.SetupQuestion
maxTokens int
defaultRequiredUserMessage string
@ -46,7 +51,11 @@ type Client struct {
}
func (an *Client) configure() (err error) {
an.client = anthropic.NewClient(an.ApiKey.Value)
if an.ApiBaseURL.Value != "" {
an.client = anthropic.NewClient(an.ApiKey.Value, anthropic.WithBaseURL(an.ApiBaseURL.Value))
} else {
an.client = anthropic.NewClient(an.ApiKey.Value)
}
return
}