chore: AI messgae pydantic model added

Backend/app/api/v1/endpoints/notes.py

@@ -8,4 +8,16 @@ from fastapi import APIRouter, Depends, HTTPException
 from chromadb.api.models.Collection import Collection # Import Collection type
 from app.api.deps import get_chroma_collection
 from app.llm import call_llm
-import uuid
+import uuid
+
+
+
+router = APIRouter(prefix="/notes")
+
+@router.post("/ai_chat", status_code=status.HTTP_201_CREATED)
+async def ai_chat(
+    Input_model: IngestRequest, 
+    collection: Collection = Depends(get_chroma_collection), 
+    current_user: User = Depends(get_current_user)
+):
+    pass

Backend/app/schema/models.py

@@ -2,6 +2,9 @@ from pydantic import BaseModel, EmailStr, Field, field_validator, ConfigDict
 from typing import  Optional, Literal, List
 from datetime import datetime
 
+
+#--------Auth models--------#
+
 class UserCreate(BaseModel):
     username: str = Field(..., min_length=3, max_length=50)
     email: EmailStr = Field(...)
@@ -24,6 +27,8 @@ class Token(BaseModel):
 class LoginResponse(Token):
     username: str
 
+
+#--------Quiz models--------#
 class Quiz_input(BaseModel):
     parsed_doc: str
     user_prompt: str
@@ -42,4 +47,21 @@ class QuizOutput(BaseModel):
 class IngestRequest(BaseModel):
     parsed_doc: str = Field(..., description="The main document content to embed")
     user_prompt: str = Field(..., description="The user prompt associated with this document")
-    id: Optional[str] = None
+    id: Optional[str] = None
+
+
+# #--------Notes models--------#
+
+class ChatMessage(BaseModel):
+    role: Literal['user', 'assistant', 'system'] = Field(description="The sender of the message. Must be 'user', 'assistant', or 'system'.")
+    content: str=Field(..., min_length=1, description="The text content of the message.")
+
+class AI_chat_input(BaseModel):
+    messages = List[ChatMessage] = Field(
+        ...,
+        min_length=1,
+        description="The complete conversation history (list of messages) to send to the LLM."
+    )
+    session_id: str | None = Field(
+        None, description="The unique ID of the current chat session (optional)."
+    )