Spaces:
Running
Running
ChenyuRabbitLove
commited on
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,
|