Adds session management for openweb ui

Author: vmesel

docker-compose.dev-container.yml

@@ -39,3 +39,21 @@ services:
       - PGPASSWORD=talkdai
     env_file:
       - .env
+  openwebui:
+    image: ghcr.io/open-webui/open-webui:main
+    ports:
+      - '3000:8080'
+    environment:
+      - OPENAI_API_KEYS=FAKE-KEY;
+      - OPENAI_API_BASE_URLS=http://dialog:8000/openai;
+      - ENABLE_OPENAI_API=true
+    volumes:
+      - open-webui:/app/backend/data
+    depends_on:
+      db:
+        condition: service_healthy
+      dialog:
+        condition: service_started
+
+volumes:
+  open-webui:

src/dialog/routers/openai.py

@@ -1,7 +1,8 @@
 # *-* coding: utf-8 *-*
-from uuid import uuid4
-import datetime
+import os
 import logging
+import datetime
+from uuid import uuid4
 
 from dialog.db import engine, get_session
 from dialog_lib.db.models import Chat as ChatEntity, ChatMessages
@@ -44,11 +45,20 @@ async def ask_question_to_llm(message: OpenAIChat, session: Session = Depends(ge
     """
     This posts a message to the LLM and returns the response in the OpenAI format.
     """
+
     start_time = datetime.datetime.now()
-    new_chat = ChatEntity(
-        session_id = f"openai-{str(uuid4())}",
-    )
-    session.add(new_chat)
+    chat_entity = session.query(ChatEntity).filter(ChatEntity.session_id == Settings().OPENWEB_UI_SESSION).first()
+
+    if not chat_entity:
+        logging.info("Creating new chat entity")
+        new_chat = ChatEntity(
+            session_id = Settings().OPENWEB_UI_SESSION,
+        )
+        session.add(new_chat)
+        session.flush()
+    else:
+        logging.info("Using old chat entity")
+        new_chat = chat_entity
 
     non_empty_messages = []
 

src/dialog/settings.py

@@ -1,12 +1,15 @@
 import tomllib
 import logging
+from uuid import uuid4
 from pathlib import Path
 from decouple import Csv, Config
 
 config = Config(".env")
 
 logger = logging.getLogger(__name__)
 
+openweb_ui_session_id_fallback = uuid4()
+
 class Settings:
 
     @property
@@ -113,4 +116,8 @@ def CORS_ALLOW_METHODS(self):
 
     @property
     def CORS_ALLOW_HEADERS(self):
-        return config.get("CORS_ALLOW_HEADERS", cast=Csv(), default="*")
+        return config.get("CORS_ALLOW_HEADERS", cast=Csv(), default="*")
+
+    @property
+    def OPENWEB_UI_SESSION(self):
+        return config.get("OPENWEB_UI_SESSION", f"dialog-openweb-ui")