feat: implement ERP AI Assistant Phase 1
Backend (FastAPI + SQLAlchemy + Claude API + RAG): - Config management with Pydantic v2 - Database engine with connection pooling and SQL injection prevention - AI engine with Claude API integration (support custom base URL) - RAG engine with ChromaDB and sentence-transformers - Requirement analysis service - Config generation service - Executor engine with SQL validation - REST API endpoints: /analyze, /generate, /execute Frontend (Vue 3 + Element Plus + Pinia): - Complete 3-step workflow: analyze → generate → execute - Step indicator with progress visualization - Analysis result display with field table - SQL preview with monospace font - Execute confirmation dialog with safety warning - Execution result display - State management with Pinia - API service integration Security: - SQL injection prevention with parameterized queries - Dangerous SQL operation blocking - Database password URL encoding - Transaction auto-rollback - Pydantic config validation Features: - Natural language requirement analysis - Automated SQL configuration generation - Safe execution with human review - LAN access support - Custom Claude API endpoint support Documentation: - README with quick start guide - Quick start guide - LAN access configuration - Dependency fixes guide - Claude API configuration - Git operation guide - Implementation report Dependencies fixed: - numpy<2.0.0 for chromadb compatibility - sentence-transformers==2.7.0 for huggingface_hub compatibility Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
102
backend/app/api/generate.py
Normal file
102
backend/app/api/generate.py
Normal file
@@ -0,0 +1,102 @@
|
||||
"""Generate API endpoint for configuration generation.
|
||||
|
||||
This module provides the /generate endpoint for generating SQL configuration.
|
||||
"""
|
||||
|
||||
from fastapi import APIRouter, HTTPException, status
|
||||
from loguru import logger
|
||||
|
||||
from app.models.request import GenerateRequest
|
||||
from app.models.response import GenerateResponse, ErrorResponse
|
||||
from app.services.config_service import ConfigService
|
||||
from app.api.execute import store_session_sql # Import SQL storage function
|
||||
|
||||
# Create router
|
||||
router = APIRouter()
|
||||
|
||||
|
||||
@router.post(
|
||||
"/generate",
|
||||
response_model=GenerateResponse,
|
||||
responses={
|
||||
400: {"model": ErrorResponse, "description": "Invalid request"},
|
||||
500: {"model": ErrorResponse, "description": "Internal server error"}
|
||||
},
|
||||
summary="Generate SQL configuration",
|
||||
description="Generate SQL configuration based on structured requirements"
|
||||
)
|
||||
async def generate_config(request: GenerateRequest) -> GenerateResponse:
|
||||
"""Generate SQL configuration based on structured requirements.
|
||||
|
||||
This endpoint takes structured requirements from the analysis phase
|
||||
and generates SQL configuration statements using Claude AI.
|
||||
|
||||
Args:
|
||||
request: GenerateRequest with session_id and requirements
|
||||
|
||||
Returns:
|
||||
GenerateResponse with session_id, status, and generated config
|
||||
|
||||
Raises:
|
||||
HTTPException: 400 for invalid input, 500 for processing errors
|
||||
"""
|
||||
try:
|
||||
# Validate requirements
|
||||
if not request.requirements:
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_400_BAD_REQUEST,
|
||||
detail={
|
||||
"code": "EMPTY_REQUIREMENTS",
|
||||
"message": "requirements cannot be empty",
|
||||
"session_id": request.session_id
|
||||
}
|
||||
)
|
||||
|
||||
logger.info(f"[{request.session_id}] Processing generate request")
|
||||
|
||||
# Create service and generate config
|
||||
service = ConfigService()
|
||||
result = await service.generate(
|
||||
requirements=request.requirements,
|
||||
session_id=request.session_id
|
||||
)
|
||||
|
||||
# Store generated SQL for later execution
|
||||
if result and result.get("配置方案") and result["配置方案"].get("sql_list"):
|
||||
sql_list = result["配置方案"]["sql_list"]
|
||||
store_session_sql(request.session_id, sql_list)
|
||||
logger.info(f"[{request.session_id}] Stored {len(sql_list)} SQL statements for execution")
|
||||
|
||||
logger.success(f"[{request.session_id}] Config generation completed")
|
||||
|
||||
return GenerateResponse(
|
||||
session_id=request.session_id,
|
||||
status="success",
|
||||
data=result
|
||||
)
|
||||
|
||||
except HTTPException:
|
||||
# Re-raise HTTP exceptions
|
||||
raise
|
||||
|
||||
except ValueError as e:
|
||||
logger.error(f"[{request.session_id}] Validation error: {e}")
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_400_BAD_REQUEST,
|
||||
detail={
|
||||
"code": "VALIDATION_ERROR",
|
||||
"message": str(e),
|
||||
"session_id": request.session_id
|
||||
}
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"[{request.session_id}] Config generation failed: {e}")
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
||||
detail={
|
||||
"code": "GENERATION_FAILED",
|
||||
"message": f"Failed to generate config: {str(e)}",
|
||||
"session_id": request.session_id
|
||||
}
|
||||
)
|
||||
Reference in New Issue
Block a user