From 4fc58e062a58421a2d75aa75e1dfa595ccab1500 Mon Sep 17 00:00:00 2001 From: LJ Date: Thu, 24 Apr 2025 18:24:42 -0700 Subject: [PATCH] fix: variable name typo --- python/cocoindex/query.py | 4 ++-- src/execution/query.rs | 4 ++-- src/py/mod.rs | 6 +++--- 3 files changed, 7 insertions(+), 7 deletions(-) diff --git a/python/cocoindex/query.py b/python/cocoindex/query.py index cf9f57fbc..befd643e4 100644 --- a/python/cocoindex/query.py +++ b/python/cocoindex/query.py @@ -66,13 +66,13 @@ def internal_handler(self) -> _engine.SimpleSemanticsQueryHandler: return self._lazy_query_handler() def search(self, query: str, limit: int, vector_field_name: str | None = None, - similarity_matric: index.VectorSimilarityMetric | None = None) -> tuple[list[QueryResult], SimpleSemanticsQueryInfo]: + similarity_metric: index.VectorSimilarityMetric | None = None) -> tuple[list[QueryResult], SimpleSemanticsQueryInfo]: """ Search the index with the given query, limit, vector field name, and similarity metric. """ internal_results, internal_info = self.internal_handler().search( query, limit, vector_field_name, - similarity_matric.value if similarity_matric is not None else None) + similarity_metric.value if similarity_metric is not None else None) fields = [field['name'] for field in internal_results['fields']] results = [QueryResult(data=dict(zip(fields, result['data'])), score=result['score']) for result in internal_results['results']] info = SimpleSemanticsQueryInfo( diff --git a/src/execution/query.rs b/src/execution/query.rs index d06ca2d74..0f8ebe5db 100644 --- a/src/execution/query.rs +++ b/src/execution/query.rs @@ -68,7 +68,7 @@ impl SimpleSemanticsQueryHandler { query: String, limit: u32, vector_field_name: Option, - similarity_matric: Option, + similarity_metric: Option, ) -> Result<(QueryResults, SimpleSemanticsQueryInfo)> { let query_results = evaluate_transient_flow( &self.query_transform_flow, @@ -106,7 +106,7 @@ impl SimpleSemanticsQueryHandler { .or(self.default_vector_field_name.clone()) .ok_or_else(|| api_error!("vector field name must be provided"))?; - let similarity_metric = similarity_matric.unwrap_or(self.default_similarity_metric); + let similarity_metric = similarity_metric.unwrap_or(self.default_similarity_metric); let info = SimpleSemanticsQueryInfo { similarity_metric, query_vector: vector.clone(), diff --git a/src/py/mod.rs b/src/py/mod.rs index e42fb9992..21400d4d8 100644 --- a/src/py/mod.rs +++ b/src/py/mod.rs @@ -255,14 +255,14 @@ impl SimpleSemanticsQueryHandler { Ok(()) } - #[pyo3(signature = (query, limit, vector_field_name = None, similarity_matric = None))] + #[pyo3(signature = (query, limit, vector_field_name = None, similarity_metric = None))] pub fn search( &self, py: Python<'_>, query: String, limit: u32, vector_field_name: Option, - similarity_matric: Option>, + similarity_metric: Option>, ) -> PyResult<( Pythonized, Pythonized, @@ -275,7 +275,7 @@ impl SimpleSemanticsQueryHandler { query, limit, vector_field_name, - similarity_matric.map(|m| m.0), + similarity_metric.map(|m| m.0), ) .await })