feat: 开放所有apikey

Author: ai-business-hql

backend/controller/conversation_api.py

@@ -239,6 +239,9 @@ async def invoke_chat(request):
         "workflow_checkpoint_id": workflow_checkpoint_id,
         "openai_api_key": request.headers.get('Openai-Api-Key'),
         "openai_base_url": request.headers.get('Openai-Base-Url'),
+        # Workflow LLM settings (optional, used by tools/agents that need a different LLM)
+        "workflow_llm_api_key": request.headers.get('Workflow-LLM-Api-Key'),
+        "workflow_llm_base_url": request.headers.get('Workflow-LLM-Base-Url'),
         "model_select": next((x['data'][0] for x in ext if x['type'] == 'model_select' and x.get('data')), None)
     }
 
@@ -504,7 +507,10 @@ async def invoke_debug(request):
         "session_id": session_id,
         "model": "gemini-2.5-flash",  # Default model for debug agents
         "openai_api_key": request.headers.get('Openai-Api-Key'),
-        "openai_base_url": request.headers.get('Openai-Base-Url', 'https://api.openai.com/v1'),
+        "openai_base_url": request.headers.get('Openai-Base-Url'),
+        # Workflow LLM settings (optional)
+        "workflow_llm_api_key": request.headers.get('Workflow-LLM-Api-Key'),
+        "workflow_llm_base_url": request.headers.get('Workflow-LLM-Base-Url'),
     }
 
     # 获取当前语言

backend/service/debug_agent.py

@@ -1,6 +1,7 @@
 '''
 Debug Agent for ComfyUI Workflow Error Analysis
 '''
+from ..utils.key_utils import workflow_config_adapt
 from ..agent_factory import create_agent
 from agents.items import ItemHelpers
 from agents.run import Runner
@@ -205,6 +206,7 @@ async def debug_workflow_errors(workflow_data: Dict[str, Any]):
         # Get session_id and config from request context
         session_id = get_session_id()
         config = get_config()
+        workflow_config_adapt(config)
 
         if not session_id:
             session_id = str(uuid.uuid4())  # Fallback if no context
@@ -253,7 +255,8 @@ async def debug_workflow_errors(workflow_data: Dict[str, Any]):
             model=WORKFLOW_MODEL_NAME,
             tools=[run_workflow, analyze_error_type, save_current_workflow],
             config={
-                "max_tokens": 8192
+                "max_tokens": 8192,
+                **config
             }
         )
 
@@ -302,7 +305,8 @@ async def debug_workflow_errors(workflow_data: Dict[str, Any]):
             tools=[get_current_workflow, get_node_info, update_workflow],
             handoffs=[agent],
             config={
-                "max_tokens": 8192
+                "max_tokens": 8192,
+                **config
             }
         )
 
@@ -393,7 +397,8 @@ async def debug_workflow_errors(workflow_data: Dict[str, Any]):
                    get_current_workflow, get_node_info],
             handoffs=[agent],
             config={
-                "max_tokens": 8192
+                "max_tokens": 8192,
+                **config
             }
         )
 
@@ -487,7 +492,8 @@ async def debug_workflow_errors(workflow_data: Dict[str, Any]):
                 suggest_model_download, update_workflow_parameter, get_current_workflow],
             handoffs=[agent],
             config={
-                "max_tokens": 8192
+                "max_tokens": 8192,
+                **config
             }
         )
 
@@ -692,16 +698,9 @@ async def debug_workflow_errors(workflow_data: Dict[str, Any]):
 
     except Exception as e:
         log.error(f"Error in debug_workflow_errors: {str(e)}")
-        error_message = current_text + f"\n\n× Error occurred during debugging: {str(e)}\n\n"
+        error_message = f"\n\n× Error occurred during debugging: {str(e)}\n\n"
 
-        # Include workflow_update ext if captured from tools before the error
-        final_error_ext = None
-        if 'workflow_update_ext' in locals() and workflow_update_ext:
-            final_error_ext = [workflow_update_ext]
-            log.info(f"-- Including latest workflow_update ext in error response")
-        
         ext_with_finished = {
-            "data": final_error_ext,
             "finished": True
         }
         yield (error_message, ext_with_finished)

backend/service/workflow_rewrite_agent.py

@@ -2,7 +2,7 @@
 Author: ai-business-hql qingli.hql@alibaba-inc.com
 Date: 2025-07-24 17:10:23
 LastEditors: ai-business-hql ai.bussiness.hql@gmail.com
