From 08c8e8c9a4d65677de6a493446a605d70efee631 Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期二, 10 十二月 2024 16:32:07 +0800 Subject: [PATCH] 12.10 16 --- app/task/fetch_agent.py | 12 ++++++++++++ 1 files changed, 12 insertions(+), 0 deletions(-) diff --git a/app/task/fetch_agent.py b/app/task/fetch_agent.py index 36b3c2e..42aa3e0 100644 --- a/app/task/fetch_agent.py +++ b/app/task/fetch_agent.py @@ -1,12 +1,14 @@ from typing import Dict, List, Tuple from sqlalchemy import create_engine, Column, String, Integer +from sqlalchemy.dialects.postgresql import array from sqlalchemy.exc import IntegrityError from sqlalchemy.orm import sessionmaker from app.config.config import settings from app.models.agent_model import AgentModel from app.models.base_model import SessionLocal, Base +from app.service.v2.initialize_data import dialog_menu_sync # 鍒涘缓鏁版嵁搴撳紩鎿庡拰浼氳瘽宸ュ巶 engine_bisheng = create_engine(settings.sgb_db_url) @@ -147,3 +149,13 @@ print("Agents synchronized successfully") except Exception as e: print(f"Failed to sync agents: {str(e)}") + + +async def sync_web_menu(): + db = SessionLocal() + await dialog_menu_sync(db) + + +async def sync_default_group(): + db = SessionLocal() + await dialog_menu_sync(db) \ No newline at end of file -- Gitblit v1.8.0