mirror of
https://github.com/danielmiessler/fabric
synced 2024-11-10 07:10:31 +00:00
42 lines
1.1 KiB
Go
42 lines
1.1 KiB
Go
|
package azure
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/danielmiessler/fabric/common"
|
||
|
|
"github.com/danielmiessler/fabric/vendors/openai"
|
||
|
|
goopenai "github.com/sashabaranov/go-openai"
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewClient() (ret *Client) {
|
||
|
|
ret = &Client{}
|
||
|
|
ret.Client = openai.NewClientCompatible("Azure", ret.configure)
|
||
|
|
|
||
|
|
ret.ApiEndpoint = ret.AddSetupQuestion("API endpoint", true)
|
||
|
|
ret.ApiDeployments = ret.AddSetupQuestionCustom("deployments", true,
|
||
|
|
"Enter your Azure deployments (comma separated)")
|
||
|
|
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
type Client struct {
|
||
|
|
*openai.Client
|
||
|
|
ApiEndpoint *common.SetupQuestion
|
||
|
|
ApiDeployments *common.SetupQuestion
|
||
|
|
|
||
|
|
apiDeployments []string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (oi *Client) configure() (err error) {
|
||
|
|
oi.apiDeployments = strings.Split(oi.ApiDeployments.Value, ",")
|
||
|
|
oi.ApiClient = goopenai.NewClientWithConfig(goopenai.DefaultAzureConfig(oi.ApiKey.Value, oi.ApiEndpoint.Value))
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func (oi *Client) ListModels() (ret []string, err error) {
|
||
|
|
// Azure Open Client has models and deployments. We need to get the deployments to use them for chat
|
||
|
|
// There is no easy way to get the deployments from the API, so we need to ask the user to provide them
|
||
|
|
ret = oi.apiDeployments
|
||
|
|
return
|
||
|
|
}
|