71 lines
2.6 KiB
Python
71 lines
2.6 KiB
Python
# Copyright 2026 zhaoxi826
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException, UploadFile, File
|
|
from pydantic import BaseModel
|
|
from kilostar.utils.access import Accessor, TokenData
|
|
from kilostar.utils.ray_hook import ray_actor_hook
|
|
import os
|
|
import anyio
|
|
from kilostar.utils.logger import get_logger
|
|
|
|
|
|
logger = get_logger("frontend")
|
|
client_router = APIRouter(prefix="/api/v1/adapter/client", tags=["client"])
|
|
|
|
|
|
class Message(BaseModel):
|
|
"""``POST /client`` 入参：来自前端的一段聊天文本。"""
|
|
|
|
message: str
|
|
|
|
|
|
@client_router.post("")
|
|
async def create_message(
|
|
message: Message, token_data: TokenData = Depends(Accessor.get_current_user)
|
|
):
|
|
"""把前端消息转交给 RegulatoryNode 处理，并把回复透传给前端。"""
|
|
logger.info("收到消息，来源：客户端")
|
|
logger.debug(f"消息内容：{message.message}")
|
|
regulatory_node = ray_actor_hook("regulatory_node").regulatory_node
|
|
reply = await regulatory_node.handle_client_message.remote(
|
|
user_id=token_data.user_id,
|
|
user_name=token_data.username,
|
|
message=message.message,
|
|
)
|
|
return {"message": reply}
|
|
|
|
|
|
@client_router.post("/upload")
|
|
async def upload_file(
|
|
file: UploadFile = File(...),
|
|
token_data: TokenData = Depends(Accessor.get_current_user),
|
|
):
|
|
"""以流式方式把上传文件落到 ``uploads/`` 目录；失败抛 500。"""
|
|
try:
|
|
upload_dir = "uploads"
|
|
os.makedirs(upload_dir, exist_ok=True)
|
|
file_path = os.path.join(upload_dir, file.filename)
|
|
async with await anyio.open_file(file_path, "wb") as buffer:
|
|
while chunk := await file.read(64 * 1024): # 64KB chunks
|
|
await buffer.write(chunk)
|
|
logger.info(f"用户 {token_data.username} 上传了文件: {file.filename}")
|
|
return {
|
|
"filename": file.filename,
|
|
"message": f"File {file.filename} uploaded successfully",
|
|
}
|
|
except Exception as e:
|
|
logger.error(f"文件上传失败: {e}")
|
|
raise HTTPException(status_code=500, detail="文件上传失败")
|