bugfix: fix minor bugs

utils/completion_reward.py

@@ -22,6 +22,18 @@ from google import auth
 from google.cloud import bigquery
 from google.cloud import storage
 
+SERVICE_ACCOUNT_INFO = os.getenv("GBQ_TOKEN")
+SCOPES = ["https://www.googleapis.com/auth/cloud-platform"]
+service_account_info_dict = json.loads(SERVICE_ACCOUNT_INFO)
+
+creds = Credentials.from_service_account_info(service_account_info_dict, scopes=SCOPES)
+
+gbq_client = bigquery.Client(
+    credentials=creds, project=service_account_info_dict["project_id"]
+)
+gcs_client = storage.Client(
+    credentials=creds, project=service_account_info_dict["project_id"]
+)
 
 class CompletionReward:
     def __init__(self):
@@ -47,17 +59,6 @@ class CompletionReward:
             "google": self.paragraph_google,
             "mtk": self.paragraph_mtk,
         }
-        SERVICE_ACCOUNT_INFO = os.getenv("GBQ_TOKEN")
-        SCOPES = ["https://www.googleapis.com/auth/cloud-platform"]
-        service_account_info_dict = json.loads(SERVICE_ACCOUNT_INFO)
-
-        creds = Credentials.from_service_account_info(service_account_info_dict, scopes=SCOPES)
-        self.gbq_client = bigquery.Client(
-            credentials=creds, project=service_account_info_dict["project_id"]
-        )
-        # self.gcs_client = storage.Client(
-        #     credentials=creds, project=service_account_info_dict["project_id"]
-        # )
 
     def get_llm_response(self, player_logs):
         agents_responses = {
@@ -163,7 +164,7 @@ class CompletionReward:
     ):
         insert_row = self.to_dict()
         self.insert_data_into_bigquery(
-            self.gbq_client, "streaming_log", "log_mes_completion_rewards", [insert_row]
+            gbq_client, "streaming_log", "log_mes_completion_rewards", [insert_row]
         )
         logging.info(
             f"Player {insert_row['player_backend_user_id']} rendered successfully."
@@ -184,7 +185,7 @@ class CompletionReward:
     ):
         """Uploads a file to the bucket and makes it publicly accessible."""
         # Initialize a storage client
-        bucket = self.gcs_client.bucket(bucket_name)
+        bucket = gcs_client.bucket(bucket_name)
         blob = bucket.blob(destination_blob_name)
 
         # Upload the file