[Improvements] Upgrade langchain-openai package and other improvements (#1372)

This commit is contained in:
Deshraj Yadav
2024-05-21 23:42:50 -07:00
committed by GitHub
parent 9ba445e623
commit 9544c69c55
9 changed files with 86 additions and 232 deletions

View File

@@ -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),
},
}
]