regcode 0.3.0__tar.gz → 0.4.1__tar.gz

{regcode-0.3.0 → regcode-0.4.1}/GOALS.md

@@ -492,11 +492,14 @@ All phases complete. 50/50 tests passing. Lint clean.
 ### LLM Hallucinated Tool Calls Fix
 - DONE: Model was hallucinating tool calls (e.g. `<tool_call>`) when budget was exhausted and `tools=None`. Root cause: tool definitions were baked into the system prompt string, so even with `tools=None` the model still saw tool names and hallucinated calling them. Fix: when budget is exhausted, replace the system prompt with a version stripped of tool definitions (`_build_system_prompt_no_tools()`). Additionally set `tool_choice="none"` and stop sequences `["<tool_call>", "</tool_use>"]`. Message history (including tool_call_ids) is preserved intact to maintain proper conversation continuity.
 
-## Top Priority To-Dos:
 
-- Currently agent could keep on loading files running into context window limitations. We need to add content compaction threshold and add automatic compaction support.
-- I think for this agent to be able to work on big repos, we need to add `add_review_note` tool which basically can write review notes into file which can survive compaction. Review notes are supposed to be preleminary findings. Add `read_review_notes` as well. This must work intune with compaction feature
-- Save tokens reported by litellm and use following logic `context_window=max_tokens + message_chain_token_size` and compact based on compaction threshold.
-- _build_system_prompt in regcode/main.py and _build_code_review_system_prompt in the same file share an identical loop that constructs a "tool descriptions" string from registered tools. The same logic appears twice. This should be extracted into a single helper method, e.g. _format_tool_descriptions(self) -> str, called by both.
+### COMPLETED (2026-06-29)
 
-Additionally, _build_code_review_system_prompt in regcode/main.py calls self.registry.list_tools() and builds tool descriptions, but it could simply accept a system prompt string and tool descriptions as parameters, instead of duplicating the logic again.
+1. ✅ **Remove model config params from AgentConfig** — `model`, `max_tokens`, `temperature` moved from `Config.agent` to `Config.provider`. `AgentConfig` now only holds `context_window`, `compaction_threshold`, and `enable_compaction`. All tests updated to use new format.
+
+2. ✅ **TUI complete overhaul** — `tui.py` rewritten from scratch. New layout: (1) top status bar showing `[RegCode <version>] [model] [$cost] [tokens]`, (2) large chat history area with clearly separated user/agent/tool/system message blocks, (3) bottom bar with progress indicator and input field. Streaming works correctly: text chunks stream live, then on completion full Markdown is rendered. Tool calls are interleaved after streaming text (stream buffer flushed before status rendering). Context compaction shows status indicators. All 137 tests pass, lint clean.
+
+
+## To Dos:
+- implement web crawl tool with https://github.com/unclecode/crawl4AI, the api should be straight forward. Use its inbuilt markdown generation and markdown integration.
+- If current Dir has `Agents.MD`, or `AGENTS.md` or `agents.md`, read it and provide it clearly marked as content of Agents.md to the model as extra context about the project.

regcode-0.4.1/PKG-INFO

