diff --git a/src/models.py b/src/models.py
index e12618a..f4422ce 100644
--- a/src/models.py
+++ b/src/models.py
@@ -84,10 +84,14 @@ class EstimationResponse(BaseModel):
     details: EstimationDetails
     results: List[EstimationResult]
 
+class InsuranceChatContext(BaseModel):
+    page: str
+
 class InsuranceChatRequest(BaseModel):
     userId: str | int | None = None
     message: str = Field(..., description="User message")
-    session_id: Optional[str] = Field(None, description="Chat session ID")
+    session_id: str | None = Field(None, description="Chat session ID")
+    context: InsuranceChatContext | None
 
 class Source(BaseModel):
     plan_name: str