openai

Documentation

Index

• Variables
• func LoadModelsHelpFiles() ([]*help.Section, error)
• type Completion
• type EngineInfoCommand
  • func NewEngineInfoCommand() (*EngineInfoCommand, error)
  • func (c *EngineInfoCommand) Run(ctx context.Context, parsedLayers map[string]*layers.ParsedParameterLayer, ...) error
• type Family
• type ListEnginesCommand
  • func NewListEngineCommand() (*ListEnginesCommand, error)
  • func (c *ListEnginesCommand) Run(ctx context.Context, parsedLayers map[string]*layers.ParsedParameterLayer, ...) error
• type ModelsJSON
• type SimpleModelsJSON

Variables

var FamiliesCmd = &cobra.Command{
	Use:   "ls-families",
	Short: "list families",
	Run: func(cmd *cobra.Command, args []string) {
		models := SimpleModelsJSON{}
		err := json.Unmarshal([]byte(modelsJson), &models)
		cobra.CheckErr(err)

		ctx := cmd.Context()

		gp, _, err := cli.CreateGlazedProcessorFromCobra(cmd)
		cobra.CheckErr(err)

		for _, family := range models.Families {
			err = gp.AddRow(ctx, family)
			cobra.CheckErr(err)
		}

		err = gp.Close(ctx)
		cobra.CheckErr(err)
	},
}

var ModelsCmd = &cobra.Command{
	Use:   "ls-models",
	Short: "list models",
	Run: func(cmd *cobra.Command, args []string) {
		ctx := cmd.Context()

		models := SimpleModelsJSON{}
		err := json.Unmarshal([]byte(modelsJson), &models)
		cobra.CheckErr(err)

		gp, _, err := cli.CreateGlazedProcessorFromCobra(cmd)
		cobra.CheckErr(err)

		for _, completion := range models.Completion {
			err = gp.AddRow(ctx, completion)
			cobra.CheckErr(err)
		}

		err = gp.Close(ctx)
		cobra.CheckErr(err)
	},
}

var OpenaiCmd = &cobra.Command{
	Use:   "openai",
	Short: "OpenAI commands",
}

Functions

func LoadModelsHelpFiles

func LoadModelsHelpFiles() ([]*help.Section, error)

Types

type Completion

type Completion struct {
	Name                   string `json:"name"`
	Family                 string `json:"family"`
	Description            string `json:"description"`
	MaxTokens              int    `json:"max_tokens"`
	TrainingDataCutoffDate string `json:"training_data_cutoff_date"`
}

type EngineInfoCommand

type EngineInfoCommand struct {
	*cmds.CommandDescription
}

func NewEngineInfoCommand

func NewEngineInfoCommand() (*EngineInfoCommand, error)

func (*EngineInfoCommand) Run

func (c *EngineInfoCommand) Run(
	ctx context.Context,
	parsedLayers map[string]*layers.ParsedParameterLayer,
	ps map[string]interface{},
	gp middlewares.Processor,
) error

type Family

type Family struct {
	Name             string   `json:"name"`
	Description      string   `json:"description"`
	PricePer1kTokens float64  `json:"price_per_1k_tokens"`
	GoodAt           []string `json:"good_at"`
	KeyPoints        []string `json:"key_points"`
	Subtitle         string   `json:"subtitle"`
	Short            string   `json:"short"`
}

type ListEnginesCommand

type ListEnginesCommand struct {
	*cmds.CommandDescription
}

func NewListEngineCommand

func NewListEngineCommand() (*ListEnginesCommand, error)

func (*ListEnginesCommand) Run

func (c *ListEnginesCommand) Run(
	ctx context.Context,
	parsedLayers map[string]*layers.ParsedParameterLayer,
	ps map[string]interface{},
	gp middlewares.Processor,
) error

type ModelsJSON

type ModelsJSON struct {
	Completion []Completion `json:"completion"`
	Families   []Family     `json:"families"`
}

type SimpleModelsJSON

type SimpleModelsJSON struct {
	Completion []types.Row `json:"completion"`
	Families   []types.Row `json:"families"`
}