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

Method list_embeddings

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

Source from the content-addressed store, hash-verified

1121 }
1122
1123 fn list_embeddings(&self, path: &str) -> PristineResult<Vec<(u32, EmbeddingRecord)>> {
1124 let table = match self.txn.open_table(EMBEDDINGS) {
1125 Ok(table) => table,
1126 Err(redb::TableError::TableDoesNotExist(_)) => return Ok(Vec::new()),
1127 Err(e) => return Err(PristineError::from(e)),
1128 };
1129
1130 let mut results = Vec::new();
1131 let prefix = format!("{}\0", path);
1132
1133 let iter = table.range::<&str>(prefix.as_str()..)?;
1134
1135 for item in iter {
1136 let (key_guard, value_guard) = item?;
1137 let key_str = key_guard.value();
1138
1139 if !key_str.starts_with(&prefix) {
1140 break;
1141 }
1142
1143 let (_, chunk_idx) = match decode_embedding_key(key_str) {
1144 Some(decoded) => decoded,
1145 None => continue,
1146 };
1147
1148 let bytes = value_guard.value();
1149 let record: EmbeddingRecord =
1150 postcard::from_bytes(bytes).map_err(|e| PristineError::Serialization {
1151 message: format!(
1152 "failed to deserialize EmbeddingRecord at '{}': {}",
1153 key_str, e
1154 ),
1155 })?;
1156 results.push((chunk_idx, record));
1157 }
1158
1159 Ok(results)
1160 }
1161
1162 fn count_embeddings(&self) -> PristineResult<usize> {
1163 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