o ȳgY @sddlmZddlZddlZddlmZmZmZmZm Z m Z ddl m Z ddl ZeeZer5ddlmZejfd'd d Zejfd(ddZGdddZd)ddZd*ddZd*dd Zd+d#d$Zd*d%d&ZdS),) annotationsN) TYPE_CHECKINGAnyListOptionalPatterncast)urlparse)Redisarray List[float]dtyperreturnbytescCst||SN)npr astypetobytes)r r r_/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/langchain_community/utilities/redis.py_array_to_buffersrbuffercCsttttj||dS)N)r )rrfloatr frombuffertolist)rr rrr_buffer_to_arraysrc@s4eZdZUdZdZded<dddd Zdd d ZdS) TokenEscaperz4 Escape punctuation within an input string. z&[,.<>{}\[\]\\\"\':;!@#$%^&*()\-+=~\/ ]strDEFAULT_ESCAPED_CHARSNescape_chars_reOptional[Pattern]cCs |r||_dSt|j|_dSr)escaped_chars_rerecompiler)selfrrrr__init__!s zTokenEscaper.__init__valuercCs4t|tstdt|ddd}|j||S) Nz:Value must be a string object for token escaping.Got type matchre.MatchrrcSs|d}d|S)Nr\)group)r'r&rrr escape_symbol.s  z*TokenEscaper.escape..escape_symbol)r'r(rr) isinstancer TypeErrortyper!sub)r$r&r+rrrescape's  zTokenEscaper.escaper)rr )r&rrr)__name__ __module__ __qualname____doc__r__annotations__r%r0rrrrrs   rclient RedisTyperequired_modules List[dict]NonecCsh|}dd|D}|D]}|d|vr(t||ddt|dkr(dSq d}t|t|)z1Check if the correct Redis modules are installed.cSsi|] }|dd|qS)snamezutf-8)decode).0modulerrr 8sz,check_redis_module_exist..namesververNzRedis cannot be used as a vector database without RediSearch >=2.4Please head to https://redis.io/docs/stack/search/quick_start/to know more about installing the RediSearch module within Redis Stack.) module_listintloggererror ValueError)r6r8installed_modulesr= error_messagerrrcheck_redis_module_exist5s   rH redis_urlrkwargscKszddl}Wn tytdw|dr!t|fi|}|S|dr=4.1.0`.zredis+sentinelzrediss+sentinelTssl ssl_cert_reqsnone)redis ImportError startswith_redis_sentinel_clientfrom_url_check_for_clusterclose_redis_cluster_client)rIrJrN redis_clientrrr get_clientJs*,    rWc KsNddl}t|}|jp d|jpdfg}|jr/|jd}|dp!d}t|dkr.|d|d <nd}i}|jr@|j|d <|j|d <|jrM|j|d <|j|d <|D]}| d sZ|d kr`||||<qO|j j |fd|i|} z| dWn-|j jy} zd| jdvrtd|j j |fi|} n| WYd} ~ nd} ~ ww| |S)ahelper method to parse an (un-official) redis+sentinel url and create a Sentinel connection to fetch the final redis client connection to a replica-master for read-write operations. If username and/or password for authentication is given the same credentials are used for the Redis Sentinel as well as Redis Server. With this implementation using a redis url only it is not possible to use different data for authentication on booth systems. rN localhosti g/mymasterdbpasswordusernamerK client_namesentinel_kwargspingzno password is setzRedis sentinel connection configured with password but Sentinel answered NO PASSWORD NEEDED - Please check Sentinel configuration)rNr hostnameportpathsplitlenr^r_rPsentinelSentinelexecute_command exceptionsAuthenticationErrorargsrCwarning master_for) rIrJrN parsed_url sentinel_list path_parts service_name sentinel_argsargsentinel_clientaerrrrQsR            rQrVboolcCs:ddl}z |d}|ddkWS|jjyYdSw)Nrclustercluster_enabledrZF)rNinfork RedisError)rVrN cluster_inforrrrSs rScKsddlm}|j|fi|S)Nr) RedisCluster) redis.clusterr~rR)rIrJr~rrrrUs rU)r r r rrr)rrr rrr )r6r7r8r9rr:)rIrrJrrr7)rVr7rrx) __future__rloggingr"typingrrrrrr urllib.parser numpyr getLoggerr1rC redis.clientr r7float32rrrrHrWrQrSrUrrrrs"        E =