From 4a331db5fc110d3ff64e6a9966b49f51e2789414 Mon Sep 17 00:00:00 2001 From: Siddhant Rai Date: Sat, 18 Jan 2025 20:00:51 +0530 Subject: [PATCH] fix: api_key attribute --- application/llm/google_ai.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/application/llm/google_ai.py b/application/llm/google_ai.py index c14c89b99..e9188987c 100644 --- a/application/llm/google_ai.py +++ b/application/llm/google_ai.py @@ -8,7 +8,7 @@ class GoogleLLM(BaseLLM): def __init__(self, api_key=None, user_api_key=None, *args, **kwargs): super().__init__(*args, **kwargs) - self.client = genai.Client(api_key="AIzaSyDmbZX65qlQKXcvfMBkJV2KwH82_0yIMlE") + self.client = genai.Client(api_key=api_key) def _clean_messages_google(self, messages): cleaned_messages = []