diff --git a/main.py b/main.py index cccd178..b38dbf4 100644 --- a/main.py +++ b/main.py @@ -129,4 +129,6 @@ async def recs(production_id: str, count: int | None = 5): print(f'time elapsed = {time.time() - time_start}') scores = [idx[0] for idx in sorted(scores, key=lambda x: x[1], reverse=True)[:count+1]] scores.remove(production_id) - return scores + return { + 'id': scores + }