Spaces:
Running
Running
File size: 1,336 Bytes
dd3763f 5750465 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
from db.crud import ingest, read, update, delete
from db.schema import Feedback, Response, ModelRatings
from datetime import datetime
def test():
# Sample data for ModelRatings
model_ratings = ModelRatings(
query_v_ratings={'v1': 5, 'v2': 4},
query_p0_ratings={'p0_1': 3, 'p0_2': 4},
query_p1_ratings={'p1_1': 2, 'p1_2': 5}
)
# Sample data for Response
response = Response(
config_id='config_123',
model_ratings={'model_1': model_ratings},
comment='This is a sample response.',
timestamp='2025-02-01T18:13:48'
)
# Sample data for Feedback
feedback = Feedback(
id=1,
user_id='user_123',
time_stamp=datetime(2025, 2, 1, 18, 13, 48),
responses=[response]
)
ingest(feedback)
print(read("user_123"))
# Read (Fetch)
feedback_data = read("ashmi")
if feedback_data:
print(feedback_data)
# Update (Modify)
# updated_feedback = Feedback(
# id=1,
# user_id="user123",
# time_stamp=datetime.now(),
# responses=[
# {"q_id": "q1", "ans": 4}, # Updated answer
# {"q_id": "q2", "ans": 3}
# ]
# )
# update(1, updated_feedback)
#
# # Delete (Remove)
# delete(1)
if __name__ == "__main__":
test()
|