Replace gopenai with go-openai and update chat command

Author: chand1012

cmd/chat.go

@@ -4,11 +4,12 @@ Copyright © 2023 Chandler <chandler@chand1012.dev>
 package cmd
 
 import (
+	"context"
 	"fmt"
 	"os"
 
-	gopenai "github.com/CasualCodersProjects/gopenai"
 	"github.com/chand1012/ottodocs/pkg/config"
+	"github.com/sashabaranov/go-openai"
 	"github.com/spf13/cobra"
 )
 
@@ -31,26 +32,35 @@ If '-q' is not specified, the user will be prompted to enter a question.
 			os.Exit(1)
 		}
 
-		openai := gopenai.NewOpenAI(&gopenai.OpenAIOpts{
-			APIKey: conf.APIKey,
-		})
+		client := openai.NewClient(conf.APIKey)
 
 		// if the question is not provided, prompt the user for it
 		if question == "" {
 			fmt.Print("What would you like to chat ChatGPT?\n> ")
 			fmt.Scanln(&question)
 		}
 
-		// chat ChatGPT the question
-		resp, err := openai.CreateChatSimple(question, 0) // 0 sets the max tokens to 1024
+		resp, err := client.CreateChatCompletion(context.Background(), openai.ChatCompletionRequest{
+			Model: conf.Model,
+			Messages: []openai.ChatCompletionMessage{
+				{
+					Content: question,
+					Role:    openai.ChatMessageRoleUser,
+				},
+			},
+		})
+
 		if err != nil {
-			log.Errorf("Error: %s", err)
+			log.Error(err)
 			os.Exit(1)
 		}
 
-		message := resp.Choices[0].Message.Content
+		if len(resp.Choices) == 0 {
+			log.Error("No choices returned")
+			os.Exit(1)
+		}
 
-		fmt.Println(message)
+		fmt.Println(resp.Choices[0].Message.Content)
 	},
 }
 

cmd/commit.go

@@ -68,6 +68,7 @@ var commitCmd = &cobra.Command{
 				}
 				fmt.Println(output)
 			}
+			os.Exit(0)
 		}
 
 		if plain {

go.mod

@@ -3,7 +3,6 @@ module github.com/chand1012/ottodocs
 go 1.20
 
 require (
-	github.com/CasualCodersProjects/gopenai v0.3.0
 	github.com/chand1012/git2gpt v0.4.1
 	github.com/chand1012/memory v0.3.0
 	github.com/charmbracelet/log v0.2.1

go.sum

@@ -1,6 +1,4 @@
 github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
-github.com/CasualCodersProjects/gopenai v0.3.0 h1:vTFe2Rl/At+LnRL0q/joNHYZGVQd19HXFaQT6aNnpWg=
-github.com/CasualCodersProjects/gopenai v0.3.0/go.mod h1:NGGTkFg60IYsXynphxExU11U4YqWDKEg/XqP9sVvwEY=
 github.com/RoaringBitmap/roaring v0.4.23/go.mod h1:D0gp8kJQgE1A4LQ5wFLggQEyvDi06Mq5mKs52e1TwOo=
 github.com/RoaringBitmap/roaring v1.2.3 h1:yqreLINqIrX22ErkKI0vY47/ivtJr6n+kMhVOVmhWBY=
 github.com/RoaringBitmap/roaring v1.2.3/go.mod h1:plvDsJQpxOC5bw8LRteu/MLWHsHez/3y6cubLI4/1yE=