From 7e8dd6bba8f3fb18c2cd682c6c5117e11303da09 Mon Sep 17 00:00:00 2001 From: ManishMadan2882 Date: Mon, 12 Aug 2024 01:06:21 +0530 Subject: [PATCH] fix: get api keys endpoint --- application/api/user/routes.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/application/api/user/routes.py b/application/api/user/routes.py index 84831a65..7c6e979c 100644 --- a/application/api/user/routes.py +++ b/application/api/user/routes.py @@ -426,9 +426,17 @@ def get_api_keys(): keys = api_key_collection.find({"user": user}) list_keys = [] for key in keys: - source_name = ( - db.dereference(key["source"])["name"] if isinstance(key["source"], DBRef) else key["source"].split("/")[0] - ) + if "source" in key and isinstance(key["source"],DBRef): + source = db.dereference(key["source"]) + if source is None: + continue + else: + source_name = source["name"] + elif "retriever" in key: + source_name = key["retriever"] + else: + continue + list_keys.append( { "id": str(key["_id"]),