From df8fc829f93e930df4dd686eab6cadcb654ecadb Mon Sep 17 00:00:00 2001 From: alekszievr <44192193+alekszievr@users.noreply.github.com> Date: Wed, 4 Dec 2024 16:25:05 +0100 Subject: [PATCH] check if repo path exists before starting the pipeline (#252) Co-authored-by: Rita Aleksziev --- .../repo_processor/get_repo_file_dependencies.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/cognee/tasks/repo_processor/get_repo_file_dependencies.py b/cognee/tasks/repo_processor/get_repo_file_dependencies.py index 352cbe19b..87669b7d8 100644 --- a/cognee/tasks/repo_processor/get_repo_file_dependencies.py +++ b/cognee/tasks/repo_processor/get_repo_file_dependencies.py @@ -1,12 +1,14 @@ +import asyncio import os +from concurrent.futures import ProcessPoolExecutor from typing import AsyncGenerator from uuid import NAMESPACE_OID, uuid5 + import aiofiles -from concurrent.futures import ProcessPoolExecutor -import asyncio from cognee.shared.CodeGraphEntities import CodeFile, Repository -from cognee.tasks.repo_processor.get_local_dependencies import get_local_script_dependencies +from cognee.tasks.repo_processor.get_local_dependencies import \ + get_local_script_dependencies async def get_py_path_and_source(file_path): @@ -54,6 +56,10 @@ def run_coroutine(coroutine_func, *args, **kwargs): async def get_repo_file_dependencies(repo_path: str) -> AsyncGenerator[list, None]: """Generate a dependency graph for Python files in the given repository path.""" + + if not os.path.exists(repo_path): + raise FileNotFoundError(f"Repository path {repo_path} does not exist.") + py_files_dict = await get_py_files_dict(repo_path) repo = Repository(