>'hdZddlmZddlZddlmZ 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.jsonc@eZdZUdZedZded<eddZy) PromptFilez0Model representing a prompt file stored on disk.T)excluderpathc||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 .z.jinja)textnameversionmetadatar)rr write_textrawr)clspromptr prompt_files a/home/kushmeetdev/Regenta/Chatbot/venv/lib/python3.12/site-packages/banks/registries/directory.pyfrom_prompt_pathzPromptFile.from_prompt_path sb }Afnn-=VDD vzz*&++v~~PVP_P_fq  N)rz type[Self]rrrrreturnr) __name__ __module__ __qualname____doc__rr__annotations__ classmethodrrrrrs&:t$D$$  rrc.eZdZUdZegZded<y)PromptFileIndexz1Index tracking all prompt files in the directory.)defaultzlist[PromptFile]filesN)r r!r"r#rr*r$r&rrr(r(3s;#B/E /rr(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 JJLrc|jS)z0Get the directory path where prompts are stored.)r1r7s rrzDirectoryPromptRegistry.pathSszzrN)rc|xst}|jjD]W}|j|k(s|j|k(s#|j j s>tdi|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*rrrr4r model_dumpr )r7rrpfs rgetzDirectoryPromptRegistry.getXsf,_++## 1Bww$2::#8RWW^^=M0 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_filertimectimerrrr1r?r*_saver r append)r7rrCridxrAr:s rsetzDirectoryPromptRegistry.setls nn7G++g+NGC04  -)3)D)DVTZZ)X !!#& *6;;-7hi(--" ,0JJLFOOL ),,VTZZ@B KK   $ $R ( JJL  sBB<!B<=D>chtj|jj|_y)z Load the prompt index from disk.N)r(model_validate_jsonr3 read_textr?r=s rr6zDirectoryPromptRegistry._loads#%99$:J:J:T:T:VW rcj|jj|jjy)zSave the prompt index to disk.N)r3rr?model_dump_jsonr=s rrIzDirectoryPromptRegistry._saves# ##DKK$?$?$ABrc 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?r1globstemrsplitr openrreadr*rJr3rrQ)r7rrrfrAs rr5zDirectoryPromptRegistry._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 )r7rrirAr:s rrFz(DirectoryPromptRegistry._get_prompt_filesft{{001 EArww$2::#8"u  /tfOG9AN!#&&r)r8strr-bool)rr)rr_r str | Nonerr)rrrCr`)rrarr_rztuple[int, PromptFile]) r r!r"r#r;propertyrrBrLr6rIr5rFr&rrr,r,9sLIEJ.7;"(8=6XCC'rr,)r# __future__rrGpathlibrtypingr ImportErrortyping_extensionspydanticrrbanksr banks.errorsr r banks.promptr r r2rr(r,r&rrrlsl# '&@5"  00i0 s's'U'&'sA A)(A)