BE-14: Session Management Implementation

[tecnodeveloper]

Description:
Implement session system for chatbot. User starts chat → session created → messages stored in MongoDB → context maintained → backend uses session history for responses → done.

---

User Story

Given user is chatting with bot
When messages are sent
Then system should store and manage conversation in a session

---

Tasks

---

Backend Setup (Flask / FastAPI)

1. Create Session Module

   • /app/routes/session.py
   • /app/controllers/session_controller.py
   • /app/services/session_service.py

---

MongoDB Setup

2. Create Sessions Collection

   • Collection: sessions
   • Index on user_id

3. Define Session Schema

   • session_id
   • user_id
   • messages (array)
   • created_at
   • updated_at

4. Define Message Schema

   • role (user / assistant)
   • content
   • timestamp

---

Session Creation

5. Create Session API

   • POST /session/create
   • Generate unique session_id
   • Save to MongoDB

6. Return Session ID

   • Send session_id to frontend

---

Message Handling

7. Add Message API

   • POST /session/{id}/message
   • Accept user message

8. Store User Message

   • Append message to session
   • Save in MongoDB

---

Chat Integration

9. Connect with LLaMA Model

   • Fetch session history
   • Build context window
   • Send to model

10. Store Bot Response

• Save assistant reply
• Append to session

---

Context Management

11. Maintain Context Window

• Limit message history
• Keep last N messages
• Remove old data if needed

12. Session Memory Handling

• Preserve conversation flow
• Ensure relevance in replies

---

Session Retrieval

13. Get Session API

• GET /session/{id}
• Return full chat history

14. Get User Sessions

• GET /sessions/user/{user_id}
• List all sessions

---

Frontend Integration

15. Store Session ID

• Save session_id after creation
• Attach to every chat request

16. Load Chat History

• Fetch session on page load
• Render messages

---

State Management

17. Frontend State Update

• Update chat UI in real-time
• Append new messages instantly

---

Postman Testing 🧪

18. Setup Postman

• Create session
• Send message request

19. Test APIs

• /session/create
• /session/{id}/message
• /session/{id}

20. Validate Responses

• Check message storage
• Verify session updates

---

Run & Validate

21. Run Backend

• Start server
• Fix errors

22. Test Full Flow

• Create session
• Send messages
• Check MongoDB updates
• Verify chatbot response

---

Acceptance Criteria

• Session creation works
• Messages stored in MongoDB
• Context maintained properly
• Chat flow connected with model
• Postman tests completed
• Frontend integrated

---

Testing Steps

1. Start backend
2. Create session
3. Send messages
4. Verify DB updates
5. Check chatbot response
6. Load session history in UI

---

Definition of Done

• Session system fully working
• MongoDB integration complete
• Chat memory working
• Backend + frontend connected