>'hvdZddlmZddlmZddlmZddlmZm Z ddl m Z m Z GddeZ Gd d Zy ) 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 eZdZUdZgZded<y)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____doc__r __annotations__\/home/kushmeetdev/Regenta/Chatbot/venv/lib/python3.12/site-packages/banks/registries/file.pyr r s "$G #rr cDeZdZdZd dZddd dZddd dZdd Zdd Zy)FilePromptRegistryz@A prompt registry storing all prompt data in a single JSON file.ct||_tg|_ tj |jj d|_y#t $r*|jjjddYywxYw)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__zFilePromptRegistry.__init__#sz#'~"6+>r+J  H-AA$BSBSB]B]gnB]BopDK  H    $ $ * *4$ * G Hs4A0B  B N)versionc^|j||\}}tdi|jS)a@ 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 rgetzFilePromptRegistry.get5s1))$85+((*++rF) overwritec |j|j|j\}}|r=tj||j j |<|jyd|jd}t|#t$rMtj|}|j j j||jYywxYw)a 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)r+r'r from_promptrr _saverrappend)r$promptr/idxp_modelmsgs rsetzFilePromptRegistry.setFs 11&++v~~NLC+6+B+B6+J ##C( *6;;-7hi(--" !--f5G KK   & &w / JJL sA'B*BACCct|jdd5}|j|jj dddy#1swYyxYw)zt Save the prompt index to the JSON file. Writes the current state of the registry to disk. wrrN)openrwritermodel_dump_json)r$fs rr2zFilePromptRegistry._save^sF $##S7 ; 3q GGDKK//1 2 3 3 3s *A  Act|jjD]*\}}|j|k(s|j|k(s&||fcSd|d|d}t |)aF 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 '') enumeraterr r+r'r)r$r+r'ir-r7s rr)z$FilePromptRegistry._get_prompt_modelgsg"$++"5"56 HAuzzT!emmw&>%x /tfOG9AN!#&&r)r%strreturnNone)r+rCr' str | NonerDr)r4rr/boolrDrE)rDrE)r+rFr'rFrDztuple[int, PromptModel]) r rrrr&r.r8r2r)rrrrr s)JH$7;,"8=03'rrN)r __future__rpathlibrpydanticr banks.errorsrr banks.promptrr r rrrrrMs3#@,$)$Z'Z'r