try custom integration

yp87 · Jan 31, 2025 · ba6b956 · ba6b956
1 parent 7627889
commit ba6b956
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/hass/custom_components/gemini_api/__init__.py b/hass/custom_components/gemini_api/__init__.py
@@ -1,10 +1,9 @@
 #imports
-import json
-import socket
-import ast
 import base64
+import requests
+import json
 
-from homeassistant.core import SupportsResponse
+from homeassistant.helpers import SupportsResponse
 
 DOMAIN = 'gemini_api'
 
@@ -52,6 +51,4 @@ def generate_text(call):
             "text": response.json()["candidates"][0]["content"]["parts"][0]["text"]
         }
 
-
     hass.services.register(DOMAIN, 'generate_text', generate_text, supports_response=SupportsResponse.ONLY)
-    return True