@@ -0,0 +1,237 @@
+Metadata-Version: 2.4
+Name: regcode
+Version: 0.4.1
+Summary: Minimalistic coding agent with Python API and CLI
+Requires-Python: >=3.12
+Requires-Dist: click>=8.0
+Requires-Dist: crawl4ai>=0.4.0
+Requires-Dist: litellm>=1.83.17
+Requires-Dist: mlflow>=3.14.0
+Requires-Dist: pydantic-monty>=0.0.18
+Requires-Dist: pydantic-settings>=2.14.2
+Requires-Dist: pytest>=8.0
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: ruff>=0.8
+Requires-Dist: textual>=8.2.7
+Provides-Extra: dev
+Description-Content-Type: text/markdown
+
+# RegCode
+
+A minimalistic coding agent with Python API bindings and CLI interface.
+
+## Features
+
+- **litellm** as the LLM backbone (supports OpenAI, Anthropic, and 100+ providers)
+- **YAML configuration** for model, tokens, temperature, and provider settings via Pydantic models
+- **Environment variable expansion** in config values (e.g., `${OPENAI_API_KEY}`)
+- **CLI** via click: `chat`, `configure`, `version`
+- **Python API**: `import regcode; agent = regcode.Agent()`
+- **Host filesystem I/O**: `write_file` and `patch_file` operate directly on the host filesystem with path traversal protection
+- **Robust tool call handling**: malformed tool calls and JSON parse errors are gracefully handled
+- **Progressive streaming**: TUI always streams assistant responses progressively rather than all at once
+- **Context window compaction**: Automatically compacts conversation history when approaching token limits, preserving system instructions and review notes
+- **Review notes**: Persistent findings stored across context compaction sessions
+- **Tool budget limits**: Configurable maximum number of tool calls per conversation turn
+
+## Quick Start
+
+```bash
+# Install dependencies
+uv sync
+
+# Configure your API key (or set OPENAI_API_KEY in your environment)
+# Edit config.yaml or use environment variables
+
+# Chat with the agent (uses Textual TUI)
+uv run regcode chat
+
+# Interactive config generator
+uv run regcode configure
+
+# View version
+uv run regcode version
+```
+
+## CLI Chat Options
+
+```bash
+# Full agent mode (enables write permissions)
+uv run regcode chat --full
+
+# Use a specific model
+uv run regcode chat --model "openai/gpt-4o"
+
+# Disable colored output
+uv run regcode chat --no-color
+
+# Use a custom config file
+uv run regcode chat --config /path/to/config.yaml
+```
+
+## TUI Slash Commands
+
+While chatting, you can use these slash commands in the input bar:
+
+- `/exit` or `/quit` — Quit the application
+- `/clear` — Clear the chat history
+- `/reset` — Reset the session (clear history and tools)
+
+## Usage as a Library
+
+```python
+import regcode
+
+# Load config (reads config.yaml, falls back to ~/.regcode/config.yaml, then defaults)
+config = regcode.Config.load("config.yaml")
+
+# Create agent
+agent = regcode.Agent(config=config)
+
+# Chat (TUI always streams responses progressively)
+reply = agent.chat("Refactor this function to use async/await.")
+print(reply)
+
+# Reset conversation
+agent.reset()
+```
+
+## Tools
+
+RegCode includes several built-in tools the agent can use:
+
+| Tool | Permission | Description |
+|------|-----------|-------------|
+| `python_code` | `EXECUTE` | Execute Python code in an isolated Monty sandbox with Rust-based security and resource limits |
+| `shell_command` | `EXECUTE` | Execute shell commands (subprocess, not sandbox). Only safe, read-only commands are allowed (ls, cat, echo, head, tail, wc, grep, sort, uniq, date, pwd, whoami, id, uname, df, free, which, type, stat, file, sha256sum, md5sum, base64, xxd, od, hexdump, find) |
+| `run_script` | `EXECUTE` | Execute a Python script file in the Monty sandbox |
+| `read_file` | `READ` | Read the contents of a file from the filesystem |
+| `write_file` | `WRITE` | Write content to a file on the host filesystem |
+| `patch_file` | `WRITE` | Replace a range of lines in an existing file |
+| `list_dir` | `READ` | List contents of a directory |
+| `search_files` | `READ` | Search for files matching a pattern |
+| `search_dir` | `READ` | Search for a pattern inside all files within a directory |
+| `fetch_git_diff` | `READ` | Fetch the git diff of the current repository |
+| `osv_cve_scan` | `READ` | Scan project dependencies for known CVEs via OSV.dev API. Optionally scans the codebase for vulnerable usage patterns |
+| `system_info` | `READ` | Get system information (OS, Python version, etc.) |
+
+> **Note**: `browse_dir` is defined in the codebase but **not included in the default tools**. It can be explicitly registered if needed, but is excluded by default because it recursively lists all files in a directory, which can fill the context window in large projects (e.g., `.venv`, `node_modules`).
+
+All file-writing tools (`write_file`, `patch_file`) include path traversal protection (`..` is blocked). The `patch_file` tool validates line number bounds and type correctness before performing the replacement.
+
+## Project Structure
+
+```
+project-root/
+├── main.py              # Unused stub (hello world only)
+├── config.yaml          # Runtime configuration
+├── config.yaml.template # Template for new configs
+├── pyproject.toml       # Project metadata and dependencies
+├── uv.lock              # Locked dependencies
+└── regcode/             # Python package
+    ├── __init__.py      # Python API entry point
+    ├── main.py          # Agent core (chat, review, etc.)
+    ├── cli.py           # CLI entry point (click)
+    ├── config.py        # Config loader (yaml + pydantic)
+    ├── permissions.py    # Tool permission definitions
+    ├── tui.py           # Textual-based terminal UI
+    ├── conversation_manager.py  # Context compaction & review notes
+    ├── sandbox.py       # Legacy sandbox support
+    ├── monty_sandbox.py # Monty sandbox integration
+    └── tools/
+        ├── __init__.py
+        ├── base.py      # Base tool classes
+        ├── builtins.py  # Built-in tool implementations
+        ├── registry.py  # Tool registry management
+        └── review_notes.py  # Review notes persistence
+├── tests/
+│   ├── conftest.py
+│   ├── test_main.py
+│   ├── test_config.py
+│   ├── test_api.py
+│   ├── test_cli.py
+│   ├── test_context_manager.py
+│   ├── test_monty_sandbox.py
+│   ├── test_provider_extra.py
+│   └── test_review_notes.py
+└── README.md
+```
+
+## Testing & Linting
+
+```bash
+uv run pytest tests/ -v
+uv run ruff check --fix regcode/ tests/
+```
+
+## Configuration
+
+Edit `config.yaml` to customize the agent's behavior. The config is loaded from the specified path, `~/.regcode/config.yaml`, or falls back to defaults.
+
+```yaml
+agent:
+  context_window: 128000        # Maximum context window size
+  compaction_threshold: 0.8     # Compact when context > 80% of window
+  enable_compaction: true       # Enable automatic context compaction
+
+provider:
+  model: "openai/gpt-4o"        # Model identifier (litellm format)
+  base_url: "https://api.openai.com/v1"
+  api_key: "${OPENAI_API_KEY}"  # Environment variable expansion supported
+  temperature: 1.0
+  max_tokens: 4096
+
+system_prompt: |              # Custom system prompt (default is provided)
+  You are a coding agent.
+
+tools: {}                     # Reserved for future tool configuration
+
+sandbox:
+  use_monty: true             # Enable Monty sandbox for Python execution
+  max_duration_secs: 10.0
+  max_memory_mb: 64
+  max_recursion_depth: 100
+  type_check: true
+  allowed_imports: []
+
+tool_budget: 20               # Max tool calls per agent turn
+
+permissions:
+  - read                      # READ: read_file, list_dir, search_files, search_dir, browse_dir, fetch_git_diff, read_review_notes, system_info
+  - execute                   # EXECUTE: python_code, shell_command, run_script
+  - write                     # WRITE: write_file, patch_file, add_review_note
+  # - delete                  # DELETE: no tools currently mapped
+  # - all                     # ALL: grants all permissions
+```
+
+### Permission Levels
+
+| Permission | Tools Granted |
+|-----------|---------------|
+| `read` | `read_file`, `list_dir`, `search_files`, `search_dir`, `fetch_git_diff`, `system_info`, `osv_cve_scan` |
+| `write` | `write_file`, `patch_file` |
+| `execute` | `python_code`, `shell_command`, `run_script` |
+| `delete` | *(no tools currently)* |
+| `all` | All permissions |
+
+### Key Settings
+
+- **`context_window`**: Maximum token budget for the conversation context.
+- **`compaction_threshold`**: Ratio of context window at which compaction triggers (e.g., `0.8` = compact at 80%).
+- **`tool_budget`**: Maximum number of tool calls the agent can make in a single turn before responding.
+- **`permissions`**: List of permission levels granted to the agent. The `chat --full` CLI flag automatically adds `WRITE` permission.
+
+### Environment Variables
+
+Values in `config.yaml` can reference environment variables using `${VAR_NAME}` syntax. For example:
+
+```yaml
+provider:
+  api_key: "${OPENAI_API_KEY}"
+```
+
+Supported formats: `${VAR}`, `${VAR:-default}`, and standard shell variable expansion.
+
+## License
+
+MIT

