From 51cd807ab85c5a932b1fc8ca073755613b0ba976 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sebasti=C3=A1n=20Est=C3=A9vez?= Date: Fri, 16 May 2025 15:15:38 -0400 Subject: [PATCH] fix: g-assist import bug (#8092) fix import bug --- .../base/langflow/components/nvidia/system_assist.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/backend/base/langflow/components/nvidia/system_assist.py b/src/backend/base/langflow/components/nvidia/system_assist.py index 2047cfcf0..d30e7b4f0 100644 --- a/src/backend/base/langflow/components/nvidia/system_assist.py +++ b/src/backend/base/langflow/components/nvidia/system_assist.py @@ -6,9 +6,6 @@ from langflow.io import MessageTextInput, Output from langflow.schema import Message from langflow.services.cache.utils import CacheMiss -with contextlib.suppress(ImportError): - from gassist.rise import register_rise_client, send_rise_command - class NvidiaSystemAssistComponent(ComponentWithCache): display_name = "NVIDIA System-Assist" @@ -52,6 +49,9 @@ class NvidiaSystemAssistComponent(ComponentWithCache): ] async def sys_assist_prompt(self) -> Message: + with contextlib.suppress(ImportError): + from gassist.rise import send_rise_command + self.maybe_register_rise_client() response = await asyncio.to_thread(send_rise_command, self.prompt)