add show_page hook

alembic/versions/2025_10_03_1608-4324550d9c83_add_user_page.py

@@ -0,0 +1,36 @@
+"""add user page
+
+Revision ID: 4324550d9c83
+Revises: 31359fcda8a7
+Create Date: 2025-10-03 16:08:13.898990
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = '4324550d9c83'
+down_revision: Union[str, Sequence[str], None] = '31359fcda8a7'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('user_session', schema=None) as batch_op:
+        batch_op.add_column(sa.Column('page_id', sa.String(), nullable=True))
+
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('user_session', schema=None) as batch_op:
+        batch_op.drop_column('page_id')
+
+    # ### end Alembic commands ###

src/api/v1/router.py

@@ -16,6 +16,7 @@ async def insurance_chat(request: models.InsuranceChatRequest):
     """Handle insurance chat requests"""
     try:
         current_page = None
+        page_id = None
         if request.context and request.context.page:
             page_id = str(request.context.page).lower()
             current_page = await get_page_description(page_id)
@@ -32,6 +33,7 @@ async def insurance_chat(request: models.InsuranceChatRequest):
             uid=str(request.userId) if request.userId else None,
             current_page=current_page,
             application=application,
+            page_id=page_id
         )
         
         return models.InsuranceChatResponse(
@@ -61,7 +63,11 @@ async def init_chat(request: models.InitializeChatRequest):
     if request.context:
         first_visit = request.context.isFirstVisit
 
-    result = await chat_service.initialize_chat(str(request.userId), application, name, first_visit)
+    page_id = None
+    if request.context and request.context.page:
+        page_id = str(request.context.page).lower()
+
+    result = await chat_service.initialize_chat(str(request.userId), application, name, first_visit, page_id)
     return models.InitializeChatResponse(
         session_id=result["session_id"],
         answer=result["answer"],

src/database.py

@@ -40,6 +40,7 @@ class UserSession(Base):
     id = Column(BigInteger, primary_key=True, index=True)
     user_id = Column(String, index=True)
     session_id = Column(String)
+    page_id = Column(String, default=None, nullable=True)
 
 
 engine = create_engine(settings.DATABASE_URL)

src/models.py

@@ -125,9 +125,12 @@ class PlansParam(BaseModel):
 class ApplicantParam(BaseModel):
     applicants: list[Applicant]
 
+class PageParam(BaseModel):
+    page: str
+
 class ChatHook(BaseModel):
     tool: str
-    params: PlansParam | ApplicantParam
+    params: PlansParam | ApplicantParam | PageParam
 
 class AIChatResponse(BaseModel):
     answer: str

src/services/chat_service.py

@@ -4,7 +4,7 @@ from typing import Dict, Any, List, Optional
 
 import httpx
 
-from src.models import ApplicantParam, ChatHook, PlansParam
+from src.models import ApplicantParam, ChatHook, PlansParam, PageParam
 from .session_service import session_service
 from ..api.v1.models import Source, HistoryItem
 from ..config import settings
@@ -87,6 +87,15 @@ class ChatService:
                     "message": f"I'm sorry, I'm experiencing technical difficulties. Please try again later. Error: {str(e)}"
                 }
 
+    async def add_message_to_history(self, session_id: str, message: list):
+        async with await self.get_client() as client:
+            response = await client.post(
+                "/messages/",
+                params={"chat_session_id": session_id,},
+                json={"content": message}
+            )
+        return response.json()
+
     async def get_chat_history(self, session_id: str) -> List[HistoryItem]:
         """Get chat history for a session and format it properly"""
         async with await self.get_client() as client:
@@ -145,7 +154,6 @@ class ChatService:
                 return None
 
 
-    
     def _extract_sources_from_response(self, response_text: str) -> List[Source]:
         """Extract sources from RAG search results if available"""
         # This is a placeholder - in a real implementation, you would:
@@ -159,10 +167,10 @@ class ChatService:
 
     async def get_user_session(self, uid: str) -> str | None:
         with Session() as session:
-            session = session.query(UserSession).filter(UserSession.user_id == uid).first()
+            user_session = session.query(UserSession).filter(UserSession.user_id == uid).first()
-            if not session:
+            if not user_session:
                 return None
-            return session.session_id
+            return user_session.session_id
 
     async def create_user_session(self, uid: str, session_id: str):
         with Session() as session:
@@ -173,50 +181,62 @@ class ChatService:
             session.add(user_session)
             session.commit()
 
-    async def initialize_chat(self, uid: str, application, name, first_visit):
+    async def update_user_page(self, uid: str, page_id: str | None):
-        session_id = await self.get_user_session(uid)
+        with Session() as session:
-        if not session_id or not await session_service.validate_session(session_id):
+            user_session = session.query(UserSession).filter(UserSession.user_id == uid).first()
-            session_id = await session_service.create_session(agent_id=settings.TALESTORM_AGENT_ID)
+            old_page_id = user_session.page_id
-        try:
+            user_session.page_id = page_id
-            await self.create_user_session(uid, session_id)
+            session.add(user_session)
-        except:
+            session.commit()
-            pass
+        return old_page_id
 
+    def get_welcome_message(self, application, name, first_visit):
         try:
             if not name:
                 name = application["applicants"][0]["firstName"]
         except:
-            return {
+            return INIT_MESSAGES[0]
-                "session_id": session_id,
-                "answer": INIT_MESSAGES[0],
-            }
 
         if first_visit:
-            return {
+            return INIT_MESSAGES[0]
-                "session_id": session_id,
-                "answer": INIT_MESSAGES[0],
-            }
 
         try:
             applicant = application["applicants"][0]
         except:
-            return {
+            return INIT_MESSAGES[2].format(name=name)
-                "session_id": session_id,
+
-                "answer": INIT_MESSAGES[2].format(name=name)
-            }
         if not applicant.get("gender") or not applicant.get("dob") or applicant.get("dob") == "-01-" or not applicant.get("weight") or applicant.get("heightFt") is None or applicant.get("heightIn") is None:
-            return {
+            return INIT_MESSAGES[2].format(name=name)
-                "session_id": session_id,
+
-                "answer": INIT_MESSAGES[2].format(name=name)
+        return INIT_MESSAGES[1].format(name=name)
-            }
+
+
+    async def initialize_chat(self, uid: str, application, name, first_visit, page_id):
+        session_id = await self.get_user_session(uid)
+        if not session_id or not await session_service.validate_session(session_id):
+            session_id = await session_service.create_session(agent_id=settings.TALESTORM_AGENT_ID)
+            try:
+                await self.create_user_session(uid, session_id)
+            except:
+                pass
+        await self.update_user_page(uid, page_id)
+
+        welcome_msg = self.get_welcome_message(application, name, first_visit)
+
+        msg_history_item = [
+            {'parts': [{'content': 'Hi', 'part_kind': 'user-prompt'}],'kind': 'request'},
+            {'parts': [{'content': welcome_msg, 'part_kind': 'text'}], 'kind': 'response'}
+        ]
+
+        await self.add_message_to_history(session_id, msg_history_item)
 
         return {
             "session_id": session_id,
-            "answer": INIT_MESSAGES[1].format(name=name)
+            "answer": welcome_msg
         }
 
 
-    async def process_insurance_chat(self, message: str, session_id: Optional[str] = None, uid: Optional[str] = None, current_page: Optional[str] = None, application: Optional[dict] = None) -> Dict[str, Any]:
+    async def process_insurance_chat(self, message: str, session_id: Optional[str] = None, uid: Optional[str] = None, current_page: Optional[str] = None, application: Optional[dict] = None, page_id = None) -> Dict[str, Any]:
         """Process an insurance chat request"""
         try:
             if not session_id or not await session_service.validate_session(session_id):
@@ -224,13 +244,14 @@ class ChatService:
                     session_id = await self.get_user_session(uid)
                     if not session_id or not await session_service.validate_session(session_id):
                         session_id = await session_service.create_session(agent_id=settings.TALESTORM_AGENT_ID)
-                    try:
+                        try:
-                        await self.create_user_session(uid, session_id)
+                            await self.create_user_session(uid, session_id)
-                    except:
+                        except:
-                        pass
+                            pass
                 else:
                     session_id = await session_service.create_session(agent_id=settings.TALESTORM_AGENT_ID)
 
+            old_page_id = await self.update_user_page(uid, page_id)
 
             instructions = ""
             if uid:
@@ -265,6 +286,7 @@ class ChatService:
             compare_plans = ai_response.get("compare_plans")
             show_plans = ai_response.get("show_plans")
             update_applicants = ai_response.get("update_applicants")
+            show_page = ai_response.get("show_page")
             hooks = []
             if update_applicants:
                 hooks.append(ChatHook(
@@ -283,6 +305,12 @@ class ChatService:
                     tool="show_plans",
                     params=PlansParam(plans=show_plans)
                 ))
+            elif show_page and old_page_id:
+                hooks.append(ChatHook(
+                    tool="show_page",
+                    params=PageParam(page=old_page_id)
+                ))
+
 
             return {
                 "session_id": session_id,