feat: add flexible configuration file support for agent and simunet services

Add command-line arguments and environment variable support for specifying custom configuration files when starting agent and simunet services, enabling flexible deployment across different environments.

Changes:
- Add CLI arguments: -c/--config, --host, -p/--port, --reload/--no-reload
- Support environment variables: NETDRIVER_AGENT_CONFIG and NETDRIVER_SIMUNET_CONFIG
- Maintain backward compatibility with default config paths
- Dynamic config reloading when custom config is specified
- Fix incorrect import in conn_req_handler.py (Session from pytest -> netdriver.client.session)

Usage examples:
  poetry run agent -c config/production.yml --no-reload
  poetry run simunet -c config/test-simunet.yml -p 9001
  NETDRIVER_AGENT_CONFIG=config/custom.yml poetry run agent

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>

Author: vincentbergman

bases/netdriver/agent/containers.py

@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 
+import os
 from dependency_injector.containers import DeclarativeContainer
 from dependency_injector.providers import Factory, Configuration
 from netdriver.agent.handlers.cmd_req_handler import CommandRequestHandler
@@ -9,8 +10,15 @@
 
 class Container(DeclarativeContainer):
     """ IoC container of netdriver agent. """
-    config = Configuration(yaml_files=["config/agent/agent.yml"])
+    config = Configuration()
     cmd_req_handler = Factory(CommandRequestHandler)
     conn_req_handler = Factory(ConnectRequestHandler)
 
+
+def get_config_file() -> str:
+    """Get config file path from environment variable or use default."""
+    return os.getenv("NETDRIVER_AGENT_CONFIG", "config/agent/agent.yml")
+
+
 container = Container()
+container.config.from_yaml(get_config_file())

bases/netdriver/agent/handlers/conn_req_handler.py

@@ -1,9 +1,9 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
-from pytest import Session
 from netdriver.agent.models.common import CommonResponse
 from netdriver.agent.models.conn import ConnectRequest
 from netdriver.client.pool import SessionPool
+from netdriver.client.session import Session
 from netdriver.exception.errors import ConnectFailed
 
 

bases/netdriver/agent/main.py

@@ -4,6 +4,9 @@
 This is the main module for the agent.
 It is responsible for starting the FastAPI server.
 """
+import os
+import sys
+import argparse
 from contextlib import asynccontextmanager
 
 import uvicorn
@@ -68,4 +71,58 @@ async def root() -> dict:
 
 
 def start():
-    uvicorn.run("netdriver.agent.main:app", host="0.0.0.0", port=8000, reload=True)
+    """Start the agent server with optional configuration file parameter."""
+    parser = argparse.ArgumentParser(description="NetDriver Agent Server")
+    parser.add_argument(
+        "-c", "--config",
+        type=str,
+        default=None,
+        help="Path to configuration file (default: config/agent/agent.yml or NETDRIVER_AGENT_CONFIG env var)"
+    )
+    parser.add_argument(
+        "--host",
+        type=str,
+        default="0.0.0.0",
+        help="Host to bind (default: 0.0.0.0)"
+    )
+    parser.add_argument(
+        "-p", "--port",
+        type=int,
+        default=8000,
+        help="Port to bind (default: 8000)"
+    )
+    parser.add_argument(
+        "--reload",
+        action="store_true",
+        default=True,
+        help="Enable auto-reload (default: True)"
+    )
+    parser.add_argument(
+        "--no-reload",
+        action="store_true",
+        help="Disable auto-reload"
+    )
+
+    args = parser.parse_args()
+
+    # Set config file path via environment variable if specified
+    if args.config:
+        os.environ["NETDRIVER_AGENT_CONFIG"] = args.config
+        # Reload container configuration with new config file
+        container.config.from_yaml(args.config)
+        # Reconfigure logging with new config
+        logman.configure_logman(
+            level=container.config.logging.level(),
+            intercept_loggers=container.config.logging.intercept_loggers(),
+            log_file=container.config.logging.log_file()
+        )
+
+    # Handle reload flag
+    reload = args.reload and not args.no_reload
+
+    uvicorn.run(
+        "netdriver.agent.main:app",
+        host=args.host,
+        port=args.port,
+        reload=reload
+    )

bases/netdriver/simunet/containers.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
+import os
 from dependency_injector.containers import DeclarativeContainer
 from dependency_injector.providers import Configuration
 
@@ -9,5 +10,10 @@ class Container(DeclarativeContainer):
     config = Configuration()
 
 
+def get_config_file() -> str:
+    """Get config file path from environment variable or use default."""
+    return os.getenv("NETDRIVER_SIMUNET_CONFIG", "config/simunet/simunet.yml")
+
+
 container = Container()
-container.config.from_yaml("config/simunet/simunet.yml", required=True)
+container.config.from_yaml(get_config_file(), required=True)

bases/netdriver/simunet/main.py

@@ -1,5 +1,7 @@
 #!/usr/bin/env python3.10.6
 # -*- coding: utf-8 -*-
+import os
+import argparse
 import asyncio
 from typing import AsyncGenerator
 
@@ -54,4 +56,58 @@ async def on_shutdown() -> None:
 
 
 def start():
-    uvicorn.run("netdriver.simunet.main:app", host="0.0.0.0", port=8001, reload=True)
+    """Start the simunet server with optional configuration file parameter."""
+    parser = argparse.ArgumentParser(description="NetDriver SimuNet Server")
+    parser.add_argument(
+        "-c", "--config",
+        type=str,
+        default=None,
+        help="Path to configuration file (default: config/simunet/simunet.yml or NETDRIVER_SIMUNET_CONFIG env var)"
+    )
+    parser.add_argument(
+        "--host",
+        type=str,
+        default="0.0.0.0",
+        help="Host to bind (default: 0.0.0.0)"
+    )
+    parser.add_argument(
+        "-p", "--port",
+        type=int,
+        default=8001,
+        help="Port to bind (default: 8001)"
+    )
+    parser.add_argument(
+        "--reload",
+        action="store_true",
+        default=True,
+        help="Enable auto-reload (default: True)"
+    )
+    parser.add_argument(
+        "--no-reload",
+        action="store_true",
+        help="Disable auto-reload"
+    )
+
+    args = parser.parse_args()
+
+    # Set config file path via environment variable if specified
+    if args.config:
+        os.environ["NETDRIVER_SIMUNET_CONFIG"] = args.config
+        # Reload container configuration with new config file
+        container.config.from_yaml(args.config, required=True)
+        # Reconfigure logging with new config
+        logman.configure_logman(
+            level=container.config.logging.level(),
+            intercept_loggers=container.config.logging.intercept_loggers(),
+            log_file=container.config.logging.log_file()
+        )
+
+    # Handle reload flag
+    reload = args.reload and not args.no_reload
+
+    uvicorn.run(
+        "netdriver.simunet.main:app",
+        host=args.host,
+        port=args.port,
+        reload=reload
+    )