gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update cognee/tasks/repo_processor/get_repo_file_dependencies.py

Browse source 
Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
This commit is contained in:
Fardeen Malik 2025-08-18 22:04:47 +05:30 • committed by GitHub
parent 6de749b39a
commit fdb0c8292a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 11 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
cognee/tasks/repo_processor/get_repo_file_dependencies.py
View file

@ -53,14 +53,20 @@ async def get_source_code_files(repo_path, language_config: dict[str, list[str]]
lang = _get_language_from_extension(file, language_config)
if lang is None:
continue
#Exclude common test files and virtual/env folders
# Exclude tests and common build/venv directories
excluded_dirs = {
".venv", "venv", "env", ".env", "site-packages",
"node_modules", "dist", "build", ".git",
"tests", "test",
}
root_parts = set(os.path.normpath(root).split(os.sep))
base_name, _ext = os.path.splitext(file)
if (
file.startswith("test_")
or file.endswith("_test")
base_name.startswith("test_")
or base_name.endswith("_test") # catches Go's *_test.go and similar
or ".test." in file
or ".spec." in file
or any(x in root for x in (".venv", "venv", "env", ".env", "site-packages"))
or any(x in root for x in ("node_modules", "dist", "build", ".git"))
or (excluded_dirs & root_parts)
):
continue
file_path = os.path.abspath(os.path.join(root, file))