feat: Add text generation endpoint with corresponding request and response models.

models/__init__.py

@@ -22,6 +22,12 @@ from .health import (
     HealthResponse,
 )
 
+# Import generate-related schemas
+from .generate import (
+    GenerateRequest,
+    GenerateResponse,
+)
+
 # Import MCP-related schemas
 from .mcp_models import (
     ToolCallRequest,
@@ -51,6 +57,9 @@ __all__ = [
     "KPAHealthResponse",
     # Health schemas
     "HealthResponse",
+    # Generate schemas
+    "GenerateRequest",
+    "GenerateResponse",
     # MCP schemas
     "ToolCallRequest",
     "ToolCallResponse",

routes/generate.py

@@ -1,29 +1,16 @@
 """Generation specific endpoints"""
 
 from fastapi import APIRouter, HTTPException
-import pydantic
 from datetime import datetime
 import logging
-from typing import Optional
 
 from services import generate_model_manager
+from models.generate import GenerateRequest, GenerateResponse
 
 router = APIRouter()
 logger = logging.getLogger(__name__)
 
 
-class GenerateRequest(pydantic.BaseModel):
-    input_text: str
-    max_length: Optional[int] = 128
-    num_beams: Optional[int] = 4
-
-
-class GenerateResponse(pydantic.BaseModel):
-    input_text: str
-    generated_text: str
-    timestamp: str
-
-
 @router.post("/generate", response_model=GenerateResponse, tags=["Text Generation"])
 async def generate_text(request: GenerateRequest):
     """