update sth. at 2024-11-26 19:27:55

pdf2zh/translator.py

@@ -259,13 +259,16 @@ class OpenAITranslator(BaseTranslator):
         self.options = {"temperature": 0}  # 随机采样可能会打断公式标记
         # OPENAI_BASE_URL
         # OPENAI_API_KEY
-        self.client = openai.OpenAI()
+        # self.client = openai.OpenAI()
+        self.api_url = "https://sanbo1200-duck2api.hf.space/completions"  # 自定义API的URL
+        self.headers = {
+            "Content-Type": "application/json"
+        }
 
     def translate(self, text) -> str:
-        response = self.client.chat.completions.create(
-            model=self.model,
-            **self.options,
-            messages=[
+        data = {
+            "model": self.model,
+            "messages": [
                 {
                     "role": "system",
                     "content": "You are a professional,authentic machine translation engine.",
@@ -275,7 +278,8 @@ class OpenAITranslator(BaseTranslator):
                     "content": f"Translate the following markdown source text to {self.lang_out}. Keep the formula notation $v*$ unchanged. Output translation directly without any additional text.\nSource Text: {text}\nTranslated Text:",  # noqa: E501
                 },
             ],
-        )
+            "stream": False
+        }
         return response.choices[0].message.content.strip()