fix: api_key attribute

arc53 · Jan 18, 2025 · 4a331db · 4a331db
1 parent 904b0bf
commit 4a331db
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 = []