File: //lib/python3.9/site-packages/redis/commands/search/__pycache__/field.cpython-39.pyc
a
��b� � @ sz d dl mZ d dlmZ G dd� d�ZG dd� de�ZG dd� de�ZG d d
� d
e�ZG d d � d e�ZG d
d� de�Z dS )� )�List)� DataErrorc @ s\ e Zd ZdZdZdZdZdZdZdZ dZ
d Z de e
e eee d �d
d�Zdd� Zdd� Zd
S )�Field�NUMERIC�TEXT�WEIGHT�GEO�TAG�VECTOR�SORTABLE�NOINDEX�ASNF)�name�args�sortable�no_index�as_namec C s^ |d u r g }|| _ || _t� | _|| _|r8| j�tj� |rJ| j�tj� |rZ|sZt d��d S )Nz-Non-Sortable non-Indexable fields are ignored)
r r �list� args_suffixr �appendr r r �
ValueError)�selfr r r r r � r �?/usr/lib/python3.9/site-packages/redis/commands/search/field.py�__init__ s zField.__init__c C s | j �|� d S �N)r r )r �valuer r r �
append_arg) s zField.append_argc C s6 | j g}| jr || j| jg7 }|| j7 }|| j7 }|S r )r r r
r r )r r r r r �
redis_args, s
zField.redis_args)NFFN)�__name__�
__module__� __qualname__r r r r r r
r r r
�strr �boolr r r r r r r r s, �� r c @ s. e Zd ZdZdZdZd
eeeed�dd �Z dS ) � TextFieldzI
TextField is used to define a text field in a schema definition
�NOSTEM�PHONETIC� �?FN)r �weight�no_stem�phonetic_matcherc K s^ t j| |fdt jt j|gi|�� |r4t �| | j� |rZ|dv rZt �| | j� t �| |� d S )Nr )zdm:enzdm:frzdm:ptzdm:es)r r r r r r% r& )r r r( r) r* �kwargsr r r r = s " zTextField.__init__)r' FN)
r r r! �__doc__r% r& r" �floatr# r r r r r r$ 5 s ��r$ c @ s e Zd ZdZed�dd�ZdS )� NumericFieldzO
NumericField is used to define a numeric field in a schema definition
�r c K s t j| |fdt jgi|�� d S �Nr )r r r �r r r+ r r r r X s zNumericField.__init__N�r r r! r, r" r r r r r r. S s r. c @ s e Zd ZdZed�dd�ZdS )�GeoFieldzP
GeoField is used to define a geo-indexing field in a schema definition
r/ c K s t j| |fdt jgi|�� d S r0 )r r r r1 r r r r a s zGeoField.__init__Nr2 r r r r r3 \ s r3 c @ s, e Zd ZdZdZdZd
eeed�dd�Zd S ) �TagFieldzx
TagField is a tag-indexing field with simpler compression and tokenization.
See http://redisearch.io/Tags/
� SEPARATOR�
CASESENSITIVE�,F)r � separator�case_sensitivec K s: t j| j|g}|r |�| j� t j| |fd|i|�� d S r0 )r r r5 r r6 r )r r r8 r9 r+ r r r r r n s zTagField.__init__N)r7 F) r r r! r, r5 r6 r" r# r r r r r r4 e s ��r4 c @ s" e Zd ZdZeeed�dd�ZdS )� VectorFieldz�
Allows vector similarity queries against the value in this attribute.
See https://oss.redis.com/redisearch/Vectors/#vector_fields.
)r � algorithm�
attributesc
K s� |� dd�}|� dd�}|s |r(td��|�� dvr<td��g }|�� D ]\}} |�|| g� qHtj| |fdtj|t|�g|�i|�� dS ) a�
Create Vector Field. Notice that Vector cannot have sortable or no_index tag,
although it's also a Field.
``name`` is the name of the field.
``algorithm`` can be "FLAT" or "HNSW".
``attributes`` each algorithm can have specific attributes. Some of them
are mandatory and some of them are optional. See
https://oss.redis.com/redisearch/master/Vectors/#specific_creation_attributes_per_algorithm
for more information.
r Fr z5Cannot set 'sortable' or 'no_index' in Vector fields.)ZFLATZHNSWzIRealtime vector indexing supporting 2 Indexing Methods:'FLAT' and 'HNSW'.r N) �getr �upper�items�extendr r r
�len)
r r r; r< r+ �sortZnoindexZattr_li�keyr r r r r ~ s$ ����zVectorField.__init__N)r r r! r, r" �dictr r r r r r: x s r: N)
�typingr Zredisr r r$ r. r3 r4 r: r r r r �<module> s /