*'&hddlmZddlZddlmZddlmZddlmZddl m Z ddl m Z GddeZ Gd d eZ Gd d eZeGd dZGdde jZdS)) annotationsN) dataclass)Enum)Optional)FloatceZdZdZdZ dZdS)VectorIndexTypezEnum representing different types of VECTOR index structures. See :ref:`oracle_vector_datatype` for background. .. versionadded:: 2.0.41 HNSWIVFN)__name__ __module__ __qualname____doc__r r \/var/www/html/reinick/venv/lib/python3.11/site-packages/sqlalchemy/dialects/oracle/vector.pyr r s1 D Crr c(eZdZdZdZ dZ dZ dZdS)VectorDistanceTypezEnum representing different types of vector distance metrics. See :ref:`oracle_vector_datatype` for background. .. versionadded:: 2.0.41 EUCLIDEANDOTCOSINE MANHATTANN)r r rrrrrrrrrrr(sII CFIrrc(eZdZdZdZ dZ dZ dZdS)VectorStorageFormatzEnum representing the data format used to store vector components. See :ref:`oracle_vector_datatype` for background. .. versionadded:: 2.0.41 INT8BINARYFLOAT32FLOAT64N)r r rrrrrrrrrrrGsI DFGGrrceZdZUdZejZded<dZded<dZ ded<dZ ded <dZ ded <dZ ded <dZ ded <dZded <dZded<dZdS)VectorIndexConfigaDefine the configuration for Oracle VECTOR Index. See :ref:`oracle_vector_datatype` for background. .. versionadded:: 2.0.41 :param index_type: Enum value from :class:`.VectorIndexType` Specifies the indexing method. For HNSW, this must be :attr:`.VectorIndexType.HNSW`. :param distance: Enum value from :class:`.VectorDistanceType` specifies the metric for calculating distance between VECTORS. :param accuracy: interger. Should be in the range 0 to 100 Specifies the accuracy of the nearest neighbor search during query execution. :param parallel: integer. Specifies degree of parallelism. :param hnsw_neighbors: interger. Should be in the range 0 to 2048. Specifies the number of nearest neighbors considered during the search. The attribute :attr:`.VectorIndexConfig.hnsw_neighbors` is HNSW index specific. :param hnsw_efconstruction: integer. Should be in the range 0 to 65535. Controls the trade-off between indexing speed and recall quality during index construction. The attribute :attr:`.VectorIndexConfig.hnsw_efconstruction` is HNSW index specific. :param ivf_neighbor_partitions: integer. Should be in the range 0 to 10,000,000. Specifies the number of partitions used to divide the dataset. The attribute :attr:`.VectorIndexConfig.ivf_neighbor_partitions` is IVF index specific. :param ivf_sample_per_partition: integer. Should be between 1 and ``num_vectors / neighbor partitions``. Specifies the number of samples used per partition. The attribute :attr:`.VectorIndexConfig.ivf_sample_per_partition` is IVF index specific. :param ivf_min_vectors_per_partition: integer. From 0 (no trimming) to the total number of vectors (results in 1 partition). Specifies the minimum number of vectors per partition. The attribute :attr:`.VectorIndexConfig.ivf_min_vectors_per_partition` is IVF index specific. r index_typeNzOptional[VectorDistanceType]distancez Optional[int]accuracyhnsw_neighborshnsw_efconstructionivf_neighbor_partitionsivf_sample_per_partitionivf_min_vectors_per_partitionparallelct|j|_dD]O}t||}|;t|ts&t |dt |jPdS)N)r$r%r&r'r(r)r#z$ must be an integer ifprovided, got )r r!getattr isinstanceint TypeErrortyper )selffieldvalues r __post_init__zVectorIndexConfig.__post_init__s)$/::   ED%((E E3)?)? <<%)%[[%9<<  r)r r rrr r r!__annotations__r"r#r$r%r&r'r(r)r3rrrr r bs00d#2"6J6666-1H1111"H""""$(N(((()------11111.2222237!7777"H""""rr ceZdZdZdZdZejdejdej dej diZ ddZ d Z d Zd ZGd d ejjZdS)VECTORzOracle VECTOR datatype. For complete background on using this type, see :ref:`oracle_vector_datatype`. .. versionadded:: 2.0.41 TbBfdNc|$t|tstd|$t|tstd||_||_dS)aPConstruct a VECTOR. :param dim: integer. The dimension of the VECTOR datatype. This should be an integer value. :param storage_format: VectorStorageFormat. The VECTOR storage type format. This may be Enum values form :class:`.VectorStorageFormat` INT8, BINARY, FLOAT32, or FLOAT64. Nzdim must be an intergerz:storage_format must be an enum of type VectorStorageFormat)r,r-r.rdimstorage_format)r0r<r=s r__init__zVECTOR.__init__so ?:c3#7#7?566 6  %j // / %L ,rcfd}|S)zT Convert a list to a array.array before binding it to the database. c|t|tjr|St|tr1j}tj||}|St d)Nz$VECTOR accepts list or array.array())r,arraylist_array_typecoder=r.)r2typecoder0s rprocessz.VECTOR._cached_bind_processor..processso} 5%+ > >} E4(( H//0CDD He44  FGGGrr)r0dialectrEs` r_cached_bind_processorzVECTOR._cached_bind_processors( H H H H Hrc d}|S)zR Convert a array.array to list before binding it to the database. cXt|tjrt|SdS)N)r,rArB)r2s rrEz0VECTOR._cached_result_processor..processs,%-- #E{{" # #rr)r0rFcoltyperEs r_cached_result_processorzVECTOR._cached_result_processors  # # #rc8|j|dS)z7 Map storage format to array typecode. r:) _typecode_mapget)r0rDs rrCzVECTOR._array_typecodes!%%h444rc eZdZdZdZdZdS)VECTOR.comparator_factorycL|dt|S)Nz<-> return_typeoprr0others r l2_distancez%VECTOR.comparator_factory.l2_distance"44775e744U;; ;rcL|dt|S)Nz<#>rRrTrVs r inner_productz'VECTOR.comparator_factory.inner_productrYrcL|dt|S)Nz<=>rRrTrVs rcosine_distancez)VECTOR.comparator_factory.cosine_distance rYrN)r r rrXr[r]rrrcomparator_factoryrPsA < < < < < < < < < < rGrKrCtypes TypeEngine Comparatorr^rrrr6r6sHN  #"C#S#S M----,(   555 <<<<risx#""""" !!!!!! """"""d&>$6 MMMMMMM M`W<W<W<W<W