xiaogpt 1.51__py3-none-any.whl → 1.60__py3-none-any.whl

xiaogpt/bot/__init__.py

@@ -5,6 +5,7 @@ from xiaogpt.bot.chatgptapi_bot import ChatGPTBot
 from xiaogpt.bot.gpt3_bot import GPT3Bot
 from xiaogpt.bot.newbing_bot import NewBingBot
 from xiaogpt.bot.glm_bot import GLMBot
+from xiaogpt.bot.bard_bot import BardBot
 from xiaogpt.config import Config
 
 BOTS: dict[str, type[BaseBot]] = {
@@ -12,6 +13,7 @@ BOTS: dict[str, type[BaseBot]] = {
     "newbing": NewBingBot,
     "chatgptapi": ChatGPTBot,
     "glm": GLMBot,
+    "bard": BardBot,
 }
 
 
@@ -22,4 +24,4 @@ def get_bot(config: Config) -> BaseBot:
         raise ValueError(f"Unsupported bot {config.bot}, must be one of {list(BOTS)}")
 
 
-__all__ = ["GPT3Bot", "ChatGPTBot", "NewBingBot", "GLMBot", "get_bot"]
+__all__ = ["GPT3Bot", "ChatGPTBot", "NewBingBot", "GLMBot", "BardBot", "get_bot"]

xiaogpt/bot/bard_bot.py

@@ -0,0 +1,32 @@
+"""ChatGLM bot"""
+from __future__ import annotations
+from typing import Any
+
+from bardapi import BardAsync
+from rich import print
+
+from xiaogpt.bot.base_bot import BaseBot
+
+
+class BardBot(BaseBot):
+    def __init__(
+        self,
+        bard_token: str,
+    ) -> None:
+        self._bot = BardAsync(token=bard_token)
+        self.history = []
+
+    @classmethod
+    def from_config(cls, config):
+        return cls(bard_token=config.bard_token)
+
+    async def ask(self, query, **options):
+        try:
+            r = await self._bot.get_answer(query)
+        except Exception as e:
+            print(str(e))
+        print(r["content"])
+        return r["content"]
+
+    def ask_stream(self, query: str, **options: Any):
+        raise Exception("Bard do not support stream")

xiaogpt/bot/chatgptapi_bot.py

@@ -48,7 +48,11 @@ class ChatGPTBot(BaseBot):
             ms.append({"role": "assistant", "content": h[1]})
         ms.append({"role": "user", "content": f"{query}"})
         kwargs = {**self.default_options, **options}
-        completion = await openai.ChatCompletion.acreate(messages=ms, **kwargs)
+        try:
+            completion = await openai.ChatCompletion.acreate(messages=ms, **kwargs)
+        except Exception as e:
+            print(str(e))
+            return ""
         message = (
             completion["choices"][0]
             .get("message")
@@ -72,9 +76,13 @@ class ChatGPTBot(BaseBot):
         kwargs = {"model": "gpt-3.5-turbo", **options}
         if openai.api_type == "azure":
             kwargs["deployment_id"] = self.deployment_id
-        completion = await openai.ChatCompletion.acreate(
-            messages=ms, stream=True, **kwargs
-        )
+        try:
+            completion = await openai.ChatCompletion.acreate(
+                messages=ms, stream=True, **kwargs
+            )
+        except Exception as e:
+            print(str(e))
+            return
 
         async def text_gen():
             async for event in completion:

xiaogpt/bot/glm_bot.py

@@ -30,7 +30,11 @@ class GLMBot(BaseBot):
         kwargs = {**self.default_options, **options}
         kwargs["prompt"] = ms
         ms.append({"role": "user", "content": f"{query}"})
-        r = zhipuai.model_api.sse_invoke(**kwargs)
+        try:
+            r = zhipuai.model_api.sse_invoke(**kwargs)
+        except Exception as e:
+            print(str(e))
+            return
         message = ""
         for i in r.events():
             message += str(i.data)
@@ -43,4 +47,4 @@ class GLMBot(BaseBot):
         return message
 
     def ask_stream(self, query: str, **options: Any):
-        pass
+        raise Exception("GLM do not support stream")

xiaogpt/bot/gpt3_bot.py

@@ -29,7 +29,11 @@ class GPT3Bot(BaseBot):
             "top_p": 1,
             **options,
         }
-        completion = await openai.Completion.acreate(**data)
+        try:
+            completion = await openai.Completion.acreate(**data)
+        except Exception as e:
+            print(str(e))
+            return ""
         print(completion["choices"][0]["text"])
         return completion["choices"][0]["text"]
 
@@ -43,7 +47,11 @@ class GPT3Bot(BaseBot):
             "stream": True,
             **options,
         }
