Update llm/apimodels/gemini_model.py

llm/apimodels/gemini_model.py

@@ -1,26 +1,10 @@
-
 import os
-import logging
 
 from llm.utils.hf_interface import HFInterface
 
 from langchain_google_genai import GoogleGenerativeAI
 from abc import ABC
 
-logger = logging.getLogger(__name__)
-logger.setLevel(logging.ERROR)
-
-file_handler = logging.FileHandler(
-    "logs/chelsea_llm_gemini.log")  # for all modules here template for logs file is "llm/logs/chelsea_{module_name}_{entity}.log"
-logger.setLevel(logging.INFO)  # informed
-
-formatted = logging.Formatter("%(asctime)s - %(levelname)s - %(message)s")
-file_handler.setFormatter(formatted)
-
-logger.addHandler(file_handler)
-logger.info("Getting information from apimodel module")
-
-
 # 429 - You've exceeded the rate limit.
 # 400 - The request body is malformed.
 # 403 - Your API key doesn't have the required permissions.
@@ -33,7 +17,7 @@ logger.info("Getting information from apimodel module")
 # якщо ж у користувача відсутнє інтернет зʼєднання, то нічим не зарадиш, 
 # хіба що пропонувати скачати репозиторій.
 
-_api = os.environ.get("GEMINI_API_TOKEN")
+_api = os.environ.get("GOOGLE_API_KEY")
 
 
 class Gemini(HFInterface, ABC):
@@ -74,7 +58,6 @@ class Gemini(HFInterface, ABC):
         try:
             return self.llm
         except Exception as e:
-            logger.error("Something wrong with Gemini api", exc_info=e)
             print(f"Something wrong with Gemini api: {e}")
 
     def model_name(self):