xiaogpt 3.5__tar.gz → 3.10__tar.gz

{xiaogpt-3.5 → xiaogpt-3.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xiaogpt
-Version: 3.5
+Version: 3.10
 Summary: Play ChatGPT or other LLM with xiaomi AI speaker
 Author-Email: yihong0618 <zouzou0208@gmail.com>
 License: MIT

{xiaogpt-3.5 → xiaogpt-3.10}/pyproject.toml

@@ -31,7 +31,7 @@ dependencies = [
     "lingua-language-detector>=2.0.2; python_version < \"3.13\"",
 ]
 dynamic = []
-version = "3.5"
+version = "3.10"
 
 [project.license]
 text = "MIT"

{xiaogpt-3.5 → xiaogpt-3.10}/xiaogpt/bot/gemini_bot.py

@@ -32,13 +32,15 @@ safety_settings = [
 class GeminiBot(ChatHistoryMixin, BaseBot):
     name = "Gemini"
 
-    def __init__(self, gemini_key: str, gemini_api_domain: str) -> None:
+    def __init__(
+        self, gemini_key: str, gemini_api_domain: str, gemini_model: str
+    ) -> None:
         import google.generativeai as genai
 
         from google.auth import api_key
 
         credentials = api_key.Credentials(gemini_key)
-        if len(gemini_api_domain) > 0:
+        if gemini_api_domain:
             print("Use custom gemini_api_domain: " + gemini_api_domain)
             credentials._universe_domain = gemini_api_domain
             genai.configure(
@@ -54,7 +56,7 @@ class GeminiBot(ChatHistoryMixin, BaseBot):
 
         self.history = []
         model = genai.GenerativeModel(
-            model_name="gemini-pro",
+            model_name=gemini_model or "gemini-2.0-flash-lite",
             generation_config=generation_config,
             safety_settings=safety_settings,
         )
@@ -63,7 +65,9 @@ class GeminiBot(ChatHistoryMixin, BaseBot):
     @classmethod
     def from_config(cls, config):
         return cls(
-            gemini_key=config.gemini_key, gemini_api_domain=config.gemini_api_domain
+            gemini_key=config.gemini_key,
+            gemini_api_domain=config.gemini_api_domain,
+            gemini_model=config.gemini_model,
         )
 
     async def ask(self, query, **options):

{xiaogpt-3.5 → xiaogpt-3.10}/xiaogpt/config.py

@@ -59,6 +59,7 @@ class Config:
     llama_api_key: str = os.getenv("GROQ_API_KEY", "")  # use groq
     glm_key: str = os.getenv("CHATGLM_KEY", "")
     gemini_key: str = os.getenv("GEMINI_KEY", "")  # keep the old rule
+    gemini_model: str = os.getenv("GEMINI_MODEL", "")  # keep the old rule
     qwen_key: str = os.getenv("DASHSCOPE_API_KEY", "")  # keep the old rule
     serpapi_api_key: str = os.getenv("SERPAPI_API_KEY", "")
     gemini_api_domain: str = os.getenv(