feat: 跟踪使用情况

Author: ai-business-hql

__init__.py

@@ -1,16 +1,17 @@
 '''
 Author: ai-business-hql [email protected]
 Date: 2025-02-17 20:53:45
-LastEditors: ai-business-hql qingli.hql@alibaba-inc.com
-LastEditTime: 2025-06-24 17:09:23
+LastEditors: ai-business-hql ai.bussiness.hql@gmail.com
+LastEditTime: 2025-11-20 20:03:20
 FilePath: /comfyui_copilot/__init__.py
 Description: 这是默认设置,请设置`customMade`, 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
 '''
 # Copyright (C) 2025 AIDC-AI
 # Licensed under the MIT License.
 
 import sys
-import os
+import asyncio
+
 
 # Ensure 'agents' resolves to openai-agents (not legacy RL package)
 try:

backend/agent_factory.py

@@ -2,13 +2,13 @@
 Author: ai-business-hql [email protected]
 Date: 2025-07-31 19:38:08
 LastEditors: ai-business-hql [email protected]
-LastEditTime: 2025-09-03 11:00:57
+LastEditTime: 2025-11-20 17:51:33
 FilePath: /comfyui_copilot/backend/agent_factory.py
 Description: 这是默认设置,请设置`customMade`, 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
 '''
-import os
+
 try:
-    from agents import Agent, OpenAIChatCompletionsModel, ModelSettings
+    from agents import Agent, OpenAIChatCompletionsModel, ModelSettings, Runner, set_trace_processors, set_tracing_disabled, set_default_openai_api
     if not hasattr(__import__('agents'), 'Agent'):
         raise ImportError
 except Exception:
@@ -28,6 +28,7 @@
 
 from agents._config import set_default_openai_api
 from agents.tracing import set_tracing_disabled
+import asyncio
 # from .utils.logger import log
 
 # def load_env_config():
@@ -46,7 +47,9 @@
 # load_env_config()
 
 set_default_openai_api("chat_completions")
-set_tracing_disabled(True)
+set_tracing_disabled(False)
+
+
 def create_agent(**kwargs) -> Agent:
     # 通过用户配置拿/环境变量
     config = kwargs.pop("config") if "config" in kwargs else {}
@@ -59,13 +62,13 @@ def create_agent(**kwargs) -> Agent:
     # Determine base URL and API key
     base_url = LLM_DEFAULT_BASE_URL
     api_key = get_comfyui_copilot_api_key() or ""
-    
+
     if config:
         if config.get("openai_base_url") and config.get("openai_base_url") != "":
             base_url = config.get("openai_base_url")
         if config.get("openai_api_key") and config.get("openai_api_key") != "":
             api_key = config.get("openai_api_key")
-    
+
     # Check if this is LMStudio and adjust API key handling
     is_lmstudio = is_lmstudio_url(base_url)
     if is_lmstudio and not api_key:

backend/service/mcp_client.py

@@ -2,7 +2,7 @@
 Author: ai-business-hql [email protected]
 Date: 2025-06-16 16:50:17
 LastEditors: ai-business-hql [email protected]
-LastEditTime: 2025-11-19 20:16:03
+LastEditTime: 2025-11-20 17:44:52
 FilePath: /comfyui_copilot/backend/service/mcp-client.py
 Description: 这是默认设置,请设置`customMade`, 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
 '''
@@ -222,6 +222,13 @@ async def on_handoff(ctx: RunContextWrapper[None], input_data: HandoffRewriteDat
             agent_input = messages
             log.info(f"-- Processing {len(messages)} messages")
 
+            from agents import Agent, Runner, set_trace_processors, set_tracing_disabled, set_default_openai_api
+            from langsmith.wrappers import OpenAIAgentsTracingProcessor
+
+            set_tracing_disabled(False)
+            set_default_openai_api("chat_completions")
+            set_trace_processors([OpenAIAgentsTracingProcessor()])
+
             result = Runner.run_streamed(
                 agent,
                 input=agent_input,

requirements.txt

@@ -11,7 +11,8 @@ httpx>=0.24.0
 pytest>=6.0.0
 pytest-asyncio>=0.18.0 
 openai-agents>=0.3.0
+langsmith>=0.4.0
 fastmcp
 
 modelscope>=1.28.0
-urllib3>=1.26,<2.0
+urllib3>=1.26,<2.0