bugfix: fix minor bugs

utils/completion_reward.py

@@ -21,8 +21,6 @@ from google.oauth2.service_account import Credentials
 from google import auth
 from google.cloud import bigquery
 from google.cloud import storage
-from google.cloud import aiplatform
-from vertexai.preview.generative_models import GenerativeModel
 
 SERVICE_ACCOUNT_INFO = os.getenv("GBQ_TOKEN")
 SCOPES = ["https://www.googleapis.com/auth/cloud-platform"]
@@ -320,16 +318,19 @@ class AWSAgent:
 
 
 class GoogleAgent:
+    from google.cloud import aiplatform
+    from vertexai.preview.generative_models import GenerativeModel
+    SERVICE_ACCOUNT_INFO = os.getenv("GBQ_TOKEN")
+    service_account_info_dict = json.loads(SERVICE_ACCOUNT_INFO)
+    logging.warning(f"Google service account info: {service_account_info_dict}")
+    logging.warning(f"{type(service_account_info_dict)}")
+    aiplatform.init(
+        project='junyiacademy',
+        service_account=service_account_info_dict,
+    )
+    gemini_pro_model = GenerativeModel("gemini-pro")
+
     def get_story(self, user_log):
-        SERVICE_ACCOUNT_INFO = os.getenv("GBQ_TOKEN")
-        service_account_info_dict = json.loads(SERVICE_ACCOUNT_INFO)
-        logging.warning(f"Google service account info: {service_account_info_dict}")
-        logging.warning(f"{type(service_account_info_dict)}")
-        aiplatform.init(
-            project='junyiacademy',
-            service_account=service_account_info_dict,
-        )
-        gemini_pro_model = GenerativeModel("gemini-pro")
         system_prompt = """
                     我正在舉辦一個學習型的活動，參與活動的學生為 1-9 年級，我為他們設計了一個獨特的故事機制，每天每個學生都會收到屬於自己獨特的冒險紀錄，現在我需要你協助我將這些冒險紀錄，製作成一段冒險故事，請
                     - 以「你」稱呼學生
@@ -348,7 +349,7 @@ class GoogleAgent:
         while retry_attempts < 5:
             try:
                 logging.info("Google Generating response...")
-                model_response = gemini_pro_model.generate_content(f"{system_prompt}, 以下是我的冒險故事 ```{user_log}```")
+                model_response = self.gemini_pro_model.generate_content(f"{system_prompt}, 以下是我的冒險故事 ```{user_log}```")
                 
                 chinese_converter = OpenCC("s2tw")