소스 검색

完成插入新列和翻译,用 groq 翻译

mrh 11 달 전
부모
커밋
3e4c91b526
3개의 변경된 파일4개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 0
      .env
  2. 2 0
      mylib/new_col_translate.py
  3. 1 1
      mylib/pdfzh_translator.py

+ 1 - 0
.env

@@ -1,4 +1,5 @@
 OPENAI_API_KEY='sk-NscqaCD1PfVm7soEF3C3E6297bE14d7fB595Be8f17F39aFf'
 OPENAI_API_BASE='https://aiapi.magong.site/v1'
 LLM_MODEL="deepseek-chat"
+TRANSLATE_MODEL="llama3-70b-8192"
 DEBUG=True

+ 2 - 0
mylib/new_col_translate.py

@@ -144,6 +144,8 @@ def process_csv(
         raise
 
 if __name__ == "__main__":
+    from dotenv import load_dotenv
+    load_dotenv()
     # 示例用法
     file_path = "/home/mrh/code/excel_tool/temp/测试.csv"
     output_path = "/home/mrh/code/excel_tool/temp/测试_processed.csv"

+ 1 - 1
mylib/pdfzh_translator.py

@@ -56,7 +56,7 @@ class GoogleTranslator(BaseTranslator):
 
 
 class OpenAITranslator(BaseTranslator):
-    def __init__(self, service='opeanai', lang_out='zh-CN', lang_in='auto', model=os.getenv('LLM_MODEL'), max_tokens=2000):
+    def __init__(self, service='opeanai', lang_out='zh-CN', lang_in='auto', model=os.getenv('TRANSLATE_MODEL'), max_tokens=2000):
         lang_out = "zh-CN" if lang_out == "auto" else lang_out
         lang_in = "en" if lang_in == "auto" else lang_in
         super().__init__(service, lang_out, lang_in, model)