@@ -86,7 +86,7 @@ def initialize_qwen(self):
8686
8787 client = OpenAI (
8888 api_key = api_key ,
89- base_url = "https://dashscope.aliyuncs.com/compatible-mode/v1"
89+ base_url = "https://dashscope.aliyuncs.com/compatible-mode/v1" ,
9090 )
9191 client .api_key = api_key
9292 client .base_url = "https://dashscope.aliyuncs.com/compatible-mode/v1"
@@ -147,6 +147,7 @@ def validation(self, model, voice_mode):
147147 self .require_api_key (
148148 "ANTHROPIC_API_KEY" , "Anthropic API key" , model == "claude-3"
149149 )
150+ self .require_api_key ("QWEN_API_KEY" , "Qwen API key" , model == "qwen-vl" )
150151
151152 def require_api_key (self , key_name , key_description , is_required ):
152153 key_exists = bool (os .environ .get (key_name ))
@@ -173,6 +174,8 @@ def prompt_and_save_api_key(self, key_name, key_description):
173174 self .google_api_key = key_value
174175 elif key_name == "ANTHROPIC_API_KEY" :
175176 self .anthropic_api_key = key_value
177+ elif key_name == "QWEN_API_KEY" :
178+ self .qwen_api_key = key_value
176179 self .save_api_key_to_env (key_name , key_value )
177180 load_dotenv () # Reload environment variables
178181 # Update the instance attribute with the new key
0 commit comments