etj?*<ddlmZddlZddlmZddlmZddlmZddlm Z ddl m Z dd l m Z Gd d eZ Gd d eZGddeZGddeZeGddZGddZGdde jZdS)) annotationsN) dataclass)Enum)Optional)Union)types)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/CA-Chatbot/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)rrrrrrrrrrrrr)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)rrrrrrr r!rrrrrHsI DFGGrrceZdZdZdZ dZdS)VectorStorageTypez}Enum representing the vector type, See :ref:`oracle_vector_datatype` for background. .. versionadded:: 2.0.43 SPARSEDENSEN)rrrrr$r%rrrr#r#cs1F Err#ceZdZUdZejZded<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: integer. 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: integer. 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/r0r*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)rrrrr r r(__annotations__r)r*r+r,r-r.r/r0r:rrrr'r'xs00d#2"6J6666-1H1111"H""""$(N(((()------11111.2222237!7777"H""""rr'c eZdZdZd dZdZd S) SparseVectorz Lightweight SQLAlchemy-side version of SparseVector. This mimics oracledb.SparseVector. .. versionadded:: 2.0.43 num_dimensionsr4indicesUnion[list, array.array]valuesc`t|tjr |jdkrtjd|}t|tjstjd|}t|t|krt d||_||_||_dS)NIdz.indices and values must be of the same length!)r3arraytypecodelenr5r>r?rA)r7r>r?rAs r__init__zSparseVector.__init__s '5;// 073Cs3J3Jk#w//G&%+.. .[f--F w<<3v;; & &LMM M,  rc\d|jdt|jd|jjdS)NzSparseVector(num_dimensions=z, size=z , typecode=))r>rGr?rArF)r7s r__str__zSparseVector.__str__sQ J4+> J J %% J J26+2F J J J rN)r>r4r?r@rAr@)rrrrrHrKrrrr=r=sA"     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 TbBfrDNc|$t|tstd|$t|tstd|$t|tstd||_||_||_dS)aConstruct 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 should be Enum values form :class:`.VectorStorageFormat` INT8, BINARY, FLOAT32, or FLOAT64. :param storage_type: VectorStorageType. The Vector storage type. This should be Enum values from :class:`.VectorStorageType` SPARSE or DENSE. Nzdim must be an integerz:storage_format must be an enum of type VectorStorageFormatz6storage_type must be an enum of type VectorStorageType)r3r4r5rr#dimstorage_format storage_type)r7rRrSrTs rrHzVECTOR.__init__s ?:c3#7#7?455 5  %j // / %L   #J +- - #H ,(rcfd}|S)z Converts a Python-side SparseVector instance into an oracledb.SparseVectormor a compatible array format before binding it to the database. cj|t|tjr|St|tr1j}tj||}|St|t r+j|j|j|j Std)Nz Invalid input for VECTOR: expected a list, an array.array, or a SparseVector object. ) r3rElist_array_typecoderSr=dbapir>r?rAr5)r9rFdialectr7s rprocessz.VECTOR._cached_bind_processor..process)s} 5%+ > >} E4(( //0CDD He44 E<00 }11(ML !rr)r7rZr[s`` r_cached_bind_processorzVECTOR._cached_bind_processor"s)      0rcfd}|S)a Converts database-returned values into Python-native representations. If the value is an oracledb.SparseVector, it is converted into the SQLAlchemy-side SparseVector class. If the value is a array.array, it is converted to a plain Python list. c|dSt|tjrt|St|jjr!t |j|j|jSdS)N)r>r?rA)r3rErWrYr=r>r?rA)r9rZs rr[z0VECTOR._cached_result_processor..processLsw}tE5;// E{{"E7=#=>> ##(#7!M <  rr)r7rZcoltyper[s ` r_cached_result_processorzVECTOR._cached_result_processorCs#     rc8|j|dS)z7 Map storage format to array typecode. rD) _typecode_mapget)r7rFs rrXzVECTOR._array_typecode]s!%%h444rc eZdZdZdZdZdS)VECTOR.comparator_factorycL|dt|S)Nz<-> return_typeopr r7others r l2_distancez%VECTOR.comparator_factory.l2_distanced"44775e744U;; ;rcL|dt|S)Nz<#>rgrirks r inner_productz'VECTOR.comparator_factory.inner_productgrnrcL|dt|S)Nz<=>rgrirks rcosine_distancez)VECTOR.comparator_factory.cosine_distancejrnrN)rrrrmrprrrrrcomparator_factoryrecsA < < < < < < < < < < r}s#""""" !!!!!!d&>$6* MMMMMMM M`        BA<A<A<A<A