final touches based on greptile reviews

Author: Pouyanpi

.gitignore

@@ -69,5 +69,5 @@ scratch.py
 
 .env
 
-# Ignoring congfiguration files generated by chainlit
+# Ignoring configuration files generated by chainlit
 .chainlit/*

nemoguardrails/cli/__init__.py

@@ -153,15 +153,15 @@ def server(
     try:
         import uvicorn
         from fastapi import FastAPI
+
+        from nemoguardrails.server import api
     except ImportError:
         typer.secho(
             "Server dependencies are missing. Install them with: pip install nemoguardrails[server]",
             fg=typer.colors.RED,
         )
         raise typer.Exit(1)
 
-    from nemoguardrails.server import api
-
     if config:
         # We make sure there is no trailing separator, as that might break things in
         # single config mode.

nemoguardrails/server/api.py

@@ -208,11 +208,7 @@ async def get_rails_configs():
         if os.path.isdir(os.path.join(app.rails_config_path, f))
         and f[0] != "."
         and f[0] != "_"
-        # We filter out all the configs for which there is no `config.yml` file.
-        and (
-            os.path.exists(os.path.join(app.rails_config_path, f, "config.yml"))
-            or os.path.exists(os.path.join(app.rails_config_path, f, "config.yaml"))
-        )
+        and _has_config_file(os.path.join(app.rails_config_path, f))
     ]
 
     return [{"id": config_id} for config_id in config_ids]
@@ -258,6 +254,16 @@ async def list_models(request: Request):
 llm_rails_events_history_cache: dict[str, dict] = {}
 
 
+def _has_config_file(path: str) -> bool:
+    """Check if a directory (or its 'config' subdirectory) contains a config.yml/yaml."""
+    for candidate in [path, os.path.join(path, "config")]:
+        if os.path.exists(os.path.join(candidate, "config.yml")) or os.path.exists(
+            os.path.join(candidate, "config.yaml")
+        ):
+            return True
+    return False
+
+
 def _generate_cache_key(config_ids: List[str], model_name: Optional[str] = None) -> str:
     """Generates a cache key for the given config ids and model name."""
     key = "-".join(config_ids)

nemoguardrails/server/app.py

@@ -22,7 +22,7 @@
 import chainlit as cl
 
 from nemoguardrails.exceptions import StreamingNotSupportedError
-from nemoguardrails.server.api import _get_rails, app, challenges
+from nemoguardrails.server.api import _get_rails, _has_config_file, app, challenges
 
 log = logging.getLogger(__name__)
 
@@ -43,16 +43,6 @@ async def set_starters(user: Optional[cl.User] = None) -> List[cl.Starter]:
     ]
 
 
-def _has_config_file(path: str) -> bool:
-    """Check if a directory (or its 'config' subdirectory) contains a config.yml/yaml."""
-    for candidate in [path, os.path.join(path, "config")]:
-        if os.path.exists(os.path.join(candidate, "config.yml")) or os.path.exists(
-            os.path.join(candidate, "config.yaml")
-        ):
-            return True
-    return False
-
-
 def _discover_configs() -> List[str]:
     """Return the list of available guardrails configuration IDs."""
     if app.single_config_mode and app.single_config_id: