diff --git a/crates/assistant/src/context.rs b/crates/assistant/src/context.rs index f14cf4776a..0cc7e6661f 100644 --- a/crates/assistant/src/context.rs +++ b/crates/assistant/src/context.rs @@ -1407,7 +1407,7 @@ impl Context { async move { let prompt_store = cx.update(|cx| PromptStore::global(cx))?.await?; - let mut prompt = prompt_store.step_resolution_prompt(); + let mut prompt = prompt_store.step_resolution_prompt()?; prompt.push_str(&step_text); request.messages.push(LanguageModelRequestMessage { diff --git a/crates/assistant/src/prompt_library.rs b/crates/assistant/src/prompt_library.rs index a121a5b316..769db54225 100644 --- a/crates/assistant/src/prompt_library.rs +++ b/crates/assistant/src/prompt_library.rs @@ -1505,15 +1505,15 @@ impl PromptStore { self.metadata_cache.read().metadata.first().cloned() } - pub fn step_resolution_prompt(&self) -> String { - String::from_utf8( + pub fn step_resolution_prompt(&self) -> Result { + let path = "prompts/step_resolution.md"; + + Ok(String::from_utf8( Assets - .load("prompts/step_resolution.md") - .unwrap() - .unwrap() + .load(path)? + .ok_or_else(|| anyhow!("{path} not found"))? .to_vec(), - ) - .unwrap() + )?) } }