106 lines
3.8 KiB
Python
106 lines
3.8 KiB
Python
from fastapi import FastAPI
|
||
from fastapi.middleware.cors import CORSMiddleware
|
||
|
||
import os
|
||
|
||
from database import init_db, SessionLocal
|
||
from models import Avatar, Authorization, Organization, TokenAccount, TokenPlan
|
||
from fastapi.staticfiles import StaticFiles
|
||
import routers.avatars
|
||
import routers.tokens
|
||
import routers.authorizations
|
||
import routers.organizations
|
||
import routers.knowledge
|
||
import routers.huihui_auth
|
||
import routers.chat
|
||
from responses import ok
|
||
|
||
app = FastAPI(title="会会数字分身 API", version="1.0.0")
|
||
|
||
app.add_middleware(
|
||
CORSMiddleware,
|
||
allow_origins=["*"],
|
||
allow_credentials=False,
|
||
allow_methods=["*"],
|
||
allow_headers=["*"],
|
||
)
|
||
|
||
app.include_router(routers.avatars.router, prefix="/api")
|
||
app.include_router(routers.tokens.router, prefix="/api")
|
||
app.include_router(routers.authorizations.router, prefix="/api")
|
||
app.include_router(routers.organizations.router, prefix="/api")
|
||
app.include_router(routers.knowledge.router, prefix="/api")
|
||
app.include_router(routers.huihui_auth.router, prefix="/api")
|
||
app.include_router(routers.chat.router, prefix="/api")
|
||
|
||
UPLOAD_DIR = routers.knowledge.UPLOAD_DIR
|
||
os.makedirs(UPLOAD_DIR, exist_ok=True)
|
||
app.mount("/api/files", StaticFiles(directory=UPLOAD_DIR), name="knowledge-files")
|
||
|
||
|
||
@app.get("/api/health")
|
||
def health():
|
||
return ok({"status": "ok"})
|
||
|
||
|
||
def seed():
|
||
db = SessionLocal()
|
||
try:
|
||
if db.query(TokenAccount).first() is None:
|
||
db.add(TokenAccount(balance=1250))
|
||
|
||
if db.query(TokenPlan).count() == 0:
|
||
plans = [
|
||
TokenPlan(id="1", name="新手体验", amount=1000, price=9.9, desc="新手体验"),
|
||
TokenPlan(id="2", name="热门套餐", amount=5000, price=39.9, badge="热门"),
|
||
TokenPlan(id="3", name="超值套餐", amount=12000, price=89.9, badge="超值"),
|
||
TokenPlan(id="4", name="企业推荐", amount=30000, price=199, badge="企业推荐", desc="适合高频使用"),
|
||
]
|
||
db.add_all(plans)
|
||
|
||
if db.query(Avatar).count() == 0:
|
||
avatar = Avatar(
|
||
name="我的数字分身",
|
||
display_name="会会助手",
|
||
description="我是您的AI数字分身，可以帮您管理日程、回复消息、处理任务。",
|
||
emoji="🤖",
|
||
status="active",
|
||
token_balance=0,
|
||
config={
|
||
"replyStyle": "professional",
|
||
"creativity": 50,
|
||
"rigor": 50,
|
||
"humor": 30,
|
||
"responseLength": "medium",
|
||
"systemPrompt": "",
|
||
},
|
||
)
|
||
db.add(avatar)
|
||
db.commit()
|
||
db.refresh(avatar)
|
||
|
||
if db.query(Authorization).count() == 0:
|
||
auths = [
|
||
Authorization(avatar_id=avatar.id, target_type="application", target_name="微信小程序", permissions=["read", "reply"], status="active"),
|
||
Authorization(avatar_id=avatar.id, target_type="user", target_name="张三", permissions=["read"], status="active"),
|
||
Authorization(avatar_id=avatar.id, target_type="organization", target_name="产品团队", permissions=["read", "edit"], status="inactive"),
|
||
]
|
||
db.add_all(auths)
|
||
|
||
if db.query(Organization).count() == 0:
|
||
orgs = [
|
||
Organization(name="会会增长团队", description="负责会会产品的增长与运营", emoji="🚀", org_type="team", member_count=12),
|
||
Organization(name="AI 实验室", description="探索前沿 AI 能力", emoji="💡", org_type="company", member_count=8),
|
||
]
|
||
db.add_all(orgs)
|
||
|
||
db.commit()
|
||
finally:
|
||
db.close()
|
||
|
||
|
||
@app.on_event("startup")
|
||
def on_startup():
|
||
init_db()
|
||
seed()
|