FdTdZddlmZddlmZddlmZddlmZGddZ dS) z ========= Workspace ========= The workspace is a directory containing configuration and working files for an AutoGPT agent. ) annotations)Path)Config)loggerceZdZdZgdZddZedd Zed Ze dd Z ddZ e dddZ e dZe d d!dZdS)" Workspacez9A class that represents a workspace for an AutoGPT agent.)r r z\zr z%00workspace_root str | Pathrestrict_to_workspaceboolcH|||_||_dSN)_sanitize_path_root_restrict_to_workspace)selfr r s OC:\Users\Administrator\Downloads\Auto-GPT-master\autogpt\workspace\workspace.py__init__zWorkspace.__init__s%((88 &;###returnrc|jS)z$The root directory of the workspace.)rrs rrootzWorkspace.roots zrc|jS)z5Whether to restrict generated paths to the workspace.)rrs rr zWorkspace.restrict_to_workspace s **rworkspace_directoryc^||}|dd|S)aCreate a workspace directory and return the path to it. Parameters ---------- workspace_directory The path to the workspace directory. Returns ------- Path The path to the workspace directory. T)exist_okparents)rmkdir)clsrargskwargss rmake_workspacezWorkspace.make_workspace%s8 "001DEE!!4!>>>""r relative_pathcF|||j|jS)aGet the full path for an item in the workspace. Parameters ---------- relative_path The relative path to resolve in the workspace. Returns ------- Path The resolved path relative to the workspace. )rrestrict_to_root)rrr )rr%s rget_pathzWorkspace.get_path9s/"" !7#   rNTrr'ctjD]3}|t|vs|t|vrtd4|!t |St jd|d|dt |t |}}t jd|d|r+| |std|d|d| |}t jd |d|r+| |std |d |d|S) a^Resolve the relative path within the given root if possible. Parameters ---------- relative_path The relative path to resolve. root The root path to resolve the relative path within. restrict_to_root Whether to restrict the path to the root. Returns ------- Path The resolved path. Raises ------ ValueError If the path is absolute and a root is provided. ValueError If the path is outside the root and the root is restricted. zembedded null byteNzResolving path 'z' in workspace ''zResolved root as 'z#Attempted to access absolute path 'z'.zJoined paths as 'zAttempted to access path 'z' outside of workspace ') r NULL_BYTESstr ValueErrorrresolverdebug is_absoluteis_relative_tojoinpath)r%rr' null_byte full_paths rrzWorkspace._sanitize_pathMsD#- 7 7IC ....)s4yy2H2H !56663I < &&..00 0 N NNtNNNOOO"4jj0022D4G4Gm 1$111222  $ $ & & }/K/KD/Q/Q ]m]]UY]]] MM-0088::  5555666  I$<$rRs#"""""!!!!!!M#M#M#M#M#M#M#M#M#M#r