feat: rename ntu img to ruby

utils/completion_reward.py

@@ -65,13 +65,7 @@ class CompletionReward:
         ]
         self.shuffled_response_order = {}
         self.pop_response_order = []
-        self.paragraph_map = {
-            "openai": self.paragraph_openai,
-            "aws": self.paragraph_aws,
-            "google": self.paragraph_google,
-            "mtk": self.paragraph_mtk,
-            "ntu": self.paragraph_ntu,
-        }
+        self.response_time_map = {}
 
     def get_llm_response_once(self, player_logs):
         if self.agent_list:
@@ -80,9 +74,10 @@ class CompletionReward:
         else:
             return "No agents left", None
 
-        story = agent.get_story(player_logs)
+        story, response_time = agent.get_story(player_logs)
         self.agents_responses[agent.name] = story
         self.pop_response_order.append(agent.name)
+        self.response_time_map[agent.name] = response_time
 
         if len(self.pop_response_order) == 5:
             self.shuffled_response_order = {
@@ -156,11 +151,11 @@ class CompletionReward:
             "paragraph_google": self.paragraph_google,
             "paragraph_mtk": self.paragraph_mtk,
             "paragraph_ntu": self.paragraph_ntu,
-            "response_time_openai": self.openai_response_time,
-            "response_time_aws": self.aws_response_time,
-            "response_time_google": self.google_response_time,
-            "response_time_mtk": self.mtk_response_time,
-            "response_time_ntu": self.ntu_response_time,
+            "response_time_openai": self.response_time_map["openai"],
+            "response_time_aws": self.response_time_map["aws"],
+            "response_time_google": self.response_time_map["google"],
+            "response_time_mtk": self.response_time_map["mtk"],
+            "response_time_ntu": self.response_time_map["ntu"],
             "player_certificate_url": self.player_certificate_url,
             "created_at": datetime.now(),
         }
@@ -269,7 +264,7 @@ class OpenAIAgent:
                 )
                 chinese_converter = OpenCC("s2tw")
                 self.openai_response_time = time.time() - start_time
-                return chinese_converter.convert(response.choices[0].message.content)
+                return chinese_converter.convert(response.choices[0].message.content), self.openai_response_time
 
             except Exception as e:
                 retry_attempts += 1
@@ -277,7 +272,7 @@ class OpenAIAgent:
                 time.sleep(1 * retry_attempts)
 
         self.openai_response_time = time.time() - start_time
-        return '星際夥伴短時間內寫了太多故事，需要休息一下，請稍後再試，或是選擇其他星際夥伴的故事。'
+        return '星際夥伴短時間內寫了太多故事，需要休息一下，請稍後再試，或是選擇其他星際夥伴的故事。', self.openai_response_time
 
     def get_background(self):
         client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
@@ -340,7 +335,7 @@ class AWSAgent:
                 chinese_converter = OpenCC("s2tw")
 
                 self.aws_response_time = time.time() - start_time
-                return chinese_converter.convert(completion.completion)
+                return chinese_converter.convert(completion.completion), self.aws_response_time
 
             except Exception as e:
                 retry_attempts += 1
@@ -348,7 +343,7 @@ class AWSAgent:
                 time.sleep(1 * retry_attempts)
 
         self.aws_response_time = time.time() - start_time
-        return '星際夥伴短時間內寫了太多故事，需要休息一下，請稍後再試，或是選擇其他星際夥伴的故事。'
+        return '星際夥伴短時間內寫了太多故事，需要休息一下，請稍後再試，或是選擇其他星際夥伴的故事。', self.aws_response_time
 
 
 class GoogleAgent:
@@ -401,7 +396,7 @@ class GoogleAgent:
                 self.google_response_time = time.time() - start_time
                 return chinese_converter.convert(
                     model_response.candidates[0].content.parts[0].text
-                )
+                ), self.google_response_time
 
             except Exception as e:
                 retry_attempts += 1
@@ -409,7 +404,7 @@ class GoogleAgent:
                 time.sleep(1 * retry_attempts)
 
         self.google_response_time = time.time() - start_time
-        return '星際夥伴短時間內寫了太多故事，需要休息一下，請稍後再試，或是選擇其他星際夥伴的故事。'
+        return '星際夥伴短時間內寫了太多故事，需要休息一下，請稍後再試，或是選擇其他星際夥伴的故事。', self.google_response_time
 
 
 class MTKAgent:
@@ -477,9 +472,9 @@ class MTKAgent:
                 chinese_converter = OpenCC("s2tw")
                 self.mtk_response_time = time.time() - start_time
                 if extracted_content:
-                    return chinese_converter.convert(extracted_content)
+                    return chinese_converter.convert(extracted_content), self.mtk_response_time
                 else:
-                    return chinese_converter.convert(response_text)
+                    return chinese_converter.convert(response_text), self.mtk_response_time
 
             except Exception as e:
                 retry_attempts += 1
@@ -487,7 +482,7 @@ class MTKAgent:
                 time.sleep(1 * retry_attempts)
 
         self.mtk_response_time = time.time() - start_time
-        return '星際夥伴短時間內寫了太多故事，需要休息一下，請稍後再試，或是選擇其他星際夥伴的故事。'
+        return '星際夥伴短時間內寫了太多故事，需要休息一下，請稍後再試，或是選擇其他星際夥伴的故事。', self.mtk_response_time
 
 class NTUAgent:
     def __init__(self):
@@ -564,9 +559,9 @@ class NTUAgent:
                 self.ntu_response_time = time.time() - start_time
                 logging.warning(f"NTU response time: {self.ntu_response_time}")
                 if extracted_content:
-                    return chinese_converter.convert(extracted_content)
+                    return chinese_converter.convert(extracted_content), self.ntu_response_time
                 else:
-                    return chinese_converter.convert(response_text)
+                    return chinese_converter.convert(response_text), self.ntu_response_time
 
             except Exception as e:
                 retry_attempts += 1
@@ -574,7 +569,7 @@ class NTUAgent:
                 time.sleep(1 * retry_attempts)
 
         self.ntu_response_time = time.time() - start_time
-        return '星際夥伴短時間內寫了太多故事，需要休息一下，請稍後再試，或是選擇其他星際夥伴的故事。'
+        return '星際夥伴短時間內寫了太多故事，需要休息一下，請稍後再試，或是選擇其他星際夥伴的故事。', self.ntu_response_time
 
 class ImageProcessor:
     @staticmethod