支持 标准Bearer格式和直接token格式(

This commit is contained in:
2025-10-27 16:31:17 +08:00
parent 8086a73f9f
commit 4b95be9762
6 changed files with 40 additions and 11 deletions

View File

@@ -23,7 +23,8 @@ from typing import List, Optional
from fastapi import APIRouter, Depends, File, Form, HTTPException, UploadFile, Query
from fastapi.responses import StreamingResponse
from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
from fastapi.security import HTTPAuthorizationCredentials
from api.utils.api_utils import security
from api import settings
from api.common.check_team_permission import check_kb_team_permission
@@ -53,7 +54,6 @@ from pydantic import BaseModel
from api.db.db_models import User
# Security
security = HTTPBearer()
# Pydantic models for request/response
class WebCrawlRequest(BaseModel):

View File

@@ -18,7 +18,8 @@ from pathlib import Path
from typing import List
from fastapi import APIRouter, Depends
from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
from fastapi.security import HTTPAuthorizationCredentials
from api.utils.api_utils import security
from api.db.services.file2document_service import File2DocumentService
from api.db.services.file_service import FileService
@@ -33,7 +34,6 @@ from api.utils.api_utils import get_json_result
from pydantic import BaseModel
# Security
security = HTTPBearer()
# Pydantic models for request/response
class ConvertRequest(BaseModel):

View File

@@ -20,7 +20,8 @@ from typing import List, Optional
from fastapi import APIRouter, Depends, File, Form, HTTPException, UploadFile, Query
from fastapi.responses import StreamingResponse
from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
from fastapi.security import HTTPAuthorizationCredentials
from api.utils.api_utils import security
from api.common.check_team_permission import check_file_team_permission
from api.db.services.document_service import DocumentService
@@ -38,7 +39,6 @@ from rag.utils.storage_factory import STORAGE_IMPL
from pydantic import BaseModel
# Security
security = HTTPBearer()
# Pydantic models for request/response
class CreateFileRequest(BaseModel):

View File

@@ -15,7 +15,8 @@
#
from typing import List, Optional, Dict, Any
from fastapi import APIRouter, Depends, HTTPException, Query
from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
from fastapi.security import HTTPAuthorizationCredentials
from api.utils.api_utils import security
from api import settings
from api.db import VALID_MCP_SERVER_TYPES
@@ -31,7 +32,6 @@ from rag.utils.mcp_tool_call_conn import MCPToolCallSession, close_multiple_mcp_
from pydantic import BaseModel
# Security
security = HTTPBearer()
# Pydantic models for request/response
class ListMCPRequest(BaseModel):

View File

@@ -21,7 +21,8 @@ from datetime import datetime
from typing import Optional, Dict, Any
from fastapi import APIRouter, Depends, HTTPException, Request, Response, status
from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
from fastapi.security import HTTPAuthorizationCredentials
from api.utils.api_utils import security
from fastapi.responses import RedirectResponse
from pydantic import BaseModel, EmailStr
try:
@@ -65,7 +66,6 @@ from api.utils.crypt import decrypt
router = APIRouter()
# 安全方案
security = HTTPBearer()
# Pydantic模型
class LoginRequest(BaseModel):