feat: 上下文8192

backend/agent_factory.py

@@ -2,11 +2,11 @@
 Author: ai-business-hql qingli.hql@alibaba-inc.com
 Date: 2025-07-31 19:38:08
 LastEditors: ai-business-hql ai.bussiness.hql@gmail.com
-LastEditTime: 2025-08-25 17:42:41
+LastEditTime: 2025-08-25 21:23:10
 FilePath: /comfyui_copilot/backend/agent_factory.py
 Description: 这是默认设置,请设置`customMade`, 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
 '''
-from agents import Agent, OpenAIChatCompletionsModel
+from agents import Agent, OpenAIChatCompletionsModel, ModelSettings
 from dotenv import dotenv_values
 from .utils.globals import LLM_DEFAULT_BASE_URL, LMSTUDIO_DEFAULT_BASE_URL, get_comfyui_copilot_api_key, is_lmstudio_url
 from openai import AsyncOpenAI
@@ -68,5 +68,7 @@ def create_agent(**kwargs) -> Agent:
     default_model_name = os.environ.get("OPENAI_MODEL", "gemini-2.5-flash")
     model_name = kwargs.pop("model") or default_model_name
     model = OpenAIChatCompletionsModel(model_name, openai_client=client)
-    
+
+    if config.get("max_tokens"):
+        return Agent(model=model, model_settings=ModelSettings(max_tokens=config.get("max_tokens") or 8192), **kwargs)
     return Agent(model=model, **kwargs)

backend/service/workflow_rewrite_agent.py

@@ -105,6 +105,9 @@ def create_workflow_rewrite_agent():
         始终以用户的实际需求为导向，提供专业、准确、高效的工作流改写服务。
         """,
         tools=[get_rewrite_expert_by_name, get_current_workflow, get_node_info, update_workflow, remove_node],
+        config={
+            "max_tokens": 8192
+        }
     )
 
 # 注意：工作流改写代理现在需要在有session context的环境中创建