Merge pull request #70 from EnvisionMindCa/codex/make-backend-production-ready

README.md

@@ -8,6 +8,14 @@
 - **Tool execution** – a built-in `execute_terminal` tool runs commands inside a Docker-based VM. Network access is enabled and both stdout and stderr are captured (up to 10,000 characters). The VM is reused across chats when `PERSIST_VMS=1` so installed packages remain available.
 - **System prompts** – every request includes a system prompt that guides the assistant to plan tool usage, verify results and avoid unnecessary jargon.
 
+## Environment Variables
+
+Several settings can be customised via environment variables:
+
+- `DB_PATH` – location of the SQLite database (default `chat.db` in the project directory).
+- `LOG_LEVEL` – logging verbosity (`DEBUG`, `INFO`, etc.).
+- `VM_IMAGE` and `VM_STATE_DIR` control the Docker-based VM.
+
 ## Quick Start
 
 ```bash

src/config.py

@@ -16,6 +16,7 @@ PERSIST_VMS: Final[bool] = os.getenv("PERSIST_VMS", "1") == "1"
 VM_STATE_DIR: Final[str] = os.getenv(
     "VM_STATE_DIR", str(Path.cwd() / "vm_state")
 )
+DB_PATH: Final[str] = os.getenv("DB_PATH", str(Path.cwd() / "chat.db"))
 
 SYSTEM_PROMPT: Final[str] = (
     "You are Starlette, the senior agent leading a two-agent team. "

src/db.py

@@ -3,6 +3,8 @@ from __future__ import annotations
 from datetime import datetime
 from pathlib import Path
 
+from .config import DB_PATH
+
 from peewee import (
     AutoField,
     CharField,
@@ -14,7 +16,7 @@ from peewee import (
 )
 
 
-_DB_PATH = Path(__file__).resolve().parent.parent / "chat.db"
+_DB_PATH = Path(DB_PATH)
 _db = SqliteDatabase(_DB_PATH)
 
 

src/log.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import logging
+import os
 from typing import Final
 
 from colorama import Fore, Style, init as colorama_init
@@ -22,10 +23,18 @@ class _ColourFormatter(logging.Formatter):
         return f"{colour}{super().format(record)}{Style.RESET_ALL}"
 
 
-def get_logger(name: str | None = None, level: int = logging.INFO) -> logging.Logger:
+def get_logger(name: str | None = None, level: int | None = None) -> logging.Logger:
+    """Return a configured logger instance."""
+
     colorama_init()
+    env_level = os.getenv("LOG_LEVEL", "INFO").upper()
+    if level is None:
+        level = getattr(logging, env_level, logging.INFO)
+
     handler = logging.StreamHandler()
-    handler.setFormatter(_ColourFormatter("%(asctime)s [%(levelname)s] %(name)s: %(message)s"))
+    handler.setFormatter(
+        _ColourFormatter("%(asctime)s [%(levelname)s] %(name)s: %(message)s")
+    )
 
     logger = logging.getLogger(name)
     logger.setLevel(level)