ChenyuRabbitLove commited on
Commit
1f7e0a7
1 Parent(s): 3ac9e34

bugfix: fix minor bugs

Browse files
Files changed (1) hide show
  1. utils/completion_reward.py +2 -0
utils/completion_reward.py CHANGED
@@ -321,6 +321,8 @@ class AWSAgent:
321
 
322
  class GoogleAgent:
323
  def get_story(self, user_log):
 
 
324
  aiplatform.init(
325
  project='junyiacademy',
326
  service_account=service_account_info_dict,
 
321
 
322
  class GoogleAgent:
323
  def get_story(self, user_log):
324
+ SERVICE_ACCOUNT_INFO = os.getenv("GBQ_TOKEN")
325
+ service_account_info_dict = json.loads(SERVICE_ACCOUNT_INFO)
326
  aiplatform.init(
327
  project='junyiacademy',
328
  service_account=service_account_info_dict,