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

Adjust SWE-bench script to code graph pipeline call

Browse source
This commit is contained in:
Rita Aleksziev 2025-01-09 14:27:37 +01:00
parent 626bc76f5c
commit d7b2186300
1 changed files with 11 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
evals/eval_swe_bench.py
View file

@ -34,9 +34,8 @@ def check_install_package(package_name):
async def generate_patch_with_cognee(instance, llm_client, search_type=SearchType.CHUNKS): async def generate_patch_with_cognee(instance, llm_client, search_type=SearchType.CHUNKS):
repo_path = download_github_repo(instance, "../RAW_GIT_REPOS") repo_path = download_github_repo(instance, "../RAW_GIT_REPOS")
pipeline = await run_code_graph_pipeline(repo_path)
async for result in pipeline: async for result in run_code_graph_pipeline(repo_path, include_docs=True):
print(result) print(result)
print("Here we have the repo under the repo_path") print("Here we have the repo under the repo_path")
@ -47,7 +46,16 @@ async def generate_patch_with_cognee(instance, llm_client, search_type=SearchTyp
instructions = read_query_prompt("patch_gen_kg_instructions.txt") instructions = read_query_prompt("patch_gen_kg_instructions.txt")
retrieved_edges = await brute_force_triplet_search( retrieved_edges = await brute_force_triplet_search(
problem_statement, top_k=3, collections=["data_point_source_code", "data_point_text"] problem_statement,
top_k=3,
collections=[
"code_summary_text",
"data_point_name",
"document_chunk_text",
"entity_name",
"entity_type_name",
"sourcecodechunk_source_code",
],
) )
retrieved_edges_str = retrieved_edges_to_string(retrieved_edges) retrieved_edges_str = retrieved_edges_to_string(retrieved_edges)