From 624216aa64cc48e5a0f1db5f65d5c2dc8581f0f1 Mon Sep 17 00:00:00 2001 From: Bhav Sardana <40908961+sardanabhav@users.noreply.github.com> Date: Mon, 10 Feb 2025 19:27:58 +0530 Subject: [PATCH] community:Fix for Pydantic model validator of GoogleApiYoutubeLoader (#29694) - **Description:** Community: bugfix for pedantic model validator for GoogleApiYoutubeLoader - **Issue:** #29165, #27432 Fix is similar to #29346 --- .../langchain_community/document_loaders/youtube.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/libs/community/langchain_community/document_loaders/youtube.py b/libs/community/langchain_community/document_loaders/youtube.py index 64871eb3caf..1b99a8d2da2 100644 --- a/libs/community/langchain_community/document_loaders/youtube.py +++ b/libs/community/langchain_community/document_loaders/youtube.py @@ -392,11 +392,11 @@ class GoogleApiYoutubeLoader(BaseLoader): @model_validator(mode="before") @classmethod - def validate_channel_or_videoIds_is_set(cls, values: Dict[str, Any]) -> Any: + def validate_channel_or_videoIds_is_set(cls, values: Any) -> Any: """Validate that either folder_id or document_ids is set, but not both.""" - if not values.get("channel_name") and not values.get("video_ids"): + if not values.kwargs.get("channel_name") and not values.kwargs.get("video_ids"): raise ValueError("Must specify either channel_name or video_ids") - return values + return values.kwargs def _get_transcripe_for_video_id(self, video_id: str) -> str: from youtube_transcript_api import NoTranscriptFound, YouTubeTranscriptApi