Skip to content

Commit 8cc211a

Browse files
author
jer
committed
chore(a2a): prioritize extensions
1 parent a0ffb8b commit 8cc211a

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

src/strands/multiagent/a2a/executor.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -222,9 +222,14 @@ def _get_file_format_from_mime_type(self, mime_type: str | None, file_type: str)
222222
extensions = mimetypes.guess_all_extensions(mime_type)
223223

224224
if extensions:
225-
# Get the most common extension (first one) and remove the dot
226-
extension = extensions[0][1:] # Remove the leading dot
225+
# Prioritize extensions that have explicit format mappings
226+
for ext in extensions:
227+
extension = ext[1:] # Remove the leading dot
228+
if extension in self.FORMAT_MAPPINGS:
229+
return self.FORMAT_MAPPINGS[extension]
227230

231+
# Fall back to first extension if no mapping found
232+
extension = extensions[0][1:] # Remove the leading dot
228233
return self.FORMAT_MAPPINGS.get(extension, extension)
229234

230235
# Fallback to defaults for unknown MIME types

0 commit comments

Comments
 (0)