Added OpenAI

Author: djthorpe

cmd/cli/main.go

@@ -11,7 +11,7 @@ import (
 
 func main() {
 	name := path.Base(os.Args[0])
-	flags, err := NewFlags(name, os.Args[1:], ElevenlabsFlags)
+	flags, err := NewFlags(name, os.Args[1:], ElevenlabsFlags, OpenAIFlags)
 	if err != nil {
 		if err != flag.ErrHelp {
 			fmt.Fprintln(os.Stderr, err)
@@ -42,6 +42,11 @@ func main() {
 		fmt.Fprintln(os.Stderr, err)
 		os.Exit(1)
 	}
+	cmd, err = OpenAIRegister(cmd, opts, flags)
+	if err != nil {
+		fmt.Fprintln(os.Stderr, err)
+		os.Exit(1)
+	}
 
 	// Run command
 	if err := Run(cmd, flags); err != nil {

cmd/cli/openai.go

@@ -0,0 +1,66 @@
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+
+	// Packages
+	"github.com/mutablelogic/go-client/pkg/client"
+	"github.com/mutablelogic/go-client/pkg/openai"
+)
+
+func OpenAIFlags(flags *Flags) {
+	flags.String("openai-api-key", "${OPENAI_API_KEY}", "OpenAI API key")
+}
+
+func OpenAIRegister(cmd []Client, opts []client.ClientOpt, flags *Flags) ([]Client, error) {
+	// Get API key
+	key, err := flags.GetString("openai-api-key")
+	if err != nil {
+		return nil, err
+	}
+
+	// Create client
+	openai, err := openai.New(key, opts...)
+	if err != nil {
+		return nil, err
+	}
+
+	// Register commands
+	cmd = append(cmd, Client{
+		ns: "openai",
+		cmd: []Command{
+			{Name: "models", Description: "Return registered models", MinArgs: 2, MaxArgs: 2, Fn: OpenAIModels(openai, flags)},
+			{Name: "model", Description: "Return model information", MinArgs: 3, MaxArgs: 3, Fn: OpenAIModel(openai, flags)},
+		},
+	})
+
+	// Return success
+	return cmd, nil
+}
+
+func OpenAIModels(client *openai.Client, flags *Flags) CommandFn {
+	return func() error {
+		if models, err := client.Models(); err != nil {
+			return err
+		} else if data, err := json.MarshalIndent(models, "", "  "); err != nil {
+			return err
+		} else {
+			fmt.Printf("Models: %s\n", data)
+		}
+		return nil
+	}
+}
+
+func OpenAIModel(client *openai.Client, flags *Flags) CommandFn {
+	return func() error {
+		if model, err := client.Model(flags.Arg(2)); err != nil {
+			return err
+		} else if data, err := json.MarshalIndent(model, "", "  "); err != nil {
+			return err
+		} else {
+			fmt.Printf("%s\n", data)
+		}
+		return nil
+	}
+}