regcode-0.4.1/README.md

@@ -0,0 +1,219 @@
+# RegCode
+
+A minimalistic coding agent with Python API bindings and CLI interface.
+
+## Features
+
+- **litellm** as the LLM backbone (supports OpenAI, Anthropic, and 100+ providers)
+- **YAML configuration** for model, tokens, temperature, and provider settings via Pydantic models
+- **Environment variable expansion** in config values (e.g., `${OPENAI_API_KEY}`)
+- **CLI** via click: `chat`, `configure`, `version`
+- **Python API**: `import regcode; agent = regcode.Agent()`
+- **Host filesystem I/O**: `write_file` and `patch_file` operate directly on the host filesystem with path traversal protection
+- **Robust tool call handling**: malformed tool calls and JSON parse errors are gracefully handled
+- **Progressive streaming**: TUI always streams assistant responses progressively rather than all at once
+- **Context window compaction**: Automatically compacts conversation history when approaching token limits, preserving system instructions and review notes
+- **Review notes**: Persistent findings stored across context compaction sessions
+- **Tool budget limits**: Configurable maximum number of tool calls per conversation turn
+
+## Quick Start
+
+```bash
+# Install dependencies
+uv sync
+
+# Configure your API key (or set OPENAI_API_KEY in your environment)
+# Edit config.yaml or use environment variables
+
+# Chat with the agent (uses Textual TUI)
+uv run regcode chat
+
+# Interactive config generator
+uv run regcode configure
+
+# View version
+uv run regcode version
+```
+
+## CLI Chat Options
+
+```bash
+# Full agent mode (enables write permissions)
+uv run regcode chat --full
+
+# Use a specific model
+uv run regcode chat --model "openai/gpt-4o"
+
+# Disable colored output
+uv run regcode chat --no-color
+
+# Use a custom config file
+uv run regcode chat --config /path/to/config.yaml
+```
+
+## TUI Slash Commands
+
+While chatting, you can use these slash commands in the input bar:
+
+- `/exit` or `/quit` — Quit the application
+- `/clear` — Clear the chat history
+- `/reset` — Reset the session (clear history and tools)
+
+## Usage as a Library
+
+```python
+import regcode
+
+# Load config (reads config.yaml, falls back to ~/.regcode/config.yaml, then defaults)
+config = regcode.Config.load("config.yaml")
+
+# Create agent
+agent = regcode.Agent(config=config)
+
+# Chat (TUI always streams responses progressively)
+reply = agent.chat("Refactor this function to use async/await.")
+print(reply)
+
+# Reset conversation
+agent.reset()
+```
+
+## Tools
+
+RegCode includes several built-in tools the agent can use:
+
+| Tool | Permission | Description |
+|------|-----------|-------------|
+| `python_code` | `EXECUTE` | Execute Python code in an isolated Monty sandbox with Rust-based security and resource limits |
+| `shell_command` | `EXECUTE` | Execute shell commands (subprocess, not sandbox). Only safe, read-only commands are allowed (ls, cat, echo, head, tail, wc, grep, sort, uniq, date, pwd, whoami, id, uname, df, free, which, type, stat, file, sha256sum, md5sum, base64, xxd, od, hexdump, find) |
+| `run_script` | `EXECUTE` | Execute a Python script file in the Monty sandbox |
+| `read_file` | `READ` | Read the contents of a file from the filesystem |
+| `write_file` | `WRITE` | Write content to a file on the host filesystem |
+| `patch_file` | `WRITE` | Replace a range of lines in an existing file |
+| `list_dir` | `READ` | List contents of a directory |
+| `search_files` | `READ` | Search for files matching a pattern |
+| `search_dir` | `READ` | Search for a pattern inside all files within a directory |
+| `fetch_git_diff` | `READ` | Fetch the git diff of the current repository |
+| `osv_cve_scan` | `READ` | Scan project dependencies for known CVEs via OSV.dev API. Optionally scans the codebase for vulnerable usage patterns |
+| `system_info` | `READ` | Get system information (OS, Python version, etc.) |
+
+> **Note**: `browse_dir` is defined in the codebase but **not included in the default tools**. It can be explicitly registered if needed, but is excluded by default because it recursively lists all files in a directory, which can fill the context window in large projects (e.g., `.venv`, `node_modules`).
+
+All file-writing tools (`write_file`, `patch_file`) include path traversal protection (`..` is blocked). The `patch_file` tool validates line number bounds and type correctness before performing the replacement.
+
+## Project Structure
+
+```
+project-root/
+├── main.py              # Unused stub (hello world only)
+├── config.yaml          # Runtime configuration
+├── config.yaml.template # Template for new configs
+├── pyproject.toml       # Project metadata and dependencies
+├── uv.lock              # Locked dependencies
+└── regcode/             # Python package
+    ├── __init__.py      # Python API entry point
+    ├── main.py          # Agent core (chat, review, etc.)
+    ├── cli.py           # CLI entry point (click)
+    ├── config.py        # Config loader (yaml + pydantic)
+    ├── permissions.py    # Tool permission definitions
+    ├── tui.py           # Textual-based terminal UI
+    ├── conversation_manager.py  # Context compaction & review notes
+    ├── sandbox.py       # Legacy sandbox support
+    ├── monty_sandbox.py # Monty sandbox integration
+    └── tools/
+        ├── __init__.py
+        ├── base.py      # Base tool classes
+        ├── builtins.py  # Built-in tool implementations
+        ├── registry.py  # Tool registry management
+        └── review_notes.py  # Review notes persistence
+├── tests/
+│   ├── conftest.py
+│   ├── test_main.py
+│   ├── test_config.py
+│   ├── test_api.py
+│   ├── test_cli.py
+│   ├── test_context_manager.py
+│   ├── test_monty_sandbox.py
+│   ├── test_provider_extra.py
+│   └── test_review_notes.py
+└── README.md
+```
+
+## Testing & Linting
+
+```bash
+uv run pytest tests/ -v
+uv run ruff check --fix regcode/ tests/
+```
+
+## Configuration
+
+Edit `config.yaml` to customize the agent's behavior. The config is loaded from the specified path, `~/.regcode/config.yaml`, or falls back to defaults.
+
+```yaml
+agent:
+  context_window: 128000        # Maximum context window size
+  compaction_threshold: 0.8     # Compact when context > 80% of window
+  enable_compaction: true       # Enable automatic context compaction
+
+provider:
+  model: "openai/gpt-4o"        # Model identifier (litellm format)
+  base_url: "https://api.openai.com/v1"
+  api_key: "${OPENAI_API_KEY}"  # Environment variable expansion supported
+  temperature: 1.0
+  max_tokens: 4096
+
+system_prompt: |              # Custom system prompt (default is provided)
+  You are a coding agent.
+
+tools: {}                     # Reserved for future tool configuration
+
+sandbox:
+  use_monty: true             # Enable Monty sandbox for Python execution
+  max_duration_secs: 10.0
+  max_memory_mb: 64
+  max_recursion_depth: 100
+  type_check: true
+  allowed_imports: []
+
+tool_budget: 20               # Max tool calls per agent turn
+
+permissions:
+  - read                      # READ: read_file, list_dir, search_files, search_dir, browse_dir, fetch_git_diff, read_review_notes, system_info
+  - execute                   # EXECUTE: python_code, shell_command, run_script
+  - write                     # WRITE: write_file, patch_file, add_review_note
+  # - delete                  # DELETE: no tools currently mapped
+  # - all                     # ALL: grants all permissions
+```
+
+### Permission Levels
+
+| Permission | Tools Granted |
+|-----------|---------------|
+| `read` | `read_file`, `list_dir`, `search_files`, `search_dir`, `fetch_git_diff`, `system_info`, `osv_cve_scan` |
+| `write` | `write_file`, `patch_file` |
+| `execute` | `python_code`, `shell_command`, `run_script` |
+| `delete` | *(no tools currently)* |
+| `all` | All permissions |
+
+### Key Settings
+
+- **`context_window`**: Maximum token budget for the conversation context.
+- **`compaction_threshold`**: Ratio of context window at which compaction triggers (e.g., `0.8` = compact at 80%).
+- **`tool_budget`**: Maximum number of tool calls the agent can make in a single turn before responding.
+- **`permissions`**: List of permission levels granted to the agent. The `chat --full` CLI flag automatically adds `WRITE` permission.
+
+### Environment Variables
+
+Values in `config.yaml` can reference environment variables using `${VAR_NAME}` syntax. For example:
+
+```yaml
+provider:
+  api_key: "${OPENAI_API_KEY}"
+```
+
+Supported formats: `${VAR}`, `${VAR:-default}`, and standard shell variable expansion.
+
+## License
+
+MIT

