ChenyuRabbitLove commited on
Commit
685a650
1 Parent(s): b294f44

feat: rename ntu img to ruby

Browse files
Files changed (1) hide show
  1. utils/completion_reward.py +13 -8
utils/completion_reward.py CHANGED
@@ -217,14 +217,15 @@ class OpenAIAgent:
217
  self.temperature = 0.8
218
  self.frequency_penalty = 0
219
  self.presence_penalty = 0
220
- self.max_tokens = 2048
221
 
222
  def get_story(self, user_log):
223
  system_prompt = """
224
  我正在舉辦一個學習型的活動,我為學生設計了一個獨特的故事機制,每天每個學生都會收到屬於自己獨特的冒險紀錄,現在我需要你協助我將這些冒險紀錄,製作成一段冒險故事,請
225
  - 以「你」稱呼學生
226
- - 可以裁減內容以將內容限制在 1024 個 token 內
227
  - 試著合併故事記錄成一段連貫、有吸引力的故事
 
228
  - 請使用 zh_TW
229
  - 請直接回覆故事內容,不需要回覆任何訊息
230
  """
@@ -305,8 +306,9 @@ class AWSAgent:
305
  system_prompt = """
306
  我正在舉辦一個學習型的活動,我為學生設計了一個獨特的故事機制,每天每個學生都會收到屬於自己獨特的冒險紀錄,現在我需要你協助我將這些冒險紀錄,製作成一段冒險故事,請
307
  - 以「你」稱呼學生
308
- - 可以裁減內容以將內容限制在 1024 個 token 內
309
  - 試著合併故事記錄成一段連貫、有吸引力的故事
 
310
  - 請使用 zh_TW
311
  - 請直接回覆故事內容,不需要回覆任何訊息
312
  """
@@ -327,7 +329,7 @@ class AWSAgent:
327
  start_time = time.time()
328
  completion = client.completions.create(
329
  model="anthropic.claude-v2",
330
- max_tokens_to_sample=2048,
331
  prompt=f"{anthropic_bedrock.HUMAN_PROMPT}{system_prompt},以下是我的故事紀錄```{user_log}``` {anthropic_bedrock.AI_PROMPT}",
332
  )
333
  chinese_converter = OpenCC("s2tw")
@@ -370,8 +372,9 @@ class GoogleAgent:
370
  system_prompt = """
371
  我正在舉辦一個學習型的活動,我為學生設計了一個獨特的故事機制,每天每個學生都會收到屬於自己獨特的冒險紀錄,現在我需要你協助我將這些冒險紀錄,製作成一段冒險故事,請
372
  - 以「你」稱呼學生
373
- - 可以裁減內容以將內容限制在 1024 個 token 內
374
  - 試著合併故事記錄成一段連貫、有吸引力的故事
 
375
  - 請使用 zh_TW
376
  - 請直接回覆故事內容,不需要回覆任何訊息
377
  """
@@ -413,8 +416,9 @@ class MTKAgent:
413
  system_prompt = """
414
  我正在舉辦一個學習型的活動,我為學生設計了一個獨特的故事機制,每天每個學生都會收到屬於自己獨特的冒險紀錄,現在我需要你協助我將這些冒險紀錄,製作成一段冒險故事,請
415
  - 以「你」稱呼學生
416
- - 可以裁減內容以將內容限制在 1024 個 token 內
417
  - 試著合併故事記錄成一段連貫、有吸引力的故事
 
418
  - 請使用 zh_TW
419
  - 請直接回覆故事內容,不需要回覆任何訊息
420
  """
@@ -490,8 +494,9 @@ class NTUAgent:
490
  system_prompt = """
491
  我正在舉辦一個學習型的活動,我為學生設計了一個獨特的故事機制,每天每個學生都會收到屬於自己獨特的冒險紀錄,現在我需要你協助我將這些冒險紀錄,製作成一段冒險故事,請
492
  - 以「你」稱呼學生
493
- - 可以裁減內容以將內容限制在 1024 個 token 內
494
  - 試著合併故事記錄成一段連貫、有吸引力的故事
 
495
  - 請使用 zh_TW
496
  - 請直接回覆故事內容,不需要回覆任何訊息
497
  """
@@ -518,7 +523,7 @@ class NTUAgent:
518
  "temperature": 0.7,
519
  "top_p": 1,
