From 0a3889ae849bcfa5ffe4b3dc9f0174772ecc76ee Mon Sep 17 00:00:00 2001
From: Eli <egsalamie@gmail.com>
Date: Sun, 6 Oct 2024 21:14:04 -0500
Subject: [PATCH] Now update google gemini dep bug.

---
 pyproject.toml            |  2 +-
 readmeai/models/gemini.py | 64 ++++++++++++++++++++++++++++++++++-----
 2 files changed, 57 insertions(+), 9 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 63130540..b41fff60 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "readmeai"
-version = "0.5.94"
+version = "0.5.95"
 description = "Automated README file generator, powered by AI."
 authors = ["Eli <egsalamie@gmail.com>"]
 license = "MIT"
diff --git a/readmeai/models/gemini.py b/readmeai/models/gemini.py
index 4fb68c47..fc504364 100644
--- a/readmeai/models/gemini.py
+++ b/readmeai/models/gemini.py
@@ -1,10 +1,9 @@
 """Google Gemini LLM API service implementation."""
 
 import os
-from typing import Any
+from typing import Any, Optional
 
 import aiohttp
-import google.generativeai as genai
 from tenacity import (
     retry,
     retry_if_exception_type,
@@ -17,6 +16,14 @@
 from readmeai.models.base import BaseModelHandler
 from readmeai.models.tokens import token_handler
 
+try:
+    import google.generativeai as genai
+
+    GENAI_AVAILABLE = True
+except ImportError:
+    genai = None
+    GENAI_AVAILABLE = False
+
 
 class GeminiHandler(BaseModelHandler):
     """
@@ -27,19 +34,47 @@ def __init__(
         self, config_loader: ConfigLoader, context: RepositoryContext
     ) -> None:
         super().__init__(config_loader, context)
-        self._model_settings()
+        self.model: Optional[Any] = None
+        self.model_name: str = "gemini-1.5-flash"
+        self.top_p: float = (
+            0.8  # Default value, override from config if available
+        )
+        if GENAI_AVAILABLE:
+            self._model_settings()
+        else:
+            self._logger.warning(
+                "Google Generative AI library is not available. Some features will be disabled."
+            )
 
     def _model_settings(self):
+        if not GENAI_AVAILABLE:
+            self._logger.error(
+                "Attempted to configure Gemini model without the required library."
+            )
+            return
+
         api_key = os.getenv("GOOGLE_API_KEY")
+        if not api_key:
+            self._logger.error(
+                "GOOGLE_API_KEY environment variable is not set."
+            )
+            return
+
         genai.configure(api_key=api_key)
-        self.model_name = "gemini-1.5-flash"
         self.model = genai.GenerativeModel(self.model_name)
-        self.top_p = self.config.llm.top_p
+
+        # Safely get top_p from config, use default if not available
+        self.top_p = getattr(self.config.llm, "top_p", self.top_p)
 
     async def _build_payload(self, prompt: str, tokens: int) -> dict[str, Any]:
         """Build payload for POST request to the Gemini API."""
+        if not GENAI_AVAILABLE:
+            raise RuntimeError(
+                "Google Generative AI library is not available."
+            )
+
         return genai.types.GenerationConfig(
-            max_output_tokens=self.max_tokens,
+            max_output_tokens=tokens,
             temperature=self.temperature,
             top_p=self.top_p,
         )
@@ -63,9 +98,18 @@ async def _make_request(
         repo_files: Any,
     ) -> Any:
         """Processes Gemini API responses and returns generated text."""
+        if not GENAI_AVAILABLE:
+            self._logger.error(
+                "Cannot make request: Google Generative AI library is not available."
+            )
+            return index, self.placeholder
+
+        if self.model is None:
+            self._logger.error("Gemini model is not properly initialized.")
+            return index, self.placeholder
+
         try:
             prompt = await token_handler(self.config, index, prompt, tokens)
-
             parameters = await self._build_payload(prompt, tokens)
 
             async with self.rate_limit_semaphore:
@@ -73,7 +117,11 @@ async def _make_request(
                     prompt,
                     generation_config=parameters,
                 )
-                data = response.text
+                data = (
+                    response.text
+                    if hasattr(response, "text")
+                    else str(response)
+                )
                 self._logger.info(
                     f"Response from Gemini for '{index}': {data}"
                 )