andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-01 23:32:39 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

feat: add URL file source support for multimodal content

Browse Source
This commit is contained in:
Greyson LaLonde
2026-01-22 14:18:16 -05:00
parent 4d0b6d834c
commit a064b84ead
1 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
lib/crewai/src/crewai/files/processing/validators.py
View File

@@ -136,17 +136,15 @@ def _get_video_duration(
format_hint = _VIDEO_FORMAT_MAP.get(content_type) if content_type else None format_hint = _VIDEO_FORMAT_MAP.get(content_type) if content_type else None
try: try:
container = av.open(io.BytesIO(content), format=format_hint) # type: ignore[attr-defined] with av.open(io.BytesIO(content), format=format_hint) as container: # type: ignore[attr-defined]
try: duration: int | None = container.duration # type: ignore[union-attr]
duration = getattr(container, "duration", None)
if duration is None: if duration is None:
return None return None
return float(duration) / 1_000_000 return float(duration) / 1_000_000
finally:
container.close()
except Exception as e: except Exception as e:
logger.debug(f"Could not determine video duration: {e}") logger.debug(f"Could not determine video duration: {e}")
return None
return None
def _format_size(size_bytes: int) -> str: def _format_size(size_bytes: int) -> str:
@@ -431,7 +429,7 @@ def validate_video(
) )
if constraints.max_duration_seconds is not None: if constraints.max_duration_seconds is not None:
duration = _get_video_duration(content, file.content_type) duration = _get_video_duration(content)
if duration is not None and duration > constraints.max_duration_seconds: if duration is not None and duration > constraints.max_duration_seconds:
msg = ( msg = (
f"Video '{filename}' duration ({duration:.1f}s) exceeds " f"Video '{filename}' duration ({duration:.1f}s) exceeds "