openrag 0.6.0.dev18__tar.gz → 0.6.0.dev20__tar.gz

{openrag-0.6.0.dev18/src/openrag.egg-info → openrag-0.6.0.dev20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openrag
-Version: 0.6.0.dev18
+Version: 0.6.0.dev20
 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.6.0.dev18 → openrag-0.6.0.dev20}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "openrag"
-version = "0.6.0.dev18"
+version = "0.6.0.dev20"
 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.6.0.dev20/src/api/flows.py

@@ -0,0 +1,117 @@
+"""Reset Flow API endpoints"""
+
+from typing import Literal
+
+from fastapi import Depends
+from fastapi.responses import JSONResponse
+from pydantic import BaseModel
+
+from dependencies import get_flows_service, require_permission
+from session_manager import User
+from utils.logging_config import get_logger
+
+logger = get_logger(__name__)
+
+FlowType = Literal["nudges", "retrieval", "ingest"]
+
+
+def _is_oss_mode() -> bool:
+    from config.settings import IBM_AUTH_ENABLED, is_dev_ibm_cos_enabled
+
+    return not (IBM_AUTH_ENABLED or is_dev_ibm_cos_enabled())
+
+
+async def reset_flow_endpoint(
+    flow_type: str,
+    flows_service=Depends(get_flows_service),
+    user: User = Depends(require_permission("flows:edit")),
+):
+    """Reset a Langflow flow by type (nudges, retrieval, or ingest)"""
+    if flow_type not in ["nudges", "retrieval", "ingest"]:
+        return JSONResponse(
+            {
+                "success": False,
+                "error": "Invalid flow type. Must be 'nudges', 'retrieval', or 'ingest'",
+            },
+            status_code=400,
+        )
+
+    try:
+        result = await flows_service.reset_langflow_flow(flow_type)
+
+        if result.get("success"):
+            logger.info("Flow reset successful", flow_type=flow_type, flow_id=result.get("flow_id"))
+            return JSONResponse(result, status_code=200)
+        else:
+            logger.error("Flow reset failed", flow_type=flow_type, error=result.get("error"))
+            return JSONResponse(result, status_code=500)
+
+    except ValueError as e:
+        logger.error("Invalid request for flow reset", error=str(e))
+        return JSONResponse({"success": False, "error": str(e)}, status_code=400)
+    except Exception as e:
+        logger.error("Unexpected error in flow reset", error=str(e))
+        return JSONResponse({"success": False, "error": "Internal server error"}, status_code=500)
+
+
+class BulkUpdateFlowsRequest(BaseModel):
+    flow_types: list[str]
+    backup_custom: bool = True
+
+
+async def get_flows_updates_endpoint(
+    flows_service=Depends(get_flows_service),
+    user: User = Depends(require_permission("flows:read")),
+):
+    """Get available updates for core flows"""
+    if not _is_oss_mode():
+        return JSONResponse({"success": True, "updates": []}, status_code=200)
+    try:
+        user_id = getattr(user, "db_user_id", None) or getattr(user, "user_id", None)
+        updates = await flows_service.get_flows_updates_available(user_id=user_id)
+        return JSONResponse({"success": True, "updates": updates}, status_code=200)
+    except Exception as e:
+        logger.error("Error getting flow updates", error=str(e))
+        return JSONResponse({"success": False, "error": "Internal server error"}, status_code=500)
+
+
+async def bulk_update_flows_endpoint(
+    request: BulkUpdateFlowsRequest,
+    flows_service=Depends(get_flows_service),
+    user: User = Depends(require_permission("flows:edit")),
+):
+    """Bulk update multiple flows and optionally backup custom flows"""
+    if not _is_oss_mode():
+        return JSONResponse(
+            {"success": False, "error": "Flow updates are only available in OSS mode"},
+            status_code=403,
+        )
+    try:
+        results = await flows_service.bulk_update_flows(request.flow_types, request.backup_custom)
+        overall_success = all(r.get("success", False) for r in results)
+        return JSONResponse({"success": overall_success, "results": results}, status_code=200)
+    except Exception as e:
+        logger.error("Error bulk updating flows", error=str(e))
+        return JSONResponse({"success": False, "error": "Internal server error"}, status_code=500)
+
+
+class DismissFlowsUpdateRequest(BaseModel):
+    flow_types: list[str] | None = None
+
+
+async def dismiss_flows_update_endpoint(
+    request: DismissFlowsUpdateRequest | None = None,
+    flows_service=Depends(get_flows_service),
+    user: User = Depends(require_permission("flows:read")),
+):
+    """Dismiss available flow update notifications ephemerally in backend memory"""
+    if not _is_oss_mode():
+        return JSONResponse({"success": True}, status_code=200)
+    try:
+        flow_types = request.flow_types if request else None
+        user_id = getattr(user, "db_user_id", None) or getattr(user, "user_id", None)
+        flows_service.dismiss_flows_updates(flow_types, user_id=user_id)
+        return JSONResponse({"success": True}, status_code=200)
+    except Exception as e:
+        logger.error("Error dismissing flow updates", error=str(e))
+        return JSONResponse({"success": False, "error": "Internal server error"}, status_code=500)

{openrag-0.6.0.dev18 → openrag-0.6.0.dev20}/src/app/routes/internal.py

@@ -426,6 +426,26 @@ def register_internal_routes(app: FastAPI):
         tags=["internal"],
     )
 
+    # Flow updates endpoints
+    app.add_api_route(
+        "/settings/flows/updates-available",
+        flows.get_flows_updates_endpoint,
+        methods=["GET"],
+        tags=["internal"],
+    )
+    app.add_api_route(
+        "/settings/flows/update",
+        flows.bulk_update_flows_endpoint,
+        methods=["POST"],
+        tags=["internal"],
+    )
+    app.add_api_route(
+        "/settings/flows/dismiss-update",
+        flows.dismiss_flows_update_endpoint,
+        methods=["POST"],
+        tags=["internal"],
+    )
+
     # Router upload ingest endpoint
     app.add_api_route(
         "/router/upload_ingest",

{openrag-0.6.0.dev18 → openrag-0.6.0.dev20/src/openrag.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openrag
-Version: 0.6.0.dev18
+Version: 0.6.0.dev20
 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