handle list-wrapped llm responses (#4056)

skyvern/forge/sdk/api/llm/utils.py

@@ -138,6 +138,32 @@ async def llm_messages_builder_with_history(
     return messages
 
 
+def _coerce_response_to_dict(response: Any) -> dict[str, Any]:
+    """Ensure parsed LLM responses expose a dict interface to callers."""
+    if isinstance(response, dict):
+        return response
+
+    if isinstance(response, list):
+        first_dict = next((item for item in response if isinstance(item, dict)), None)
+        LOG.warning(
+            "Parsed LLM response is a list; using first dict element",
+            response_length=len(response),
+            first_item_type=type(response[0]).__name__ if response else None,
+            first_item_keys=list(first_dict.keys()) if first_dict else None,
+        )
+        if first_dict is not None:
+            return first_dict
+
+        LOG.warning("List response contained no dict entries; returning empty dict")
+        return {}
+
+    LOG.warning(
+        "Parsed LLM response is not a dict; returning empty dict",
+        response_type=type(response).__name__,
+    )
+    return {}
+
+
 def parse_api_response(response: litellm.ModelResponse, add_assistant_prefix: bool = False) -> dict[str, Any]:
     content = None
     try:
@@ -146,7 +172,8 @@ def parse_api_response(response: litellm.ModelResponse, add_assistant_prefix: bo
         if add_assistant_prefix:
             content = "{" + content
 
-        return json_repair.loads(content)
+        parsed = json_repair.loads(content)
+        return _coerce_response_to_dict(parsed)
 
     except Exception:
         LOG.warning(
@@ -157,7 +184,8 @@ def parse_api_response(response: litellm.ModelResponse, add_assistant_prefix: bo
             if not content:
                 raise EmptyLLMResponseError(str(response))
             content = _try_to_extract_json_from_markdown_format(content)
-            return commentjson.loads(content)
+            parsed = commentjson.loads(content)
+            return _coerce_response_to_dict(parsed)
         except Exception as e:
             if content:
                 LOG.warning(
@@ -166,7 +194,8 @@ def parse_api_response(response: litellm.ModelResponse, add_assistant_prefix: bo
                     content=content,
                 )
                 try:
-                    return _fix_and_parse_json_string(content)
+                    parsed = _fix_and_parse_json_string(content)
+                    return _coerce_response_to_dict(parsed)
                 except Exception as e2:
                     LOG.exception("Failed to auto-fix LLM response.", error=str(e2))
                     raise InvalidLLMResponseFormat(str(response)) from e2