From 207eb94a51f385b319360bfa72b77fef386653d0 Mon Sep 17 00:00:00 2001
From: Gagan Nagaraj <gagandeep.konana@gmail.com>
Date: Sun, 25 Jan 2026 20:31:22 -0700
Subject: [PATCH] check if samplers size > seq id before access

---
 common/common.cpp               | 3 +++
 tools/completion/completion.cpp | 4 ++++
 2 files changed, 7 insertions(+)

diff --git a/common/common.cpp b/common/common.cpp
index 26250abb6c..0b8047c480 100644
--- a/common/common.cpp
+++ b/common/common.cpp
@@ -1195,6 +1195,9 @@ llama_context * common_init_result::context() {
 }
 
 common_sampler * common_init_result::sampler(llama_seq_id seq_id) {
+    if (pimpl->samplers.size() <= (size_t) seq_id) {
+        return nullptr;
+    }
     return pimpl->samplers[seq_id].get();
 }
 
diff --git a/tools/completion/completion.cpp b/tools/completion/completion.cpp
index f368a2f4c6..d9ac94babb 100644
--- a/tools/completion/completion.cpp
+++ b/tools/completion/completion.cpp
@@ -144,6 +144,10 @@ int main(int argc, char ** argv) {
     ctx   = llama_init->context();
     model = llama_init->model();
     smpl  = llama_init->sampler(0);
+    if (smpl == NULL) {
+        LOG_ERR("%s: error: unable to create sampler\n", __func__);
+        return 1;
+    }
 
     if (ctx == NULL) {
         LOG_ERR("%s: error: unable to create context\n", __func__);