feat:.env适配test

This commit is contained in:
ai-business-hql
2025-10-11 16:33:42 +08:00
parent 6efde26efd
commit 0ec79b8b42
5 changed files with 24 additions and 15 deletions
+1 -1
View File
@@ -206,7 +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)
config = workflow_config_adapt(config)
if not session_id:
session_id = str(uuid.uuid4()) # Fallback if no context
+1 -4
View File
@@ -2,7 +2,7 @@
Author: ai-business-hql qingli.hql@alibaba-inc.com
Date: 2025-06-16 16:50:17
LastEditors: ai-business-hql ai.bussiness.hql@gmail.com
LastEditTime: 2025-09-02 17:11:50
LastEditTime: 2025-10-11 16:32:59
FilePath: /comfyui_copilot/backend/service/mcp-client.py
Description: 这是默认设置,请设置`customMade`, 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
'''
@@ -80,9 +80,6 @@ async def comfyui_agent_invoke(messages: List[Dict[str, Any]], images: List[Imag
cache_tools_list=True,
client_session_timeout_seconds=300.0
) as server:
# tools = await server.list_tools()
# Model selection is handled inside create_agent via config / kwargs precedence
# 创建workflow_rewrite_agent实例 (session_id通过context获取)
workflow_rewrite_agent_instance = create_workflow_rewrite_agent()
+1 -1
View File
@@ -45,7 +45,7 @@ def create_workflow_rewrite_agent():
language = get_language()
session_id = get_session_id() or "unknown_session"
config = get_config()
workflow_config_adapt(config)
config = workflow_config_adapt(config)
return create_agent(
name="Workflow Rewrite Agent",
@@ -83,7 +83,7 @@ ComfyUI API格式工作流是一个JSON对象,其中:
}
"""
config = get_config()
workflow_config_adapt(config)
config = workflow_config_adapt(config)
# 创建OpenAI客户端
client = OpenAI(
base_url = config.get("openai_base_url") or LLM_DEFAULT_BASE_URL,
+20 -8
View File
@@ -1,10 +1,22 @@
from copy import deepcopy
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
def workflow_config_adapt(config: dict) -> dict:
"""Return a deep-copied and adapted workflow config without mutating input.
- Map workflow_llm_api_key -> openai_api_key (and null out original key)
- Map workflow_llm_base_url -> openai_base_url (and null out original key)
"""
if not config:
return {}
new_config = deepcopy(config)
if new_config.get("workflow_llm_api_key"):
new_config["openai_api_key"] = new_config.get("workflow_llm_api_key")
new_config["workflow_llm_api_key"] = None
if new_config.get("workflow_llm_base_url"):
new_config["openai_base_url"] = new_config.get("workflow_llm_base_url")
new_config["workflow_llm_base_url"] = None
return new_config