فهرست منبع

deep_seek_api function rename

luojunhui 4 ماه پیش
والد
کامیت
86fc04e018
3فایلهای تغییر یافته به همراه4 افزوده شده و 4 حذف شده
  1. 1 1
      applications/api/__init__.py
  2. 1 1
      applications/api/deep_seek_by_byte_dance_api.py
  3. 2 2
      tasks/article_summary_task.py

+ 1 - 1
applications/api/__init__.py

@@ -1,7 +1,7 @@
 """
 @author: luojunhui
 """
-from .deep_seek_by_byte_dance_api import deep_seek_api
+from .deep_seek_by_byte_dance_api import get_response_by_deep_seek_api
 from .google_ai_api import GoogleAIAPI
 from .moon_shot_api import generate_mini_program_title
 from .nlp_api import similarity_between_title_list

+ 1 - 1
applications/api/deep_seek_by_byte_dance_api.py

@@ -8,7 +8,7 @@ from config import deep_seek_default_model
 from config import deep_seek_api_key_byte_dance
 
 
-def deep_seek_api(model, prompt):
+def get_response_by_deep_seek_api(model, prompt):
     """
     deep_seek方法
     """

+ 2 - 2
tasks/article_summary_task.py

@@ -4,7 +4,7 @@
 from pymysql.cursors import DictCursor
 from tqdm import tqdm
 
-from applications.api import deep_seek_api
+from applications.api import get_response_by_deep_seek_api
 from applications.const import VideoToTextConst
 from applications.db import DatabaseConnector
 from config import long_articles_config
@@ -79,7 +79,7 @@ class ArticleSummaryTask(object):
         try:
             # 生成prompt
             prompt = generate_prompt(video_text)
-            response = deep_seek_api(model="DeepSeek-R1", prompt=prompt)
+            response = get_response_by_deep_seek_api(model="DeepSeek-R1", prompt=prompt)
             if response:
                 update_sql = f"""
                     update publish_single_video_source