-        completion = await openai.Completion.acreate(**data)
+        try:
+            completion = await openai.Completion.acreate(**data)
+        except Exception as e:
+            print(str(e))
+            return
 
         async def text_gen():
             async for event in completion:

xiaogpt/bot/newbing_bot.py

@@ -40,13 +40,20 @@ class NewBingBot(BaseBot):
     async def ask(self, query, **options):
         kwargs = {"conversation_style": ConversationStyle.balanced, **options}
         completion = await self._bot.ask(prompt=query, **kwargs)
-        text = self.clean_text(completion["item"]["messages"][1]["text"])
+        try:
+            text = self.clean_text(completion["item"]["messages"][1]["text"])
+        except Exception as e:
+            print(str(e))
+            return
         print(text)
         return text
 
     async def ask_stream(self, query, **options):
         kwargs = {"conversation_style": ConversationStyle.balanced, **options}
-        completion = self._bot.ask_stream(prompt=query, **kwargs)
+        try:
+            completion = self._bot.ask_stream(prompt=query, **kwargs)
+        except Exception as e:
+            return
 
         async def text_gen():
             current = ""

xiaogpt/cli.py

@@ -32,6 +32,11 @@ def main():
         dest="glm_key",
         help="chatglm api key",
     )
+    parser.add_argument(
+        "--bard_token",
+        dest="bard_token",
+        help="google bard token see https://github.com/dsdanielpark/Bard-API",
+    )
     parser.add_argument(
         "--proxy",
         dest="proxy",
@@ -106,6 +111,13 @@ def main():
         const="glm",
         help="if use chatglm",
     )
+    group.add_argument(
+        "--use_bard",
+        dest="bot",
+        action="store_const",
+        const="bard",
+        help="if use bard",
+    )
     parser.add_argument(
         "--bing_cookie_path",
         dest="bing_cookie_path",
@@ -115,7 +127,7 @@ def main():
         "--bot",
         dest="bot",
         help="bot type",
-        choices=["gpt3", "chatgptapi", "newbing", "glm"],
+        choices=["gpt3", "chatgptapi", "newbing", "glm", "bard"],
     )
     parser.add_argument(
         "--config",
@@ -144,7 +156,7 @@ def main():
     )
 
     options = parser.parse_args()
-    if options.bot == "glm" and options.stream:
+    if options.bot in ["glm", "bard"] and options.stream:
         raise Exception("For now ChatGLM do not support stream")
     config = Config.from_options(options)
 

xiaogpt/config.py

@@ -61,6 +61,7 @@ class Config:
     password: str = os.getenv("MI_PASS", "")
     openai_key: str = os.getenv("OPENAI_API_KEY", "")
     glm_key: str = os.getenv("CHATGLM_KEY", "")
+    bard_token: str = os.getenv("BARD_TOKEN", "")
     proxy: str | None = None
     mi_did: str = os.getenv("MI_DID", "")
     keyword: Iterable[str] = KEY_WORD
@@ -139,5 +140,9 @@ class Config:
                         key, value = "bot", "gpt3"
                     elif key == "use_newbing":
                         key, value = "bot", "newbing"
+                    elif key == "use_glm":
+                        key, value = "bot", "glm"
+                    elif key == "use_bard":
+                        key, value = "bot", "bard"
                     result[key] = value
         return result

xiaogpt/xiaogpt.py

@@ -421,6 +421,7 @@ class MiGPT:
         )
 
     async def run_forever(self):
+        ask_name = self.config.bot.upper()
         async with ClientSession() as session:
             await self.init_all_data(session)
             task = asyncio.create_task(self.poll_latest_ask())
@@ -470,7 +471,7 @@ class MiGPT:
                 else:
                     # waiting for xiaoai speaker done
                     await asyncio.sleep(8)
