iSrSSKJr SSKrSSKJr SSKJr SSK J r J r SSK Jr SSKJrJr SS KJrJr S r"S S \5r"S S\ 5r"SS5rg!\a SSK Jr NOf=f)zN Directory-based prompt registry implementation that stores prompts as files. ) annotationsN)Path)Self) BaseModelField)Prompt)InvalidPromptErrorPromptNotFoundError)DEFAULT_VERSION PromptModelz index.jsoncF\rSrSr%Sr\"SSS9rS\S'\S Sj5r S r g) PromptFilez0Model representing a prompt file stored on disk.NT)defaultexcludez Path | NonepathcX!RSURS3- nURUR5 U"URURURURUS9$)z 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 T/home/james-whalen/.local/lib/python3.13/site-packages/banks/registries/directory.pyfrom_prompt_pathPromptFile.from_prompt_path s` }Afnn-=VDD vzz*&++v~~PVP_P_fq  )rz type[Self]rrrrreturnr) __name__ __module__ __qualname____firstlineno____doc__rr__annotations__ classmethodr __static_attributes__r#r"rrrs(:dD9D+9  r"rc0\rSrSr%Sr\"/S9rS\S'Srg)PromptFileIndex3z1Index tracking all prompt files in the directory.)rzlist[PromptFile]filesr#N) r%r&r'r(r)rr0r*r,r#r"rr.r.3s;#B/E /r"r.c~\rSrSrSrSS.SSjjr\SSj5rSS.SS jjrSS .SS jjr S r S r Sr SSjr Srg)DirectoryPromptRegistry9z?Registry that stores prompts as files in a directory structure.F) force_reindexc([U5nUR5(d Sn[U5eX0lUR[- UlUR R 5(aU(aUR5 gUR5 g)z 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_pathr4dir_pathmsgs r__init__ DirectoryPromptRegistry.__init__<sj'  9CS/ ! ::(::&&((M JJL JJLr"cUR$)z0Get the directory path where prompts are stored.)r8r>s rrDirectoryPromptRegistry.pathSszzr"N)rcHU=(d [nURRHtnURU:XdMURU:XdM'UR (dM:UR R 5(dM[[S0UR5D6s $ [e)z 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 _indexr0rrrr;r model_dumpr )r>rrpfs rgetDirectoryPromptRegistry.getXsl,_++##Bww$2::#8RWWWIYIY0 00$"!r") overwritecUR=(d [nURURUS9upEU(ai[R "5UR S'[RXR5URRU'UR5 gSURS3n[U5e![ay [R "5UR S'[RXR5nURRRU5 UR5 gf=f)z 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_filertimectimerrr r8rHr0_saver r append)r>rrMridxrJrAs rsetDirectoryPromptRegistry.setls nn7G++g+NGC04  -)3)D)DVZZ)X !!#& *6;;-7hi(--" ,0JJLFOOL ),,VZZ@B KK   $ $R ( JJL  sB$C'CBEEc[RURR55UlURR H@nUR bMURURSURS3- UlMB g)z Load the prompt index from disk.Nrr) r.model_validate_jsonr: read_textrHr0rr8rr)r>rJs rr=DirectoryPromptRegistry._loadse%99$:J:J:T:T:VW ++##Bww**"''!BJJrrrfrJs rr<DirectoryPromptRegistry._scans'6'8 JJOOJ/D8;tyy8HDII,,S!4tyyZiNjMD31QVVXDPT_ab !!((, 0 ##DKK$?$?$AB s ?AC?? D c[URR5H,up4URU:XdMURU:XdM)X44s $ SUSUS3n[ U5e)z 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 '') enumeraterHr0rrr )r>rrirJrAs rrP(DirectoryPromptRegistry._get_prompt_files]t{{001EAww$2::#8u 2/tfOG9AN!#&&r")rHr:r8)r?strr4bool)r$r)rrnr str | Noner$r)rrrMro)rrprrnr$ztuple[int, PromptFile])r%r&r'r(r)rBpropertyrrKrVr=rSr<rPr,r#r"rr2r29sLIEJ.7;"(8=6GCC'r"r2)r) __future__rrQpathlibrtypingr ImportErrortyping_extensionspydanticrrbanksr banks.errorsr r banks.promptr r r9rr.r2r#r"rr{sl# '&@5"  00i0 w'w'U'&'sA A*)A*