newtw

run.py

@@ -23,11 +23,9 @@ _root = Path(__file__).resolve().parent
 if str(_root) not in sys.path:
     sys.path.insert(0, str(_root))
 
-logging.basicConfig(
-    level=logging.INFO,
-    format='[%(asctime)s] %(levelname)s: %(message)s'
-)
-logger = logging.getLogger(__name__)
+from core.websocket_logger_setup import setup_logger
+
+logger = setup_logger()
 
 DEFAULT_HTTP_PORT = 6060
 DEFAULT_HTTP_HOST = "127.0.0.1"
@@ -65,7 +63,7 @@ def run_tianwang(enable_agent: bool, host: str, port: int):
     """完整版: HTTP API + WebSocket + AI Agent"""
     import asyncio
     from api.http_server import start_http_server
-    from core.websocket_client import QingjianAPIClient
+    from core.websocket_client_v2 import QingjianAPIClient
 
     logger.info("=" * 60)
     logger.info("AI 客服系统 - 天网协作版（完整）")