-                await self.do_tts("正在问GPT请耐心等待")
+                await self.do_tts(f"正在问{ask_name}请耐心等待")
                 try:
                     print(
                         "以下是小爱的回答: ",
@@ -478,7 +479,7 @@ class MiGPT:
                     )
                 except IndexError:
                     print("小爱没回")
-                print("以下是GPT的回答: ", end="")
+                print(f"以下是 {ask_name} 的回答: ", end="")
                 try:
                     if not self.config.enable_edge_tts:
                         async for message in self.ask_gpt(query):
@@ -492,7 +493,7 @@ class MiGPT:
                         await self.edge_tts(self.ask_gpt(query), tts_lang)
                     print("回答完毕")
                 except Exception as e:
-                    print(f"GPT回答出错 {str(e)}")
+                    print(f"{ask_name} 回答出错 {str(e)}")
                 if self.in_conversation:
                     print(f"继续对话, 或用`{self.config.end_conversation}`结束对话")
                     await self.wakeup_xiaoai()

{xiaogpt-1.51.dist-info → xiaogpt-1.60.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.1
 Name: xiaogpt
-Version: 1.51
-Summary: Play ChatGPT with xiaomi AI speaker
+Version: 1.60
+Summary: Play ChatGPT or other LLM with xiaomi AI speaker
 Author-Email: yihong0618 <zouzou0208@gmail.com>
 License: MIT
 Classifier: License :: OSI Approved :: MIT License
@@ -14,6 +14,7 @@ Requires-Dist: openai
 Requires-Dist: aiohttp
 Requires-Dist: rich
 Requires-Dist: zhipuai
+Requires-Dist: bardapi
 Requires-Dist: edge-tts>=6.1.3
 Requires-Dist: EdgeGPT==0.1.26
 Description-Content-Type: text/markdown
@@ -115,6 +116,8 @@ python3 xiaogpt.py --hardware LX06  --mute_xiaoai --use_gpt3
 
 # 如果你想使用 ChatGLM api
 python3 xiaogpt.py --hardware LX06  --mute_xiaoai --use_glm --glm_key ${glm_key}
+# 如果你想使用 google 的 bard
+python3 xiaogpt.py --hardware LX06  --mute_xiaoai --use_bard --bard_token ${bard_token}
 ```
 
 ## config.json
@@ -147,6 +150,7 @@ python3 xiaogpt.py
 
 具体参数作用请参考 [Open AI API 文档](https://platform.openai.com/docs/api-reference/chat/create)。
 ChatGLM [文档](http://open.bigmodel.cn/doc/api#chatglm_130b)
+Bard-API [参考](https://github.com/dsdanielpark/Bard-API)
 ## 配置项说明
 
 | 参数                  | 说明                                              | 默认值                              |
@@ -156,6 +160,7 @@ ChatGLM [文档](http://open.bigmodel.cn/doc/api#chatglm_130b)
 | password              | 小爱账户密码                                      |                                     |
 | openai_key            | openai的apikey                                    |                                     |
 | glm_key               | chatglm 的 apikey                                    |                                     |
+| bard_token            | bard 的 token 参考 [Bard-API](https://github.com/dsdanielpark/Bard-API)                                 |                                     |
 | cookie                | 小爱账户cookie （如果用上面密码登录可以不填）     |                                     |
 | mi_did                | 设备did                                           |                                     |
 | use_command           | 使用 MI command 与小爱交互                        | `false`                             |

xiaogpt-1.60.dist-info/RECORD

@@ -0,0 +1,18 @@
+xiaogpt-1.60.dist-info/METADATA,sha256=R29SaPwyUtu65J7LcGCchYGca_RmwwTzoZLuEuCrDO4,12819
+xiaogpt-1.60.dist-info/WHEEL,sha256=0QD4Fi8spmPDA9YIWsctxI6ukFf44Zgk0L9495lbqs0,90
+xiaogpt-1.60.dist-info/entry_points.txt,sha256=zLFzA72qQ_eWBepdA2YU5vdXFqORH8wXhv2Ox1vnYP8,46
+xiaogpt-1.60.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=CrxzcQXn4DZVw83l8qXWQOZxSFbo7570mjyTcZTi6Jk,780
+xiaogpt/bot/bard_bot.py,sha256=UKgfLLKwte8XBZFl7H7bAQRZM17qbI8Sa2AUrtzBrwM,774
+xiaogpt/bot/base_bot.py,sha256=kzbUw1eQrgFqkE_eZZrWmcbPdwFrQezJBLm7cJAM8qw,554
+xiaogpt/bot/chatgptapi_bot.py,sha256=cq0CwXD6ynNw7hB-bYubP0Nsz7pT49JYKp0_UjRoIps,3477
+xiaogpt/bot/glm_bot.py,sha256=aenSgy6w5GKCAlJ9vrmNUliHOgVqen8OHsDpcLYWzj8,1385
+xiaogpt/bot/gpt3_bot.py,sha256=Ju_c9izW09XbCCSgZTp-3Rjz1zBMeBGj19zH0UwqGFw,1827
+xiaogpt/bot/newbing_bot.py,sha256=fRRrNnnw6xVFlkp4aQ37XRrOCnZmc2tCyMmW7BFgrOE,2156
+xiaogpt/cli.py,sha256=DXSH-5YEejQdPkrfbrKoykCglkvyUzvVN7-sRHk2kEI,4152
+xiaogpt/config.py,sha256=6_GMtOx2_CMHfZwlC0q7JKm9QI1oF04P1G3zpB1Fhks,5604
+xiaogpt/utils.py,sha256=kJ8nOBkilEoo1i2uUSp9tkUWV4pWrd5_EzEicZXzNjY,2071
+xiaogpt/xiaogpt.py,sha256=KeRQqXL6DV1_nG0tw0sxCiZFQwXQ-mpULvnia43JboU,18769
+xiaogpt-1.60.dist-info/RECORD,,

{xiaogpt-1.51.dist-info → xiaogpt-1.60.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: pdm-backend (2.1.2)
+Generator: pdm-backend (2.1.4)
 Root-Is-Purelib: true
 Tag: py3-none-any

xiaogpt-1.51.dist-info/RECORD

@@ -1,17 +0,0 @@
-xiaogpt-1.51.dist-info/METADATA,sha256=7OOFi4_MDsknspD3Dd5w7kZ2lhFzc551vS9Saa-60Jw,12425
-xiaogpt-1.51.dist-info/WHEEL,sha256=g58cvAsF5_9d0VXJuk7F0rhl6nAjUc5P9vx880WkZtI,90
-xiaogpt-1.51.dist-info/entry_points.txt,sha256=zLFzA72qQ_eWBepdA2YU5vdXFqORH8wXhv2Ox1vnYP8,46
-xiaogpt-1.51.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=A_kVOZY4tiW8V7V_MP85Ij8PyjS3KaUuCzRKMGeJr7Y,707
-xiaogpt/bot/base_bot.py,sha256=kzbUw1eQrgFqkE_eZZrWmcbPdwFrQezJBLm7cJAM8qw,554
-xiaogpt/bot/chatgptapi_bot.py,sha256=XcePzPpTo5ofVJtT6IBjSRjj0aoppdJ8Zs18spAc76o,3280
-xiaogpt/bot/glm_bot.py,sha256=u6vXAvsveHgqGcrDlY2TUnTWxMc8gI1VoXU4x2zlkaE,1252
-xiaogpt/bot/gpt3_bot.py,sha256=c_37SMs_karXH3lKuyeyNZA5iGyj5uzErVdbuGA1GtA,1638
-xiaogpt/bot/newbing_bot.py,sha256=OBTZYWEqd3sfPl_0BrQxXZC79l8-l-2ikYFIhc4plqI,1996
-xiaogpt/cli.py,sha256=qcVekNdynS-sxn5WqZGDcUweYd12JD4rUZm6FAIrxaA,3821
-xiaogpt/config.py,sha256=CioOI4UMHmFcvsEoGIZI41Gc1Z-oI0Yh3i3wc-1C93o,5366
-xiaogpt/utils.py,sha256=kJ8nOBkilEoo1i2uUSp9tkUWV4pWrd5_EzEicZXzNjY,2071
-xiaogpt/xiaogpt.py,sha256=_IQAxihPqjhVAhAuYcUFf1QlPZaGELqus0D9ioEUeeI,18700
-xiaogpt-1.51.dist-info/RECORD,,