|
@@ -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)
|