ivSrSSKJr SSKJr SSKJr SSKJrJ r SSK J r J r "SS\5r "S S 5rg ) a# File-based prompt registry implementation that stores all prompts in a single JSON file. This module provides functionality to store and retrieve prompts using a single JSON file as the storage backend. The file contains an index of all prompts with their associated metadata and content. ) annotations)Path) BaseModel)InvalidPromptErrorPromptNotFoundError)Prompt PromptModelc(\rSrSr%Sr/rS\S'Srg)PromptRegistryIndexz Model representing the registry index containing all prompts. Stores a list of PromptModel objects that represent all prompts in the registry. zlist[PromptModel]promptsN)__name__ __module__ __qualname____firstlineno____doc__r __annotations____static_attributes__rO/home/james-whalen/.local/lib/python3.13/site-packages/banks/registries/file.pyr r s "$G #rr cZ\rSrSrSrS SjrSS.SSjjrSS.SS jjrSS jrSS jr S r g)FilePromptRegistry z@A prompt registry storing all prompt data in a single JSON file.c [U5Ul[/S9Ul[R URR SS95Ulg![ a' URRRSSS9 gf=f)z Initialize the file prompt registry. Args: registry_index: Path to the JSON file that will store the prompts Note: Creates parent directories if they don't exist. )r utf-8encodingT)parentsexist_okN) r _index_fpathr _indexmodel_validate_json read_textFileNotFoundErrorparentmkdir)selfregistry_indexs r__init__FilePromptRegistry.__init__#sz#'~"6+>r+J  H-AA$BSBSB]B]gnB]BopDK  H    $ $ * *4$ * G Hs1A.BBN)versioncZURX5up4[S0UR5D6$)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 the requested prompt doesn't exist r)_get_prompt_modelr model_dump)r(namer,_models rgetFilePromptRegistry.get5s-))$8+((*++rF) overwritecURURUR5up4U(a>[R"U5UR R U'UR5 gSURS3n[U5e![aN [R"U5nUR R RU5 UR5 gf=f)z Store a prompt in the registry. Args: prompt: The Prompt object to store overwrite: Whether to overwrite an existing prompt Raises: InvalidPromptError: If prompt exists and overwrite=False zPrompt with name 'z1' already exists. Use overwrite=True to overwriteN) r.r0r,r from_promptr"r _saverrappend)r(promptr5idxp_modelmsgs rsetFilePromptRegistry.setFs 11&++v~~NLC+6+B+B6+J ##C( *6;;-7hi(--" !--f5G KK   & &w / JJL sA,B /B AC"!C"c[URSSS9nURURR 55 SSS5 g!,(df  g=f)z\ Save the prompt index to the JSON file. Writes the current state of the registry to disk. wrrN)openr!writer"model_dump_json)r(fs rr8FilePromptRegistry._save^s= $##S7 ;q GGDKK//1 2< ; ;s *A  Ac[URR5H,up4URU:XdMURU:XdM)X44s $ SUSUS3n[ U5e)z Find a prompt model in the index by name and version. Args: name: Name of the prompt version: Version of the prompt Returns: Tuple of (index position, PromptModel) Raises: PromptNotFoundError: If the prompt doesn't exist in the index zcannot find prompt with name 'z' and version '') enumerater"r r0r,r)r(r0r,ir2r=s rr.$FilePromptRegistry._get_prompt_modelgs^"$++"5"56HAzzT!emmw&>x7/tfOG9AN!#&&r)r"r!)r)strreturnNone)r0rLr, str | NonerMr)r:rr5boolrMrN)rMrN)r0rOr,rOrMztuple[int, PromptModel]) rrrrrr*r3r>r8r.rrrrrr s)JH$7;,"8=03'rrN)r __future__rpathlibrpydanticr banks.errorsrr banks.promptrr r rrrrrVs3#@,$)$Z'Z'r