Skip to content

Commit b2b6a8e

Browse files
committed
fix: update gemini model close #574
Signed-off-by: yihong0618 <[email protected]>
1 parent 355ebc6 commit b2b6a8e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

xiaogpt/bot/gemini_bot.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ def __init__(
4040
from google.auth import api_key
4141

4242
credentials = api_key.Credentials(gemini_key)
43-
if len(gemini_api_domain) > 0:
43+
if gemini_api_domain:
4444
print("Use custom gemini_api_domain: " + gemini_api_domain)
4545
credentials._universe_domain = gemini_api_domain
4646
genai.configure(
@@ -56,7 +56,7 @@ def __init__(
5656

5757
self.history = []
5858
model = genai.GenerativeModel(
59-
model_name=gemini_model or "gemini-pro",
59+
model_name=gemini_model or "gemini-2.0-flash-lite",
6060
generation_config=generation_config,
6161
safety_settings=safety_settings,
6262
)

0 commit comments

Comments
 (0)