Skip to content

Commit 7538fc0

Browse files
committed
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
1 parent c234b13 commit 7538fc0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

litellm/llms/bedrock/count_tokens/transformation.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ def _transform_to_converse_format(
9292
system_messages.append({"text": message.get("content", "")})
9393
else:
9494
# Transform message content to Bedrock format
95-
transformed_message = {"role": message.get("role"), "content": []}
95+
transformed_message: Dict[str, Any] = {"role": message.get("role"), "content": []}
9696

9797
# Handle content - ensure it's in the correct array format
9898
content = message.get("content", "")

0 commit comments

Comments
 (0)