From 03291c018a6bd138ee5cd73821ba5a709a1b3541 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jos=C3=A9=20Ulises=20Ni=C3=B1o=20Rivera?= Date: Sun, 13 Oct 2024 15:41:28 -0600 Subject: [PATCH] Fix potential panic in create_http_context MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: José Ulises Niño Rivera --- arch/src/filter_context.rs | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/arch/src/filter_context.rs b/arch/src/filter_context.rs index ff6342b3..c365d8bb 100644 --- a/arch/src/filter_context.rs +++ b/arch/src/filter_context.rs @@ -282,7 +282,10 @@ impl RootContext for FilterContext { // No StreamContext can be created until the Embedding Store is fully initialized. let embedding_store = match self.mode { GatewayMode::Llm => None, - GatewayMode::Prompt => Some(Rc::clone(self.embeddings_store.as_ref().unwrap())), + GatewayMode::Prompt => match self.embeddings_store.as_ref() { + None => return None, + Some(store) => Some(Rc::clone(store)), + }, }; Some(Box::new(StreamContext::new( context_id,