Refactor Gemini provider to handle missing llm_model key

This commit is contained in:
2024-11-01 08:28:53 -04:00
parent ad1800840d
commit 3dd2e1b248
+3 -1
View File
@@ -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}],