openrag 0.5.0.dev37__tar.gz → 0.5.1.dev1__tar.gz

{openrag-0.5.0.dev37/src/openrag.egg-info → openrag-0.5.1.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openrag
-Version: 0.5.0.dev37
+Version: 0.5.1.dev1
 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.5.0.dev37 → openrag-0.5.1.dev1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "openrag"
-version = "0.5.0.dev37"
+version = "0.5.1.dev1"
 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.5.0.dev37 → openrag-0.5.1.dev1}/src/api/settings.py

@@ -1676,10 +1676,7 @@ async def _update_langflow_model_values(config, flows_service, llm_model=None, l
 async def _update_langflow_system_prompt(config, flows_service):
     """Update system prompt in chat flow"""
     try:
-        llm_provider = config.agent.llm_provider.lower()
-        await flows_service.update_chat_flow_system_prompt(
-            config.agent.system_prompt, llm_provider
-        )
+        await flows_service.update_chat_flow_system_prompt(config.agent.system_prompt)
         logger.info("Successfully updated chat flow system prompt")
     except Exception as e:
         logger.error(f"Failed to update chat flow system prompt: {str(e)}")

{openrag-0.5.0.dev37 → openrag-0.5.1.dev1}/src/main.py

@@ -1619,7 +1619,7 @@ async def create_app():
     """Create and configure the FastAPI application"""
     services = await initialize_services()
 
-    app = FastAPI(title="OpenRAG API", version=OPENRAG_VERSION, debug=True)
+    app = FastAPI(title="OpenRAG API", version=OPENRAG_VERSION, debug=True, lifespan=lifespan)
     app.state.services = services  # Store services for cleanup
     app.state.background_tasks = set()
 
@@ -2139,102 +2139,9 @@ async def create_app():
     # FastMCP requires its own lifespan to be run so that the
     # StreamableHTTPSessionManager task group is initialized before requests arrive.
     # FastAPI does not automatically propagate lifespan to mounted sub-apps,
-    # so we wire it in manually via startup/shutdown handlers.
+    # so we wire it in manually via the parent's lifespan (using app.state).
     _mcp_lifespan_ctx = mcp_http_app.router.lifespan_context(mcp_http_app)
-
-    async def _start_mcp_lifespan():
-        await _mcp_lifespan_ctx.__aenter__()
-        logger.info("FastMCP lifespan started")
-
-    async def _stop_mcp_lifespan():
-        await _mcp_lifespan_ctx.__aexit__(None, None, None)
-        logger.info("FastMCP lifespan stopped")
-
-    app.add_event_handler("startup", _start_mcp_lifespan)
-    app.add_event_handler("shutdown", _stop_mcp_lifespan)
-
-    # Add startup event handler
-    @app.on_event("startup")
-    async def startup_event():
-        await TelemetryClient.send_event(
-            Category.APPLICATION_STARTUP, MessageId.ORB_APP_STARTED
-        )
-        # Start index initialization in background to avoid blocking OIDC endpoints
-        t1 = asyncio.create_task(startup_tasks(services))
-        app.state.background_tasks.add(t1)
-        t1.add_done_callback(app.state.background_tasks.discard)
-
-        # Start periodic task cleanup scheduler
-        services["task_service"].start_cleanup_scheduler()
-
-        # Start periodic flow backup task (every 5 minutes)
-        async def periodic_backup():
-            """Periodic backup task that runs every 15 minutes"""
-            while True:
-                try:
-                    await asyncio.sleep(5 * 60)  # Wait 5 minutes
-
-                    # Check if onboarding has been completed
-                    config = get_openrag_config()
-                    if not config.edited:
-                        logger.debug(
-                            "Onboarding not completed yet, skipping periodic backup"
-                        )
-                        continue
-
-                    flows_service = services.get("flows_service")
-                    if flows_service:
-                        logger.info("Running periodic flow backup")
-                        backup_results = await flows_service.backup_all_flows(
-                            only_if_changed=True
-                        )
-                        if backup_results["backed_up"]:
-                            logger.info(
-                                "Periodic backup completed",
-                                backed_up=len(backup_results["backed_up"]),
-                                skipped=len(backup_results["skipped"]),
-                            )
-                        else:
-                            logger.debug(
-                                "Periodic backup: no flows changed",
-                                skipped=len(backup_results["skipped"]),
-                            )
-                except asyncio.CancelledError:
-                    logger.info("Periodic backup task cancelled")
-                    break
-                except Exception as e:
-                    logger.error(f"Error in periodic backup task: {str(e)}")
-                    # Continue running even if one backup fails
-
-        backup_task = asyncio.create_task(periodic_backup())
-        app.state.background_tasks.add(backup_task)
-        backup_task.add_done_callback(app.state.background_tasks.discard)
-
-    # Add shutdown event handler
-    @app.on_event("shutdown")
-    async def shutdown_event():
-        await TelemetryClient.send_event(
-            Category.APPLICATION_SHUTDOWN, MessageId.ORB_APP_SHUTDOWN
-        )
-        logger.info("Application shutdown initiated")
-
-        # Gracefully shutdown OpenSearch connection first
-        try:
-            from utils.opensearch_utils import graceful_opensearch_shutdown
-            await graceful_opensearch_shutdown(clients.opensearch)
-        except Exception as e:
-            logger.error("Error during graceful OpenSearch shutdown", error=str(e))
-
-        await cleanup_subscriptions_proper(services)
-        # Cleanup task service (cancels background tasks and process pool)
-        await services["task_service"].shutdown()
-        # Cleanup async clients (this will also close OpenSearch client if not already closed)
-        await clients.cleanup()
-        # Cleanup telemetry client
-        from utils.telemetry.client import cleanup_telemetry_client
-
-        await cleanup_telemetry_client()
-        logger.info("Application shutdown completed")
+    app.state.mcp_lifespan_ctx = _mcp_lifespan_ctx
 
     return app
 

{openrag-0.5.0.dev37 → openrag-0.5.1.dev1/src/openrag.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openrag
-Version: 0.5.0.dev37
+Version: 0.5.1.dev1
 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.5.0.dev37 → openrag-0.5.1.dev1}/src/services/flows_service.py

@@ -673,16 +673,14 @@ class FlowsService:
         await self._update_flow_field(LANGFLOW_CHAT_FLOW_ID, "model_name", model_name,
                                 node_display_name=target_llm_id)
 
-    async def update_chat_flow_system_prompt(self, system_prompt: str, provider: str):
+    async def update_chat_flow_system_prompt(self, system_prompt: str):
         """Helper function to update the system prompt in the chat flow"""
         if not LANGFLOW_CHAT_FLOW_ID:
             raise ValueError("LANGFLOW_CHAT_FLOW_ID is not configured")
 
-        # Determine target component IDs based on provider
-        target_agent_id = self._get_provider_component_ids(provider)[1]
-
-        await self._update_flow_field(LANGFLOW_CHAT_FLOW_ID, "system_prompt", system_prompt,
-                                node_display_name=target_agent_id)
+        await self._update_flow_field(
+            LANGFLOW_CHAT_FLOW_ID, "system_prompt", system_prompt, node_display_name=AGENT_COMPONENT_DISPLAY_NAME
+        )
 
     async def update_flow_docling_preset(self, preset: str, preset_config: dict):
         """Helper function to update docling preset in the ingest flow"""