{regcode-0.3.0 → regcode-0.4.1}/pyproject.toml

@@ -1,10 +1,11 @@
 [project]
 name = "regcode"
-version = "0.3.0"
+version = "0.4.1"
 description = "Minimalistic coding agent with Python API and CLI"
 readme = "README.md"
 requires-python = ">=3.12"
 dependencies = [
+    "crawl4ai>=0.4.0",
     "litellm>=1.83.17",
     "pydantic-settings>=2.14.2",
     "pyyaml>=6.0",
@@ -13,15 +14,16 @@ dependencies = [
     "ruff>=0.8",
     "pydantic-monty>=0.0.18",
     "textual>=8.2.7",
+    "mlflow>=3.14.0",
 ]
 
-[build-system]
-requires = ["hatchling"]
-build-backend = "hatchling.build"
-
 [project.scripts]
 regcode = "regcode.cli:cli"
 
+[build-system]
+requires = ["hatchling", "setuptools"]
+build-backend = "hatchling.build"
+
 [project.optional-dependencies]
 dev = []
 

{regcode-0.3.0 → regcode-0.4.1}/regcode/__init__.py

@@ -2,4 +2,4 @@ from regcode.config import Config
 from regcode.main import Agent, AgentStatus, ToolCall
 
 __all__ = ["Agent", "Config", "AgentStatus", "ToolCall"]
-__version__ = "0.1.0"
+__version__ = "0.4.1"

{regcode-0.3.0 → regcode-0.4.1}/regcode/cli.py

@@ -1,3 +1,4 @@
+import logging
 import sys
 from pathlib import Path
 
@@ -5,9 +6,14 @@ import click
 import yaml
 
 import regcode
+from regcode.dependencies import ensure_playwright_browsers
 from regcode.permissions import AgentPermission
 from regcode.tui import ChatUI
 
+# Configure logging
+logging.basicConfig(level=logging.INFO, format="%(levelname)s: %(message)s")
+_logger = logging.getLogger(__name__)
+
 
 def _confirm(prompt_text):
     """Ask yes/no question, accepting y/n (case insensitive)."""
@@ -43,6 +49,14 @@ def cli(ctx):
 )
 def chat(model, config_path, stream, full, no_color):
     """Chat with the agent using a Textual TUI with sticky bottom bar."""
