rust/lancedb
1
0
Fork 0
mirror of https://github.com/lancedb/lancedb.git synced 2026-05-18 20:40:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

more code

Browse Source
This commit is contained in:
albertlockett
2024-01-31 18:46:38 -05:00
parent 65c9c0ba9b
commit 2bbc56b9f9
2 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
rust/vectordb/src/merge_insert.rs
View File

@@ -90,8 +90,8 @@ impl MergeInsertBuilder {
// }
let job = builder.try_build().unwrap(); // TODO no unwrap
let bitches = reader_to_stream(batches).await.unwrap().0; // TODO no unwrap
let ds2 = job.execute(bitches).await.unwrap(); // TODO no unwrap
let batches = reader_to_stream(batches).await.unwrap().0; // TODO no unwrap
let ds2 = job.execute(batches).await.unwrap(); // TODO no unwrap
native_table.reset_dataset(ds2.as_ref().clone());

7
rust/vectordb/src/table.rs
View File

@@ -34,6 +34,7 @@ use log::info;
use crate::error::{Error, Result};
use crate::index::vector::{VectorIndex, VectorIndexStatistics};
use crate::index::IndexBuilder;
use crate::merge_insert::MergeInsertBuilder;
use crate::query::Query;
use crate::utils::{PatchReadParam, PatchWriteParam};
use crate::WriteMode;
@@ -241,6 +242,8 @@ pub trait Table: std::fmt::Display + Send + Sync {
/// Modeled after ``VACCUM`` in PostgreSQL.
/// Not all implementations support explicit optimization.
async fn optimize(&self, action: OptimizeAction) -> Result<OptimizeStats>;
fn merge_insert(&self) -> MergeInsertBuilder;
}
/// Reference to a Table pointer.
@@ -698,6 +701,10 @@ impl Table for NativeTable {
}
Ok(stats)
}
fn merge_insert(&self) -> MergeInsertBuilder {
MergeInsertBuilder::new(Arc::new(self.clone()))
}
}
#[cfg(test)]