|
@@ -76,14 +76,14 @@ make_title_safe_prompt = """
|
|
|
"""
|
|
|
|
|
|
|
|
|
-def fetch_moon_shot_response(task, input_text, out_put_type="text"):
|
|
|
+def fetch_moon_shot_response(task, input_text, output_type="text"):
|
|
|
"""
|
|
|
调用kimi的api获取结果
|
|
|
"""
|
|
|
# generate prompt
|
|
|
match task:
|
|
|
case "generate_kimi_title":
|
|
|
- prompt = input_text + generate_program_title_prompt
|
|
|
+ prompt = input_text + '\n' + generate_program_title_prompt
|
|
|
case "get_title_safe_score":
|
|
|
prompt = get_title_safe_score_prompt + input_text
|
|
|
case "make_title_safe":
|
|
@@ -98,7 +98,7 @@ def fetch_moon_shot_response(task, input_text, out_put_type="text"):
|
|
|
)
|
|
|
|
|
|
# get response
|
|
|
- if out_put_type == "json":
|
|
|
+ if output_type == "json":
|
|
|
chat_completion = client.chat.completions.create(
|
|
|
messages=[
|
|
|
{
|