o |!g@sdZddlmZddlZddlmZzddlmZWney)ddl mZYnwddl m Z m Z ddl mZddlmZmZdd lmZmZd ZGd d d eZGd dde ZGdddZdS)zN Directory-based prompt registry implementation that stores prompts as files. ) annotationsN)Path)Self) BaseModelField)Prompt)InvalidPromptErrorPromptNotFoundError)DEFAULT_VERSION PromptModelz index.jsonc@s2eZdZUdZeddZded<edd d ZdS) PromptFilez0Model representing a prompt file stored on disk.T)excluderpathcls type[Self]promptrreturnrcCs>||jd|jd}||j||j|j|j|j|dS)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)rrr prompt_filer\/mnt/skqttb/ctump_chatbot/chatbot/lib/python3.10/site-packages/banks/registries/directory.pyfrom_prompt_path s  zPromptFile.from_prompt_pathN)rrrrrrrr) __name__ __module__ __qualname____doc__rr__annotations__ classmethodrrrrrr s r c@s$eZdZUdZegdZded<dS)PromptFileIndexz1Index tracking all prompt files in the directory.)defaultzlist[PromptFile]filesN)rrr r!rr&r"rrrrr$3s r$c@speZdZdZddd$dd Zed%d d Zd dd&ddZddd'ddZddZ ddZ dd Z d(d"d#Z d S))DirectoryPromptRegistryz?Registry that stores prompts as files in a directory structure.F) force_reindexdirectory_pathrr(boolcCsL|s d}t|||_|jt|_|jr|r |dS|dS)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)is_dir ValueError_pathDEFAULT_INDEX_NAME _index_pathexists_scan_load)selfr)r(msgrrr__init__<s    z DirectoryPromptRegistry.__init__rcCs|jS)z0Get the directory path where prompts are stored.)r-r3rrrrRszDirectoryPromptRegistry.pathN)rrstrr str | NonercCsN|pt}|jjD]}|j|kr$|j|kr$|jr$tdi|Sqt )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 Nr) r _indexr&rrrr0r model_dumpr )r3rrpfrrrgetWs  zDirectoryPromptRegistry.get) overwriterr=cCsz5|jpt}|j|j|d\}}|r+t|jd<t||j |j j |<| WdSd|jd}t |tyXt|jd<t||j }|j j || YdSw)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_filertimectimerr rr-r9r&_saverr append)r3rr=ridxr;r4rrrsetks  zDirectoryPromptRegistry.setcCst|j|_dS)z Load the prompt index from disk.N)r$model_validate_jsonr/ read_textr9r6rrrr2zDirectoryPromptRegistry._loadcCs|j|jdS)zSave the prompt index to disk.N)r/rr9model_dump_jsonr6rrrrBrHzDirectoryPromptRegistry._savec Cst|_|jdD]<}d|jvr|jddn|jtf\}}|d}t| |||id}|jj |Wdn1sAwYq |j |jdS)z4Scan directory for prompt files and build the index.z*.jinja*rr)rrrrrN)r$r9r-globstemrsplitr openr readr&rCr/rrI)r3rrrfr;rrrr1s& zDirectoryPromptRegistry._scantuple[int, PromptFile]cCsPt|jjD]\}}|j|kr|j|kr||fSqd|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 '') enumerater9r&rrr )r3rrir;r4rrrr?s  z(DirectoryPromptRegistry._get_prompt_file)r)rr(r*)rr)rr7rr8rr)rrr=r*)rr8rr7rrR) rrr r!r5propertyrr<rEr2rBr1r?rrrrr'9s  r')r! __future__rr@pathlibrtypingr ImportErrortyping_extensionspydanticrrbanksr banks.errorsrr banks.promptr r r.r r$r'rrrrs"