From 3dd2e1b24825252741d31d5ec3aa31a095968a22 Mon Sep 17 00:00:00 2001
From: Kenneth Reitz <me@kennethreitz.org>
Date: Fri, 1 Nov 2024 08:28:53 -0400
Subject: [PATCH] Refactor Gemini provider to handle missing llm_model key

---
 simplemind/providers/gemini.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/simplemind/providers/gemini.py b/simplemind/providers/gemini.py
index abd39db..aa08db8 100644
--- a/simplemind/providers/gemini.py
+++ b/simplemind/providers/gemini.py
@@ -81,7 +81,9 @@ class Gemini(BaseProvider):
     @logger
     def structured_response(self, prompt: str, response_model: Type[T], **kwargs) -> T:
         """Send a structured response to the Gemini API."""
-        kwargs.pop("llm_model")
+        # Only try to pop if the key exists
+        kwargs.pop("llm_model", None)  # Add default value of None
+
         try:
             response = self.structured_client.chat.completions.create(
                 messages=[{"role": "user", "content": prompt}],