wt002 commited on
Commit
b65c6f7
·
verified ·
1 Parent(s): 0636b11

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +6 -0
agent.py CHANGED
@@ -28,6 +28,8 @@ import requests
28
  import json
29
  from langchain_core.documents import Document
30
  from langchain_community.vectorstores import FAISS
 
 
31
  from langchain_community.embeddings import HuggingFaceEmbeddings
32
 
33
  from youtube_transcript_api import YouTubeTranscriptApi
@@ -62,6 +64,10 @@ from typing import Union
62
  from functools import reduce
63
  from youtube_transcript_api import YouTubeTranscriptApi
64
  from youtube_transcript_api._errors import TranscriptsDisabled, VideoUnavailable
 
 
 
 
65
 
66
 
67
  load_dotenv()
 
28
  import json
29
  from langchain_core.documents import Document
30
  from langchain_community.vectorstores import FAISS
31
+ from langchain.vectorstores import FAISS
32
+ from langchain.embeddings import BERTEmbeddings
33
  from langchain_community.embeddings import HuggingFaceEmbeddings
34
 
35
  from youtube_transcript_api import YouTubeTranscriptApi
 
64
  from functools import reduce
65
  from youtube_transcript_api import YouTubeTranscriptApi
66
  from youtube_transcript_api._errors import TranscriptsDisabled, VideoUnavailable
67
+ import faiss
68
+ from langchain.vectorstores import FAISS
69
+ from langchain.embeddings import BERTEmbeddings
70
+ from langchain.schema import Document
71
 
72
 
73
  load_dotenv()