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

Added a few fixes and refactored the base app

Browse source
This commit is contained in:
Vasilije 2023-10-25 20:54:43 +02:00
parent 9936248942
commit 5f6f2974b1
1 changed files with 5 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
level_3/rag_test_manager.py
View file

@ -405,24 +405,20 @@ async def start_test(data, test_set=None, user_id=None, params=None, job_id=None
results = []
if retriever_type:
if retriever_type == "llm_context":
test_id, result = await run_test(test=None, loader_settings=loader_settings, metadata=metadata,
retriever_type=retriever_type) # No params for this case
results.append([result, "No params"])
for param in test_params:
test_id, result = await run_test(param, loader_settings, metadata, retriever_type=retriever_type) # Add the params to the result
results.append([result, param])
elif retriever_type == "single_document_context":
for param in test_params:
test_id, result = await run_test(param, loader_settings, metadata, retriever_type=retriever_type) # Add the params to the result
results.append([result, param])
for b, r in results:
print("Here is the result", r)
for result_list in b:
for result in result_list:
print("here is the result", result)
# for result_list in results[0]:
# for result in result_list:
print("Here is one result", result)
await add_entity(session, TestOutput(
id=test_id,