diff --git a/embedchain/embedchain.py b/embedchain/embedchain.py index 033efc0b..1cd01ea3 100644 --- a/embedchain/embedchain.py +++ b/embedchain/embedchain.py @@ -8,7 +8,7 @@ from langchain.embeddings.openai import OpenAIEmbeddings from embedchain.loaders.youtube_video import YoutubeVideoLoader from embedchain.loaders.pdf_file import PdfFileLoader from embedchain.loaders.web_page import WebPageLoader -from embedchain.loaders_local.qna_pair import QnaPairLoader +from embedchain.loaders.local_qna_pair import LocalQnaPairLoader from embedchain.chunkers.youtube_video import YoutubeVideoChunker from embedchain.chunkers.pdf_file import PdfFileChunker from embedchain.chunkers.web_page import WebPageChunker @@ -49,7 +49,7 @@ class EmbedChain: 'youtube_video': YoutubeVideoLoader(), 'pdf_file': PdfFileLoader(), 'web_page': WebPageLoader(), - 'qna_pair': QnaPairLoader() + 'qna_pair': LocalQnaPairLoader() } if data_type in loaders: return loaders[data_type] diff --git a/embedchain/loaders_local/qna_pair.py b/embedchain/loaders/local_qna_pair.py similarity index 91% rename from embedchain/loaders_local/qna_pair.py rename to embedchain/loaders/local_qna_pair.py index de3479cc..76151d98 100644 --- a/embedchain/loaders_local/qna_pair.py +++ b/embedchain/loaders/local_qna_pair.py @@ -1,4 +1,4 @@ -class QnaPairLoader: +class LocalQnaPairLoader: def load_data(self, content): question, answer = content diff --git a/embedchain/loaders_local/__init__.py b/embedchain/loaders_local/__init__.py deleted file mode 100644 index e69de29b..00000000