diff --git a/app/storage/mongo/configs.py b/app/storage/mongo/configs.py index e3e374a..154bd59 100644 --- a/app/storage/mongo/configs.py +++ b/app/storage/mongo/configs.py @@ -35,4 +35,5 @@ async def get(project: str, stage: str) -> list[Config]: result = [] async for item in collection.find({'stage': stage, 'project': project}): result.append(Config.model_validate(item)) + result[-1]._id = item['_id'] return result diff --git a/app/storage/mongo/experiments.py b/app/storage/mongo/experiments.py index eba4246..ed3f33f 100644 --- a/app/storage/mongo/experiments.py +++ b/app/storage/mongo/experiments.py @@ -36,4 +36,5 @@ async def get(project: str, stage: str) -> list[Experiment]: result = [] async for item in collection.find({'stage': stage, 'project': project}): result.append(Experiment.model_validate(item)) + result[-1]._id = item['_id'] return result