Add WebSocket routing for conversation handling and refactor controllers

src/app.py

@@ -4,7 +4,7 @@ from contextlib import asynccontextmanager
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 
-from src.controllers import api_router
+from src.controllers import api_router, websocket_router
 from src.config import logger, DatabaseConfig
 from src.middlewares import AuthenticationMiddleware
 
@@ -48,3 +48,4 @@ async def check_health():
 
 
 app.include_router(api_router, prefix="/api")
+app.include_router(websocket_router, prefix="/ws")

src/controllers/__init__.py

@@ -6,9 +6,11 @@ from ._file_controller import FileController
 from ._conversation_controller import ConversationController
 
 api_router = APIRouter()
+websocket_router = APIRouter()
 
 auth_router = AuthController().api_router
-conversation_router = ConversationController().api_router
+conversation_api_router = ConversationController().api_router
+conversation_websocket_router = ConversationController().websocket_router
 file_router = FileController().api_router
 
 API_KEY_HEADER = APIKeyHeader(name="Authorization", auto_error=False)
@@ -20,11 +22,16 @@ api_router.include_router(
     dependencies=[Depends(API_KEY_HEADER)],
 )
 api_router.include_router(
-    conversation_router,
+    conversation_api_router,
     prefix="/v1",
     tags=["Conversation"],
     dependencies=[Depends(API_KEY_HEADER)],
 )
+websocket_router.include_router(
+    conversation_websocket_router,
+    prefix="/v1",
+    tags=["Conversation"],
+)
 api_router.include_router(
     file_router,
     prefix="/v1",
@@ -32,6 +39,6 @@ api_router.include_router(
     dependencies=[Depends(API_KEY_HEADER)],
 )
 
-__all__ = ["api_router"]
+__all__ = ["api_router", "websocket_router"]
 __version__ = "0.1.0"
 __author__ = "Ramanjit Singh"

src/controllers/_conversation_controller.py

@@ -74,6 +74,7 @@ class ConversationController:
         self.redis_client = RedisClient().client
         self.service = ConversationService
         self.api_router = APIRouter()
+        self.websocket_router = APIRouter()
         self.api_router.add_api_route(
             "/conversations",
             self.create_conversation,
@@ -92,7 +93,7 @@ class ConversationController:
             methods=["POST"],
             response_model=CreateConversationSummaryResponse,
         )
-        self.api_router.add_websocket_route("/conversations", self.conversation)
+        self.websocket_router.add_websocket_route("/conversations", self.conversation)
 
     async def create_conversation(
         self, request: Request, data: CreateConversationSchema

src/services/_web_rtc_service.py

@@ -24,9 +24,10 @@ class WebRTCService:
             data = await websocket.receive_text()
             message = json.loads(data)
 
-            user_status = await redis_client.get(f"session:{user_id}{conversation_id}")
-            if not user_status:
-                raise Exception("A max 25 minute session has ended")
+            # user_status = await redis_client.get(f"session:{user_id}{conversation_id}")
+            # if not user_status:
+            #     logger.error("User session has ended")
+            #     raise Exception("A max 25 minute session has ended")
 
             openai_server_events = {
                 "session_created": message["type"] == "session.created",