jW ddlmZddlZddlZddlmZmZmZmZm Z m Z ddl m Z ddl ZejeZerddlmZej*fddZej*fddZGdd Zdd Zdd Zdd Zdd ZddZy)) annotationsN) TYPE_CHECKINGAnyListOptionalPatterncast)urlparse)Rediscftj|j|jSN)nparrayastypetobytes)rdtypes m/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/langchain_community/utilities/redis.py_array_to_bufferrs$ 88E? ! !% ( 0 0 22cvttttj||j S)N)r)r rfloatr frombuffertolist)bufferrs r_buffer_to_arrayrs' U R]]6?FFH IIrc2eZdZUdZdZded<dd dZd dZy) TokenEscaperz4 Escape punctuation within an input string. z&[,.<>{}\[\]\\\"\':;!@#$%^&*()\-+=~\/ ]strDEFAULT_ESCAPED_CHARSNc`|r||_ytj|j|_yr )escaped_chars_rerecompiler)selfescape_chars_res r__init__zTokenEscaper.__init__!s$ $3D !$&JJt/I/I$JD !rct|tstdt|dd}|jj ||S)Nz:Value must be a string object for token escaping.Got type c.|jd}d|S)Nr\)group)matchvalues r escape_symbolz*TokenEscaper.escape..escape_symbol.sKKNEw< r)r+zre.Matchreturnr) isinstancer TypeErrortyper!sub)r$r,r-s rescapezTokenEscaper.escape'sL%% K=*   $$((>>rr )r%zOptional[Pattern])r,rr.r)__name__ __module__ __qualname____doc__r__annotations__r&r3rrrrs  "K3JK ?rrc|j}|Dcic]}|djd|}}|D]/}|d|vs t||ddt|dk\s/yd}tj |t |cc}w)z1Check if the correct Redis modules are installed.snamezutf-8namesververNzRedis 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_listdecodeintloggererror ValueError)clientrequired_modulesinstalled_modulesmodule error_messages rcheck_redis_module_existrH5s**,>O>OFww'/>O# &>. .3 fVn -f 54  4   # R  LL ] ##sB c L ddl}|jdrt|fi|}|S|jdrd|d<d|vrd |d<t|fi|}|S|j|fi|}t |r|j t|fi|}|S#t$r tdwxYw) aGet a redis client from the connection url given. This helper accepts urls for Redis server (TCP with/without TLS or UnixSocket) as well as Redis Sentinel connections. Redis Cluster is not supported. Before creating a connection the existence of the database driver is checked an and ValueError raised otherwise To use, you should have the ``redis`` python package installed. Example: .. code-block:: python from langchain_community.utilities.redis import get_client redis_client = get_client( redis_url="redis://username:password@localhost:6379" index_name="my-index", embedding_function=embeddings.embed_query, ) To use a redis replication setup with multiple redis server and redis sentinels set "redis_url" to "redis+sentinel://" scheme. With this url format a path is needed holding the name of the redis service within the sentinels to get the correct redis server connection. The default service name is "mymaster". The optional second part of the path is the redis db number to connect to. An optional username or password is used for both connections to the rediserver and the sentinel, different passwords for server and sentinel are not supported. And as another constraint only one sentinel instance can be given: Example: .. code-block:: python from langchain_community.utilities.redis import get_client redis_client = get_client( redis_url="redis+sentinel://username:password@sentinelhost:26379/mymaster/0" index_name="my-index", embedding_function=embeddings.embed_query, ) rNzYCould not import redis python package. Please install it with `pip install redis>=4.1.0`.zredis+sentinelzrediss+sentinelTssl ssl_cert_reqsnone)redis ImportError startswith_redis_sentinel_clientfrom_url_check_for_clusterclose_redis_cluster_client) redis_urlkwargsrM redis_clients r get_clientrXJsX ,--iB6B     / 0u & (&,F? #-iB6B   &u~~i:6: l +    0EfEL )   A   s BB#c Hddl}t|}|jxsd|jxsdfg}|jr;|jj d}|dxsd}t |dkDr |d|d <nd}i}|jr|j|d <|j|d <|jr|j|d <|j|d <|D]!}|jd s|d k(s||||<#|jj|fd|i|} | jd| j%|S#|jj$rP} d| jdvr3t j#d|jj|fi|} n| Yd} ~ ud} ~ wwxYw)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 both systems. rN localhosti g/mymasterdbpasswordusernamerJ client_namesentinel_kwargspingzno password is setzRedis sentinel connection configured with password but Sentinel answered NO PASSWORD NEEDED - Please check Sentinel configuration)rMr hostnameportpathsplitlenr`rarOsentinelSentinelexecute_command exceptionsAuthenticationErrorargsr@warning master_for) rUrVrM parsed_url sentinel_list path_parts service_name sentinel_argsargsentinel_clientaes rrPrPs)$J ))8[*//:RUSTM__**3/ !!}2 z?Q %a=F4L! M$.$7$7 j!'00z$.$7$7 j!'00z >>% C=$8!'M#  .enn--'48>O ''/  % %l 33    / / 2771: - NNC 6enn55mNvNOH  sD88F!AFF!c|ddl} |jd}|ddk(S#|jj$rYywxYw)Nrclustercluster_enabledr\F)rMinform RedisError)rWrM cluster_infos rrRrRsI#((3 -.!33    & &s ;;c 4ddlm}|j|fi|S)Nr) RedisCluster) redis.clusterrrQ)rUrVrs rrTrTs* <  5f 55r)r List[float]rrr.bytes)rrrrr.r)rC RedisTyperDz List[dict]r.None)rUrrVrr.r)rWrr.bool) __future__rloggingr"typingrrrrrr urllib.parser numpyr getLoggerr4r@ redis.clientr rfloat32rrrrHrXrPrRrTr9rrrsu" DD!   8 $/79jj324J??:$*BJ:4z6r