o |!g @sbddlmZddlZddlmZddlmZddlmZm Z ddl m Z m Z dZ Gdd d ZdS) ) annotationsN)cast)Prompt)InvalidPromptErrorPromptNotFoundError)DEFAULT_VERSION PromptModelzCredis is not installed. Please install it with `pip install redis`.c@sJeZdZdZ  ddd d Zdd dZdddddZddd ddZdS)!RedisPromptRegistryz/A prompt registry that stores prompts in Redis.redis://localhost:6379 banks:prompt: redis_urlstrprefixreturnNonec CsJzddl}Wnty}ztt|d}~ww|j|dd|_||_dS)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)selfr rrerX/mnt/skqttb/ctump_chatbot/chatbot/lib/python3.10/site-packages/banks/registries/redis.py__init__s   zRedisPromptRegistry.__init__nameversioncCs|j|d|S)zCreate Redis key for a prompt.:)r)rrrrrr _make_key$szRedisPromptRegistry._make_keyN)r str | NonercCs\|pt}|||}|j|}|sd|d|d}t|ttt|}t di|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 ''Nr) rr rgetrjsonloadsrr r)rrrkeydatamsg prompt_datarrrr#(s  zRedisPromptRegistry.getF) overwritepromptr*boolcCsh|jpt}||j|}|j|r|sd|jd}t|t|}| }|j |t |dS)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) rrr rrexistsrr from_prompt model_dumpsetr$dumps)rr+r*rr&r( prompt_modelr)rrrr0As  zRedisPromptRegistry.set)r r )r r rr rr)rr rr rr )rr rr!rr)r+rr*r,rr)__name__ __module__ __qualname____doc__rr r#r0rrrrr s r ) __future__rr$typingrbanksr banks.errorsrr banks.promptrrrr rrrrs