From 74f3156c476abcb34088e0f465e24b50b6c50c5a Mon Sep 17 00:00:00 2001
From: ipu <uwu@ipu.moe>
Date: Mon, 1 Sep 2025 21:47:38 +0300
Subject: [PATCH] add chat context field

---
 src/models.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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