@@ -86,7 +86,7 @@ def initialize_qwen(self):
86
86
87
87
client = OpenAI (
88
88
api_key = api_key ,
89
- base_url = "https://dashscope.aliyuncs.com/compatible-mode/v1"
89
+ base_url = "https://dashscope.aliyuncs.com/compatible-mode/v1" ,
90
90
)
91
91
client .api_key = api_key
92
92
client .base_url = "https://dashscope.aliyuncs.com/compatible-mode/v1"
@@ -147,6 +147,7 @@ def validation(self, model, voice_mode):
147
147
self .require_api_key (
148
148
"ANTHROPIC_API_KEY" , "Anthropic API key" , model == "claude-3"
149
149
)
150
+ self .require_api_key ("QWEN_API_KEY" , "Qwen API key" , model == "qwen-vl" )
150
151
151
152
def require_api_key (self , key_name , key_description , is_required ):
152
153
key_exists = bool (os .environ .get (key_name ))
@@ -173,6 +174,8 @@ def prompt_and_save_api_key(self, key_name, key_description):
173
174
self .google_api_key = key_value
174
175
elif key_name == "ANTHROPIC_API_KEY" :
175
176
self .anthropic_api_key = key_value
177
+ elif key_name == "QWEN_API_KEY" :
178
+ self .qwen_api_key = key_value
176
179
self .save_api_key_to_env (key_name , key_value )
177
180
load_dotenv () # Reload environment variables
178
181
# Update the instance attribute with the new key
0 commit comments