Bläddra i källkod

修改deepseek火山引擎--> 官方版本

luojunhui 6 timmar sedan
förälder
incheckning
bc5af597e4

+ 1 - 1
applications/api/__init__.py

@@ -3,7 +3,7 @@
 """
 from .aigc_system_api import AigcSystemApi
 from .apollo_api import ApolloApi
-from .deep_seek_api_by_volcanoengine import fetch_deepseek_response
+# from .deep_seek_api_by_volcanoengine import fetch_deepseek_response
 from .deep_seek_api_official import fetch_deepseek_completion
 from .es_api import ElasticSearchClient
 from .moon_shot_api import fetch_moon_shot_response

+ 2 - 2
tasks/ai_tasks/generate_search_keys.py

@@ -5,7 +5,7 @@ import datetime
 from tqdm import tqdm
 from pymysql.cursors import DictCursor
 from concurrent.futures import ThreadPoolExecutor
-from applications.api import fetch_deepseek_response
+from applications.api import fetch_deepseek_completion
 from applications.db import DatabaseConnector
 from config import long_articles_config
 
@@ -82,7 +82,7 @@ def generate_single_title(task):
         return
 
     prompt = generate_prompt(title)
-    response = fetch_deepseek_response(model="DeepSeek-R1", prompt=prompt)
+    response = fetch_deepseek_completion(model="DeepSeek-R1", prompt=prompt)
     title_list_str = response.strip()
     try:
         title_list = json.loads(title_list_str)

+ 2 - 2
tasks/ai_tasks/title_rewrite_task.py

@@ -9,7 +9,7 @@ from pymysql.cursors import DictCursor
 from tqdm import tqdm
 
 from applications import log
-from applications.api import fetch_deepseek_response
+from applications.api import fetch_deepseek_completion
 from applications.const import TitleRewriteTaskConst
 from applications.db import DatabaseConnector
 from config import long_articles_config
@@ -211,7 +211,7 @@ class TitleRewriteTask:
 
         try:
             prompt = generate_prompt(article_title)
-            new_title = fetch_deepseek_response(model="default", prompt=prompt)
+            new_title = fetch_deepseek_completion(model="default", prompt=prompt)
 
             # insert into rewrite table
             self.insert_into_rewrite_table(