rocksdb

Author: snowmead

.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+  "rust-analyzer.cargo.allFeatures": true,
+}

Cargo.lock

@@ -35,6 +35,9 @@ num-traits = "0.2.17"
 bounded-integer = { version = "0.5.7", features = ["types", "num-traits02"] }
 aquamarine = "0.3.2"
 tiktoken-rs = "0.5.8"
+clap = "4.5.18"
+num_cpus = "1.16.0"
+lazy_static = "1.5.0"
 
 [dev-dependencies]
 futures = "0.3"

Cargo.toml

@@ -15,6 +15,7 @@ use crate::{
 /// that needs to prompt LLM and receive a response.
 ///
 /// This is implemented over the [`Config`] trait.
+#[derive(Debug)]
 pub struct Loom<T: Config> {
 	pub chest: T::Chest,
 	_phantom: PhantomData<T>,
@@ -248,7 +249,7 @@ impl<T: Config> Loom<T> {
 	}
 
 	/// Helper method to build a [`ContextMessage`]
-	fn build_context_message(
+	pub fn build_context_message(
 		role: WrapperRole,
 		content: String,
 		account_id: Option<String>,

src/loom.rs

@@ -35,7 +35,10 @@ impl TapestryChestHandler<MockConfig> for MockChest {
 		Ok(())
 	}
 
-	async fn get_tapestry<TID: TapestryId>(&self, _tapestry_id: TID) -> crate::Result<Option<u16>> {
+	async fn get_instance_index<TID: TapestryId>(
+		&self,
+		_tapestry_id: TID,
+	) -> crate::Result<Option<u16>> {
 		Ok(Some(0))
 	}
 

src/mock.rs

@@ -50,10 +50,13 @@ pub trait TapestryChestHandler<T: Config> {
 		tapestry_id: TID,
 		metadata: M,
 	) -> crate::Result<()>;
-	/// Retrieves the number of instances of a tapestry.
+	/// Retrieves the index of a tapestry.
 	///
 	/// Returns None if the tapestry does not exist.
-	async fn get_tapestry<TID: TapestryId>(&self, tapestry_id: TID) -> crate::Result<Option<u16>>;
+	async fn get_instance_index<TID: TapestryId>(
+		&self,
+		tapestry_id: TID,
+	) -> crate::Result<Option<u16>>;
 	/// Retrieves the last tapestry fragment, or a fragment at a specified instance.
 	///
 	/// # Parameters