fix: do not fatal

jackmcguire1 · Dec 14, 2023 · 2f55ec2 · 2f55ec2
1 parent cfc1791
commit 2f55ec2
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/internal/dom/chatmodels/gemini_api.go b/internal/dom/chatmodels/gemini_api.go
@@ -2,7 +2,6 @@ package chatmodels
 
 import (
 	"context"
-	"log"
 
 	"github.com/google/generative-ai-go/genai"
 	"github.com/sashabaranov/go-openai"
@@ -23,7 +22,7 @@ func NewGeminiApiClient(token string) *GeminiApiClient {
 func (api *GeminiApiClient) GeminiChat(ctx context.Context, prompt string) (*genai.GenerateContentResponse, error) {
 	client, err := genai.NewClient(ctx, option.WithCredentialsJSON(api.Token))
 	if err != nil {
-		log.Fatal(err)
+		return nil, err
 	}
 	defer client.Close()
 	model := client.GenerativeModel("gemini-pro")