From a5f208f951a6bb1eb829e4153ca9db2a02f1033e Mon Sep 17 00:00:00 2001 From: harry Date: Wed, 3 Apr 2024 11:03:06 +0800 Subject: [PATCH] optimize pexels_api_keys config --- app/services/material.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/app/services/material.py b/app/services/material.py index f9c108e..29f61eb 100644 --- a/app/services/material.py +++ b/app/services/material.py @@ -17,6 +17,10 @@ if not pexels_api_keys: def round_robin_api_key(): + # if only one key is provided, return it + if isinstance(pexels_api_keys, str): + return pexels_api_keys + global requested_count requested_count += 1 return pexels_api_keys[requested_count % len(pexels_api_keys)]