| """ |
| 主要API模块 |
| 集成所有工具的API接口 |
| """ |
|
|
| from typing import Any |
|
|
| from ninja import File, Router |
| from ninja.files import UploadedFile |
|
|
| from apps.automation.schemas import OllamaChatIn, OllamaChatOut |
| from apps.core.infrastructure.throttling import rate_limit_from_settings |
|
|
| from .performance_monitor_api import router as performance_router |
|
|
| router = Router(tags=["Main API"]) |
|
|
|
|
| def _get_ai_service() -> Any: |
| from apps.core.dependencies.automation_adapters import build_ai_service |
|
|
| return build_ai_service() |
|
|
|
|
| def _get_document_processor_service() -> Any: |
| from apps.core.dependencies.automation_adapters import build_document_processing_service |
|
|
| return build_document_processing_service() |
|
|
|
|
| def _get_config_service() -> Any: |
| from apps.core.dependencies.automation_adapters import build_automation_config_service |
|
|
| return build_automation_config_service() |
|
|
|
|
| |
| router.add_router("/performance", performance_router) |
|
|
| |
| |
| |
|
|
|
|
| @router.post("/ai/ollama", response=OllamaChatOut) |
| def ai_ollama(request: Any, payload: OllamaChatIn) -> OllamaChatOut: |
| """Ollama AI接口""" |
| |
| service = _get_ai_service() |
|
|
| |
| result = service.chat_with_ollama(model=payload.model, prompt=payload.prompt, text=payload.text) |
|
|
| return OllamaChatOut(data=result) |
|
|
|
|
| |
| |
| |
|
|
|
|
| @router.post("/file/upload", response=dict) |
| @rate_limit_from_settings("UPLOAD") |
| def upload_file( |
| request: Any, |
| file: UploadedFile = File(...), |
| limit: int | None = None, |
| preview_page: int | None = None, |
| ) -> dict[str, Any]: |
| """通用文件上传和预处理API""" |
| |
| service = _get_document_processor_service() |
|
|
| |
| result = service.process_uploaded_file(uploaded_file=file, limit=limit, preview_page=preview_page) |
|
|
| return { |
| "success": result.success, |
| "file_info": result.file_info, |
| "extraction": result.extraction, |
| "processing_params": result.processing_params, |
| "error": result.error, |
| } |
|
|
|
|
| |
| |
| |
|
|
|
|
| @router.get("/config") |
| @rate_limit_from_settings("ADMIN") |
| def get_config(request: Any) -> Any: |
| """获取当前配置信息""" |
| from apps.core.security.admin_access import ensure_admin_request |
|
|
| ensure_admin_request(request) |
| |
| service = _get_config_service() |
|
|
| |
| config = service.get_automation_config() |
|
|
| return config |
|
|
|
|
| @router.get("/status") |
| def get_status(request: Any) -> Any: |
| """获取系统状态""" |
| |
| service = _get_config_service() |
|
|
| |
| status = service.get_system_status() |
|
|
| return status |
|
|
