From d123a67da1081c42fee4345109db7d1e188d8e8d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9C=B1=E6=BD=AE?= Date: Wed, 8 Oct 2025 18:01:11 +0800 Subject: [PATCH] apikey in header --- .gitignore | 1 + Dockerfile | 5 +- chat.html | 1008 -------------------------------------------- docker-compose.yml | 5 +- fastapi_app.py | 32 +- 5 files changed, 25 insertions(+), 1026 deletions(-) delete mode 100644 chat.html diff --git a/.gitignore b/.gitignore index bf1fc80..0fd5091 100644 --- a/.gitignore +++ b/.gitignore @@ -2,3 +2,4 @@ projects/* workspace __pycache__ +public diff --git a/Dockerfile b/Dockerfile index acee6f8..cf8b138 100644 --- a/Dockerfile +++ b/Dockerfile @@ -29,16 +29,17 @@ COPY . . # 创建必要的目录 RUN mkdir -p /app/projects +RUN mkdir -p /app/public # 设置权限 RUN chmod +x /app/mcp/json_reader_server.py # 暴露端口 -EXPOSE 8000 +EXPOSE 8001 # 健康检查 HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \ - CMD curl -f http://localhost:8000/ || exit 1 + CMD curl -f http://localhost:8001/ || exit 1 # 启动命令 CMD ["python", "fastapi_app.py"] diff --git a/chat.html b/chat.html deleted file mode 100644 index b1a1147..0000000 --- a/chat.html +++ /dev/null @@ -1,1008 +0,0 @@ - - - - - - AI聊天助手 - - - - - - - - -
-
-
AI聊天助手
-
- - 已连接 -
- -
- -
-
-
- - -
-
- - -
-
- - -
-
- - -
-
- - -
-
- - -
-
-
- -
-
-
AI
-
- 您好！我是AI助手，有什么可以帮助您的吗？ -
-
-
- -
-
- - -
-
-
- - - - \ No newline at end of file diff --git a/docker-compose.yml b/docker-compose.yml index 2b48421..80160e1 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -5,7 +5,7 @@ services: build: . container_name: qwen-agent-api ports: - - "8000:8000" + - "8001:8001" environment: # 应用配置 - PYTHONPATH=/app @@ -14,10 +14,11 @@ services: volumes: # 挂载项目数据目录 - ./projects:/app/projects + - ./public:/app/public restart: unless-stopped healthcheck: - test: ["CMD", "curl", "-f", "http://localhost:8000/"] + test: ["CMD", "curl", "-f", "http://localhost:8001/"] interval: 30s timeout: 10s retries: 3 diff --git a/fastapi_app.py b/fastapi_app.py index bcd888c..76176b2 100644 --- a/fastapi_app.py +++ b/fastapi_app.py @@ -3,7 +3,7 @@ import os from typing import AsyncGenerator, Dict, List, Optional, Union import uvicorn -from fastapi import FastAPI, HTTPException +from fastapi import FastAPI, HTTPException, Depends, Header from fastapi.responses import StreamingResponse, HTMLResponse, FileResponse from fastapi.staticfiles import StaticFiles from fastapi.middleware.cors import CORSMiddleware @@ -51,6 +51,9 @@ agent_manager = init_global_agent_manager(max_cached_agents=max_cached_agents) app = FastAPI(title="Database Assistant API", version="1.0.0") +# 挂载public文件夹为静态文件服务 +app.mount("/public", StaticFiles(directory="public"), name="static") + # 添加CORS中间件，支持前端页面 app.add_middleware( CORSMiddleware, @@ -69,7 +72,6 @@ class Message(BaseModel): class ChatRequest(BaseModel): messages: List[Message] model: str = "qwen3-next" - api_key: Optional[str] = None model_server: Optional[str] = None zip_url: Optional[str] = None generate_cfg: Optional[Dict] = None @@ -156,18 +158,28 @@ async def generate_stream_response(agent, messages, request) -> AsyncGenerator[s yield f"data: {json.dumps(error_data, ensure_ascii=False)}\n\n" -@app.post("/chat/completions") -async def chat_completions(request: ChatRequest): +@app.post("/api/v1/chat/completions") +async def chat_completions(request: ChatRequest, authorization: Optional[str] = Header(None)): """ Chat completions API similar to OpenAI, supports both streaming and non-streaming Args: request: ChatRequest containing messages, model, zip_url, etc. + authorization: Authorization header containing API key (Bearer ) Returns: Union[ChatResponse, StreamingResponse]: Chat completion response or stream """ try: + # 从Authorization header中提取API key + api_key = None + if authorization: + # 移除 "Bearer " 前缀 + if authorization.startswith("Bearer "): + api_key = authorization[7:] + else: + api_key = authorization + # 从最外层获取zip_url参数 zip_url = request.zip_url @@ -191,7 +203,7 @@ async def chat_completions(request: ChatRequest): zip_url=zip_url, files=document_files, model_name=request.model, - api_key=request.api_key, + api_key=api_key, model_server=request.model_server, generate_cfg=request.generate_cfg ) @@ -258,14 +270,6 @@ async def chat_completions(request: ChatRequest): raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}") - - -@app.get("/") -async def root(): - """Chat page endpoint""" - return FileResponse("chat.html", media_type="text/html") - - @app.get("/api/health") async def health_check(): """Health check endpoint""" @@ -353,4 +357,4 @@ async def remove_project_cache(zip_url: str): if __name__ == "__main__": - uvicorn.run(app, host="0.0.0.0", port=8000) + uvicorn.run(app, host="0.0.0.0", port=8001)