MCPcopy Create free account
hub / github.com/atomicdotdev/atomic / list_embeddings

Method list_embeddings

atomic-core/src/pristine/txn/read.rs:1012–1049  ·  view source on GitHub ↗
(&self, path: &str)

Source from the content-addressed store, hash-verified

1010 }
1011
1012 fn list_embeddings(&self, path: &str) -> PristineResult<Vec<(u32, EmbeddingRecord)>> {
1013 let table = match self.txn.open_table(EMBEDDINGS) {
1014 Ok(table) => table,
1015 Err(redb::TableError::TableDoesNotExist(_)) => return Ok(Vec::new()),
1016 Err(e) => return Err(PristineError::from(e)),
1017 };
1018
1019 let mut results = Vec::new();
1020 let prefix = format!("{}\0", path);
1021
1022 let iter = table.range::<&str>(prefix.as_str()..)?;
1023
1024 for item in iter {
1025 let (key_guard, value_guard) = item?;
1026 let key_str = key_guard.value();
1027
1028 if !key_str.starts_with(&prefix) {
1029 break;
1030 }
1031
1032 let (_, chunk_idx) = match decode_embedding_key(key_str) {
1033 Some(decoded) => decoded,
1034 None => continue,
1035 };
1036
1037 let bytes = value_guard.value();
1038 let record: EmbeddingRecord =
1039 postcard::from_bytes(bytes).map_err(|e| PristineError::Serialization {
1040 message: format!(
1041 "failed to deserialize EmbeddingRecord at '{}': {}",
1042 key_str, e
1043 ),
1044 })?;
1045 results.push((chunk_idx, record));
1046 }
1047
1048 Ok(results)
1049 }
1050
1051 fn count_embeddings(&self) -> PristineResult<usize> {
1052 let table = match self.txn.open_table(EMBEDDINGS) {

Callers

nothing calls this directly

Calls 3

decode_embedding_keyFunction · 0.85
as_strMethod · 0.45
pushMethod · 0.45

Tested by

no test coverage detected