feat: 新增图片视频并发设置

新增token导入导出为json
chore: 完善token刷新日志输出
fix: 修复自动更新时无法根据AT有效期禁用token问题

src/main.py

@@ -14,6 +14,7 @@ from .services.proxy_manager import ProxyManager
 from .services.load_balancer import LoadBalancer
 from .services.sora_client import SoraClient
 from .services.generation_handler import GenerationHandler
+from .services.concurrency_manager import ConcurrencyManager
 from .api import routes as api_routes
 from .api import admin as admin_routes
 
@@ -37,13 +38,14 @@ app.add_middleware(
 db = Database()
 token_manager = TokenManager(db)
 proxy_manager = ProxyManager(db)
-load_balancer = LoadBalancer(token_manager)
+concurrency_manager = ConcurrencyManager()
+load_balancer = LoadBalancer(token_manager, concurrency_manager)
 sora_client = SoraClient(proxy_manager)
-generation_handler = GenerationHandler(sora_client, token_manager, load_balancer, db, proxy_manager)
+generation_handler = GenerationHandler(sora_client, token_manager, load_balancer, db, proxy_manager, concurrency_manager)
 
 # Set dependencies for route modules
 api_routes.set_generation_handler(generation_handler)
-admin_routes.set_dependencies(token_manager, proxy_manager, db, generation_handler)
+admin_routes.set_dependencies(token_manager, proxy_manager, db, generation_handler, concurrency_manager)
 
 # Include routers
 app.include_router(api_routes.router)
@@ -127,6 +129,11 @@ async def startup_event():
     token_refresh_config = await db.get_token_refresh_config()
     config.set_at_auto_refresh_enabled(token_refresh_config.at_auto_refresh_enabled)
 
+    # Initialize concurrency manager with all tokens
+    all_tokens = await db.get_all_tokens()
+    await concurrency_manager.initialize(all_tokens)
+    print(f"✓ Concurrency manager initialized with {len(all_tokens)} tokens")
+
     # Start file cache cleanup task
     await generation_handler.file_cache.start_cleanup_task()