codegraphcontext 0.2.6__py3-none-any.whl → 0.2.7__py3-none-any.whl

codegraphcontext/cli/config_manager.py

@@ -16,7 +16,7 @@ CONFIG_DIR = Path.home() / ".codegraphcontext"
 CONFIG_FILE = CONFIG_DIR / ".env"
 
 # Database credential keys (stored in same .env file but not managed as config)
-DATABASE_CREDENTIAL_KEYS = {"NEO4J_URI", "NEO4J_USERNAME", "NEO4J_PASSWORD"}
+DATABASE_CREDENTIAL_KEYS = {"NEO4J_URI", "NEO4J_USERNAME", "NEO4J_PASSWORD", "NEO4J_DATABASE"}
 
 # Default configuration values
 DEFAULT_CONFIG = {
@@ -28,6 +28,7 @@ DEFAULT_CONFIG = {
     "DEBUG_LOGS": "false",
     "DEBUG_LOG_PATH": str(Path.home() / "mcp_debug.log"),
     "ENABLE_APP_LOGS": "CRITICAL",
+    "LIBRARY_LOG_LEVEL": "WARNING",
     "LOG_FILE_PATH": str(CONFIG_DIR / "logs" / "cgc.log"),
     "MAX_FILE_SIZE_MB": "10",
     "IGNORE_TEST_FILES": "false",
@@ -55,6 +56,7 @@ CONFIG_DESCRIPTIONS = {
     "DEBUG_LOGS": "Enable debug logging (for development/troubleshooting)",
     "DEBUG_LOG_PATH": "Path to debug log file",
     "ENABLE_APP_LOGS": "Application log level (DEBUG|INFO|WARNING|ERROR|CRITICAL|DISABLED)",
+    "LIBRARY_LOG_LEVEL": "Log level for third-party libraries (neo4j, asyncio, urllib3) (DEBUG|INFO|WARNING|ERROR|CRITICAL)",
     "LOG_FILE_PATH": "Path to application log file",
     "MAX_FILE_SIZE_MB": "Maximum file size to index (in MB)",
     "IGNORE_TEST_FILES": "Skip test files during indexing",
@@ -78,6 +80,7 @@ CONFIG_VALIDATORS = {
     "ALLOW_DB_DELETION": ["true", "false"],
     "DEBUG_LOGS": ["true", "false"],
     "ENABLE_APP_LOGS": ["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL", "DISABLED"],
+    "LIBRARY_LOG_LEVEL": ["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"],
     "IGNORE_TEST_FILES": ["true", "false"],
     "IGNORE_HIDDEN_FILES": ["true", "false"],
     "ENABLE_AUTO_WATCH": ["true", "false"],
@@ -86,13 +89,15 @@ CONFIG_VALIDATORS = {
     "SCIP_INDEXER": ["true", "false"],
     "SKIP_EXTERNAL_RESOLUTION": ["true", "false"],
 }
+def ensure_config_dir(path: Path = CONFIG_DIR):
+    """
+    Ensure that the configuration directory exists.
+    Creates the directory and a logs subdirectory if they do not already exist.
+    """
+    path.mkdir(parents=True, exist_ok=True)
+    (path / "logs").mkdir(parents=True, exist_ok=True)
 
 
-def ensure_config_dir():
-    """Ensure configuration directory exists."""
-    CONFIG_DIR.mkdir(parents=True, exist_ok=True)
-    (CONFIG_DIR / "logs").mkdir(parents=True, exist_ok=True)
-
 
 def load_config() -> Dict[str, str]:
     """

codegraphcontext/cli/main.py

@@ -44,10 +44,24 @@ from .cli_helpers import (
     list_watching_helper,
 )
 
-# Set the log level for the noisy neo4j, asyncio, and urllib3 loggers to WARNING to keep the output clean.
-logging.getLogger("neo4j").setLevel(logging.WARNING)
-logging.getLogger("asyncio").setLevel(logging.WARNING)
-logging.getLogger("urllib3").setLevel(logging.WARNING)
+# Set the log level for the noisy neo4j, asyncio, and urllib3 loggers to keep the output clean.
+# Get the log level from config, defaulting to WARNING
+def _configure_library_loggers():
+    """Configure third-party library loggers based on config setting."""
+    try:
+        log_level_str = config_manager.get_config_value('LIBRARY_LOG_LEVEL')
+        if log_level_str is None:
+            log_level_str = 'WARNING'
+        log_level_str = str(log_level_str).upper()
+        log_level = getattr(logging, log_level_str, logging.WARNING)
+    except (AttributeError, Exception):
+        log_level = logging.WARNING
+    
+    logging.getLogger("neo4j").setLevel(log_level)
+    logging.getLogger("asyncio").setLevel(log_level)
+    logging.getLogger("urllib3").setLevel(log_level)
+
+_configure_library_loggers()
 
 
 # Import visualization module
@@ -288,11 +302,24 @@ def _load_credentials():
             os.environ.get("NEO4J_PASSWORD")
         ])
         if has_neo4j_creds:
-            console.print("[cyan]Using database: Neo4j[/cyan]")
+            neo4j_db = os.environ.get("NEO4J_DATABASE")
+            if neo4j_db:
+                console.print(f"[cyan]Using database: Neo4j (database: {neo4j_db})[/cyan]")
+            else:
+                console.print("[cyan]Using database: Neo4j[/cyan]")
         else:
             console.print("[yellow]⚠ DEFAULT_DATABASE=neo4j but credentials not found. Falling back to FalkorDB.[/yellow]")
+    elif default_db == "falkordb-remote":
+        host = os.environ.get("FALKORDB_HOST")
+        if host:
+            console.print(f"[cyan]Using database: FalkorDB Remote ({host})[/cyan]")
+        else:
+            console.print("[yellow]⚠ DATABASE_TYPE=falkordb-remote but FALKORDB_HOST not set.[/yellow]")
     else:
-        console.print("[cyan]Using database: FalkorDB[/cyan]")
+        if os.environ.get("FALKORDB_HOST"):
+            console.print(f"[cyan]Using database: FalkorDB Remote ({os.environ.get('FALKORDB_HOST')})[/cyan]")
+        else:
+            console.print("[cyan]Using database: FalkorDB[/cyan]")
 
 # ============================================================================
 # CONFIG COMMAND GROUP
@@ -352,9 +379,9 @@ def config_db(backend: str = typer.Argument(..., help="Database backend: 'neo4j'
         cgc config db falkordb
     """
     backend = backend.lower()
-    if backend not in ['falkordb', 'neo4j']:
+    if backend not in ['falkordb', 'falkordb-remote', 'neo4j']:
         console.print(f"[bold red]Invalid backend: {backend}[/bold red]")
-        console.print("Must be 'falkordb' or 'neo4j'")
+        console.print("Must be 'falkordb', 'falkordb-remote', or 'neo4j'")
         raise typer.Exit(code=1)
     
     config_manager.set_config_value("DEFAULT_DATABASE", backend)
@@ -700,7 +727,7 @@ def doctor():
             
             if uri and username and password:
                 console.print(f"   [cyan]Testing Neo4j connection to {uri}...[/cyan]")
-                is_connected, error_msg = DatabaseManager.test_connection(uri, username, password)
+                is_connected, error_msg = DatabaseManager.test_connection(uri, username, password, database=os.environ.get("NEO4J_DATABASE"))
                 if is_connected:
                     console.print(f"   [green]✓[/green] Neo4j connection successful")
                 else:

codegraphcontext/cli/setup_wizard.py

@@ -377,7 +377,7 @@ def configure_mcp_client():
                     if line and not line.startswith("#") and "=" in line:
                         key, value = line.split("=", 1)
                         key = key.strip()
-                        if key in ["NEO4J_URI", "NEO4J_USERNAME", "NEO4J_PASSWORD"]:
+                        if key in ["NEO4J_URI", "NEO4J_USERNAME", "NEO4J_PASSWORD", "NEO4J_DATABASE"]:
                             env_vars[key] = value.strip()
         except Exception:
             pass