🩹 fix: I think I finally found the pyproject.toml file. Hopefully.

Author: Falamarcao

.pre-commit-config.yaml

@@ -1,12 +1,12 @@
 repos:
   - repo: https://github.com/astral-sh/uv-pre-commit
-    rev: 0.8.5
+    rev: 0.8.8
     hooks:
       - id: uv-lock
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.12.7
+    rev: v0.12.8
     hooks:
-      - id: ruff
+      - id: ruff-check
         args: [--fix]
       - id: ruff-format
         args: ["--line-length", "100", "--target-version", "py312"]

pyproject.toml

@@ -13,7 +13,7 @@
 
 [project]
 name = "mcp-selenium-grid"
-version = "0.1.0.dev3"
+version = "0.1.0.dev5"
 description = "MCP Server for managing Selenium Grid"
 readme = "README.md"
 license = { file = "LICENSE" }

src/app/common/toml.py

@@ -1,43 +1,55 @@
-from os import getcwd
+from contextlib import contextmanager
+from importlib.resources import as_file, files
 from pathlib import Path
 from tomllib import load
-from typing import Any
+from typing import Any, Iterator
 
-ROOT_DIR = Path(getcwd()).resolve()
 
-
-def load_value_from_toml(
-    keys: list[str], file_path: Path = ROOT_DIR / "pyproject.toml", default: Any = None
-) -> Any:
+@contextmanager
+def find_pyproject_toml() -> Iterator[Path]:
     """
-    Load a nested value from a TOML file.
+    Finds the pyproject.toml file and yields its Path.
+    Uses a context manager to handle temporary files created by as_file().
+    """
+    try:
+        # Get the Traversable object for the pyproject.toml file.
+        resource_path = files(__package__).joinpath("pyproject.toml")
+        if resource_path.is_file():
+            # Use as_file() to get a Path object to the resource.
+            # This is done within a context manager for proper cleanup.
+            with as_file(resource_path) as file_path:
+                yield file_path
+            return
+    except (ImportError, TypeError, AttributeError):
+        # Fallback for non-packaged or single-script scenarios.
+        pass
 
-    Args:
-        keys: List of nested keys to traverse.
-        file_path: Path to the TOML file.
-        default: Value to return if keys not found.
+    # Fallback logic
+    current_dir = Path(__file__).resolve()
+    for parent in current_dir.parents:
+        potential_path = parent / "pyproject.toml"
+        if potential_path.is_file():
+            yield potential_path
+            return
 
-    Returns:
-        The value from the TOML file.
+    raise FileNotFoundError("Could not find pyproject.toml")
 
-    Raises:
-        FileNotFoundError: If the file doesn't exist and no default is provided.
-        ValueError: If the keys are missing and no default is provided.
-    """
-    if not file_path.exists():
-        if default is not None:
-            return default
-        raise FileNotFoundError(f"{file_path} not found")
 
-    try:
+def load_value_from_toml(keys: list[str], default: Any = None) -> Any:
+    """
+    Load a nested value from a TOML file.
+    """
+    with find_pyproject_toml() as file_path:
         with file_path.open("rb") as f:
             data = load(f)
-        for key in keys:
-            data = data[key]
-        return data
-    except KeyError:
-        if default is not None:
-            return default
-        raise ValueError(f"Keys {'.'.join(keys)} not found in {file_path}")
-    except Exception as e:
-        raise ValueError(f"Error reading {file_path}: {e}") from e
+
+        try:
+            for key in keys:
+                data = data[key]
+            return data
+        except KeyError:
+            if default is not None:
+                return default
+            raise ValueError(f"Keys {'.'.join(keys)} not found in {file_path}")
+        except Exception as e:
+            raise ValueError(f"Error reading {file_path}: {e}") from e