From edf770c54dc7e5d534a39afbd2241fc33b9091fa Mon Sep 17 00:00:00 2001 From: yuqi Date: Sat, 9 Sep 2023 11:04:18 +0800 Subject: [PATCH] =?UTF-8?q?CLI=20=E5=8F=82=E6=95=B0=E6=B7=BB=E5=8A=A0=20ap?= =?UTF-8?q?i=5Fkey?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- langchain/openai-translator/ai_translator/flask_server.py | 3 +++ langchain/openai-translator/ai_translator/gradio_server.py | 3 +++ langchain/openai-translator/ai_translator/main.py | 3 +++ .../openai-translator/ai_translator/utils/argument_parser.py | 1 + 4 files changed, 10 insertions(+) diff --git a/langchain/openai-translator/ai_translator/flask_server.py b/langchain/openai-translator/ai_translator/flask_server.py index 7b5bed03..fcf1d161 100644 --- a/langchain/openai-translator/ai_translator/flask_server.py +++ b/langchain/openai-translator/ai_translator/flask_server.py @@ -56,6 +56,9 @@ def initialize_translator(): argument_parser = ArgumentParser() args = argument_parser.parse_arguments() + # 设置 OpenAI API Key + os.environ["OPENAI_API_KEY"] = args.api_key + # 初始化配置单例 config = TranslationConfig() config.initialize(args) diff --git a/langchain/openai-translator/ai_translator/gradio_server.py b/langchain/openai-translator/ai_translator/gradio_server.py index 8f7d8569..d148f906 100644 --- a/langchain/openai-translator/ai_translator/gradio_server.py +++ b/langchain/openai-translator/ai_translator/gradio_server.py @@ -39,6 +39,9 @@ def initialize_translator(): argument_parser = ArgumentParser() args = argument_parser.parse_arguments() + # 设置 OpenAI API Key + os.environ["OPENAI_API_KEY"] = args.api_key + # 初始化配置单例 config = TranslationConfig() config.initialize(args) diff --git a/langchain/openai-translator/ai_translator/main.py b/langchain/openai-translator/ai_translator/main.py index 33ae7282..1ac6b2bb 100644 --- a/langchain/openai-translator/ai_translator/main.py +++ b/langchain/openai-translator/ai_translator/main.py @@ -11,6 +11,9 @@ argument_parser = ArgumentParser() args = argument_parser.parse_arguments() + # 设置 OpenAI API Key + os.environ["OPENAI_API_KEY"] = args.api_key + # 初始化配置单例 config = TranslationConfig() config.initialize(args) diff --git a/langchain/openai-translator/ai_translator/utils/argument_parser.py b/langchain/openai-translator/ai_translator/utils/argument_parser.py index 57684d86..16e3470c 100644 --- a/langchain/openai-translator/ai_translator/utils/argument_parser.py +++ b/langchain/openai-translator/ai_translator/utils/argument_parser.py @@ -5,6 +5,7 @@ def __init__(self): self.parser = argparse.ArgumentParser(description='A translation tool that supports translations in any language pair.') self.parser.add_argument('--config_file', type=str, default='config.yaml', help='Configuration file with model and API settings.') self.parser.add_argument('--model_name', type=str, help='Name of the Large Language Model.') + self.parser.add_argument('--api_key', type=str, help='The API key for OpenAIModel.') self.parser.add_argument('--input_file', type=str, help='PDF file to translate.') self.parser.add_argument('--output_file_format', type=str, help='The file format of translated book. Now supporting PDF and Markdown') self.parser.add_argument('--source_language', type=str, help='The language of the original book to be translated.')