chore/change-gemini-1.5-pro-to-1.5-flash

AureliusIvan · Jul 26, 2024 · 4e34cca · 4e34cca
1 parent 94d1b2b
commit 4e34cca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/server/domain/service/RememberService.py b/server/domain/service/RememberService.py
@@ -99,7 +99,7 @@ def __init__(self):
 
         self.memory = Memory.from_config(config)
         self.client = genai.GenerativeModel(
-            model_name="gemini-1.5-pro",
+            model_name="gemini-1.5-flash",
             generation_config=generation_config,
             safety_settings={
                 HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,