Fd% dZddlmZddlZddlZddlZddlmZmZddl m Z ddl m Z ddl mZddlmZddlm Z dd lmZdd lmZmZed Zd@dZdAdZdBdZ dCdDdZ dCdEd!Zed"d#dd$d%d&d'idFd(ZdGd+Zed,d-d$d.d&d'd$d/d&d'd0dHd1Zed2d3d$d.d&d'd$d/d&d'd0 dIdJd5Z ed6d7dd$d8d&d'idFd9Z!ed:d;d#t $r1tjd |d|d}}YnwxYw|||fV|dkr||dfV| dS)zKParse the file operations log and return a tuple containing the log entriesrrencodingNzFile Operation Logger: )maxsplitr rc3>K|]}|VdSN)strip).0xs r z&operations_from_log..+s*!S!S!''))!S!S!S!S!S!Sr #z File log entry lacks checksum: ''r) openFileNotFoundErrorreplacer)splitrsplit ValueErrorr warnclose)rlogline operationtailpathchecksums roperations_from_logr=s~8S7333 22||3R88>>@@  **TA*66 4OO%% + + + 4!S!ST[[PQ[5R5R!S!S!Shh 4 4 4 FtFFFGGG!%th 4dH- - - - - ( " "djjllD1 1 1 1IIKKKKKs %% $B//8C*)C*dict[str, str]c\i}t|D]\}}}|dvr|||<|dkr||=|S)aIterates over the operations log and returns the expected state. Parses a log file at config.file_logger_path to construct a dictionary that maps each file path written or appended to its checksum. Deleted files are removed from the dictionary. Returns: A dictionary mapping file paths to their checksums. Raises: FileNotFoundError: If config.file_logger_path is not found. ValueError: If the log file content is not in the expected format. r&r)r=)rstater9r;r<s rfile_operations_staterA6sV E%8%B%B! 4 + + +"E$KK ( " "d Lrr9 Operationfilenameconfigrr< str | Noneboolct|j}|dkr||vrdS|dkr|||krdSdS)aTCheck if the operation has already been performed Args: operation: The operation to check for filename: The name of the file to check for config: The agent config checksum: The checksum of the contents to be written Returns: True if the operation has already been performed on the file rTr F)rAfile_logger_pathget)r9rCrDr<r@s ris_duplicate_operationrJMsY "&"9 : :EH!6!6tG ( 3 3x ? ?t 5ragentrNonec|d|}||d|z }tjd|t|jj|d|ddS)zLog the file operation to the file_logger.txt Args: operation: The operation to log filename: The name of the file the operation was performed on checksum: The checksum of the contents to be written r#Nr-zLogging file operation:  F) should_log)r debugappend_to_filerDrH)r9rCrKr< log_entrys r log_operationrScs~****I_(__$  L7I77888 %)'7'7'75r read_filezRead an existing filestringzThe path of the file to readT)type descriptionrequiredc t|t}tj|||j}t |jdkr|jS|S#t$r}dt|cYd}~Sd}~wwxYw)zRead a file and return the contents Args: filename (str): The name of the file to read Returns: str: The contents of the file r$Error: N) r r r from_text_filerDlenchunkssummary Exceptionr)rCrKcontent file_memoryes rrTrTvs( "#Hf55!/5<PP {! " "Q & && & """!Q!!!!!!!!"s$AAA A9A4.A94A9memoryr c tjd|t|}tj||}tjd|d||tjdt|j d|dS#t$r%}tj d|d|Yd}~dSd}~wwxYw) a/ Ingest a file by reading its content, splitting it into chunks with a specified maximum length and overlap, and adding the chunks to the memory storage. Args: filename: The name of the file to ingest memory: An object with an add() method to store the chunks in memory zIngesting file zCreated memory: Tz Ingested z chunks from zError while ingesting file 'z': N) r inforTr r[rPdumpaddr\e_chunksr_r5)rCrcr`raerrs r ingest_filerjs G 0h00111H%%!/BB  @ (8(8(>(>@@AAA ; RK$8 9 9RRRRSSSSS GGG E8EEEEFFFFFFFFFGsB&B** C4CC write_to_filezWrites to a filez The name of the file to write tozThe text to write to the file)rCrct|}td||j|rdS tj|}tj|dt|dd5}||dddn #1swxYwYtd|||d S#t$r}d |cYd}~Sd}~wwxYw) zWrite text to a file Args: filename (str): The name of the file to write to text (str): The text to write to the file Returns: str: A message indicating success or failure r z%Error: File has already been updated.Texist_okwrr NzFile written to successfully.rZ) rrJrDosr;dirnamemakedirsr/r rSr_)rCrrKr< directoryfris rrkrks&4T""HgxxHH766GOOH--  I---- (C' 2 2 2 a GGDMMM               gx999.. sBAB21B B2BB2BB22 C <CC C rQzAppends to a filerOc tj|}tj|dt |dd5}||dddn #1swxYwY|r^t |dd5}t |}dddn #1swxYwYtd||| d S#t$r}d |cYd}~Sd}~wwxYw) aAppend text to a file Args: filename (str): The name of the file to append to text (str): The text to append to the file should_log (bool): Should log output Returns: str: A message indicating success or failure Trmarr Nrr)r<zText appended successfully.rZ) rpr;rqrrr/r rreadrSr_)rCrrKrOrsrtr<ris rrQrQs: GOOH--  I---- (C' 2 2 2 a GGDMMM                 Hhg666 3!(22 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 (Heh G G G G,, sfAC A+ C+A//C2A/3C "B8, C8B<<C?B<C C1"C,&C1,C1 delete_filezDeletes a filezThe name of the file to deletectd||jrdS tj|t d||dS#t $r}d|cYd}~Sd}~wwxYw)zDelete a file Args: filename (str): The name of the file to delete Returns: str: A message indicating success or failure rz%Error: File has already been deleted.zFile deleted successfully.rZN)rJrDrpremoverSr_)rCrKris rrxrxs(h%,??766 (h%000++ s%A A AAA list_fileszLists Files in a DirectoryrszThe directory to list files in list[str]c0g}tj|D]~\}}}|D]u}|drtjtj|||jj}||v|S)zlists files in a directory recursively Args: directory (str): The directory to search in Returns: list[str]: A list of files found in the directory .) rpwalk startswithr;relpathjoinrDworkspace_pathr)rsrK found_filesroot_filesfile relative_paths rr{r{$s(K'),,..a . .Ds## GOO T4((%,*EM   } - - - -  . r)rrrr)rrrr)rrrr>r() r9rBrCrrDrr<rErrF) r9rrCrrKrr<rErrL)rCrrKrrr)rCrrcr rrL)rCrrrrKrrr)T) rCrrrrKrrOrFrr)rsrrKrrr|)#__doc__ __future__rrrpos.pathtypingrr confectionrautogpt.agent.agentrautogpt.command_decoratorr&autogpt.commands.file_operations_utilsr autogpt.config autogpt.logsr autogpt.memory.vectorr r rBrr=rArJrSrTrjrkrQrxr{rrrsK!!"""""" %%%%%%%%%%%%%%------DDDDDD!!!!!!:::::::: / 0 9999 80QU.IM& 9    """  ",GGGG4 =  :    ! 0 =  :   "@D! : ;      &  ;      r