diff --git a/crates/assistant/src/context.rs b/crates/assistant/src/context.rs index f14cf4776a5b6965a8c6cf59fd683992862456ec..0cc7e6661f29c685150c4071e78bb1d23dda7494 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 a121a5b3169180ee3b264f1c114a8d3f0d5c80a6..769db5422524c7c63cb678c1ffe60fc83a0505e0 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() + )?) } }