From dd785b85aed38f2df1b08e9577ac52a83d540868 Mon Sep 17 00:00:00 2001 From: Vasilije <8619304+Vasilije1990@users.noreply.github.com> Date: Tue, 24 Oct 2023 16:03:55 +0200 Subject: [PATCH] Added a few fixes and refactored the base app --- level_3/models/testoutput.py | 3 ++- level_3/rag_test_manager.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/level_3/models/testoutput.py b/level_3/models/testoutput.py index 0921df7e0..4a03253f9 100644 --- a/level_3/models/testoutput.py +++ b/level_3/models/testoutput.py @@ -23,7 +23,8 @@ class TestOutput(Base): """ __tablename__ = 'test_outputs' - id = Column(String, primary_key=True) + set_id = Column(String, primary_key=True) + id = Column(String, nullable=True) user_id = Column(String, ForeignKey('users.id'), index=True) # Added user_id field test_set_id = Column(String, ForeignKey('test_sets.id'), index=True) operation_id = Column(String, ForeignKey('operations.id'), index=True) diff --git a/level_3/rag_test_manager.py b/level_3/rag_test_manager.py index 43984fb25..47f07269a 100644 --- a/level_3/rag_test_manager.py +++ b/level_3/rag_test_manager.py @@ -412,7 +412,7 @@ async def start_test(data, test_set=None, user_id=None, params=None, job_id=None 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, user_id=user_id, test_results=result['success'], test_score=str(result['score']), test_metric_name=result['metric_name'], test_query=result['query'], test_output=result['output'], test_expected_output=str(['expected_output']), test_context=result['context'][0])) + await add_entity(session, TestOutput(id=test_id ,set_id=str(uuid.uuid4()), user_id=user_id, test_results=result['success'], test_score=str(result['score']), test_metric_name=result['metric_name'], test_query=result['query'], test_output=result['output'], test_expected_output=str(['expected_output']), test_context=result['context'][0])) print(results)