aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat（新功能）:更换翻译模型，关闭语种判断逻辑 fix（修复bug）: refactor（重构）: test(增加测试):

Browse Source
This commit is contained in:
zhh
2025-09-26 14:46:03 +08:00
parent 95647be610
commit b8cf3d25b4
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
app/service/prompt_generation/chatgpt_for_translation.py
View File

@@ -90,20 +90,20 @@ def get_response(messages):
def get_translation_from_llama3(text): def get_translation_from_llama3(text):
start_time = time.time() start_time = time.time()
url = "http://10.1.1.240:11434/api/generate" # url = "http://10.1.1.240:11434/api/generate"
# url = "http://10.1.1.240:1143/api/generate" url = "http://10.1.1.243:11434/api/generate"
# prompt = f"System: {prefix_for_llama}\nUser:[{text}]" # prompt = f"System: {prefix_for_llama}\nUser:[{text}]"
# 先获取用户输入文本的语言 # 先获取用户输入文本的语言
language = get_language(text) # language = get_language(text)
if 'English' in language: # if 'English' in language:
return text # return text
# 创建请求的负载 translator是自定义的翻译模型 # 创建请求的负载 translator是自定义的翻译模型
payload = { payload = {
"model": "translator", "model": "zcr_gemma3_translator:4b",
"prompt": f"[{text}]", "prompt": f"[{text}]",
"stream": False "stream": False
} }
@@ -178,7 +178,7 @@ def get_prompt_from_image(image_path, text):
def main(): def main():
"""Main function""" """Main function"""
text = get_translation_from_llama3("[火焰]") text = get_translation_from_llama3("[123]")
print(text) print(text)