>'h hddlmZddlZddlmZddlmZddlmZm Z ddl m Z m Z dZ Gdd Zy) ) annotationsN)cast)Prompt)InvalidPromptErrorPromptNotFoundError)DEFAULT_VERSION PromptModelzCredis is not installed. Please install it with `pip install redis`.cLeZdZdZ d d dZd dZddd dZdddd Zy)RedisPromptRegistryz/A prompt registry that stores prompts in Redis.c ddl}|j|d|_||_y#t$r}tt|d}~wwxYw)z Initialize the Redis prompt registry. Parameters: redis_url: Redis connection URL prefix: Key prefix for storing prompts in Redis rNT)decode_responses)redis ImportErrorREDIS_INSTALL_MSGfrom_url_redis_prefix)self redis_urlprefixres ]/home/kushmeetdev/Regenta/Chatbot/venv/lib/python3.12/site-packages/banks/registries/redis.py__init__zRedisPromptRegistry.__init__sG 8 nnYnF   8/0a 7 8s& A?Ac(|j|d|S)zCreate Redis key for a prompt.:)r)rnameversions r _make_keyzRedisPromptRegistry._make_key$s,,vQwi00N)rc|xst}|j||}|jj|}|sd|d|d}t |t j tt|}tdi|S)a Get a prompt by name and version. Parameters: name: Name of the prompt version: Version of the prompt (optional) Returns: Prompt instance Raises: PromptNotFoundError: If prompt doesn't exist zCannot find prompt with name 'z' and version '') rrrgetrjsonloadsrstrr)rrrkeydatamsg prompt_datas rr#zRedisPromptRegistry.get(su,_nnT7+{{s#24&yPQRC%c* *jjc41 $ $$rF) overwritec||jxst}|j|j|}|jj |r|sd|jd}t |tj|}|j}|jj|tj|y)z Store a prompt in Redis. Parameters: prompt: Prompt instance to store overwrite: Whether to overwrite existing prompt Raises: InvalidPromptError: If prompt exists and overwrite=False zPrompt with name 'z1' already exists. Use overwrite=True to overwriteN) rrrrrexistsrr from_prompt model_dumpsetr$dumps)rpromptr+rr'r) prompt_modelr*s rr0zRedisPromptRegistry.setAs..3OnnV[['2 ;;  c "9&v{{m3deC$S) )#..v6 "--/  TZZ 45r)zredis://localhost:6379z banks:prompt:)rr&rr&returnNone)rr&rr&r4r&)rr&rz str | Noner4r)r2rr+boolr4r5)__name__ __module__ __qualname____doc__rrr#r0r"rrr r sG92%  (17;%28=6rr ) __future__rr$typingrbanksr banks.errorsrr banks.promptrr rr r"rrr@s)" @5YL6L6r