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

feat translator 切换ollama

Browse Source 
fix
This commit is contained in:
zhouchengrong
2024-12-02 20:31:46 +08:00
parent dbea3bc975
commit bf8b3b417b
2 changed files with 75 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/api/api_prompt_generation.py
View File

@@ -6,7 +6,7 @@ from fastapi import APIRouter, HTTPException
from app.schemas.prompt_generation import PromptGenerationImageModel
from app.schemas.response_template import ResponseModel
from app.service.prompt_generation.chatgpt_for_translation import translate_to_en
from app.service.prompt_generation.chatgpt_for_translation import translate_to_en, get_translation_from_llama3
router = APIRouter()
logger = logging.getLogger()
@@ -26,7 +26,7 @@ def prompt_generation(request_data: PromptGenerationImageModel):
"""
try:
logger.info(f"prompt_generation request item is : @@@@@@:{request_data}")
data = translate_to_en("[" + request_data.text + "]")
data = get_translation_from_llama3("[" + request_data.text + "]")
logger.info(f"prompt_generation response @@@@@@:{data}")
except Exception as e:
logger.warning(f"prompt_generation Run Exception @@@@@@:{e}")