Skip to content

Commit 207fabb

Browse files
committed
refactor: clean up file filter formatting in load_documents function
1 parent 639d6ab commit 207fabb

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

embeddings.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22
import os
33
import logging
44
from dotenv import load_dotenv
5-
from langchain_community.document_loaders import DirectoryLoader
65
from langchain_community.document_loaders import GithubFileLoader
76
from langchain.text_splitter import RecursiveCharacterTextSplitter
87
from langchain.schema import Document
@@ -42,7 +41,8 @@ def load_documents(directory: str) -> list[Document]:
4241
loader = GithubFileLoader(
4342
repo="rndmcodeguy20/model-builder",
4443
branch="master",
45-
file_filter=lambda file_path: file_path.endswith(".md") and file_path.startswith("docs/"),
44+
file_filter=lambda file_path: file_path.endswith(".md")
45+
and file_path.startswith("docs/"),
4646
access_token=GITHUB_TOKEN,
4747
)
4848
documents = loader.load()

0 commit comments

Comments
 (0)