-LastEditTime: 2025-08-28 11:15:49
+LastEditTime: 2025-09-30 11:03:37
 FilePath: /comfyui_copilot/backend/service/workflow_rewrite_agent.py
 Description: 这是默认设置,请设置`customMade`, 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
 '''
@@ -14,11 +14,13 @@
 import os
 from typing import Dict, Any
 
+from ..utils.key_utils import workflow_config_adapt
+
 from ..dao.expert_table import list_rewrite_experts_short, get_rewrite_expert_by_name_list
 
 from ..agent_factory import create_agent
 from ..utils.globals import WORKFLOW_MODEL_NAME, get_language
-from ..utils.request_context import get_session_id
+from ..utils.request_context import get_config, get_session_id
 
 from ..service.workflow_rewrite_tools import *
 
@@ -42,7 +44,9 @@ def create_workflow_rewrite_agent():
 
     language = get_language()
     session_id = get_session_id() or "unknown_session"
-    
+    config = get_config()
+    workflow_config_adapt(config)
+
     return create_agent(
         name="Workflow Rewrite Agent",
         model=WORKFLOW_MODEL_NAME,

backend/service/workflow_rewrite_agent_simple.py

@@ -11,11 +11,12 @@
 from typing import Dict, Any, Optional
 import asyncio
 from pydantic import BaseModel
+from ..utils.key_utils import workflow_config_adapt
 from openai import OpenAI
 
 from ..agent_factory import create_agent
 from ..utils.globals import WORKFLOW_MODEL_NAME, get_comfyui_copilot_api_key, LLM_DEFAULT_BASE_URL
-from ..utils.request_context import get_rewrite_context, RewriteContext
+from ..utils.request_context import get_config, get_rewrite_context, RewriteContext
 from ..utils.logger import log
 
 
@@ -81,11 +82,12 @@ def rewrite_workflow_simple(rewrite_context: RewriteContext) -> str:
   "workflow_data": "这里是完整的工作流JSON字符串"
 }
 """
-
+        config = get_config()
+        workflow_config_adapt(config)
         # 创建OpenAI客户端
         client = OpenAI(
-            base_url = LLM_DEFAULT_BASE_URL,
-            api_key = get_comfyui_copilot_api_key() or ""
+            base_url = config.get("openai_base_url") or LLM_DEFAULT_BASE_URL,
+            api_key = config.get("openai_api_key") or get_comfyui_copilot_api_key() or ""
         )
 
         # 调用LLM

backend/utils/globals.py

@@ -2,11 +2,7 @@
 Author: ai-business-hql qingli.hql@alibaba-inc.com
 Date: 2025-08-08 17:14:52
 LastEditors: ai-business-hql ai.bussiness.hql@gmail.com
-<<<<<<< HEAD
-LastEditTime: 2025-08-25 17:41:14
-=======
-LastEditTime: 2025-08-29 11:25:52
->>>>>>> mcp
+LastEditTime: 2025-09-30 10:18:44
 FilePath: /comfyui_copilot/backend/utils/globals.py
 Description: 这是默认设置,请设置`customMade`, 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
 '''
@@ -15,8 +11,16 @@
 Global utilities for managing application-wide state and configuration.
 """
 
+import os
 import threading
 from typing import Optional, Dict, Any
+from pathlib import Path
+from dotenv import load_dotenv
+
+# Load .env file if it exists
+env_path = Path(__file__).parent.parent.parent / '.env'
+if env_path.exists():
+    load_dotenv(env_path)
 
 class GlobalState:
     """Thread-safe global state manager for application-wide configuration."""
@@ -95,11 +99,11 @@ def set_comfyui_copilot_api_key(api_key: str) -> None:
     _global_state.set('comfyui_copilot_api_key', api_key)
 
 
-BACKEND_BASE_URL = "https://comfyui-copilot-server.onrender.com"
+BACKEND_BASE_URL = os.getenv("BACKEND_BASE_URL", "https://comfyui-copilot-server.onrender.com")
 LMSTUDIO_DEFAULT_BASE_URL = "http://localhost:1234/v1"
-WORKFLOW_MODEL_NAME = "us.anthropic.claude-sonnet-4-20250514-v1:0"
+WORKFLOW_MODEL_NAME = os.getenv("WORKFLOW_MODEL_NAME", "us.anthropic.claude-sonnet-4-20250514-v1:0")
 # WORKFLOW_MODEL_NAME = "gpt-5-2025-08-07-GlobalStandard"
-LLM_DEFAULT_BASE_URL = BACKEND_BASE_URL + "/v1"
+LLM_DEFAULT_BASE_URL = os.getenv("LLM_DEFAULT_BASE_URL", BACKEND_BASE_URL + "/v1")
 
 
 def is_lmstudio_url(base_url: str) -> bool:

backend/utils/key_utils.py

@@ -0,0 +1,10 @@
+
+
+def workflow_config_adapt(config: dict):
+    if config:
+        if config.get("workflow_llm_api_key"):
+            config["openai_api_key"] = config.get("workflow_llm_api_key")
+            config["workflow_llm_api_key"] = None
+        if config.get("workflow_llm_base_url"):
+            config["openai_base_url"] = config.get("workflow_llm_base_url")
+            config["workflow_llm_base_url"] = None