32 lines
1.3 KiB
Python
32 lines
1.3 KiB
Python
from openai import AsyncOpenAI
|
|
|
|
|
|
class OpenAITranslate:
|
|
def __init__(self, key: str | None = None, base_url: str | None = None):
|
|
self.client = AsyncOpenAI(api_key=key, base_url=base_url)
|
|
|
|
async def translate_text(self, text, target_lang="English") -> str | None:
|
|
try:
|
|
response = await self.client.chat.completions.create(
|
|
messages=[
|
|
{
|
|
"role": "system",
|
|
"content": f"Now I ask you to be the translator. "
|
|
f"Your goal is to understand the Chinese "
|
|
f"I provided you and translate it into {target_lang}. "
|
|
f"Please do not use a translation accent when translating, "
|
|
f"but translate naturally, smoothly and authentically, "
|
|
f"using beautiful and elegant words. way of expression.",
|
|
},
|
|
{
|
|
"role": "user",
|
|
"content": text,
|
|
},
|
|
],
|
|
model="gpt-4",
|
|
)
|
|
except Exception as e:
|
|
print("Open AI Error: ", e)
|
|
return
|
|
return response.choices[0].message.content.strip()
|