"""Pydantic schemas for API request/response models."""
from pydantic import BaseModel, Field
from typing import Optional, Any
class ChatRequest(BaseModel):
"""Request model for chat endpoint."""
user_id: str = Field(..., description="Unique user identifier")
message: str = Field(..., min_length=1, description="User message text")
class ChatResponse(BaseModel):
"""Response model for chat endpoint."""
response: str = Field(..., description="Assistant response text")
tool_calls_count: int = Field(default=0, description="Number of tool calls made")
mcp_used: bool = Field(default=False, description="Whether MCP tools were used")
class HealthResponse(BaseModel):
"""Response model for health check endpoint."""
status: str = Field(default="healthy")
mcp_connected: bool = Field(default=False)
tools_count: int = Field(default=0)
class ErrorResponse(BaseModel):
"""Response model for error cases."""
error: str = Field(..., description="Error message")
detail: Optional[str] = Field(None, description="Detailed error information")
class ReviewPRRequest(BaseModel):
"""Request model for PR review endpoint."""
pr_number: int = Field(..., description="Pull Request number to review")
class ReviewPRResponse(BaseModel):
"""Response model for PR review endpoint."""
review: str = Field(..., description="Code review text with file:line references")
tool_calls_count: int = Field(default=0, description="Number of MCP tool calls made")
class ProfileUpdateRequest(BaseModel):
"""Request model for updating user profile."""
# Allow partial updates with any fields from UserProfile
data: dict[str, Any] = Field(..., description="Profile fields to update")
class ProfileResponse(BaseModel):
"""Response model for profile endpoints."""
message: str = Field(..., description="Success message")
profile: Optional[dict] = Field(None, description="User profile data")
class VoiceResponse(BaseModel):
"""Response model for voice endpoint."""
transcription: Optional[str] = Field(default=None, description="Recognized text from audio (may be None for gpt-audio-mini)")
response: str = Field(..., description="AI assistant response")
latency_ms: int = Field(..., description="Processing time in milliseconds")
audio_tokens: int = Field(default=0, description="Audio tokens consumed")
cost_usd: float = Field(default=0.0, description="Estimated cost in USD")
