147 lines
4.7 KiB
Python
147 lines
4.7 KiB
Python
"""Viewer 文件系统路由
|
|
|
|
提供 Viewer UI 使用的文件系统 API 端点。
|
|
- /viewer/filesystem/* - Viewer UI 使用
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from fastapi import APIRouter, Depends, File, Form, Query, UploadFile
|
|
from pydantic import BaseModel
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from server.utils.auth_middleware import get_db, get_required_user
|
|
from yuxi.services.viewer_filesystem_service import (
|
|
create_viewer_directory,
|
|
delete_viewer_file,
|
|
download_viewer_file,
|
|
list_viewer_filesystem_tree,
|
|
read_viewer_file_content,
|
|
upload_viewer_file,
|
|
)
|
|
from yuxi.storage.postgres.models_business import User
|
|
|
|
filesystem_router = APIRouter(prefix="/viewer/filesystem", tags=["viewer-filesystem"])
|
|
|
|
|
|
class CreateViewerDirectoryRequest(BaseModel):
|
|
thread_id: str
|
|
parent_path: str
|
|
name: str
|
|
agent_id: str | None = None
|
|
agent_config_id: int | None = None
|
|
|
|
|
|
@filesystem_router.get("/tree", response_model=dict)
|
|
async def get_viewer_tree(
|
|
thread_id: str = Query(..., description="线程 ID"),
|
|
path: str = Query("/", description="目录路径"),
|
|
agent_id: str | None = Query(None, description="智能体 ID"),
|
|
agent_config_id: int | None = Query(None, description="智能体配置 ID"),
|
|
current_user: User = Depends(get_required_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
return await list_viewer_filesystem_tree(
|
|
thread_id=thread_id,
|
|
path=path,
|
|
agent_id=agent_id,
|
|
agent_config_id=agent_config_id,
|
|
current_user=current_user,
|
|
db=db,
|
|
)
|
|
|
|
|
|
@filesystem_router.get("/file", response_model=dict)
|
|
async def get_viewer_file(
|
|
thread_id: str = Query(..., description="线程 ID"),
|
|
path: str = Query(..., description="文件路径"),
|
|
agent_id: str | None = Query(None, description="智能体 ID"),
|
|
agent_config_id: int | None = Query(None, description="智能体配置 ID"),
|
|
current_user: User = Depends(get_required_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
return await read_viewer_file_content(
|
|
thread_id=thread_id,
|
|
path=path,
|
|
agent_id=agent_id,
|
|
agent_config_id=agent_config_id,
|
|
current_user=current_user,
|
|
db=db,
|
|
)
|
|
|
|
|
|
@filesystem_router.delete("/file", response_model=dict)
|
|
async def delete_viewer_file_route(
|
|
thread_id: str = Query(..., description="线程 ID"),
|
|
path: str = Query(..., description="文件路径"),
|
|
agent_id: str | None = Query(None, description="智能体 ID"),
|
|
agent_config_id: int | None = Query(None, description="智能体配置 ID"),
|
|
current_user: User = Depends(get_required_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
return await delete_viewer_file(
|
|
thread_id=thread_id,
|
|
path=path,
|
|
agent_id=agent_id,
|
|
agent_config_id=agent_config_id,
|
|
current_user=current_user,
|
|
db=db,
|
|
)
|
|
|
|
|
|
@filesystem_router.post("/directory", response_model=dict)
|
|
async def create_viewer_directory_route(
|
|
payload: CreateViewerDirectoryRequest,
|
|
current_user: User = Depends(get_required_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
return await create_viewer_directory(
|
|
thread_id=payload.thread_id,
|
|
parent_path=payload.parent_path,
|
|
name=payload.name,
|
|
agent_id=payload.agent_id,
|
|
agent_config_id=payload.agent_config_id,
|
|
current_user=current_user,
|
|
db=db,
|
|
)
|
|
|
|
|
|
@filesystem_router.post("/upload", response_model=dict)
|
|
async def upload_viewer_file_route(
|
|
thread_id: str = Form(..., description="线程 ID"),
|
|
parent_path: str = Form(..., description="父目录路径"),
|
|
agent_id: str | None = Form(None, description="智能体 ID"),
|
|
agent_config_id: int | None = Form(None, description="智能体配置 ID"),
|
|
file: UploadFile = File(..., description="上传文件"),
|
|
current_user: User = Depends(get_required_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
return await upload_viewer_file(
|
|
thread_id=thread_id,
|
|
parent_path=parent_path,
|
|
file=file,
|
|
agent_id=agent_id,
|
|
agent_config_id=agent_config_id,
|
|
current_user=current_user,
|
|
db=db,
|
|
)
|
|
|
|
|
|
@filesystem_router.get("/download")
|
|
async def download_viewer(
|
|
thread_id: str = Query(..., description="线程 ID"),
|
|
path: str = Query(..., description="文件路径"),
|
|
agent_id: str | None = Query(None, description="智能体 ID"),
|
|
agent_config_id: int | None = Query(None, description="智能体配置 ID"),
|
|
current_user: User = Depends(get_required_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
return await download_viewer_file(
|
|
thread_id=thread_id,
|
|
path=path,
|
|
agent_id=agent_id,
|
|
agent_config_id=agent_config_id,
|
|
current_user=current_user,
|
|
db=db,
|
|
)
|