翻译--添加连接失败时的重试

app/core/config.py

@@ -175,3 +175,5 @@ PRIORITY_DICT = {
     'bag_back': -98,
     'earring_back': -99,
 }
+
+QWEN_API_KEY = "sk-a6bdf594e1f54a4aa3e9d4d48f8c661f"

app/service/chat_robot/script/service/CallQWen.py

@@ -2,6 +2,8 @@ import json
 from typing import Dict, Any
 
 from dashscope import Generation
+from retry import retry
+from urllib3.exceptions import NewConnectionError
 
 from app.core.config import *
 from app.service.chat_robot.script.callbacks.qwen_callback_handler import QWenCallbackHandler
@@ -107,8 +109,7 @@ qwen = QWenCallbackHandler()
 def search_from_internet(message):
     response = Generation.call(
         model='qwen-turbo',
-        # api_key='sk-7658298c6b99443c98184a5e634fe6ab',
+        api_key=QWEN_API_KEY,
-        api_key='sk-a6bdf594e1f54a4aa3e9d4d48f8c661f',
         messages=message,
         tools=tools,
         # seed=random.randint(1, 10000),  # 设置随机数种子seed，如果没有设置，则随机数种子默认为1234
@@ -126,12 +127,11 @@ def get_table_info(table_names):
 def query_database(sql_string):
     return CustomDatabase.run(db, sql_string)
 
-
+@retry(exceptions=NewConnectionError, tries=3, delay=1)
 def get_response(messages):
     response = Generation.call(
         model='qwen-max',
-        # api_key='sk-7658298c6b99443c98184a5e634fe6ab',
+        api_key=QWEN_API_KEY,
-        api_key='sk-a6bdf594e1f54a4aa3e9d4d48f8c661f',
         messages=messages,
         tools=tools,
         # seed=random.randint(1, 10000),  # 设置随机数种子seed，如果没有设置，则随机数种子默认为1234

app/service/prompt_generation/chatgpt_for_translation.py

@@ -1,13 +1,11 @@
 import logging
 
 from dashscope import Generation
-# from langchain.chains import LLMChain
+from requests import RequestException
-from langchain_community.chat_models import QianfanChatEndpoint, ChatTongyi
+from retry import retry
-# from langchain.chat_models import ChatOpenAI
+
-from langchain_core.prompts import SystemMessagePromptTemplate, HumanMessagePromptTemplate, ChatPromptTemplate
+from app.core.config import QWEN_API_KEY
-from langchain_core.runnables import RunnableSequence
 
-from app.core.config import OPENAI_MODEL, OPENAI_API_KEY
 
 # os.environ["http_proxy"] = "http://127.0.0.1:7890"
 # os.environ["https_proxy"] = "http://127.0.0.1:7890"
@@ -44,12 +42,11 @@ def translate_to_en(text):
     return assistant_output.content
 
 
-
+@retry(exceptions=RequestException, tries=3, delay=1)
 def get_response(messages):
     response = Generation.call(
         model='qwen-max',
-        # api_key='sk-7658298c6b99443c98184a5e634fe6ab',
+        api_key= QWEN_API_KEY,
-        api_key='sk-a6bdf594e1f54a4aa3e9d4d48f8c661f',
         messages=messages,
         # seed=random.randint(1, 10000),  # 设置随机数种子seed，如果没有设置，则随机数种子默认为1234
         result_format='message',  # 将输出设置为message形式