feat: use gemini as name not google

jackmcguire1 · Dec 14, 2023 · f2d2bca · f2d2bca
1 parent 2cce81d
commit f2d2bca
Show file tree

Hide file tree

Showing 8 changed files with 14 additions and 14 deletions.
diff --git a/cmd/sqs/main.go b/cmd/sqs/main.go
@@ -81,7 +81,7 @@ func main() {
 	h := &SqsHandler{
 		ChatModelSvc: chatmodels.NewClient(&chatmodels.Resources{
 			GPTApi:    chatmodels.NewOpenAiApiClient(os.Getenv("OPENAI_API_KEY")),
-			GoogleAPI: chatmodels.NewGeminiApiClient(os.Getenv("GEMINI_API_KEY")),
+			GeminiAPI: chatmodels.NewGeminiApiClient(os.Getenv("GEMINI_API_KEY")),
 		}),
 		ResponseQueue: queue.NewQueue(os.Getenv("RESPONSES_QUEUE_URI")),
 		Logger:        logger,

diff --git a/internal/api/handler.go b/internal/api/handler.go
@@ -35,8 +35,8 @@ func (h *Handler) DispatchIntents(ctx context.Context, req alexa.Request) (res a
 		h.Logger.With("prompt", prompt).Info("found phrase to autocomplete")
 
 		switch prompt {
-		case "use google":
-			h.Model = chatmodels.CHAT_MODEL_GOOGLE
+		case "use gemini":
+			h.Model = chatmodels.CHAT_MODEL_GEMINI
 			res = alexa.NewResponse("Autocomplete", "ok", false)
 			return
 		case "use gpt":

diff --git a/internal/dom/chatmodels/api.go b/internal/dom/chatmodels/api.go
@@ -12,8 +12,8 @@ type GptAPI interface {
 	AutoComplete(context.Context, string) (openai.ChatCompletionResponse, error)
 }
 
-type GoogleApi interface {
-	GoogleChat(context.Context, string) (*genai.GenerateContentResponse, error)
+type GeminiAPI interface {
+	GeminiChat(context.Context, string) (*genai.GenerateContentResponse, error)
 }
 
 type mockAPI struct {
@@ -26,7 +26,7 @@ func (api *mockAPI) AutoComplete(ctx context.Context, prompt string) (res openai
 	return res, args.Error(1)
 }
 
-func (api *mockAPI) GoogleChat(ctx context.Context, prompt string) (res *genai.GenerateContentResponse, err error) {
+func (api *mockAPI) GeminiChat(ctx context.Context, prompt string) (res *genai.GenerateContentResponse, err error) {
 	args := api.Called(ctx, prompt)
 	res = args.Get(0).(*genai.GenerateContentResponse)
 	return res, args.Error(1)

diff --git a/internal/dom/chatmodels/gemini_api.go b/internal/dom/chatmodels/gemini_api.go
@@ -20,7 +20,7 @@ func NewGeminiApiClient(token string) *GeminiApiClient {
 	}
 }
 
-func (api *GeminiApiClient) GoogleChat(ctx context.Context, prompt string) (*genai.GenerateContentResponse, error) {
+func (api *GeminiApiClient) GeminiChat(ctx context.Context, prompt string) (*genai.GenerateContentResponse, error) {
 	client, err := genai.NewClient(ctx, option.WithAPIKey(api.Token))
 	if err != nil {
 		log.Fatal(err)

diff --git a/internal/dom/chatmodels/models.go b/internal/dom/chatmodels/models.go
@@ -3,7 +3,7 @@ package chatmodels
 type ChatModel string
 
 const (
-	CHAT_MODEL_GOOGLE ChatModel = "google"
+	CHAT_MODEL_GEMINI ChatModel = "gemini"
 	CHAT_MODEL_GPT    ChatModel = "gpt"
 )
 

diff --git a/internal/dom/chatmodels/prompts.go b/internal/dom/chatmodels/prompts.go
@@ -9,8 +9,8 @@ import (
 
 func (client *Client) AutoComplete(ctx context.Context, prompt string, model ChatModel) (string, error) {
 	switch model {
-	case CHAT_MODEL_GOOGLE:
-		res, err := client.GoogleAPI.GoogleChat(ctx, prompt)
+	case CHAT_MODEL_GEMINI:
+		res, err := client.GeminiAPI.GeminiChat(ctx, prompt)
 		if err != nil {
 			return "", err
 		}

diff --git a/internal/dom/chatmodels/prompts_test.go b/internal/dom/chatmodels/prompts_test.go
@@ -67,9 +67,9 @@ func TestGeminiChat(t *testing.T) {
 		},
 		PromptFeedback: nil,
 	}
-	api.On("GoogleChat", mock.Anything, "steve").Return(mockResponse, nil)
-	c := Client{&Resources{GoogleAPI: api}}
-	resp, err := c.AutoComplete(context.Background(), "steve", CHAT_MODEL_GOOGLE)
+	api.On("GeminiChat", mock.Anything, "steve").Return(mockResponse, nil)
+	c := Client{&Resources{GeminiAPI: api}}
+	resp, err := c.AutoComplete(context.Background(), "steve", CHAT_MODEL_GEMINI)
 	assert.NoError(t, err)
 	assert.EqualValues(t, "is the best", resp)
 }
diff --git a/internal/dom/chatmodels/service.go b/internal/dom/chatmodels/service.go
@@ -4,7 +4,7 @@ import "context"
 
 type Resources struct {
 	GPTApi    GptAPI
-	GoogleAPI GoogleApi
+	GeminiAPI GeminiAPI
 }
 
 type Service interface {