[Improvements] Upgrade langchain-openai package and other improvements (#1372)
This commit is contained in:
@@ -1,4 +1,5 @@
|
||||
import hashlib
|
||||
import json
|
||||
from unittest.mock import MagicMock, Mock, patch
|
||||
|
||||
import pytest
|
||||
@@ -22,7 +23,11 @@ def test_load_data(youtube_video_loader):
|
||||
)
|
||||
]
|
||||
|
||||
with patch("embedchain.loaders.youtube_video.YoutubeLoader.from_youtube_url", return_value=mock_loader):
|
||||
mock_transcript = [{"text": "sample text", "start": 0.0, "duration": 5.0}]
|
||||
|
||||
with patch("embedchain.loaders.youtube_video.YoutubeLoader.from_youtube_url", return_value=mock_loader), patch(
|
||||
"embedchain.loaders.youtube_video.YouTubeTranscriptApi.get_transcript", return_value=mock_transcript
|
||||
):
|
||||
result = youtube_video_loader.load_data(video_url)
|
||||
|
||||
expected_doc_id = hashlib.sha256((mock_page_content + video_url).encode()).hexdigest()
|
||||
@@ -32,7 +37,11 @@ def test_load_data(youtube_video_loader):
|
||||
expected_data = [
|
||||
{
|
||||
"content": "This is a YouTube video content.",
|
||||
"meta_data": {"url": video_url, "title": "Test Video"},
|
||||
"meta_data": {
|
||||
"url": video_url,
|
||||
"title": "Test Video",
|
||||
"transcript": json.dumps(mock_transcript, ensure_ascii=True),
|
||||
},
|
||||
}
|
||||
]
|
||||
|
||||
|
||||
Reference in New Issue
Block a user