From 18bb282fbc7fb3c7e770641bd0b64fa38af7dd92 Mon Sep 17 00:00:00 2001 From: Rita Aleksziev Date: Thu, 9 Jan 2025 14:27:37 +0100 Subject: [PATCH] Adjust SWE-bench script to code graph pipeline call --- evals/eval_swe_bench.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/evals/eval_swe_bench.py b/evals/eval_swe_bench.py index 789c95ab4..20e005751 100644 --- a/evals/eval_swe_bench.py +++ b/evals/eval_swe_bench.py @@ -34,9 +34,8 @@ def check_install_package(package_name): async def generate_patch_with_cognee(instance, llm_client, search_type=SearchType.CHUNKS): 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("Here we have the repo under the repo_path") @@ -47,7 +46,9 @@ async def generate_patch_with_cognee(instance, llm_client, search_type=SearchTyp instructions = read_query_prompt("patch_gen_kg_instructions.txt") 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"], ) retrieved_edges_str = retrieved_edges_to_string(retrieved_edges)