+    # Ensure Playwright browsers are available for web_crawl tool
+    _logger.info("Checking startup dependencies...")
+    if not ensure_playwright_browsers():
+        _logger.warning(
+            "Playwright browsers are not available. "
+            "The web_crawl tool will not function."
+        )
+
     # Pass None if the file doesn't exist so Config.load falls back
     # to ~/.regcode/config.yaml (matching the TUI's expected behaviour).
     _effective_path = config_path if Path(config_path).exists() else None
@@ -61,6 +75,7 @@ def chat(model, config_path, stream, full, no_color):
         status_callback=ui.handle_status,
         result_callback=ui.handle_result,
         context_callback=ui.set_context_progress,
+        cost_callback=ui.handle_cost,
     )
 
     # Attach agent to the Textual app

{regcode-0.3.0 → regcode-0.4.1}/regcode/config.py

@@ -44,7 +44,15 @@ default_system_prompt = dedent(
 logger = logging.getLogger(__name__)
 
 class AgentConfig(BaseModel):
-    model_config = ConfigDict(extra="allow")
+    """Configuration specific to the agent's behavior (not provider params).
+
+    Model, temperature, max_tokens belong in ProviderConfig where
+    litellm.completion expects them. AgentConfig only holds compaction
+    and context-window tuning — passing model params through this object
+    caused conflicts with different providers.
+    """
+
+    model_config = ConfigDict(extra="forbid")
     context_window: int = 128000
     compaction_threshold: float = 0.8  # Compact when context > 80% of window
     enable_compaction: bool = True
@@ -81,11 +89,30 @@ class SandboxConfig(BaseModel):
     type_check: bool = True
     allowed_imports: list[str] = []
 
+class TracingConfig(BaseModel):
+    """Configuration for tracing and logging."""
+
+    model_config = ConfigDict(extra="allow")
+    enable_tracing: bool = False
+    mlflow_url: str | None = None
+    mlflow_experiment_name: str | None = None
+    mlflow_async: bool = True
+    mlflow_requires_auth: bool = True
+    mlflow_username: str | None = None
+    mlflow_password: str | None = None
+
+    def tracing_enabled(self) -> bool:
+        """Check if tracing is fully configured and enabled."""
+        return (
+            self.enable_tracing
+            and self.mlflow_url is not None
+        )
 
 class Config(BaseModel):
     model_config = ConfigDict(extra="allow")
 
     agent: AgentConfig = AgentConfig()
+    tracing: TracingConfig = TracingConfig()
     provider: ProviderConfig = ProviderConfig()
     system_prompt: str = default_system_prompt
     tools: dict = {}
@@ -96,19 +123,29 @@ class Config(BaseModel):
     @classmethod
     def load(cls, path: str | None = None) -> "Config":
         # get default home dir for user first
+        _path = None
         if path is None:
-            logger.warning(
-                "No config path provided, looking for ~/.regcode/config.yaml"
-            )
-            alt_config_path = Path.home() / ".regcode" / "config.yaml"
-            if alt_config_path.exists():
-                logger.warning(f"Loading from config at {alt_config_path}")
-                _path = alt_config_path
-            else:
-                _path = Path("config.yaml")
+            possible_paths = [
+                Path("./config.yaml"), Path.home() / ".regcode" / "config.yaml"
+            ]
+            for p in possible_paths:
+                if p.exists():
+                    logger.info(f"Loading from config at {p}")
+                    _path = p
+                    break
+            if _path is None:
+                logger.warning(
+                    "No config file found, using default config, "
+                    "Use `regcode configure` to configure."
+                    )
+                return cls()
         else:
             _path = Path(path)
         if not _path.exists():
+            logger.warning(
+                f"Config file {path} not found, using default config, "
+                "Use `regcode configure` to configure."
+            )
             return cls()
         with open(_path) as f:
             data = yaml.safe_load(f)
@@ -135,17 +172,6 @@ class Config(BaseModel):
         result = self.provider.model_dump()
         if extra_kwargs:
             result.update(extra_kwargs)
-        # Agent-level config overrides provider defaults for known keys
-        if self.agent.context_window is not None:
-            result["context_window"] = self.agent.context_window
-        if hasattr(self.agent, "compaction_threshold"):
-            result["compaction_threshold"] = self.agent.compaction_threshold
-        if hasattr(self.agent, "enable_compaction"):
-            result["enable_compaction"] = self.agent.enable_compaction
-        # Apply agent-level extra fields (e.g. model, max_tokens, temperature
-        # from YAML agent section) as overrides over provider defaults
-        if hasattr(self.agent, "model_extra") and self.agent.model_extra:
-            result.update(self.agent.model_extra)
         return result