diff --git a/custom_components/ha_text_ai/__init__.py b/custom_components/ha_text_ai/__init__.py
index 40da9c8..c0944bd 100644
--- a/custom_components/ha_text_ai/__init__.py
+++ b/custom_components/ha_text_ai/__init__.py
@@ -40,13 +40,16 @@ from .const import (
     API_PROVIDER_OPENAI,
     API_PROVIDER_ANTHROPIC,
     API_PROVIDER_DEEPSEEK,
+    API_PROVIDER_GEMINI,
     DEFAULT_MODEL,
     DEFAULT_DEEPSEEK_MODEL,
+    DEFAULT_GEMINI_MODEL,
     DEFAULT_TEMPERATURE,
     DEFAULT_MAX_TOKENS,
     DEFAULT_OPENAI_ENDPOINT,
     DEFAULT_ANTHROPIC_ENDPOINT,
     DEFAULT_DEEPSEEK_ENDPOINT,
+    DEFAULT_GEMINI_ENDPOINT,
     DEFAULT_REQUEST_INTERVAL,
     DEFAULT_CONTEXT_MESSAGES,
     API_TIMEOUT,
diff --git a/custom_components/ha_text_ai/config_flow.py b/custom_components/ha_text_ai/config_flow.py
index 493b6ac..501b9b1 100644
--- a/custom_components/ha_text_ai/config_flow.py
+++ b/custom_components/ha_text_ai/config_flow.py
@@ -33,6 +33,7 @@ from .const import (
     API_PROVIDERS,
     DEFAULT_MODEL,
     DEFAULT_DEEPSEEK_MODEL,
+    DEFAULT_GEMINI_MODEL,
     DEFAULT_TEMPERATURE,
     DEFAULT_MAX_TOKENS,
     DEFAULT_REQUEST_INTERVAL,
diff --git a/custom_components/ha_text_ai/const.py b/custom_components/ha_text_ai/const.py
index 2f90eeb..177e086 100644
--- a/custom_components/ha_text_ai/const.py
+++ b/custom_components/ha_text_ai/const.py
@@ -12,6 +12,8 @@ from typing import Final
 import voluptuous as vol
 from homeassistant.const import Platform, CONF_API_KEY, CONF_NAME
 from homeassistant.helpers import config_validation as cv
+import logging
+_LOGGER = logging.getLogger(__name__)
 
 # Domain and platforms
 DOMAIN: Final = "ha_text_ai"