add the code

schemas.py

@@ -0,0 +1,58 @@
+from enum import Enum
+from typing import List, Optional
+
+from pydantic import BaseModel, Field
+
+
+class CodeTaskType(str, Enum):
+    GENERATE = "generate"
+    FIX = "fix"
+    EXPLAIN = "explain"
+    REFACTOR = "refactor"
+    REVIEW = "review"
+    UNKNOWN = "unknown"
+
+
+class SourceType(str, Enum):
+    STACKOVERFLOW = "stackoverflow"
+    GITHUB = "github"
+    MODEL = "model"
+
+
+class RetrievedEvidence(BaseModel):
+    source_type: SourceType
+    title: str
+    snippet: str
+    url: Optional[str] = None
+    score: Optional[float] = None
+
+
+class ResponseMeta(BaseModel):
+    used_model: str
+    fallback_used: bool = False
+    retrieval_used: bool = False
+    source_count: int = 0
+    processing_time_ms: Optional[int] = None
+
+
+class CodeXRequest(BaseModel):
+    message: str = Field(..., min_length=1)
+    mode: Optional[CodeTaskType] = None
+    language: Optional[str] = None
+    code: Optional[str] = None
+    error_message: Optional[str] = None
+    framework: Optional[str] = None
+    file_name: Optional[str] = None
+    previous_context: Optional[str] = None
+    use_retrieval: bool = True
+
+
+class CodeXResponse(BaseModel):
+    answer: str
+    task_type: CodeTaskType
+    code_output: Optional[str] = None
+    explanation: Optional[str] = None
+    warnings: List[str] = []
+    sources: List[RetrievedEvidence] = []
+    needs_clarification: bool = False
+    meta: ResponseMeta