From eef24913e09506cfdc7c12f4b650f67dfe87b336 Mon Sep 17 00:00:00 2001
From: Sakurasan <26715255+Sakurasan@users.noreply.github.com>
Date: Sat, 14 Sep 2024 00:31:07 +0800
Subject: [PATCH] fix bug

---
 store/keydb.go | 15 +++------------
 1 file changed, 3 insertions(+), 12 deletions(-)

diff --git a/store/keydb.go b/store/keydb.go
index 87ba3c1..02722f5 100644
--- a/store/keydb.go
+++ b/store/keydb.go
@@ -2,14 +2,11 @@ package store
 
 import (
 	"encoding/json"
-	"errors"
 	"fmt"
 	"log"
 	"opencatd-open/pkg/vertexai"
 	"os"
 	"time"
-
-	"gorm.io/gorm"
 )
 
 func init() {
@@ -27,15 +24,9 @@ func init() {
 			Key:       vertex_auth,
 			ApiSecret: vertex_auth,
 		}
-		if err := db.Table("keys").Where("name = ?", Vertex.ProjectID).Find(&key).Error; err != nil {
-			if errors.Is(err, gorm.ErrRecordNotFound) {
-				if err := db.Create(&key).Error; err != nil {
-					log.Fatalln(fmt.Errorf("import vertex_auth json error: %w", err))
-				}
-			} else {
-				log.Fatalln(fmt.Errorf("import vertex_auth json error: %w", err))
-				return
-			}
+		if err := db.Where("name = ?", Vertex.ProjectID).FirstOrCreate(&key).Error; err != nil {
+			log.Fatalln(fmt.Errorf("import vertex_auth json error: %w", err))
+			return
 		}
 	}
 	LoadKeysCache()