hz*ddlmZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z Gdd eZ Gd d eZGd d eZGddeZeGddZGddZGdde j&Zy)) annotationsN) dataclass)Enum)Optional)Union)FloatceZdZdZdZ dZy)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/eduruby.in/venv/lib/python3.12/site-packages/sqlalchemy/dialects/oracle/vector.pyr r s D Crr c&eZdZdZdZ dZ dZ dZy)VectorDistanceTypezEnum representing different types of vector distance metrics. See :ref:`oracle_vector_datatype` for background. .. versionadded:: 2.0.41 EUCLIDEANDOTCOSINE MANHATTANN)r rrrrrrrrrrrr)s7I CFIrrc&eZdZdZdZ dZ dZ dZy)VectorStorageFormatzEnum representing the data format used to store vector components. See :ref:`oracle_vector_datatype` for background. .. versionadded:: 2.0.41 INT8BINARYFLOAT32FLOAT64N)r rrrrrrrrrrrrHs7 DFGGrrceZdZdZdZ dZy)VectorStorageTypez}Enum representing the vector type, See :ref:`oracle_vector_datatype` for background. .. versionadded:: 2.0.43 SPARSEDENSEN)r rrrr"r#rrrr!r!csF Err!ceZdZUdZej Zded<dZded<dZ ded<dZ ded <dZ ded <dZ ded <dZ ded <dZded <dZded<dZy)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]D}t||}|t|tr#t |dt |jy)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__sl)$//:  ED%(E E3)?g%%)%[%9%9$:< r)r rrrr r r&__annotations__r'r(r)r*r+r,r-r.r8rrrr%r%xst0d#2"6"6J6-1H*1"Hm"$(NM()---1]1.2m237!=7"Hm"rr%c*eZdZdZ ddZdZy) SparseVectorz Lightweight SQLAlchemy-side version of SparseVector. This mimics oracledb.SparseVector. .. versionadded:: 2.0.43 cPt|tjr|jdk7rtjd|}t|tjstjd|}t|t|k7r t d||_||_||_y)NIdz.indices and values must be of the same length!)r1arraytypecodelenr3num_dimensionsindicesvalues)r5rBrCrDs r__init__zSparseVector.__init__s '5;;/73C3Cs3Jkk#w/G&%++.[[f-F w<3v; &LM M,  rc|d|jdt|jd|jjdS)NzSparseVector(num_dimensions=z, size=z , typecode=))rBrArCrDr@)r5s r__str__zSparseVector.__str__sC*4+>+>*?@ %&k$++2F2F1Gq J rN)rBr2rCUnion[list, array.array]rDrI)r rrrrErHrrrr;r;s-*) " rr;ceZdZdZdZdZejdejdejdejdiZ ddZ d Z d Zd ZGd d ej"j$Zy)VECTORzOracle VECTOR datatype. For complete background on using this type, see :ref:`oracle_vector_datatype`. .. versionadded:: 2.0.41 TbBfr>Nc|t|ts td|t|ts td|t|ts td||_||_||_y)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 intergerz:storage_format must be an enum of type VectorStorageFormatz6storage_type must be an enum of type VectorStorageType)r1r2r3rr!dimstorage_format storage_type)r5rPrQrRs rrEzVECTOR.__init__s ?:c3#756 6  %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. cr|t|tjr|St|tr3jj}tj||}|St|t r;j j |j|j|jStd)Nz Invalid input for VECTOR: expected a list, an array.array, or a SparseVector object. ) r1r?list_array_typecoderQr;dbapirBrCrDr3)r7r@dialectr5s rprocessz.VECTOR._cached_bind_processor..process(s} 5%++ > E4(//0C0CD He4 E<0}}11((MMLL rr)r5rXrYs`` r_cached_bind_processorzVECTOR._cached_bind_processor!s 4rcfd}|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|yt|tjr t|St|jjr,t |j |j |jSy)N)rBrCrD)r1r?rUrWr;rBrCrD)r7rXs rrYz0VECTOR._cached_result_processor..processMsb}E5;;/E{"E7==#=#=>##(#7#7!MM <<?rr)r5rXcoltyperYs ` r_cached_result_processorzVECTOR._cached_result_processorDs rc:|jj|dS)z7 Map storage format to array typecode. r>) _typecode_mapget)r5r@s rrVzVECTOR._array_typecode^s!!%%h44rceZdZdZdZdZy)VECTOR.comparator_factoryc<|jdt|S)Nz<-> return_typeoprr5others r l2_distancez%VECTOR.comparator_factory.l2_distancee44775e74U; ;rc<|jdt|S)Nz<#>rergris r inner_productz'VECTOR.comparator_factory.inner_producthrlrc<|jdt|S)Nz<=>rergris rcosine_distancez)VECTOR.comparator_factory.cosine_distancekrlrN)r rrrkrnrprrrcomparator_factoryrcds < < r|s# ! "d&>$6* MM M`  BB