76 lines
2.5 KiB
Python
76 lines
2.5 KiB
Python
import logging
|
||
|
||
from dashscope import Generation
|
||
from requests import RequestException
|
||
from retry import retry
|
||
|
||
from app.core.config import QWEN_API_KEY
|
||
|
||
|
||
# os.environ["http_proxy"] = "http://127.0.0.1:7890"
|
||
# os.environ["https_proxy"] = "http://127.0.0.1:7890"
|
||
|
||
|
||
# llm = ChatOpenAI(model_name=OPENAI_MODEL,
|
||
# openai_api_key=OPENAI_API_KEY,
|
||
# temperature=0)
|
||
|
||
|
||
def translate_to_en(text):
|
||
template = (
|
||
"""You are a translation expert, proficient in various languages.
|
||
And can translate various languages into English.
|
||
Please translate to grammatically correct English regardless of the input language.
|
||
If the input is already in English, or consists of letters or numbers such as "cat", "abc", or "1",
|
||
output the input text exactly as it is without any modifications or additions.
|
||
If there are grammatical errors, correct them and then output the sentence."""
|
||
)
|
||
|
||
prefix = (
|
||
"""
|
||
Translate everything within the brackets [] into English.
|
||
Never translate or modify any English input.
|
||
The input must be fully translated into coherent English sentences.
|
||
Never present the translation results in the format
|
||
"The translation of \"Material suave\" into English would be \"Smooth material.\"". Instead, directly output "Smooth material".
|
||
"""
|
||
)
|
||
messages = [
|
||
# {
|
||
# Translate the entire text and ensure the output is a complete and coherent sentence in English.
|
||
# "content": template, # 系统message
|
||
# "role": "system"
|
||
# },
|
||
{
|
||
# "content": input('请输入：'), # 用户message
|
||
"content": prefix + text, # 用户message
|
||
"role": "user"
|
||
}
|
||
]
|
||
first_response = get_response(messages)
|
||
assistant_output = first_response.output.choices[0].message
|
||
print("input : {}, translate result : {}".format(text, assistant_output.content))
|
||
return assistant_output.content
|
||
|
||
|
||
@retry(exceptions=RequestException, tries=3, delay=1)
|
||
def get_response(messages):
|
||
response = Generation.call(
|
||
model='qwen-turbo',
|
||
api_key= QWEN_API_KEY,
|
||
messages=messages,
|
||
# seed=random.randint(1, 10000), # 设置随机数种子seed，如果没有设置，则随机数种子默认为1234
|
||
result_format='message', # 将输出设置为message形式
|
||
enable_search='True'
|
||
)
|
||
return response
|
||
|
||
def main():
|
||
"""Main function"""
|
||
text = translate_to_en("fire")
|
||
print(text)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|