Spaces:
Sleeping
Sleeping
Commit
·
685a650
1
Parent(s):
b294f44
feat: rename ntu img to ruby
Browse files- 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 =
|
| 221 |
|
| 222 |
def get_story(self, user_log):
|
| 223 |
system_prompt = """
|
| 224 |
我正在舉辦一個學習型的活動,我為學生設計了一個獨特的故事機制,每天每個學生都會收到屬於自己獨特的冒險紀錄,現在我需要你協助我將這些冒險紀錄,製作成一段冒險故事,請
|
| 225 |
- 以「你」稱呼學生
|
| 226 |
-
- 可以裁減內容以將內容限制在
|
| 227 |
- 試著合併故事記錄成一段連貫、有吸引力的故事
|
|
|
|
| 228 |
- 請使用 zh_TW
|
| 229 |
- 請直接回覆故事內容,不需要回覆任何訊息
|
| 230 |
"""
|
|
@@ -305,8 +306,9 @@ class AWSAgent:
|
|
| 305 |
system_prompt = """
|
| 306 |
我正在舉辦一個學習型的活動,我為學生設計了一個獨特的故事機制,每天每個學生都會收到屬於自己獨特的冒險紀錄,現在我需要你協助我將這些冒險紀錄,製作成一段冒險故事,請
|
| 307 |
- 以「你」稱呼學生
|
| 308 |
-
- 可以裁減內容以將內容限制在
|
| 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=
|
| 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 |
-
- 可以裁減內容以將內容限制在
|
| 374 |
- 試著合併故事記錄成一段連貫、有吸引力的故事
|
|
|
|
| 375 |
- 請使用 zh_TW
|
| 376 |
- 請直接回覆故事內容,不需要回覆任何訊息
|
| 377 |
"""
|
|
@@ -413,8 +416,9 @@ class MTKAgent:
|
|
| 413 |
system_prompt = """
|
| 414 |
我正在舉辦一個學習型的活動,我為學生設計了一個獨特的故事機制,每天每個學生都會收到屬於自己獨特的冒險紀錄,現在我需要你協助我將這些冒險紀錄,製作成一段冒險故事,請
|
| 415 |
- 以「你」稱呼學生
|
| 416 |
-
- 可以裁減內容以將內容限制在
|
| 417 |
- 試著合併故事記錄成一段連貫、有吸引力的故事
|
|
|
|
| 418 |
- 請使用 zh_TW
|
| 419 |
- 請直接回覆故事內容,不需要回覆任何訊息
|
| 420 |
"""
|
|
@@ -490,8 +494,9 @@ class NTUAgent:
|
|
| 490 |
system_prompt = """
|
| 491 |
我正在舉辦一個學習型的活動,我為學生設計了一個獨特的故事機制,每天每個學生都會收到屬於自己獨特的冒險紀錄,現在我需要你協助我將這些冒險紀錄,製作成一段冒險故事,請
|
| 492 |
- 以「你」稱呼學生
|
| 493 |
-
- 可以裁減內容以將內容限制在
|
| 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":
|
| 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,
|