Skip to content

Commit 3e2fc61

Browse files
feat: list_models env
1 parent 7ef55ef commit 3e2fc61

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

backend/controller/llm_api.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
Author: ai-business-hql qingli.hql@alibaba-inc.com
33
Date: 2025-07-14 16:46:20
44
LastEditors: ai-business-hql ai.bussiness.hql@gmail.com
5-
LastEditTime: 2025-10-15 17:17:28
5+
LastEditTime: 2025-12-12 16:32:50
66
FilePath: /comfyui_copilot/backend/controller/llm_api.py
77
Description: 这是默认设置,请设置`customMade`, 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
88
'''
@@ -12,7 +12,7 @@
1212
import json
1313
from typing import List, Dict, Any
1414
from aiohttp import web
15-
from ..utils.globals import LLM_DEFAULT_BASE_URL, LMSTUDIO_DEFAULT_BASE_URL, is_lmstudio_url
15+
from ..utils.globals import LLM_DEFAULT_BASE_URL, LMSTUDIO_DEFAULT_BASE_URL, OPENAI_API_KEY, OPENAI_BASE_URL, is_lmstudio_url
1616
import server
1717
import requests
1818
from ..utils.logger import log
@@ -34,8 +34,8 @@ async def list_models(request):
3434
"""
3535
try:
3636
log.info("Received list_models request")
37-
openai_api_key = request.headers.get('Openai-Api-Key') or ""
38-
openai_base_url = request.headers.get('Openai-Base-Url') or LLM_DEFAULT_BASE_URL
37+
openai_api_key = request.headers.get('Openai-Api-Key') or OPENAI_API_KEY or ""
38+
openai_base_url = request.headers.get('Openai-Base-Url') or OPENAI_BASE_URL or LLM_DEFAULT_BASE_URL
3939

4040
request_url = f"{openai_base_url}/models"
4141

0 commit comments

Comments
 (0)