Hass/ha-text-ai
1
0
Fork 0
mirror of https://github.com/smkrv/ha-text-ai.git synced 2026-07-22 07:03:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(google-gemini): rewrite integration using google-genai 1.16.0

Browse Source 
Completely rewrote the Google Gemini integration logic based on google-genai 1.16.0 to fix issue #6.
Key changes:
- Updated to the latest google-genai library
- Made API endpoint abstract while retaining option for custom endpoint configuration
- Refactored logic and classes exclusively within Google Gemini implementation
- All changes are limited to Google Gemini integration refactoring with no impact on other functionality.
This commit is contained in:
SMKRV
2025-05-21 01:27:47 +03:00
parent 7958bd010b
commit 76c5629fa0
1 changed files with 0 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
custom_components/ha_text_ai/api_client.py
-3
View File
@@ -263,7 +263,6 @@ class APIClient:
Dictionary with response content and token usage Dictionary with response content and token usage
""" """
try: try:
# Импортируем библиотеку в отдельном потоке, чтобы избежать блокировки event loop
def import_genai(): def import_genai():
from google import genai from google import genai
return genai return genai
@@ -273,7 +272,6 @@ class APIClient:
# Extract API key from headers (Bearer token) # Extract API key from headers (Bearer token)
api_key = self.headers.get("Authorization", "").replace("Bearer ", "") api_key = self.headers.get("Authorization", "").replace("Bearer ", "")
# Создаем клиент в отдельном потоке
def create_client(): def create_client():
if self.endpoint and self.endpoint != "https://generativelanguage.googleapis.com/v1beta": if self.endpoint and self.endpoint != "https://generativelanguage.googleapis.com/v1beta":
return genai.Client(api_key=api_key, transport="rest", return genai.Client(api_key=api_key, transport="rest",
@@ -314,7 +312,6 @@ class APIClient:
config = await asyncio.to_thread(create_config) config = await asyncio.to_thread(create_config)
# Выполняем запрос в отдельном потоке
def generate_content(): def generate_content():
# For single message without history, use generate_content # For single message without history, use generate_content
if len(contents) <= 1: if len(contents) <= 1: