Просмотр исходного кода

单视频供给,小程序优化

luojunhui 2 месяцев назад
Родитель
Сommit
e3827eb853

+ 3 - 3
applications/api/moon_shot_api.py

@@ -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=[
                 {

+ 1 - 1
coldStartTasks/publish/publish_video_to_pq_for_audit.py

@@ -160,7 +160,7 @@ class PublishVideosForAudit(object):
             kimi_safe_title = None
             title_safe_score = fetch_moon_shot_response(task='get_title_safe_score', input_text=mini_program_title)
             if int(title_safe_score) > const.TITLE_SAFE_SCORE_THRESHOLD:
-                kimi_safe_title_obj = fetch_moon_shot_response(task='make_title_safe', input_text=title, out_put_type='json')
+                kimi_safe_title_obj = fetch_moon_shot_response(task='make_title_safe', input_text=title, output_type='json')
                 kimi_safe_title = kimi_safe_title_obj['title_v2']
 
             mini_program_title = kimi_safe_title if kimi_safe_title else mini_program_title