From 222d3025b154edcee6d79071fb3ff882a638b2f7 Mon Sep 17 00:00:00 2001
From: Kenneth Reitz <me@kennethreitz.org>
Date: Thu, 31 Oct 2024 12:09:23 -0400
Subject: [PATCH] fix: update Gemini provider to handle unimplemented features
 and improve error handling

---
 simplemind/providers/gemini.py | 69 ++++++++++++++++++----------------
 1 file changed, 37 insertions(+), 32 deletions(-)

diff --git a/simplemind/providers/gemini.py b/simplemind/providers/gemini.py
index b2f145d..0d0ad1e 100644
--- a/simplemind/providers/gemini.py
+++ b/simplemind/providers/gemini.py
@@ -1,3 +1,6 @@
+# TODO: this is a placeholder file for the Gemini provider
+# IT is not currently working as desired.
+
 from functools import cached_property
 from typing import Type, TypeVar
 
@@ -37,34 +40,38 @@ class Gemini(BaseProvider):
 
     def send_conversation(self, conversation: "Conversation") -> "Message":
         """Send a conversation to the Gemini API."""
-        from ..models import Message
-
-        messages = [
-            {
-                "role": msg.role,
-                "content": msg.text,
-                "metadata": msg.meta or {},
-            }
-            for msg in conversation.messages
-        ]
-
-        try:
-            response = self.structured_client.chat.completions.create(
-                messages=messages, response_model=None
-            )
-        except Exception as e:
-            # Handle the exception appropriately, e.g., log the error or raise a custom exception
-            raise RuntimeError(f"Failed to send conversation to Gemini API: {e}") from e
-
-        # Create and return a properly formatted Message instance
-        return Message(
-            role="assistant",
-            text=str(response),
-            raw=response,
-            llm_model=self.model_name,
-            llm_provider=PROVIDER_NAME,
+        raise NotImplementedError(
+            "Gemini does not support conversation-based completions"
         )
 
+        # from ..models import Message
+
+        # messages = [
+        #     {
+        #         "role": msg.role,
+        #         "content": msg.text,
+        #         "metadata": msg.meta or {},
+        #     }
+        #     for msg in conversation.messages
+        # ]
+
+        # try:
+        #     response = self.client.chat.completions.create(
+        #         messages=messages, response_model=None
+        #     )
+        # except Exception as e:
+        #     # Handle the exception appropriately, e.g., log the error or raise a custom exception
+        #     raise RuntimeError(f"Failed to send conversation to Gemini API: {e}") from e
+
+        # # Create and return a properly formatted Message instance
+        # return Message(
+        #     role="assistant",
+        #     text=str(response),
+        #     raw=response,
+        #     llm_model=self.model_name,
+        #     llm_provider=PROVIDER_NAME,
+        # )
+
     def structured_response(self, prompt: str, response_model: Type[T], **kwargs) -> T:
         """Send a structured response to the Gemini API."""
         try:
@@ -82,13 +89,11 @@ class Gemini(BaseProvider):
 
     def generate_text(self, prompt: str, **kwargs) -> str:
         """Generate text using the Gemini API."""
+        llm_model = kwargs.pop("llm_model", self.model_name)
+
         try:
-            response = self.structured_client.chat.completions.create(
-                messages=[{"role": "user", "content": prompt}],
-                response_model=None,
-                **kwargs,
-            )
+            response = self.client.generate_content(prompt, **kwargs)
         except Exception as e:
             # Handle the exception appropriately, e.g., log the error or raise a custom exception
             raise RuntimeError(f"Failed to generate text with Gemini API: {e}") from e
-        return str(response)
+        return response.result