mirror of
https://github.com/danielmiessler/Fabric.git
synced 2026-01-10 06:48:04 -05:00
4004c51b9e
### CHANGES - Introduce `cmd` directory for all main application binaries. - Move all Go packages into the `internal` directory. - Rename the `restapi` package to `server` for clarity. - Consolidate patterns and strategies into a new `data` directory. - Group all auxiliary scripts into a new `scripts` directory. - Move all documentation and images into a `docs` directory. - Update all Go import paths to reflect the new structure. - Adjust CI/CD workflows and build commands for new layout.
78 lines
1.8 KiB
Go
78 lines
1.8 KiB
Go
package tools
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/danielmiessler/fabric/internal/plugins"
|
|
"github.com/danielmiessler/fabric/internal/plugins/ai"
|
|
)
|
|
|
|
func NeeDefaults(getVendorsModels func() (*ai.VendorsModels, error)) (ret *Defaults) {
|
|
vendorName := "Default"
|
|
ret = &Defaults{
|
|
PluginBase: &plugins.PluginBase{
|
|
Name: vendorName,
|
|
SetupDescription: "Default AI Vendor and Model [required]",
|
|
EnvNamePrefix: plugins.BuildEnvVariablePrefix(vendorName),
|
|
},
|
|
GetVendorsModels: getVendorsModels,
|
|
}
|
|
|
|
ret.Vendor = ret.AddSetting("Vendor", true)
|
|
|
|
ret.Model = ret.AddSetupQuestionCustom("Model", true,
|
|
"Enter the index the name of your default model")
|
|
|
|
ret.ModelContextLength = ret.AddSetupQuestionCustom("Model Context Length", false,
|
|
"Enter model context length")
|
|
|
|
return
|
|
}
|
|
|
|
type Defaults struct {
|
|
*plugins.PluginBase
|
|
|
|
Vendor *plugins.Setting
|
|
Model *plugins.SetupQuestion
|
|
ModelContextLength *plugins.SetupQuestion
|
|
GetVendorsModels func() (*ai.VendorsModels, error)
|
|
}
|
|
|
|
func (o *Defaults) Setup() (err error) {
|
|
var vendorsModels *ai.VendorsModels
|
|
if vendorsModels, err = o.GetVendorsModels(); err != nil {
|
|
return
|
|
}
|
|
|
|
vendorsModels.Print(false)
|
|
|
|
if err = o.Ask(o.Name); err != nil {
|
|
return
|
|
}
|
|
|
|
index, parseErr := strconv.Atoi(o.Model.Value)
|
|
if parseErr == nil {
|
|
if o.Vendor.Value, o.Model.Value, err = vendorsModels.GetGroupAndItemByItemNumber(index); err != nil {
|
|
return
|
|
}
|
|
} else {
|
|
o.Vendor.Value = vendorsModels.FindGroupsByItemFirst(o.Model.Value)
|
|
}
|
|
|
|
//verify
|
|
vendorNames := vendorsModels.FindGroupsByItem(o.Model.Value)
|
|
if len(vendorNames) == 0 {
|
|
err = errors.Errorf("You need to chose an available default model.")
|
|
return
|
|
}
|
|
|
|
fmt.Println()
|
|
o.Vendor.Print()
|
|
o.Model.Print()
|
|
|
|
return
|
|
}
|