OpenRAG 0.3.1__tar.gz → 0.3.2__tar.gz

{openrag-0.3.1/src/openrag.egg-info → openrag-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openrag
-Version: 0.3.1
+Version: 0.3.2
 Summary: OpenRAG is a comprehensive Retrieval-Augmented Generation platform that enables intelligent document search and AI-powered conversations.
 Classifier: Development Status :: 4 - Beta
 Classifier: Environment :: Console

{openrag-0.3.1 → openrag-0.3.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "openrag"
-version = "0.3.1"
+version = "0.3.2"
 description = "OpenRAG is a comprehensive Retrieval-Augmented Generation platform that enables intelligent document search and AI-powered conversations."
 readme = "README.md"
 requires-python = ">=3.13"

{openrag-0.3.1 → openrag-0.3.2}/src/api/search.py

@@ -4,6 +4,7 @@ from fastapi import Depends
 from pydantic import BaseModel, Field
 from fastapi.responses import JSONResponse
 from utils.logging_config import get_logger
+from utils.opensearch_utils import OpenSearchDiskSpaceError, DISK_SPACE_ERROR_MESSAGE
 
 from dependencies import get_search_service, get_session_manager, get_current_user
 from session_manager import User
@@ -50,6 +51,8 @@ async def search(
             score_threshold=body.scoreThreshold,
         )
         return JSONResponse(result, status_code=200)
+    except OpenSearchDiskSpaceError:
+        return JSONResponse({"error": DISK_SPACE_ERROR_MESSAGE}, status_code=507)
     except Exception as e:
         error_msg = str(e)
         if (

{openrag-0.3.1 → openrag-0.3.2}/src/api/v1/search.py

@@ -10,6 +10,7 @@ from fastapi import Depends
 from pydantic import BaseModel
 from fastapi.responses import JSONResponse
 from utils.logging_config import get_logger
+from utils.opensearch_utils import OpenSearchDiskSpaceError, DISK_SPACE_ERROR_MESSAGE
 from dependencies import get_search_service, get_api_key_user_async
 from session_manager import User
 
@@ -65,6 +66,9 @@ async def search_endpoint(
 
         return JSONResponse({"results": results})
 
+    except OpenSearchDiskSpaceError as e:
+        logger.error("Search blocked by disk space constraint", error=str(e), user_id=user.user_id)
+        return JSONResponse({"error": DISK_SPACE_ERROR_MESSAGE}, status_code=507)
     except Exception as e:
         error_msg = str(e)
         logger.error("Search failed", error=error_msg, user_id=user.user_id)

{openrag-0.3.1 → openrag-0.3.2/src/openrag.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openrag
-Version: 0.3.1
+Version: 0.3.2
 Summary: OpenRAG is a comprehensive Retrieval-Augmented Generation platform that enables intelligent document search and AI-powered conversations.
 Classifier: Development Status :: 4 - Beta
 Classifier: Environment :: Console

{openrag-0.3.1 → openrag-0.3.2}/src/services/search_service.py

@@ -392,6 +392,7 @@ class SearchService:
         )
 
         from opensearchpy.exceptions import RequestError
+        from utils.opensearch_utils import OpenSearchDiskSpaceError, is_disk_space_error, DISK_SPACE_ERROR_MESSAGE
 
         search_params = {"terminate_after": 0}
 
@@ -403,6 +404,12 @@ class SearchService:
             )
         except RequestError as e:
             error_message = str(e)
+            if is_disk_space_error(e):
+                logger.error(
+                    "OpenSearch query blocked by disk space constraint",
+                    error=error_message,
+                )
+                raise OpenSearchDiskSpaceError(DISK_SPACE_ERROR_MESSAGE) from e
             if (
                 fallback_search_body is not None
                 and "unknown field [num_candidates]" in error_message.lower()
@@ -417,6 +424,12 @@ class SearchService:
                         params=search_params,
                     )
                 except RequestError as retry_error:
+                    if is_disk_space_error(retry_error):
+                        logger.error(
+                            "OpenSearch retry blocked by disk space constraint",
+                            error=str(retry_error),
+                        )
+                        raise OpenSearchDiskSpaceError(DISK_SPACE_ERROR_MESSAGE) from retry_error
                     logger.error(
                         "OpenSearch retry without num_candidates failed",
                         error=str(retry_error),
@@ -428,7 +441,15 @@ class SearchService:
                     "OpenSearch query failed", error=error_message, search_body=search_body
                 )
                 raise
+        except OpenSearchDiskSpaceError:
+            raise
         except Exception as e:
+            if is_disk_space_error(e):
+                logger.error(
+                    "OpenSearch query blocked by disk space constraint",
+                    error=str(e),
+                )
+                raise OpenSearchDiskSpaceError(DISK_SPACE_ERROR_MESSAGE) from e
             logger.error(
                 "OpenSearch query failed", error=str(e), search_body=search_body
             )

{openrag-0.3.1 → openrag-0.3.2}/src/tui/_assets/docker-compose.yml

@@ -49,7 +49,6 @@ services:
       context: .
       dockerfile: Dockerfile.backend
     container_name: openrag-backend
-    user: "${OPENRAG_UID:-103000}:${OPENRAG_GID:-103000}"
     depends_on:
       - langflow
     environment:

{openrag-0.3.1 → openrag-0.3.2}/src/utils/opensearch_utils.py

@@ -5,9 +5,50 @@ from utils.logging_config import get_logger
 
 logger = get_logger(__name__)
 
+DISK_SPACE_ERROR_MESSAGE = (
+    "OpenSearch has run out of available disk space. "
+    "Search and indexing operations are blocked. "
+    "Please free up disk space to restore OpenRAG functionality."
+)
+
+# Error strings emitted by OpenSearch when disk watermark thresholds are breached
+_DISK_SPACE_INDICATORS = [
+    "disk watermark",
+    "flood_stage",
+    "flood stage",
+    "disk usage exceeded",
+    "index read-only",
+    "no space left on device",
+    "cluster_block_exception",
+    "forbidden/12",
+    "too_many_requests/12",
+]
+
+
 class OpenSearchNotReadyError(Exception):
     """Raised when OpenSearch fails to become ready within the retry limit."""
 
+
+class OpenSearchDiskSpaceError(Exception):
+    """Raised when OpenSearch operations fail due to insufficient disk space."""
+
+
+def is_disk_space_error(error: Exception) -> bool:
+    """Check whether an exception is caused by OpenSearch disk space constraints.
+
+    OpenSearch blocks write and search operations when disk usage crosses
+    the high-watermark or flood-stage watermark thresholds.
+    This function detects those error signatures.
+
+    Args:
+        error: The exception to inspect.
+
+    Returns:
+        True if the error is disk-space related, False otherwise.
+    """
+    error_str = str(error).lower()
+    return any(indicator in error_str for indicator in _DISK_SPACE_INDICATORS)
+
 async def wait_for_opensearch(
     opensearch_client: AsyncOpenSearch,
     max_retries: int = 15,