Fix mypy type annotations in transformation.py

- Add explicit type annotation for transformed_message dict
- Resolves mypy errors while maintaining functionality
- All tests continue to pass
This commit is contained in:
Tim Elfrink
2025-09-18 08:34:05 +02:00
parent c234b13275
commit 7538fc0def
@@ -92,7 +92,7 @@ class BedrockCountTokensConfig(BaseAWSLLM):
system_messages.append({"text": message.get("content", "")})
else:
# Transform message content to Bedrock format
transformed_message = {"role": message.get("role"), "content": []}
transformed_message: Dict[str, Any] = {"role": message.get("role"), "content": []}
# Handle content - ensure it's in the correct array format
content = message.get("content", "")