fix(bedrock_agent): fix querystring field resolution (#6777)

* fix: prevent splitting Bedrock Agent parameters with commas

This fixes issue #6520 where Bedrock Agent parameters containing commas (like SQL queries) were being truncated because the resolved_query_string_parameters method was splitting them by commas.

The fix overrides the resolved_query_string_parameters method in the BedrockAgentEvent class to preserve parameter values without splitting them by commas.

* style: format test files

* fix: address PR feedback for Bedrock Agent parameters with commas

- Change @cached_property to @Property for resolved_query_string_parameters
- Simplify the functional test by directly returning the query parameter
- Remove redundant unit test

---------

Co-authored-by: Leandro Damascena <[email protected]>

Author: matteofigus

aws_lambda_powertools/utilities/data_classes/bedrock_agent_event.py

@@ -112,6 +112,19 @@ def query_string_parameters(self) -> dict[str, str]:
         parameters = self.get("parameters") or []
         return {x["name"]: x["value"] for x in parameters}
 
+    @property
+    def resolved_query_string_parameters(self) -> dict[str, list[str]]:
+        """
+        Override the base implementation to prevent splitting parameter values by commas.
+
+        For Bedrock Agent events, parameters are already properly structured and should not
+        be split by commas as they might contain commas as part of their actual values
+        (e.g., SQL queries).
+        """
+        # Return each parameter value as a single-item list without splitting by commas
+        parameters = self.get("parameters") or []
+        return {x["name"]: [x["value"]] for x in parameters}
+
     @property
     def resolved_headers_field(self) -> dict[str, Any]:
         return {}

aws_lambda_powertools/utilities/data_classes/common.py

@@ -168,7 +168,7 @@ def query_string_parameters(self) -> dict[str, str]:
     def multi_value_query_string_parameters(self) -> dict[str, list[str]]:
         return self.get("multiValueQueryStringParameters") or {}
 
-    @cached_property
+    @property
     def resolved_query_string_parameters(self) -> dict[str, list[str]]:
         """
         This property determines the appropriate query string parameter to be used

tests/functional/event_handler/_pydantic/test_bedrock_agent.py

@@ -5,7 +5,7 @@
 from typing_extensions import Annotated
 
 from aws_lambda_powertools.event_handler import BedrockAgentResolver, BedrockResponse, Response, content_types
-from aws_lambda_powertools.event_handler.openapi.params import Body
+from aws_lambda_powertools.event_handler.openapi.params import Body, Query
 from aws_lambda_powertools.utilities.data_classes import BedrockAgentEvent
 from tests.functional.utils import load_event
 
@@ -343,3 +343,43 @@ def handler() -> Optional[Dict]:
 
     # THEN the OpenAPI schema must contain the "x-requireConfirmation" extension at the operation level
     assert schema["paths"]["/"]["get"]["x-requireConfirmation"] == "ENABLED"
+
+
+def test_bedrock_agent_with_comma_parameters():
+    # GIVEN a Bedrock Agent resolver
+    app = BedrockAgentResolver()
+
+    @app.post("/sql-query", description="Run a SQL query")
+    def run_sql_query(query: Annotated[str, Query()]):
+        return {"result": query}
+
+    # WHEN calling the event handler with a parameter containing commas
+    event = {
+        "actionGroup": "TestActionGroup",
+        "messageVersion": "1.0",
+        "sessionId": "12345678912345",
+        "sessionAttributes": {},
+        "promptSessionAttributes": {},
+        "inputText": "Run a SQL query",
+        "agent": {
+            "alias": "TEST",
+            "name": "test",
+            "version": "1",
+            "id": "test123",
+        },
+        "httpMethod": "POST",
+        "apiPath": "/sql-query",
+        "parameters": [
+            {
+                "name": "query",
+                "type": "string",
+                "value": "SELECT a.source_name, b.thing FROM table",
+            },
+        ],
+    }
+
+    result = app(event, {})
+
+    # THEN the parameter with commas should be correctly passed to the handler
+    body = json.loads(result["response"]["responseBody"]["application/json"]["body"])
+    assert body["result"] == "SELECT a.source_name, b.thing FROM table"