mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-18 06:24:20 +01:00
44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
from pathlib import Path
|
|
|
|
# Set a dedicated folder for file I/O
|
|
WORKSPACE_PATH = Path(os.getcwd()) / "auto_gpt_workspace"
|
|
|
|
# Create the directory if it doesn't exist
|
|
if not os.path.exists(WORKSPACE_PATH):
|
|
os.makedirs(WORKSPACE_PATH)
|
|
|
|
|
|
def path_in_workspace(relative_path: str | Path) -> Path:
|
|
"""Get full path for item in workspace
|
|
|
|
Parameters:
|
|
relative_path (str | Path): Path to translate into the workspace
|
|
|
|
Returns:
|
|
Path: Absolute path for the given path in the workspace
|
|
"""
|
|
return safe_path_join(WORKSPACE_PATH, relative_path)
|
|
|
|
|
|
def safe_path_join(base: Path, *paths: str | Path) -> Path:
|
|
"""Join one or more path components, asserting the resulting path is within the workspace.
|
|
|
|
Args:
|
|
base (Path): The base path
|
|
*paths (str): The paths to join to the base path
|
|
|
|
Returns:
|
|
Path: The joined path
|
|
"""
|
|
joined_path = base.joinpath(*paths).resolve()
|
|
|
|
if not str(joined_path.absolute()).startswith(str(base.absolute())):
|
|
raise ValueError(
|
|
f"Attempted to access path '{joined_path}' outside of working directory '{base}'."
|
|
)
|
|
|
|
return joined_path
|