Upload mcp_server.py

mcp_server.py

@@ -3,11 +3,12 @@ MCP Server for SEC Financial Report Data Query
 Based on FastAPI framework
 """
 
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, HTTPException, Request
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import HTMLResponse
+from fastapi.responses import HTMLResponse, JSONResponse
 from fastapi.staticfiles import StaticFiles
-from pydantic import BaseModel, Field
+from fastapi.exceptions import RequestValidationError
+from pydantic import BaseModel, Field, ValidationError
 from typing import Optional, List, Dict, Any
 from edgar_client import EdgarDataClient
 from financial_analyzer import FinancialAnalyzer
@@ -40,6 +41,51 @@ financial_analyzer = FinancialAnalyzer(
 )
 
 
+# Custom error handler for validation errors (422)
+@app.exception_handler(RequestValidationError)
+async def validation_exception_handler(request: Request, exc: RequestValidationError):
+    """
+    Custom handler for 422 Unprocessable Entity errors
+    Provides user-friendly error messages
+    """
+    errors = exc.errors()
+    
+    # Build friendly error message
+    error_messages = []
+    for error in errors:
+        field = " -> ".join(str(loc) for loc in error['loc'])
+        msg = error['msg']
+        error_type = error['type']
+        
+        if error_type == 'missing':
+            error_messages.append(f"Missing required field: {field}")
+        elif 'type_error' in error_type:
+            error_messages.append(f"Invalid type for field '{field}': {msg}")
+        else:
+            error_messages.append(f"Validation error in '{field}': {msg}")
+    
+    return JSONResponse(
+        status_code=422,
+        content={
+            "error": "Request validation failed",
+            "details": error_messages,
+            "help": {
+                "note": "Please ensure you send data in the request BODY (not URL parameters)",
+                "example": {
+                    "cik": "1577552",
+                    "period": "2024"  # if applicable
+                },
+                "common_issues": [
+                    "Don't put parameters in the URL query string",
+                    "Ensure CIK has no extra spaces",
+                    "Send Content-Type: application/json header",
+                    "Use POST method with JSON body"
+                ]
+            }
+        }
+    )
+
+
 # Request/Response Models
 class UserAgentRequest(BaseModel):
     user_agent: str = Field(
@@ -54,6 +100,16 @@ class CompanySearchRequest(BaseModel):
 
 class CompanyInfoRequest(BaseModel):
     cik: str = Field(..., description="Company CIK code")
+    
+    class Config:
+        # Enable validation for strings
+        str_strip_whitespace = True
+        
+    def __init__(self, **data):
+        # Strip whitespace and clean CIK
+        if 'cik' in data and data['cik']:
+            data['cik'] = str(data['cik']).strip()
+        super().__init__(**data)
 
 
 class CompanyFilingsRequest(BaseModel):
@@ -62,10 +118,30 @@ class CompanyFilingsRequest(BaseModel):
         None, 
         description="List of form types (e.g., ['10-K', '10-Q']), None for all types"
     )
+    
+    class Config:
+        # Enable validation for strings
+        str_strip_whitespace = True
+        
+    def __init__(self, **data):
+        # Strip whitespace and clean CIK
+        if 'cik' in data and data['cik']:
+            data['cik'] = str(data['cik']).strip()
+        super().__init__(**data)
 
 
 class CompanyFactsRequest(BaseModel):
     cik: str = Field(..., description="Company CIK code")
+    
+    class Config:
+        # Enable validation for strings
+        str_strip_whitespace = True
+        
+    def __init__(self, **data):
+        # Strip whitespace and clean CIK
+        if 'cik' in data and data['cik']:
+            data['cik'] = str(data['cik']).strip()
+        super().__init__(**data)
 
 
 class FinancialDataRequest(BaseModel):
@@ -74,6 +150,18 @@ class FinancialDataRequest(BaseModel):
         ..., 
         description="Period in format 'YYYY' or 'YYYYQX' (e.g., '2025' or '2025Q3')"
     )
+    
+    class Config:
+        # Enable validation for strings
+        str_strip_whitespace = True
+        
+    def __init__(self, **data):
+        # Strip whitespace and clean inputs
+        if 'cik' in data and data['cik']:
+            data['cik'] = str(data['cik']).strip()
+        if 'period' in data and data['period']:
+            data['period'] = str(data['period']).strip()
+        super().__init__(**data)
 
 
 class AdvancedSearchRequest(BaseModel):
@@ -83,10 +171,30 @@ class AdvancedSearchRequest(BaseModel):
 class ExtractMetricsRequest(BaseModel):
     cik: str = Field(..., description="Company CIK code")
     years: int = Field(default=3, description="Number of years to extract", ge=1, le=10)
+    
+    class Config:
+        # Enable validation for strings
+        str_strip_whitespace = True
+        
+    def __init__(self, **data):
+        # Strip whitespace and clean CIK
+        if 'cik' in data and data['cik']:
+            data['cik'] = str(data['cik']).strip()
+        super().__init__(**data)
 
 
 class LatestDataRequest(BaseModel):
     cik: str = Field(..., description="Company CIK code")
+    
+    class Config:
+        # Enable validation for strings
+        str_strip_whitespace = True
+        
+    def __init__(self, **data):
+        # Strip whitespace and clean CIK
+        if 'cik' in data and data['cik']:
+            data['cik'] = str(data['cik']).strip()
+        super().__init__(**data)
 
 
 class CompanySearchResponse(BaseModel):