feat: load google secret from b64

jackmcguire1 · Dec 14, 2023 · bdb4b97 · bdb4b97
1 parent 7aaca8b
commit bdb4b97
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/.github/workflows/deploy.yaml b/.github/workflows/deploy.yaml
@@ -29,4 +29,4 @@ jobs:
             token: ${{ secrets.CODECOV_TOKEN }}
             verbose: true
       - run: sam build --parameter-overrides Runtime=provided.al2 Handler=bootstrap Architecture=arm64 ApiKey=${{ secrets.OPENAI_API_KEY }}
-      - run: sam deploy --stack-name chat-gpt --s3-bucket ${{ secrets.S3_BUCKET }} --parameter-overrides Runtime=provided.al2 Handler=bootstrap Architecture=arm64 GptApiKey=${{ secrets.OPENAI_API_KEY }} GeminiApiKey='${{ secrets.GEMINI_API_KEY }}' --capabilities CAPABILITY_IAM
+      - run: sam deploy --stack-name chat-gpt --s3-bucket ${{ secrets.S3_BUCKET }} --parameter-overrides Runtime=provided.al2 Handler=bootstrap Architecture=arm64 GptApiKey=${{ secrets.OPENAI_API_KEY }} GeminiApiKey=${{ secrets.GEMINI_API_KEY }} --capabilities CAPABILITY_IAM
diff --git a/internal/dom/chatmodels/gemini_api.go b/internal/dom/chatmodels/gemini_api.go
@@ -2,6 +2,7 @@ package chatmodels
 
 import (
 	"context"
+	"encoding/base64"
 
 	"github.com/google/generative-ai-go/genai"
 	"github.com/sashabaranov/go-openai"
@@ -14,8 +15,10 @@ type GeminiApiClient struct {
 }
 
 func NewGeminiApiClient(token string) *GeminiApiClient {
+	tkn, _ := base64.StdEncoding.DecodeString(token)
+
 	return &GeminiApiClient{
-		Token: []byte(token),
+		Token: tkn,
 	}
 }