From 1a549de552d4ef4e8e5b4983f81948ac46aa89e9 Mon Sep 17 00:00:00 2001 From: "jinli.yl" Date: Tue, 16 Sep 2025 17:09:09 +0800 Subject: [PATCH] refactor(reme_ai): move FLOW_APP_NAME check to __init__.py - Move the FLOW_APP_NAME environment variable check from app.py to __init__.py - This change ensures that the environment variable is checked as soon as the package is imported - Removed redundant import of os in app.py since it's no longer needed there --- reme_ai/__init__.py | 4 ++++ reme_ai/app.py | 4 ---- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/reme_ai/__init__.py b/reme_ai/__init__.py index 50195757..38a68e34 100644 --- a/reme_ai/__init__.py +++ b/reme_ai/__init__.py @@ -1,3 +1,7 @@ +import os + +assert "FLOW_APP_NAME" in os.environ, "please set FLOW_APP_NAME in `.env`" + from reme_ai import react from reme_ai import retrieve from reme_ai import summary diff --git a/reme_ai/app.py b/reme_ai/app.py index 94df01ae..f4931b25 100644 --- a/reme_ai/app.py +++ b/reme_ai/app.py @@ -1,13 +1,9 @@ -import os import sys import warnings warnings.filterwarnings("ignore", category=DeprecationWarning, module="websockets") warnings.filterwarnings("ignore", category=DeprecationWarning, module="uvicorn") -assert "FLOW_APP_NAME" in os.environ, "please set FLOW_APP_NAME in `.env`" - - from flowllm.app import FlowLLMApp from reme_ai.config.config_parser import ConfigParser