CalvinBackup/Dioscuri
1
0
Fork 0
mirror of https://github.com/elder-plinius/Dioscuri.git synced 2026-06-07 06:43:59 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update gemini_chat.py

Browse Source
This commit is contained in:
pliny
2024-04-02 21:10:50 -07:00
committed by GitHub
parent 40d3376b78
commit e37a31183c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
gemini_chat.py
+2 -2
View File
@@ -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: