From dd8c992dad5983a9d536da3baa8edaaba8b8b170 Mon Sep 17 00:00:00 2001 From: Ori Kotek Date: Wed, 6 Sep 2023 18:27:31 +0300 Subject: [PATCH] Add ability to work with litellm debugger --- pr_agent/algo/ai_handler.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pr_agent/algo/ai_handler.py b/pr_agent/algo/ai_handler.py index fcc5f04c..ac73a325 100644 --- a/pr_agent/algo/ai_handler.py +++ b/pr_agent/algo/ai_handler.py @@ -48,6 +48,8 @@ class AiHandler: litellm.replicate_key = get_settings().replicate.key if get_settings().get("HUGGINGFACE.KEY", None): litellm.huggingface_key = get_settings().huggingface.key + if get_settings().get("LITELLM.USER_EMAIL"): + litellm.email = get_settings().get("LITELLM.USER_EMAIL", None) except AttributeError as e: raise ValueError("OpenAI key is required") from e