From e58a5355941e83b69654f24b7e4e49c6ceb245c7 Mon Sep 17 00:00:00 2001 From: Trung Dinh Date: Fri, 17 Jan 2025 11:43:06 +0700 Subject: [PATCH] Inject deepseek key to DEEPSEEK_API_KEY environment variable --- pr_agent/algo/ai_handlers/litellm_ai_handler.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/pr_agent/algo/ai_handlers/litellm_ai_handler.py b/pr_agent/algo/ai_handlers/litellm_ai_handler.py index c8a752f8..453973f6 100644 --- a/pr_agent/algo/ai_handlers/litellm_ai_handler.py +++ b/pr_agent/algo/ai_handlers/litellm_ai_handler.py @@ -90,6 +90,10 @@ class LiteLLMAIHandler(BaseAiHandler): if get_settings().get("GOOGLE_AI_STUDIO.GEMINI_API_KEY", None): os.environ["GEMINI_API_KEY"] = get_settings().google_ai_studio.gemini_api_key + # Support deepseek models + if get_settings().get("DEEPSEEK.KEY", None): + os.environ['DEEPSEEK_API_KEY'] = get_settings().get("DEEPSEEK.KEY") + def prepare_logs(self, response, system, user, resp, finish_reason): response_log = response.dict().copy() response_log['system'] = system