From e37a31183cee4f71a805154c9158cfdc669ad621 Mon Sep 17 00:00:00 2001
From: pliny <133052465+elder-plinius@users.noreply.github.com>
Date: Tue, 2 Apr 2024 21:10:50 -0700
Subject: [PATCH] Update gemini_chat.py

---
 gemini_chat.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/gemini_chat.py b/gemini_chat.py
index 2e30c4c..ead9905 100644
--- a/gemini_chat.py
+++ b/gemini_chat.py
@@ -15,12 +15,12 @@ class GeminiChat:
         genai.configure(api_key=api_key)
         self.model = genai.GenerativeModel('gemini-pro')
 
-    def send_message(self, prompt, temp=0.2, top_p=1.0):
+    def send_message(self, prompt, temperature=0.9, top_p=1.0):
         """
         Send a message to the Gemini model with specified temperature and top probability.
         """
         try:
-            generation_config = genai.types.GenerationConfig(temperature=temp, top_p=top_p)
+            generation_config = genai.types.GenerationConfig(temperature=temperature, top_p=top_p)
             response = self.model.generate_content(prompt, generation_config=generation_config)
             return response.text.strip()
         except Exception as e: