Skip to content

Commit 7416b12

Browse files
committed
add QWEN_API_KEY to prompt_and_save_api_key
1 parent 4c471cc commit 7416b12

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

operate/config.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)