diff --git a/custom_components/ha_text_ai/sensor.py b/custom_components/ha_text_ai/sensor.py
index 43ec019..655f1e9 100644
--- a/custom_components/ha_text_ai/sensor.py
+++ b/custom_components/ha_text_ai/sensor.py
@@ -56,3 +56,4 @@ class HATextAISensor(CoordinatorEntity, SensorEntity):
             ATTR_RESPONSE: last_response,
             ATTR_LAST_UPDATED: self.coordinator.last_update_success_time,
         }
+        
diff --git a/ha_text_ai.zip b/ha_text_ai.zip
index 591e34a..1e94ad9 100644
Binary files a/ha_text_ai.zip and b/ha_text_ai.zip differ
diff --git a/structure.md b/structure.md
new file mode 100644
index 0000000..181a28a
--- /dev/null
+++ b/structure.md
@@ -0,0 +1,27 @@
+ha_text_ai/
+├── custom_components/
+│   └── ha_text_ai/
+│       ├── __init__.py
+│       ├── manifest.json
+│       ├── config_flow.py
+│       ├── const.py
+│       ├── coordinator.py
+│       ├── sensor.py
+│       ├── services.yaml
+│       └── translations/
+│           ├── en.json
+│           └── ru.json
+├── .github/
+│   ├── ISSUE_TEMPLATE/
+│   │   ├── bug_report.md
+│   │   └── feature_request.md
+│   └── workflows/
+│       ├── hassfest.yaml
+│       └── validate.yaml
+├── tests/
+│   └── test_init.py
+├── .gitignore
+├── LICENSE
+├── README.md
+├── hacs.json
+└── info.md
diff --git a/tests/test_init.py b/tests/test_init.py
index 66fd5a4..6c431fb 100644
--- a/tests/test_init.py
+++ b/tests/test_init.py
@@ -1,23 +1,65 @@
 """Tests for the HA text AI integration."""
-from unittest.mock import patch
+from unittest.mock import AsyncMock, patch
 import pytest
+
+from custom_components.ha_text_ai.const import DOMAIN
 from homeassistant.core import HomeAssistant
 from homeassistant.setup import async_setup_component
-from custom_components.ha_text_ai.const import DOMAIN
 
-@pytest.mark.asyncio
-async def test_setup(hass: HomeAssistant):
-    """Test the setup."""
-    with patch('custom_components.ha_text_ai.coordinator.HATextAICoordinator'):
-        assert await async_setup_component(hass, DOMAIN, {
-            DOMAIN: {
-                "api_key": "test_key",
-                "model": "gpt-3.5-turbo",
-                "temperature": 0.7,
-                "max_tokens": 1000,
-                "api_endpoint": "https://api.openai.com/v1",
-                "request_interval": 1.0
-            }
-        })
-        await hass.async_block_till_done()
-        assert DOMAIN in hass.data
+@pytest.fixture
+def mock_setup_entry() -> AsyncMock:
+    """Override async_setup_entry."""
+    with patch(
+        "custom_components.ha_text_ai.async_setup_entry",
+        return_value=True,
+    ) as mock_setup_entry:
+        yield mock_setup_entry
+
+@pytest.fixture
+def mock_coordinator() -> AsyncMock:
+    """Override coordinator."""
+    with patch(
+        "custom_components.ha_text_ai.coordinator.HATextAICoordinator",
+        return_value=AsyncMock(),
+    ) as mock_coordinator:
+        yield mock_coordinator
+
+async def test_async_setup(hass: HomeAssistant, mock_setup_entry: AsyncMock) -> None:
+    """Test the initial setup."""
+    assert await async_setup_component(hass, DOMAIN, {
+        DOMAIN: {
+            "api_key": "test_key",
+            "model": "gpt-3.5-turbo",
+            "temperature": 0.7,
+            "max_tokens": 1000,
+            "api_endpoint": "https://api.openai.com/v1",
+            "request_interval": 1.0
+        }
+    })
+    await hass.async_block_till_done()
+    assert DOMAIN in hass.data
+
+async def test_async_setup_entry(
+    hass: HomeAssistant,
+    mock_coordinator: AsyncMock
+) -> None:
+    """Test setup entry."""
+    entry = MockConfigEntry(
+        domain=DOMAIN,
+        data={
+            "api_key": "test_key",
+            "model": "gpt-3.5-turbo",
+            "temperature": 0.7,
+            "max_tokens": 1000,
+            "api_endpoint": "https://api.openai.com/v1",
+            "request_interval": 1.0
+        },
+    )
+    entry.add_to_hass(hass)
+
+    assert await hass.config_entries.async_setup(entry.entry_id)
+    await hass.async_block_till_done()
+
+    assert len(mock_coordinator.mock_calls) == 1
+    assert DOMAIN in hass.data
+    assert entry.entry_id in hass.data[DOMAIN]