3fiUdZddlZddlZddlZddlmZmZddlmZm Z ddl m Z eGddZ Gdd e ejZejZeed <d Zeed <d eddfdZeGddeZeGddeZeGddeZeGddeZeGddeZeGddeZeGddeZy)zIndex class to add vector indexes on the PGVectorStore. Learn more about vector indexes at https://github.com/pgvector/pgvector?tab=readme-ov-file#indexing N)ABCabstractmethod) dataclassfield)Optionalc,eZdZUeed<eed<eed<y) StrategyMixinoperatorsearch_functionindex_functionN)__name__ __module__ __qualname__str__annotations__[/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain_postgres/v2/indexes.pyr r sMrr ceZdZdZdZdZdZy)DistanceStrategyz&Enumerator of the Distance strategies.)z<-> l2_distance vector_l2_ops)z<=>cosine_distancevector_cosine_ops)z<#> inner_product vector_ip_opsN)r rr__doc__ EUCLIDEANCOSINE_DISTANCE INNER_PRODUCTrrrrrs05ICO;MrrDEFAULT_DISTANCE_STRATEGYlangchainvectorindexDEFAULT_INDEX_NAME_SUFFIX identifierreturncNtjd|td|dy)Nz^[a-zA-Z_][a-zA-Z0-9_]*$zInvalid identifier: zw. Identifiers must start with a letter or underscore, and subsequent characters can be letters, digits, or underscores.)rematch ValueError)r$s rvalidate_identifierr*!s: xx+Z8@":,/f g  ArceZdZUdZdZeeed<dZeed<e dZ e ed<dZ ee eed <dZeeed <ed efd Zd efd ZddZy) BaseIndexa Abstract base class for defining vector indexes. Attributes: name (Optional[str]): A human-readable name for the index. Defaults to None. index_type (str): A string identifying the type of index. Defaults to "base". distance_strategy (DistanceStrategy): The strategy used to calculate distances between vectors in the index. Defaults to DistanceStrategy.COSINE_DISTANCE. partial_indexes (Optional[list[str]]): A list of names of partial indexes. Defaults to None. extension_name (Optional[str]): The name of the extension to be created for the index, if any. Defaults to None. Nnamebase index_typec"tjSN)rrrrrzBaseIndex.9s 0 @ @r)default_factorydistance_strategypartial_indexesextension_namer%ctd)8Set index query options for vector store initialization.z4index_options method must be implemented by subclassNotImplementedErrorselfs r index_optionszBaseIndex.index_options>s" B  rc.|jjSr1)r4r r;s rget_index_functionzBaseIndex.get_index_functionEs%%444rc|jrt|j|jrt|jyy)zCheck if initialization parameters are valid. Raises: ValueError: extension_name is a valid postgreSQL identifier N)r6r*r/r;s r __post_init__zBaseIndex.__post_init__Hs3     3 3 4 ??  0 r)r%N)r rrrr-rrrr/rr4rr5listr6rr=r?rArrrr,r,(s D(3-J*/@+',0OXd3i(/$(NHSM( s  5C5 1rr,ceZdZUdZeed<y)ExactNearestNeighborexactnearestneighborr/N)r rrr/rrrrrrDrDUs ,J,rrDc>eZdZedeefdZedefdZy) QueryOptionsr%ctd)3Convert index attributes to list of configurations.z3to_parameter method must be implemented by subclassr9r;s r to_parameterzQueryOptions.to_parameter\s""WXXrctd)#Convert index attributes to string.z0to_string method must be implemented by subclassr9r;s r to_stringzQueryOptions.to_stringas""TUUrN)r rrrrBrrJrMrrrrGrGZs@Yd3iYYV3VVrrGcDeZdZUdZeed<dZeed<dZeed<defdZ y ) HNSWIndexhnswr/m@ef_constructionr%c<d|jd|jdS)r8z(m = z, ef_construction = ))rRrTr;s rr=zHNSWIndex.index_optionsms#tvvh243G3G2HJJrN) r rrr/rrrRintrTr=rrrrOrOgs-J AsKOSKsKrrOc:eZdZUdZeed<deefdZdefdZ y)HNSWQueryOptions( ef_searchr%c"d|jgS)rIhnsw.ef_search = )r[r;s rrJzHNSWQueryOptions.to_parametervs#DNN#3455rcTtjdtd|jS)rL2to_string is deprecated, use to_parameter instead.r])warningswarnDeprecationWarningr[r;s rrMzHNSWQueryOptions.to_stringzs( @  #4>>"233rN) r rrr[rWrrBrrJrMrrrrYrYrs(Is6d3i6434rrYc6eZdZUdZeed<dZeed<defdZy) IVFFlatIndexivfflatr/dlistsr%c"d|jdS)r8z (lists = rV)rgr;s rr=zIVFFlatIndex.index_optionss4::,a((rN) r rrr/rrrgrWr=rrrrdrds"JE3)s)rrdc:eZdZUdZeed<deefdZdefdZ y)IVFFlatQueryOptionsprobesr%c"d|jgS)rIivfflat.probes = )rlr;s rrJz IVFFlatQueryOptions.to_parameters#DKK=122rcTtjdtd|jS)rLr_rn)r`rarbrlr;s rrMzIVFFlatQueryOptions.to_strings' @  #4;;-00rN) r rrrlrWrrBrrJrMrrrrjrjs'FCO3d3i3131rrj)renumr'r`abcrr dataclassesrrtypingrr Enumrrr!rr#rr*r,rDrGrOrYrdrjrrrrus?  #(   <}dii</?.N.N+N!737 C D  )1)1 )1X -9- -  V3 V  V K K K  4| 4  4  )9) )  1, 1  1r