ChenyuRabbitLove commited on
Commit
1a69cfc
1 Parent(s): 1141bac

bugfix: fix minor bugs

Browse files
Files changed (1) hide show
  1. utils/completion_reward.py +8 -4
utils/completion_reward.py CHANGED
@@ -62,11 +62,15 @@ class CompletionReward:
62
  }
63
 
64
  def get_llm_response(self, player_logs):
 
 
 
 
65
  agents_responses = {
66
- "openai": self.openai_agent.get_story(player_logs),
67
- "aws": self.aws_agent.get_story(player_logs),
68
- "google": self.google_agent.get_story(player_logs),
69
- "mtk": self.mtk_agent.get_story(player_logs),
70
  }
71
  self.paragraph_openai = agents_responses["openai"]
72
  self.paragraph_aws = agents_responses["aws"]
 
62
  }
63
 
64
  def get_llm_response(self, player_logs):
65
+ openai_story = self.openai_agent.get_story(player_logs)
66
+ aws_story = self.aws_agent.get_story(player_logs)
67
+ google_story = self.google_agent.get_story(player_logs)
68
+ mtk_story = self.mtk_agent.get_story(player_logs)
69
  agents_responses = {
70
+ 'openai': openai_story,
71
+ 'aws': aws_story,
72
+ 'google': google_story,
73
+ 'mtk': mtk_story,
74
  }
75
  self.paragraph_openai = agents_responses["openai"]
76
  self.paragraph_aws = agents_responses["aws"]