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

Merge branch 'main' into feature/cog-717-create-edge-embeddings-in-vector-databases

Browse source
This commit is contained in:
Vasilije 2024-12-04 16:33:08 +01:00 • committed by GitHub
commit 080143cdad
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -1,12 +1,14 @@
import asyncio
import os import os
from concurrent.futures import ProcessPoolExecutor
from typing import AsyncGenerator from typing import AsyncGenerator
from uuid import NAMESPACE_OID, uuid5 from uuid import NAMESPACE_OID, uuid5
import aiofiles import aiofiles
from concurrent.futures import ProcessPoolExecutor
import asyncio
from cognee.shared.CodeGraphEntities import CodeFile, Repository 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): 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]: async def get_repo_file_dependencies(repo_path: str) -> AsyncGenerator[list, None]:
"""Generate a dependency graph for Python files in the given repository path.""" """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) py_files_dict = await get_py_files_dict(repo_path)
repo = Repository( repo = Repository(