hyperstreamdb 0.3.2__tar.gz → 0.3.3__tar.gz

{hyperstreamdb-0.3.2 → hyperstreamdb-0.3.3}/Cargo.lock

@@ -3375,7 +3375,7 @@ dependencies = [
 
 [[package]]
 name = "hyperstreamdb"
-version = "0.3.2"
+version = "0.3.3"
 dependencies = [
  "ahash 0.8.12",
  "anyhow",

{hyperstreamdb-0.3.2 → hyperstreamdb-0.3.3}/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "hyperstreamdb"
-version = "0.3.2"
+version = "0.3.3"
 edition = "2021"
 license = "MIT AND Apache-2.0"
 description = "HyperStreamDB - Serverless Index-Streaming Database with Overlay Indexing and Vector Search"

{hyperstreamdb-0.3.2 → hyperstreamdb-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyperstreamdb
-Version: 0.3.2
+Version: 0.3.3
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
 Classifier: Programming Language :: Rust

{hyperstreamdb-0.3.2 → hyperstreamdb-0.3.3}/src/core/query.rs

@@ -184,12 +184,14 @@ pub fn merge_and_rerank_vector_results(
             })
             .collect::<Result<Vec<_>>>()?;
         
-        // Add distance column
-        columns.push(Arc::new(arrow::array::Float32Array::from(distances)));
-        
-        // Create new schema with distance column
         let mut fields: Vec<arrow::datatypes::Field> = batch.schema().fields().iter().map(|f| f.as_ref().clone()).collect();
-        fields.push(arrow::datatypes::Field::new("distance", arrow::datatypes::DataType::Float32, false));
+        
+        // Add distance column only if it doesn't already exist
+        if batch.schema().column_with_name("distance").is_none() {
+            columns.push(Arc::new(arrow::array::Float32Array::from(distances)));
+            fields.push(arrow::datatypes::Field::new("distance", arrow::datatypes::DataType::Float32, false));
+        }
+        
         let schema_with_distance = Arc::new(arrow::datatypes::Schema::new(fields));
         
         let result_batch = RecordBatch::try_new(schema_with_distance, columns)?;

{hyperstreamdb-0.3.2 → hyperstreamdb-0.3.3}/src/python_binding.rs

@@ -821,7 +821,7 @@ impl PyTable {
 
         let filter_str = filter.clone();
         let table_schema = self.table.arrow_schema();
-        let projected_schema = if let Some(cols) = &columns {
+        let mut projected_schema = if let Some(cols) = &columns {
             let mut fields = Vec::new();
             for c in cols {
                 if let Some((_, field)) = table_schema.column_with_name(c) {
@@ -833,6 +833,12 @@ impl PyTable {
             table_schema
         };
 
+        if vs_params_combined.is_some() && projected_schema.column_with_name("distance").is_none() {
+            let mut fields = projected_schema.fields().to_vec();
+            fields.push(std::sync::Arc::new(arrow::datatypes::Field::new("distance", arrow::datatypes::DataType::Float32, true)));
+            projected_schema = std::sync::Arc::new(arrow::datatypes::Schema::new(fields));
+        }
+
         let ctx = device.as_ref().or(self.device.as_ref()).map(|c| c.clone_ref(py));
         let rust_context = if let Some(py_ctx) = ctx {
             let ctx_borrow = py_ctx.bind(py).borrow();