Hermes Agent
95 lines
3.9 KiB
Python
95 lines
3.9 KiB
Python
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.core.config import settings
|
|
from app.core.security import hash_password
|
|
from app.models import Agent, AgentTeam, Announcement, LLMModel, LLMProvider, Menu, Permission, Role, User, Workflow
|
|
|
|
|
|
async def seed_database(db: AsyncSession) -> None:
|
|
exists = await db.scalar(select(User.id).limit(1))
|
|
if exists:
|
|
return
|
|
|
|
admin = User(
|
|
email=settings.seed_admin_email,
|
|
username="admin",
|
|
nickname="系统管理员",
|
|
password_hash=hash_password(settings.seed_admin_password),
|
|
is_superuser=True,
|
|
)
|
|
role = Role(name="超级管理员", code="admin", description="系统内置管理员")
|
|
permissions = [
|
|
Permission(name="用户管理", code="core:user", resource="user", action="manage"),
|
|
Permission(name="角色管理", code="core:role", resource="role", action="manage"),
|
|
Permission(name="AI 管理", code="ai:manage", resource="ai", action="manage"),
|
|
]
|
|
role.permissions = permissions
|
|
admin.roles = [role]
|
|
|
|
menus = [
|
|
Menu(title="工作台", path="/", icon="Monitor"),
|
|
Menu(title="用户管理", path="/system/users", icon="User"),
|
|
Menu(title="角色管理", path="/system/roles", icon="Lock"),
|
|
Menu(title="菜单管理", path="/system/menus", icon="Menu"),
|
|
Menu(title="权限管理", path="/system/permissions", icon="Key"),
|
|
Menu(title="公告管理", path="/system/announcements", icon="Bell"),
|
|
Menu(title="Provider", path="/ai/providers", icon="Connection"),
|
|
Menu(title="Model", path="/ai/models", icon="Cpu"),
|
|
Menu(title="Agent", path="/ai/agents", icon="Avatar"),
|
|
Menu(title="Agent Chat", path="/ai/chat", icon="ChatLineRound"),
|
|
Menu(title="Workflow", path="/ai/workflows", icon="Share"),
|
|
Menu(title="Workflow Runs", path="/ai/workflow-runs", icon="Tickets"),
|
|
Menu(title="Knowledge Base", path="/ai/knowledge", icon="Collection"),
|
|
Menu(title="Agent Team", path="/ai/teams", icon="Operation"),
|
|
]
|
|
|
|
provider = LLMProvider(
|
|
name="OpenAI Compatible",
|
|
code="openai_compatible",
|
|
provider_type="openai_compatible",
|
|
base_url="",
|
|
api_key="",
|
|
)
|
|
model = LLMModel(provider=provider, name="gpt-compatible", display_name="默认兼容模型")
|
|
agent = Agent(
|
|
name="Planner",
|
|
code="planner",
|
|
status="published",
|
|
system_prompt="你是任务规划智能体，负责拆解目标并给出执行计划。",
|
|
model=model,
|
|
)
|
|
workflow = Workflow(
|
|
name="最小 LLM 工作流",
|
|
code="minimal_llm",
|
|
status="published",
|
|
definition={
|
|
"nodes": [
|
|
{"id": "start", "type": "start", "data": {}},
|
|
{"id": "plan", "type": "agent", "data": {"agent_id": ""}},
|
|
{"id": "end", "type": "end", "data": {}},
|
|
],
|
|
"edges": [],
|
|
},
|
|
)
|
|
team = AgentTeam(
|
|
name="默认协作团队",
|
|
code="default_team",
|
|
description="Planner -> Executor -> Reviewer 的最小协作演示",
|
|
members=[{"agent_id": "", "role": "planner"}, {"agent_id": "", "role": "reviewer"}],
|
|
)
|
|
announcement = Announcement(title="AI Agent Admin 已初始化", content="轻量管理后台已可用。", status="published")
|
|
|
|
db.add_all([admin, role, *permissions, *menus, provider, model, agent, workflow, team, announcement])
|
|
await db.flush()
|
|
workflow.definition = {
|
|
"nodes": [
|
|
{"id": "start", "type": "start", "data": {}},
|
|
{"id": "plan", "type": "agent", "data": {"agent_id": agent.id}},
|
|
{"id": "end", "type": "end", "data": {}},
|
|
],
|
|
"edges": [],
|
|
}
|
|
team.members = [{"agent_id": agent.id, "role": "planner"}, {"agent_id": agent.id, "role": "reviewer"}]
|
|
await db.commit()
|