Merge branch 'mem'

Author: ai-business-hql

backend/controller/llm_api.py

@@ -2,7 +2,7 @@
 Author: ai-business-hql [email protected]
 Date: 2025-07-14 16:46:20
 LastEditors: ai-business-hql [email protected]
-LastEditTime: 2025-12-12 16:32:50
+LastEditTime: 2025-12-15 15:03:28
 FilePath: /comfyui_copilot/backend/controller/llm_api.py
 Description: 这是默认设置,请设置`customMade`, 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
 '''
@@ -12,7 +12,7 @@
 import json
 from typing import List, Dict, Any
 from aiohttp import web
-from ..utils.globals import LLM_DEFAULT_BASE_URL, LMSTUDIO_DEFAULT_BASE_URL, OPENAI_API_KEY, OPENAI_BASE_URL, is_lmstudio_url
+from ..utils.globals import LLM_DEFAULT_BASE_URL, LMSTUDIO_DEFAULT_BASE_URL, OPENAI_API_KEY, OPENAI_BASE_URL, TENANT_ID, is_lmstudio_url
 import server
 import requests
 from ..utils.logger import log
@@ -34,6 +34,19 @@ async def list_models(request):
     """
     try:
         log.info("Received list_models request")
+        if TENANT_ID:
+            model_list = [ "gemini-2.5-flash", "gpt-5-nano", "gpt-5-mini", "gpt-5" ]
+            llm_config = []
+            for model in model_list:
+                llm_config.append({
+                    "label": model,
+                    "name": model,
+                    "image_enable": True
+                })
+            return web.json_response({
+                "models": llm_config
+            })
+        
         openai_api_key = request.headers.get('Openai-Api-Key') or OPENAI_API_KEY or ""
         openai_base_url = request.headers.get('Openai-Base-Url') or OPENAI_BASE_URL or LLM_DEFAULT_BASE_URL
 

backend/utils/globals.py

@@ -2,7 +2,7 @@
 Author: ai-business-hql [email protected]
 Date: 2025-08-08 17:14:52
 LastEditors: ai-business-hql [email protected]
-LastEditTime: 2025-11-27 11:12:18
+LastEditTime: 2025-12-15 14:49:18
 FilePath: /comfyui_copilot/backend/utils/globals.py
 Description: 这是默认设置,请设置`customMade`, 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
 '''
@@ -114,6 +114,7 @@ def set_comfyui_copilot_api_key(api_key: str) -> None:
 WORKFLOW_LLM_MODEL = os.getenv("WORKFLOW_LLM_MODEL") or WORKFLOW_MODEL_NAME
 DISABLE_WORKFLOW_GEN = os.getenv("DISABLE_WORKFLOW_GEN") or False
 
+TENANT_ID = os.getenv("TENANT_ID") or None
 
 def apply_llm_env_defaults(config: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
     """