ragbits-chat 1.4.0.dev202601310254__tar.gz → 1.4.0.dev202602030301__tar.gz

{ragbits_chat-1.4.0.dev202601310254 → ragbits_chat-1.4.0.dev202602030301}/CHANGELOG.md

@@ -1,6 +1,7 @@
 # CHANGELOG
 
 ## Unreleased
+- Autoreload support for UI during development (#924)
 - Decoupling of components from ragbits specific logic. Introduction of slot based plugin architecture. Minimal history store implementation (#917)
 - Add timezone field to ChatContext, automatically populated from browser (#916)
 - Fix PostgreSQL conversation persistence by ensuring session flush after creating new conversation in SQL storage (#903)

{ragbits_chat-1.4.0.dev202601310254 → ragbits_chat-1.4.0.dev202602030301}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragbits-chat
-Version: 1.4.0.dev202601310254
+Version: 1.4.0.dev202602030301
 Summary: Building blocks for rapid development of GenAI applications
 Project-URL: Homepage, https://github.com/deepsense-ai/ragbits
 Project-URL: Bug Reports, https://github.com/deepsense-ai/ragbits/issues
@@ -26,7 +26,7 @@ Requires-Dist: bcrypt>=4.2.0
 Requires-Dist: fastapi<1.0.0,>=0.115.0
 Requires-Dist: httpx<1.0.0,>=0.28.1
 Requires-Dist: python-jose[cryptography]>=3.5.0
-Requires-Dist: ragbits-core==1.4.0.dev202601310254
+Requires-Dist: ragbits-core==1.4.0.dev202602030301
 Requires-Dist: uvicorn<1.0.0,>=0.31.0
 Provides-Extra: sql
 Requires-Dist: sqlalchemy<3.0.0,>=2.0.39; extra == 'sql'

{ragbits_chat-1.4.0.dev202601310254 → ragbits_chat-1.4.0.dev202602030301}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ragbits-chat"
-version = "1.4.0.dev202601310254"
+version = "1.4.0.dev202602030301"
 description = "Building blocks for rapid development of GenAI applications"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -31,7 +31,7 @@ classifiers = [
     "Topic :: Scientific/Engineering :: Artificial Intelligence",
     "Topic :: Software Development :: Libraries :: Python Modules",
 ]
-dependencies = ["fastapi>=0.115.0,<1.0.0", "uvicorn>=0.31.0,<1.0.0", "httpx>=0.28.1,<1.0.0", "bcrypt>=4.2.0", "python-jose[cryptography]>=3.5.0", "ragbits-core==1.4.0.dev202601310254"]
+dependencies = ["fastapi>=0.115.0,<1.0.0", "uvicorn>=0.31.0,<1.0.0", "httpx>=0.28.1,<1.0.0", "bcrypt>=4.2.0", "python-jose[cryptography]>=3.5.0", "ragbits-core==1.4.0.dev202602030301"]
 
 [project.urls]
 "Homepage" = "https://github.com/deepsense-ai/ragbits"

{ragbits_chat-1.4.0.dev202601310254 → ragbits_chat-1.4.0.dev202602030301}/src/ragbits/chat/api.py

@@ -3,6 +3,7 @@ import importlib
 import json
 import logging
 import re
+import tempfile
 import time
 from collections.abc import AsyncGenerator
 from contextlib import asynccontextmanager, suppress
@@ -944,9 +945,68 @@ class RagbitsAPI:
     def run(self, host: str = "127.0.0.1", port: int = 8000) -> None:
         """
         Used for starting the API
+
+        Args:
+            host: Host to bind the API server to
+            port: Port to bind the API server to
         """
         uvicorn.run(self.app, host=host, port=port)
 
+    @staticmethod
+    def run_with_reload(
+        host: str,
+        port: int,
+        chat_interface: str | None = None,
+        cors_origins: list[str] | None = None,
+        ui_build_dir: str | None = None,
+        debug_mode: bool = False,
+        auth_backend: str | None = None,
+        theme_path: str | None = None,
+    ) -> None:
+        """
+        Run the API server with auto-reload enabled for development.
+
+        This method creates a temporary Python file with the API configuration
+        that uvicorn can import and reload when code changes are detected.
+
+        Args:
+            host: Host to bind the API server to
+            port: Port to bind the API server to
+            chat_interface: Path to chat interface module
+            cors_origins: List of allowed CORS origins
+            ui_build_dir: Path to custom UI build directory
+            debug_mode: Enable debug mode
+            auth_backend: Path to authentication backend module
+            theme_path: Path to theme configuration file
+        """
+        temp_file_path: Path | None = None
+        try:
+            with tempfile.NamedTemporaryFile(
+                mode="w", suffix=".py", delete=False, dir=".", prefix="_ragbits_app_"
+            ) as temp_file:
+                temp_file_path = Path(temp_file.name)
+                temp_file.write(
+                    f"""# Auto-generated file for ragbits reload mode - DO NOT EDIT
+from ragbits.chat.api import RagbitsAPI
+
+api = RagbitsAPI(
+    chat_interface={repr(chat_interface)},
+    cors_origins={repr(cors_origins)},
+    ui_build_dir={repr(ui_build_dir)},
+    debug_mode={repr(debug_mode)},
+    auth_backend={repr(auth_backend)},
+    theme_path={repr(theme_path)},
+)
+app = api.app
+"""
+                )
+
+            module_name = temp_file_path.stem
+            uvicorn.run(f"{module_name}:app", host=host, port=port, reload=True)
+        finally:
+            if temp_file_path:
+                temp_file_path.unlink(missing_ok=True)
+
     @staticmethod
     def _convert_heroui_json_to_css(json_content: str) -> str:
         """Convert HeroUI JSON theme configuration to CSS variables."""

{ragbits_chat-1.4.0.dev202601310254 → ragbits_chat-1.4.0.dev202602030301}/src/ragbits/chat/cli.py

@@ -37,16 +37,29 @@ def run(
         "--theme",
         help="Path to a HeroUI theme JSON file from heroui.com/themes",
     ),
+    reload: bool = typer.Option(False, "--reload", help="Enable auto-reload on code changes for debugging"),
 ) -> None:
     """
     Run API service with UI demo
     """
-    api = RagbitsAPI(
-        chat_interface=chat_interface,
-        cors_origins=cors_origins,
-        ui_build_dir=ui_build_dir,
-        debug_mode=debug_mode,
-        auth_backend=auth,
-        theme_path=theme,
-    )
-    api.run(host=host, port=port)
+    if reload:
+        RagbitsAPI.run_with_reload(
+            host=host,
+            port=port,
+            chat_interface=chat_interface,
+            cors_origins=cors_origins,
+            ui_build_dir=ui_build_dir,
+            debug_mode=debug_mode,
+            auth_backend=auth,
+            theme_path=theme,
+        )
+    else:
+        api = RagbitsAPI(
+            chat_interface=chat_interface,
+            cors_origins=cors_origins,
+            ui_build_dir=ui_build_dir,
+            debug_mode=debug_mode,
+            auth_backend=auth,
+            theme_path=theme,
+        )
+        api.run(host=host, port=port)

{ragbits_chat-1.4.0.dev202601310254 → ragbits_chat-1.4.0.dev202602030301}/tests/unit/test_api.py

@@ -1,5 +1,6 @@
 import json
 from collections.abc import AsyncGenerator
+from pathlib import Path
 from typing import Literal
 from unittest.mock import MagicMock, mock_open, patch
 
@@ -337,6 +338,48 @@ def test_run_method() -> None:
         mock_run.assert_called_once_with(api.app, host="localhost", port=9000)
 
 
+def test_run_with_reload_method() -> None:
+    """Test the run_with_reload method creates temp file and starts uvicorn with reload."""
+
+    class TempFileCheck:
+        """Helper class to verify temp file exists when uvicorn.run is called."""
+
+        existed_at_run_time: bool = False
+        temp_file_path: Path | None = None
+
+        @classmethod
+        def check_and_record(cls, app_path: str, **kwargs: dict) -> None:
+            module_name = app_path.split(":")[0]
+            cls.temp_file_path = Path(f"./{module_name}.py")
+            cls.existed_at_run_time = cls.temp_file_path.exists()
+
+    with patch("uvicorn.run", side_effect=TempFileCheck.check_and_record) as mock_run:
+        RagbitsAPI.run_with_reload(
+            host="localhost",
+            port=9000,
+            chat_interface="test:TestChat",
+            debug_mode=True,
+        )
+
+        # Verify uvicorn.run was called with reload=True and an import string
+        mock_run.assert_called_once()
+        call_args = mock_run.call_args
+        assert call_args[1]["reload"] is True
+        assert call_args[1]["host"] == "localhost"
+        assert call_args[1]["port"] == 9000
+        assert isinstance(call_args[0][0], str)
+        # Module name format: _ragbits_app_<random>:app
+        assert ":app" in call_args[0][0]
+        assert "_ragbits_app_" in call_args[0][0]
+
+        # Verify temp file existed when uvicorn.run was called
+        assert TempFileCheck.existed_at_run_time, "Temp file was not created before uvicorn.run"
+
+        # Verify temp file was cleaned up after
+        assert TempFileCheck.temp_file_path is not None
+        assert not TempFileCheck.temp_file_path.exists(), "Temp file was not cleaned up"
+
+
 def test_login_endpoint(client: TestClient) -> None:
     """Test the login endpoint with valid credentials sets session cookie."""
     response = client.post("/api/auth/login", json={"username": "testuser", "password": "testpass"})