From 71d00f083fb59bda34c82b82eea85602c1710265 Mon Sep 17 00:00:00 2001
From: tgasser-nv <200644301+tgasser-nv@users.noreply.github.com>
Date: Tue, 2 Sep 2025 11:17:40 -0500
Subject: [PATCH 1/2] Dummy commit to set up the chore/type-clean-guardrails PR
 and branch

---
 nemoguardrails/actions/llm/generation.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nemoguardrails/actions/llm/generation.py b/nemoguardrails/actions/llm/generation.py
index 2a57e1c26..cd11e70a7 100644
--- a/nemoguardrails/actions/llm/generation.py
+++ b/nemoguardrails/actions/llm/generation.py
@@ -137,7 +137,7 @@ async def init(self):
             self._init_flows_index(),
         )
 
-    def _extract_user_message_example(self, flow: Flow):
+    def _extract_user_message_example(self, flow: Flow) -> None:
         """Heuristic to extract user message examples from a flow."""
         elements = [
             item

From de37e626a48e3cd88c980632d1cfe0b741c6de78 Mon Sep 17 00:00:00 2001
From: tgasser-nv <200644301+tgasser-nv@users.noreply.github.com>
Date: Tue, 9 Sep 2025 21:19:50 -0500
Subject: [PATCH 2/2] Check for None before saving basic_index embeddings

---
 nemoguardrails/kb/kb.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/nemoguardrails/kb/kb.py b/nemoguardrails/kb/kb.py
index 685f8a9a0..bd440f842 100644
--- a/nemoguardrails/kb/kb.py
+++ b/nemoguardrails/kb/kb.py
@@ -167,6 +167,8 @@ async def build(self):
                 # We also save the file for future use
                 os.makedirs(CACHE_FOLDER, exist_ok=True)
                 basic_index = cast(BasicEmbeddingsIndex, self.index)
+                if not basic_index.embeddings_index:
+                    raise Exception("Couldn't create basic embeddings index")
                 basic_index.embeddings_index.save(cache_file)
 
                 # And, explicitly save the size as we need it when we reload