JidZddlmZddlZddlmZ ddlmZddl m Z m Z ddl mZddlmZmZdd lmZmZd ZGd d eZGd de ZGddZy#e$r ddl mZYNwxYw)zN Directory-based prompt registry implementation that stores prompts as files. ) annotationsN)Path)Self) BaseModelField)Prompt)InvalidPromptErrorPromptNotFoundError)DEFAULT_VERSION PromptModelz index.jsoncBeZdZUdZeddZded<eddZy) PromptFilez0Model representing a prompt file stored on disk.NT)defaultexcludez Path | Nonepathc||jd|jdz }|j|j||j|j|j|j|S)a Create a PromptFile instance from a Prompt object and save it to disk. Args: prompt: The Prompt object to save path: Directory path where the prompt file should be stored Returns: A new PromptFile instance ..jinja)textnameversionmetadatar)rr write_textrawr)clspromptr prompt_files l/var/www/html/BankruptcyAI-uat/bankruptcy-ai/venv/lib/python3.12/site-packages/banks/registries/directory.pyfrom_prompt_pathzPromptFile.from_prompt_path sb }Afnn-=VDD vzz*&++v~~PVP_P_fq  )rz type[Self]rrrrreturnr) __name__ __module__ __qualname____doc__rr__annotations__ classmethodrr rrrs(:dD9D+9  r rc.eZdZUdZegZded<y)PromptFileIndexz1Index tracking all prompt files in the directory.)rzlist[PromptFile]filesN)r"r#r$r%rr+r&r(r rr*r*3s;#B/E /r r*cfeZdZdZddddZeddZddddZdd dd Zd Z d Z d Z ddZ y)DirectoryPromptRegistryz?Registry that stores prompts as files in a directory structure.F) force_reindexc t|}|js d}t|||_|jtz |_|j j r|r|jy|jy)a. Initialize the directory prompt registry. Args: directory_path: Path to directory where prompts will be stored force_reindex: Whether to force rebuilding the index from disk Raises: ValueError: If directory_path is not a directory z%{directory_path} must be a directory.N) ris_dir ValueError_pathDEFAULT_INDEX_NAME _index_pathexists_scan_load)selfdirectory_pathr.dir_pathmsgs r__init__z DirectoryPromptRegistry.__init__<sf' 9CS/ ! ::(::&&(M JJL JJLr c|jS)z0Get the directory path where prompts are stored.)r2r8s rrzDirectoryPromptRegistry.pathSszzr N)rc|xst}|jjD]d}|j|k(s|j|k(s#|j s0|j j sKtdi|jcSt)a2 Retrieve a prompt by name and version. Args: name: Name of the prompt to retrieve version: Version of the prompt (optional) Returns: The requested Prompt object Raises: PromptNotFoundError: If prompt doesn't exist r() r _indexr+rrrr5r model_dumpr )r8rrpfs rgetzDirectoryPromptRegistry.getXsl,_++## 1Bww$2::#8RWWIY0 00 1"!r ) overwritec |jxst}|j|j|\}}|rit j |j d<tj||j|jj|<|jyd|jd}t|#t$ryt j |j d<tj||j}|jjj||jYywxYw)a Store a prompt in the registry. Args: prompt: The Prompt object to store overwrite: Whether to overwrite existing prompt Raises: InvalidPromptError: If prompt exists and overwrite=False )rr created_atzPrompt with name 'z1' already exists. Use overwrite=True to overwriteN)rr _get_prompt_filertimectimerrrr2r@r+_saver r append)r8rrDridxrBr;s rsetzDirectoryPromptRegistry.setls nn7G++g+NGC04  -)3)D)DVTZZ)X !!#& *6;;-7hi(--" ,0JJLFOOL ),,VTZZ@B KK   $ $R ( JJL  sBB<!B<=D>ctj|jj|_|jj D]=}|j |j|jd|jdz |_?y)z Load the prompt index from disk.Nrr) r*model_validate_jsonr4 read_textr@r+rr2rr)r8rBs rr7zDirectoryPromptRegistry._loadsl%99$:J:J:T:T:VW ++## GBww**"''!BJJs rrJzDirectoryPromptRegistry._saves# ##DKK$?$?$ABr c t|_|jjdD]}d|jvr|jj ddn|jt f\}}|jd5}t|j|||i}|jjj|ddd|jj|jjy#1swYxYw)z4Scan directory for prompt files and build the index.z*.jinja*rr)rrrrrN)r*r@r2globstemrsplitr openrreadr+rKr4rrR)r8rrrfrBs rr6zDirectoryPromptRegistry._scans'6'8 JJOOJ/ -D8;tyy8HDII,,S!4tyyZiNjMD'3 -1QVVXD'PT_ab !!((, - - - ##DKK$?$?$AB - -s >ADD ct|jjD]*\}}|j|k(s|j|k(s&||fcSd|d|d}t |)a( Find a prompt file in the index. Args: name: Name of the prompt version: Version of the prompt Returns: Tuple of (index position, PromptFile) Raises: PromptNotFoundError: If prompt doesn't exist in index zcannot find prompt with name 'z' and version '') enumerater@r+rrr )r8rrirBr;s rrGz(DirectoryPromptRegistry._get_prompt_filesft{{001 EArww$2::#8"u  /tfOG9AN!#&&r )r9strr.bool)r!r)rr`r str | Noner!r)rrrDra)rrbrr`r!ztuple[int, PromptFile]) r"r#r$r%r<propertyrrCrMr7rJr6rGr(r rr-r-9sLIEJ.7;"(8=6GCC'r r-)r% __future__rrHpathlibrtypingr ImportErrortyping_extensionspydanticrrbanksr banks.errorsr r banks.promptr r r3rr*r-r(r rrmsl# '&@5"  00i0 w'w'U'&'sA A)(A)