fix(bedrock): use dict literal instead of set in AI21 response parse default (#5527)

2026-06-15 12:09:34 +05:30
parent de471799d1
commit a1eefc31bc
2 changed files with 29 additions and 1 deletions
@@ -409,7 +409,7 @@ class AWSBedrockLLM(LLMBase):
            elif self.provider == "cohere":
                return response_json.get("generations", [{"text": ""}])[0].get("text", "")
            elif self.provider == "ai21":
-                return response_json.get("completions", [{"data", {"text": ""}}])[0].get("data", {}).get("text", "")
+                return response_json.get("completions", [{"data": {"text": ""}}])[0].get("data", {}).get("text", "")
            else:
                # Generic parsing - try common response fields
                for field in ["content", "text", "completion", "generation"]: