OpenRAG 0.3.1__tar.gz → 0.3.3__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openrag
-Version: 0.3.1
+Version: 0.3.3
 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
@@ -15,7 +15,6 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.13
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: agentd>=0.2.2
 Requires-Dist: aiofiles>=24.1.0
 Requires-Dist: cryptography>=45.0.6
 Requires-Dist: google-api-python-client>=2.143.0
@@ -36,6 +35,9 @@ Requires-Dist: python-dotenv>=1.0.0
 Requires-Dist: textual-fspicker>=0.6.0
 Requires-Dist: structlog>=25.4.0
 Requires-Dist: zxcvbn>=4.5.0
+Requires-Dist: openai>=1.0.0
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: tiktoken>=0.7.0
 Dynamic: license-file
 
 <div align="center">

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

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "openrag"
-version = "0.3.1"
+version = "0.3.3"
 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"
@@ -21,7 +21,6 @@ classifiers = [
     "Topic :: Software Development :: Libraries :: Python Modules",
 ]
 dependencies = [
-    "agentd>=0.2.2",
     "aiofiles>=24.1.0",
     "cryptography>=45.0.6",
     "google-api-python-client>=2.143.0",
@@ -41,7 +40,10 @@ dependencies = [
     "python-dotenv>=1.0.0",
     "textual-fspicker>=0.6.0",
     "structlog>=25.4.0",
-    "zxcvbn>=4.5.0"
+    "zxcvbn>=4.5.0",
+    "openai>=1.0.0",
+    "pyyaml>=6.0",
+    "tiktoken>=0.7.0"
 ]
 
 [dependency-groups]

{openrag-0.3.1 → openrag-0.3.3}/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.3}/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.3}/src/config/config_manager.py

@@ -49,6 +49,10 @@ class ProvidersConfig:
     watsonx: WatsonXConfig
     ollama: OllamaConfig
 
+    def any_configured(self) -> bool:
+        """Return True if at least one provider is marked as configured."""
+        return any(p.configured for p in (self.openai, self.anthropic, self.watsonx, self.ollama))
+
     def get_provider_config(self, provider: str):
         """Get configuration for a specific provider."""
         provider_lower = provider.lower()

{openrag-0.3.1 → openrag-0.3.3}/src/config/settings.py

@@ -3,7 +3,6 @@ import os
 from utils.env_utils import get_env_int, get_env_float
 
 import httpx
-from agentd.patch import patch_openai_with_mcp
 from dotenv import load_dotenv
 from openai import AsyncOpenAI
 from opensearchpy import AsyncOpenSearch
@@ -535,16 +534,14 @@ class AppClients:
                     use_http2 = future.result(timeout=15)
 
                 if use_http2:
-                    self._patched_async_client = patch_openai_with_mcp(AsyncOpenAI())
+                    self._patched_async_client = AsyncOpenAI()
                     logger.info("OpenAI client initialized with HTTP/2")
                 else:
                     http_client = httpx.AsyncClient(
                         http2=False,
                         timeout=httpx.Timeout(60.0, connect=10.0)
                     )
-                    self._patched_async_client = patch_openai_with_mcp(
-                        AsyncOpenAI(http_client=http_client)
-                    )
+                    self._patched_async_client = AsyncOpenAI(http_client=http_client)
                     logger.info("OpenAI client initialized with HTTP/1.1 (fallback)")
                 logger.info("Successfully initialized OpenAI client")
             except Exception as e:

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openrag
-Version: 0.3.1
+Version: 0.3.3
 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
@@ -15,7 +15,6 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.13
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: agentd>=0.2.2
 Requires-Dist: aiofiles>=24.1.0
 Requires-Dist: cryptography>=45.0.6
 Requires-Dist: google-api-python-client>=2.143.0
@@ -36,6 +35,9 @@ Requires-Dist: python-dotenv>=1.0.0
 Requires-Dist: textual-fspicker>=0.6.0
 Requires-Dist: structlog>=25.4.0
 Requires-Dist: zxcvbn>=4.5.0
+Requires-Dist: openai>=1.0.0
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: tiktoken>=0.7.0
 Dynamic: license-file
 
 <div align="center">

{openrag-0.3.1 → openrag-0.3.3}/src/openrag.egg-info/requires.txt

@@ -1,4 +1,3 @@
-agentd>=0.2.2
 aiofiles>=24.1.0
 cryptography>=45.0.6
 google-api-python-client>=2.143.0
@@ -19,3 +18,6 @@ python-dotenv>=1.0.0
 textual-fspicker>=0.6.0
 structlog>=25.4.0
 zxcvbn>=4.5.0
+openai>=1.0.0
+pyyaml>=6.0
+tiktoken>=0.7.0

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

@@ -1,7 +1,6 @@
 import copy
 import json
 from typing import Any, Dict
-from agentd.tool_decorator import tool
 from config.settings import EMBED_MODEL, clients, get_embedding_model, get_index_name, WATSONX_EMBEDDING_DIMENSIONS
 from auth_context import get_auth_context
 from utils.logging_config import get_logger
@@ -17,7 +16,6 @@ class SearchService:
     def __init__(self, session_manager=None):
         self.session_manager = session_manager
 
-    @tool
     async def search_tool(self, query: str, embedding_model: str = None) -> Dict[str, Any]:
         """
         Use this tool to search for documents relevant to the query.
@@ -392,6 +390,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 +402,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 +422,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 +439,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.3}/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.3}/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,