520
  "n": 1,
521
- "max_tokens": 2048,
522
  "stop": ["string"],
523
  "stream": False,
524
  "presence_penalty": 0,
 
217
  self.temperature = 0.8
218
  self.frequency_penalty = 0
219
  self.presence_penalty = 0
220
+ self.max_tokens = 1024
221
 
222
  def get_story(self, user_log):
223
  system_prompt = """
224
  我正在舉辦一個學習型的活動,我為學生設計了一個獨特的故事機制,每天每個學生都會收到屬於自己獨特的冒險紀錄,現在我需要你協助我將這些冒險紀錄,製作成一段冒險故事,請
225
  - 以「你」稱呼學生
226
+ - 可以裁減內容以將內容限制在 500 字內
227
  - 試著合併故事記錄成一段連貫、有吸引力的故事
228
+ - 請勿突然中斷故事,請讓故事有一個完整的結局
229
  - 請使用 zh_TW
230
  - 請直接回覆故事內容,不需要回覆任何訊息
231
  """
 
306
  system_prompt = """
307
  我正在舉辦一個學習型的活動,我為學生設計了一個獨特的故事機制,每天每個學生都會收到屬於自己獨特的冒險紀錄,現在我需要你協助我將這些冒險紀錄,製作成一段冒險故事,請
308
  - 以「你」稱呼學生
309
+ - 可以裁減內容以將內容限制在 500 字內
310
  - 試著合併故事記錄成一段連貫、有吸引力的故事
311
+ - 請勿突然中斷故事,請讓故事有一個完整的結局
312
  - 請使用 zh_TW
313
  - 請直接回覆故事內容,不需要回覆任何訊息
314
  """
 
329
  start_time = time.time()
330
  completion = client.completions.create(
331
  model="anthropic.claude-v2",
332
+ max_tokens_to_sample=1024,
333
  prompt=f"{anthropic_bedrock.HUMAN_PROMPT}{system_prompt},以下是我的故事紀錄```{user_log}``` {anthropic_bedrock.AI_PROMPT}",
334
  )
335
  chinese_converter = OpenCC("s2tw")
 
372
  system_prompt = """
373
  我正在舉辦一個學習型的活動,我為學生設計了一個獨特的故事機制,每天每個學生都會收到屬於自己獨特的冒險紀錄,現在我需要你協助我將這些冒險紀錄,製作成一段冒險故事,請
374
  - 以「你」稱呼學生
375
+ - 可以裁減內容以將內容限制在 500 字內
376
  - 試著合併故事記錄成一段連貫、有吸引力的故事
377
+ - 請勿突然中斷故事,請讓故事有一個完整的結局
378
  - 請使用 zh_TW
379
  - 請直接回覆故事內容,不需要回覆任何訊息
380
  """
 
416
  system_prompt = """
417
  我正在舉辦一個學習型的活動,我為學生設計了一個獨特的故事機制,每天每個學生都會收到屬於自己獨特的冒險紀錄,現在我需要你協助我將這些冒險紀錄,製作成一段冒險故事,請
418
  - 以「你」稱呼學生
419
+ - 可以裁減內容以將內容限制在 500 字內
420
  - 試著合併故事記錄成一段連貫、有吸引力的故事
421
+ - 請勿突然中斷故事,請讓故事有一個完整的結局
422
  - 請使用 zh_TW
423
  - 請直接回覆故事內容,不需要回覆任何訊息
424
  """
 
494
  system_prompt = """
495
  我正在舉辦一個學習型的活動,我為學生設計了一個獨特的故事機制,每天每個學生都會收到屬於自己獨特的冒險紀錄,現在我需要你協助我將這些冒險紀錄,製作成一段冒險故事,請
496
  - 以「你」稱呼學生
497
+ - 可以裁減內容以將內容限制在 500 字內
498
  - 試著合併故事記錄成一段連貫、有吸引力的故事
499
+ - 請勿突然中斷故事,請讓故事有一個完整的結局
500
  - 請使用 zh_TW
501
  - 請直接回覆故事內容,不需要回覆任何訊息
502
  """
 
523
  "temperature": 0.7,
524
  "top_p": 1,
525
  "n": 1,
526
+ "max_tokens": 1024,
527
  "stop": ["string"],
528
  "stream": False,
529
  "presence_penalty": 0,