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.')