o |!g@sjdZddlmZddlmZddlmZddlmZm Z ddl m Z m Z GdddeZ Gd d d Zd S) 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@seZdZUdZgZded<dS)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__rrW/mnt/skqttb/ctump_chatbot/chatbot/lib/python3.10/site-packages/banks/registries/file.pyr s r c@sNeZdZdZdddZdd dddZdddddZd ddZd!ddZdS)"FilePromptRegistryz@A prompt registry storing all prompt data in a single JSON file.registry_indexrreturnNonecCsR||_tgd|_z t|j|_WdSty(|jjjdddYdSw)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 T)parentsexist_okN) _index_fpathr _indexmodel_validate_json read_textFileNotFoundErrorparentmkdir)selfrrrr__init__#s  zFilePromptRegistry.__init__N)versionnamestrr! str | NonercCs"|||\}}tdi|S)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 Nr)_get_prompt_modelr model_dump)rr"r!_modelrrrget5szFilePromptRegistry.getF) overwritepromptr*boolcCsz'||j|j\}}|rt||jj|<|WdSd|jd}t|t yAt|}|jj ||YdSw)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)rr+r*idxp_modelmsgrrrsetFs   zFilePromptRegistry.setcCsFt|jddd}||jWddS1swYdS)zt Save the prompt index to the JSON file. Writes the current state of the registry to disk. wzutf-8)encodingN)openrwritermodel_dump_json)rfrrrr.^s"zFilePromptRegistry._savetuple[int, PromptModel]cCsPt|jjD]\}}|j|kr|j|kr||fSqd|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)rr"r!ir(r2rrrr%gs  z$FilePromptRegistry._get_prompt_model)rrrr)r"r#r!r$rr)r+rr*r,rr)rr)r"r$r!r$rr:) r r r rr r)r3r.r%rrrrr s   rN)r __future__rpathlibrpydanticr banks.errorsrr banks.promptrrr rrrrrs