github/Fabric
1
1
Fork 0
mirror of https://github.com/danielmiessler/Fabric.git synced 2026-01-10 06:48:04 -05:00
Code Issues Packages Projects Releases Wiki Activity

Update azure.go

Browse Source
This commit is contained in:
Eckii24
2025-03-04 17:36:11 +01:00
committed by GitHub
parent f2b85af0ea
commit 629c1b3e11
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
plugins/ai/azure/azure.go
View File

@@ -13,6 +13,8 @@ func NewClient() (ret *Client) {
ret.Client = openai.NewClientCompatible("Azure", "", ret.configure)
ret.ApiDeployments = ret.AddSetupQuestionCustom("deployments", true,
"Enter your Azure deployments (comma separated)")
ret.ApiVersion = ret.AddSetupQuestionCustom("API Version", false,
"Enter the Azure API version (optional, default is latest)")
return
}
@@ -20,13 +22,18 @@ func NewClient() (ret *Client) {
type Client struct {
*openai.Client
ApiDeployments *plugins.SetupQuestion
ApiVersion *plugins.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.ApiBaseURL.Value))
config := goopenai.DefaultAzureConfig(oi.ApiKey.Value, oi.ApiBaseURL.Value)
if oi.ApiVersion.Value != "" {
config.APIVersion = oi.ApiVersion.Value
}
oi.ApiClient = goopenai.NewClientWithConfig(config)
return
}