Bump to version 2.1.6

- Change JSON field names from camelCase to snake_case as required by Gemini API
  (generation_config, max_output_tokens, system_instruction)
- Improve message handling to ensure proper role alternation (user/model)
- Add safety checks for empty contents and ensure first message is always from user
- Implement robust error handling and response parsing
- Handle edge cases where candidatesTokenCount might be returned as a list

Fixes #6

custom_components/ha_text_ai/api_client.py

@@ -259,40 +259,83 @@ class APIClient:
         contents = []
         system_instruction = ""
 
+        # Process messages
         for msg in messages:
             if msg['role'] == 'system':
                 system_instruction += msg['content'] + "\n"
             else:
+                # Convert role: 'user' stays 'user', anything else becomes 'model'
+                role = "user" if msg['role'] == 'user' else "model"
                 contents.append({
-                    "role": "user" if msg['role'] == 'user' else "model",
+                    "role": role,
                     "parts": [{"text": msg['content']}]
                 })
 
+        # Ensure contents starts with a user message if not empty
+        if contents and contents[0]["role"] != "user":
+            # Add a placeholder user message
+            contents.insert(0, {
+                "role": "user",
+                "parts": [{"text": "I need your assistance."}]
+            })
+
+        # Ensure contents is not empty
+        if not contents:
+            contents.append({
+                "role": "user",
+                "parts": [{"text": "I need your assistance."}]
+            })
+
+        # Create payload with snake_case keys as required by Gemini API
         payload = {
             "contents": contents,
-            "generationConfig": {
+            "generation_config": {  # Changed from camelCase to snake_case
                 "temperature": temperature,
-                "maxOutputTokens": max_tokens
+                "max_output_tokens": max_tokens  # Changed from camelCase to snake_case
             }
         }
+
         if system_instruction:
-            payload["systemInstruction"] = {
+            payload["system_instruction"] = {  # Changed from camelCase to snake_case
-                "parts": [{"text": system_instruction}]
+                "parts": [{"text": system_instruction.strip()}]
             }
 
-        data = await self._make_request(url, payload)
+        try:
-        return {
+            data = await self._make_request(url, payload)
-            "choices": [{
+
-                "message": {
+            # Safely extract response data
-                    "content": data["candidates"][0]["content"]["parts"][0]["text"]
+            candidates = data.get("candidates", [])
+            if not candidates:
+                raise HomeAssistantError("Gemini API returned no candidates")
+
+            content = candidates[0].get("content", {})
+            parts = content.get("parts", [])
+            if not parts:
+                raise HomeAssistantError("Gemini API response contains no content parts")
+
+            answer_text = parts[0].get("text", "")
+
+            # Safely extract usage data
+            usage = data.get("usageMetadata", {})
+            prompt_tokens = usage.get("promptTokenCount", 0)
+            completion_tokens = usage.get("candidatesTokenCount", 0)
+            total_tokens = usage.get("totalTokenCount", prompt_tokens + completion_tokens)
+
+            return {
+                "choices": [{
+                    "message": {
+                        "content": answer_text
+                    }
+                }],
+                "usage": {
+                    "prompt_tokens": prompt_tokens,
+                    "completion_tokens": completion_tokens,
+                    "total_tokens": total_tokens
                 }
-            }],
-            "usage": {
-                "prompt_tokens": data["usageMetadata"]["promptTokenCount"],
-                "completion_tokens": data["usageMetadata"]["candidatesTokenCount"],
-                "total_tokens": data["usageMetadata"]["totalTokenCount"]
             }
-        }
+        except Exception as e:
+            _LOGGER.error(f"Gemini API error: {str(e)}")
+            raise HomeAssistantError(f"Gemini API error: {str(e)}")
 
     async def shutdown(self) -> None:
         """Shutdown API client."""

custom_components/ha_text_ai/manifest.json

@@ -24,6 +24,6 @@
   "single_config_entry": false,
   "ssdp": [],
   "usb": [],
-  "version": "2.1.4",
+  "version": "2.1.6",
   "zeroconf": []
 }