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

Merge branch 'COG-949' of https://github.com/topoteretes/cognee into COG-949

Browse source
This commit is contained in:
hajdul88 2025-01-09 15:25:19 +01:00
commit 7e103496a1
1 changed files with 1 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
evals/eval_swe_bench.py
View file

@ -48,14 +48,7 @@ async def generate_patch_with_cognee(instance, llm_client, search_type=SearchTyp
retrieved_edges = await brute_force_triplet_search( retrieved_edges = await brute_force_triplet_search(
problem_statement, problem_statement,
top_k=3, top_k=3,
collections=[ collections=["code_summary_text"],
"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)