deng/translator
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add openai mode

Browse Source
This commit is contained in:
deng
2025-06-19 06:45:03 +00:00
parent e4a1786501
commit 8d444df917
4 changed files with 344 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
translator/app.py
View File

@ -8,6 +8,7 @@ from langchain.chains import LLMChain
from langchain.prompts import PromptTemplate
from langchain_ollama import ChatOllama
from langchain_google_genai import ChatGoogleGenerativeAI
from langchain_openai import ChatOpenAI
from utils import parse_config
@ -39,7 +40,15 @@ class TranslatorApp:
llm = ChatGoogleGenerativeAI(
model=self._config['app']['gemini']['model_name'],
temperature=self._config['app']['gemini']['temperature'],
max_tokens=self._config['app']['gemini']['max_tokens']
max_tokens=self._config['app']['gemini']['max_tokens'],
top_p=self._config['app']['gemini']['top_p']
)
elif self._config['app']['llm_mode'] == 'openai':
llm = ChatOpenAI(
model=self._config['app']['openai']['model_name'],
temperature=self._config['app']['openai']['temperature'],
max_tokens=self._config['app']['openai']['max_tokens'],
top_p=self._config['app']['openai']['top_p']
)
else:
raise ValueError(f'Unsupported llm model: {self._config['app']['llm_mode']}')