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

Deleting code_graph_pipeline not working entrypoint

Browse source 
From now on eval_swe_bench contains and rung the updated version of the pipeline
This commit is contained in:
hajdul88 2024-11-28 12:19:08 +01:00
parent c094898d15
commit 72a8bc43a1
1 changed files with 0 additions and 36 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
examples/python/code_graph_pipeline.py
View file

@ -1,36 +0,0 @@
import asyncio
from cognee.modules.pipelines import Task, run_tasks
from cognee.tasks.repo_processor import (
enrich_dependency_graph,
expand_dependency_graph,
get_repo_file_dependencies,
)
from cognee.tasks.storage import add_data_points
from cognee.tasks.summarization import summarize_code
async def print_results(pipeline):
async for result in pipeline:
print(result)
if __name__ == "__main__":
'''
parser = argparse.ArgumentParser(description="Process a file path")
parser.add_argument("path", help="Path to the file")
args = parser.parse_args()
abspath = os.path.abspath(args.path or ".")
'''
abspath = '/Users/laszlohajdu/Documents/Github/RAW_GIT_REPOS/astropy__astropy-12907'
tasks = [
Task(get_repo_file_dependencies),
Task(add_data_points),
Task(enrich_dependency_graph),
Task(expand_dependency_graph),
Task(add_data_points),
# Task(summarize_code, summarization_model = SummarizedContent),
]
pipeline = run_tasks(tasks, abspath, "cognify_code_pipeline")
asyncio.run(print_results(pipeline))