xiaogpt 3.5__py3-none-any.whl → 3.10__py3-none-any.whl

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/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(

{xiaogpt-3.5.dist-info → xiaogpt-3.10.dist-info}/METADATA

@@ -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.dist-info → xiaogpt-3.10.dist-info}/RECORD

@@ -1,14 +1,14 @@
-xiaogpt-3.5.dist-info/METADATA,sha256=fqymHqLmsFnNBE2G332D8ZbdmEFJ0gI3q08OXn6M3j0,31516
-xiaogpt-3.5.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-xiaogpt-3.5.dist-info/entry_points.txt,sha256=q4WRS7kS4kQ5kZX57Fq40VrhCi74NZcyRPRX4JP2veo,61
-xiaogpt-3.5.dist-info/licenses/LICENSE,sha256=XdClh516MvlnOf9749JZHCxSB7y6_fyXcWmLDz6IkZY,1063
+xiaogpt-3.10.dist-info/METADATA,sha256=7RFe3n9WrY2Zc9V3d_lnBWlJCZLfQBD7PtuLtm-WSNU,31517
+xiaogpt-3.10.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+xiaogpt-3.10.dist-info/entry_points.txt,sha256=q4WRS7kS4kQ5kZX57Fq40VrhCi74NZcyRPRX4JP2veo,61
+xiaogpt-3.10.dist-info/licenses/LICENSE,sha256=XdClh516MvlnOf9749JZHCxSB7y6_fyXcWmLDz6IkZY,1063
 xiaogpt/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 xiaogpt/__main__.py,sha256=MSmt_5Xg84uHqzTN38JwgseJK8rsJn_11A8WD99VtEo,61
 xiaogpt/bot/__init__.py,sha256=BDGvj1JuWVw47qfREWGKnSXeiFg6DVJJAz2rHVryqmc,1160
 xiaogpt/bot/base_bot.py,sha256=oKn6LLFHXol4hKrSrjnxknrOqrcGICtT_GPPYRNxpkw,1467
 xiaogpt/bot/chatgptapi_bot.py,sha256=TpAL_Rarz535PcbJhYQUNYurZ_zoXGMAvBl0H8fiqI8,3654
 xiaogpt/bot/doubao_bot.py,sha256=UufQmYcPbwTLTYDZUQwRy4Hg24vgPEa3hdeZWMWw9YM,2773
-xiaogpt/bot/gemini_bot.py,sha256=vX-fTWyPwdB4N0HDQ9uIRCB4KvV-YgBqXjkrqgg4WHs,2516
+xiaogpt/bot/gemini_bot.py,sha256=UrLMhPi79PJ2fFiIDmizdBPmOZ3-glpcqcSsU7TVcKg,2626
 xiaogpt/bot/glm_bot.py,sha256=A90QbbU2UU7pr0o4_5-fxF4Gkz7gFB0LADIGx7N5aO4,1713
 xiaogpt/bot/langchain_bot.py,sha256=4Uz5iOYzA2ongCklS-9zBse2fw-7kEE_9wITH7wdVCc,1944
 xiaogpt/bot/llama_bot.py,sha256=HRR_ycuC6DY5MQTKauXEayQ0o_JKk9t-ea3mblscm8E,708
@@ -16,7 +16,7 @@ xiaogpt/bot/moonshot_bot.py,sha256=PrVRBskZx-U0lH_3RVe89QJa7WKHYqhpft0089pYQz0,8
 xiaogpt/bot/qwen_bot.py,sha256=rFCOz5uiUsuhePjPozdCecNv5HGiUTNEhHYNw3Exexs,3590
 xiaogpt/bot/yi_bot.py,sha256=D7JEIh8KPVMvlOLaEVr9ahvyMaJLGToHP_gWU3RoYPc,784
 xiaogpt/cli.py,sha256=iUPQwiZqIpzu1HoinneijBEfVKV7F_f_YMgqPPZM5lA,5932
-xiaogpt/config.py,sha256=GpK3RXJI_B5YBKMxErAh9wUH1FWqVbQppZM4f7B-EoI,7043
+xiaogpt/config.py,sha256=0ygn73UkWhLSIayb-8sS3qV2tajhWRlE3AaM-cFO898,7118
 xiaogpt/langchain/callbacks.py,sha256=yR9AXQt9OHVYBWC47Q1I_BUT4Xg9iM44vnW2vv0BLpE,2616
 xiaogpt/langchain/chain.py,sha256=z0cqRlL0ElWnf31ByxZBN7AKOT-svXQDt5_NDft_nYc,1495
 xiaogpt/langchain/examples/email/mail_box.py,sha256=xauqrjE4-G4XPQnokUPE-MZgAaHQ_VrUDLlbfYTdCoo,6372
@@ -28,4 +28,4 @@ xiaogpt/tts/live.py,sha256=7-NpuOU7E6GNht1PHjAA8_miXdFVDk8C7aL-sY1xqPE,3299
 xiaogpt/tts/mi.py,sha256=1MzCB27DBohPQ_4Xz4W_FV9p-chJFDavOHB89NviLcM,1095
 xiaogpt/utils.py,sha256=YYmRDNtccxqB9gyN_xhKZwgL1_PNYEp7So_-jt2tiVg,2668
 xiaogpt/xiaogpt.py,sha256=zHZFijx-tH6vi3roB1uluZUhqLUnhmYNUR367GXVixg,16696
-xiaogpt-3.5.dist-info/RECORD,,
+xiaogpt-3.10.dist-info/RECORD,,