loom error in api

src/lib.rs

@@ -77,7 +77,7 @@ use types::{LoomError, SummaryModelTokens, WeaveError};
 
 use crate::types::{PromptModelTokens, WrapperRole};
 
-pub type Result<T> = std::result::Result<T, Box<dyn std::error::Error + Send + Sync>>;
+pub type Result<T> = std::result::Result<T, LoomError>;
 
 /// Represents a unique identifier for any arbitrary entity.
 ///

src/loom.rs

@@ -52,10 +52,7 @@ impl<T: Config> Loom<T> {
 		tapestry_id: TID,
 		instructions: String,
 		mut msgs: Vec<ContextMessage<T>>,
-	) -> Result<
-		(<<T as Config>::PromptModel as Llm<T>>::Response, u64, bool),
-		Box<dyn std::error::Error + Send + Sync>,
-	> {
+	) -> Result<(<<T as Config>::PromptModel as Llm<T>>::Response, u64, bool), LoomError> {
 		let instructions_ctx_msg =
 			Self::build_context_message(SYSTEM_ROLE.into(), instructions, None);
 		let instructions_req_msg: PromptModelRequest<T> = instructions_ctx_msg.clone().into();
@@ -211,7 +208,7 @@ impl<T: Config> Loom<T> {
 		summary_model_config: &LlmConfig<T, T::SummaryModel>,
 		tapestry_fragment: &TapestryFragment<T>,
 		summary_max_tokens: SummaryModelTokens<T>,
-	) -> Result<String, Box<dyn std::error::Error + Send + Sync>> {
+	) -> Result<String, LoomError> {
 		trace!(
 			"Generating summary with max tokens: {:?}, for tapestry fragment: {:?}",
 			summary_max_tokens,

src/types.rs

@@ -65,16 +65,7 @@ pub enum LoomError {
 	#[error("Storage error: {0}")]
 	Storage(#[from] StorageError),
 	#[error("Error: {0}")]
-	Error(String),
-}
-
-impl From<Box<dyn std::error::Error + Send + Sync>> for LoomError {
-	fn from(error: Box<dyn std::error::Error + Send + Sync>) -> Self {
-		match error.downcast::<LoomError>() {
-			Ok(loom_error) => *loom_error,
-			Err(error) => LoomError::Error(error.to_string()),
-		}
-	}
+	UnknownError(String),
 }
 
 #[derive(Debug, thiserror::Error)]
@@ -83,6 +74,10 @@ pub enum WeaveError {
 	MaxCompletionTokensIsZero,
 	#[error("Bad configuration: {0}")]
 	BadConfig(String),
+	#[error("Not enough credits to cover cost")]
+	NotEnoughCredits,
+	#[error("Unknown error: {0}")]
+	Unknown(String),
 }
 
 #[derive(Debug, thiserror::Error)]