问题优化

zhaoqingang

2024-12-13 f2c43713b16e821f4a1ae97804c60f18734d5f6a

 app/task/fetch_agent.py

@@ -1,14 +1,12 @@
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, default_group_sync, default_role_sync

# 创建数据库引擎和会话工厂
engine_bisheng = create_engine(settings.sgb_db_url)
@@ -151,13 +149,5 @@
        print(f"Failed to sync agents: {str(e)}")


async def sync_web_menu():
    db = SessionLocal()
    await dialog_menu_sync(db)


async def sync_default_data():
    db = SessionLocal()
    await default_group_sync(db)
    await default_role_sync(db)