xiaozhi-sdk 0.0.9__py3-none-any.whl → 0.0.10__py3-none-any.whl

xiaozhi_sdk/__init__.py

@@ -1,3 +1,3 @@
-__version__ = "0.0.9"
+__version__ = "0.0.10"
 
 from xiaozhi_sdk.core import XiaoZhiWebsocket  # noqa

xiaozhi_sdk/cli.py

@@ -12,6 +12,34 @@ import sounddevice as sd
 from xiaozhi_sdk import XiaoZhiWebsocket
 from xiaozhi_sdk.config import INPUT_SERVER_AUDIO_SAMPLE_RATE
 
+# 定义自定义日志级别
+INFO1 = 21
+INFO2 = 22
+INFO3 = 23
+
+# 添加自定义日志级别到logging模块
+logging.addLevelName(INFO1, "INFO1")
+logging.addLevelName(INFO2, "INFO2")
+logging.addLevelName(INFO3, "INFO3")
+
+# 为logger添加自定义方法
+def info1(self, message, *args, **kwargs):
+    if self.isEnabledFor(INFO1):
+        self._log(INFO1, message, args, **kwargs)
+
+def info2(self, message, *args, **kwargs):
+    if self.isEnabledFor(INFO2):
+        self._log(INFO2, message, args, **kwargs)
+
+def info3(self, message, *args, **kwargs):
+    if self.isEnabledFor(INFO3):
+        self._log(INFO3, message, args, **kwargs)
+
+# 将自定义方法添加到Logger类
+logging.Logger.info1 = info1
+logging.Logger.info2 = info2
+logging.Logger.info3 = info3
+
 # 配置彩色logging
 handler = colorlog.StreamHandler()
 handler.setFormatter(
@@ -19,8 +47,11 @@ handler.setFormatter(
         "%(log_color)s%(asctime)s - %(name)s - %(levelname)s - %(message)s",
         datefmt="%Y-%m-%d %H:%M:%S",
         log_colors={
-            "DEBUG": "green",
-            "INFO": "white",
+            "DEBUG": "white",
+            "INFO": "green",
+            "INFO1": "green",
+            "INFO2": "cyan",
+            "INFO3": "blue",
             "WARNING": "yellow",
             "ERROR": "red",
             "CRITICAL": "red,bg_white",
@@ -41,7 +72,15 @@ is_end = False
 async def handle_message(message):
     """处理接收到的消息"""
     global is_end
-    logger.info("message received: %s", message)
+    if message["type"] == "stt":
+        logger.info1("message received: %s", message)
+    elif message["type"] == "tts":
+        logger.info2("message received: %s", message)
+    elif message["type"] == "llm":
+        logger.info3("message received: %s", message)
+    else:
+        logger.info("message received: %s", message)
+
     if message["type"] == "websocket" and message["state"] == "close":
         is_end = True
 

xiaozhi_sdk/iot.py

@@ -12,7 +12,7 @@ from xiaozhi_sdk.config import OTA_URL
 # 常量定义
 BOARD_TYPE = "xiaozhi-sdk-box"
 USER_AGENT = "xiaozhi-sdk/{}".format(__version__)
-BOARD_NAME = "xiaozhi-sdk-{}".format(__version__)
+BOARD_NAME = "xiaozhi-sdk"
 
 logger = logging.getLogger("xiaozhi_sdk")
 
@@ -60,6 +60,9 @@ class OtaDevice:
 
         async with aiohttp.ClientSession() as session:
             async with session.post(self.ota_url + "/", headers=headers, data=json.dumps(payload)) as response:
+                if response.status != 200:
+                    err_text = await response.text()
+                    raise Exception(err_text)
                 response.raise_for_status()
                 return await response.json()
 

{xiaozhi_sdk-0.0.9.dist-info → xiaozhi_sdk-0.0.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xiaozhi-sdk
-Version: 0.0.9
+Version: 0.0.10
 Summary: 一个用于连接和控制小智智能设备的Python SDK，支持实时音频通信、MCP工具集成和设备管理功能。
 Author-email: dairoot <623815825@qq.com>
 License-Expression: MIT

{xiaozhi_sdk-0.0.9.dist-info → xiaozhi_sdk-0.0.10.dist-info}/RECORD

@@ -8,19 +8,19 @@ file/opus/linux-x64-libopus.so,sha256=FmXJqkxLpDzNFOHYkmOzmsp1hP0eIS5b6x_XfOs-IQ
 file/opus/macos-arm64-libopus.dylib,sha256=H7wXwkrGwb-hesMMZGFxWb0Ri1Y4m5GWiKsd8CfOhE8,357584
 file/opus/macos-x64-libopus.dylib,sha256=MqyL_OjwSACF4Xs_-KrGbcScy4IEprr5Rlkk3ddZye8,550856
 file/opus/windows-opus.dll,sha256=kLfhioMvbJhOgNMAldpWk3DCZqC5Xd70LRbHnACvAnw,463360
-xiaozhi_sdk/__init__.py,sha256=-gx5Vs8YIZLaWKtMWumKSkCiYu2Nu8Z7rFjFnYbkDhQ,77
+xiaozhi_sdk/__init__.py,sha256=4-2cLsMk3Bq03wsMvSLb-DzFWsOqPFUKgKf4oeheg2Y,78
 xiaozhi_sdk/__main__.py,sha256=i0ZJdHUqAKg9vwZrK_w0TJkzdotTYTK8aUeSPcJc1ks,210
-xiaozhi_sdk/cli.py,sha256=LFZE4f4A44EIPXLsowxQ4z_EskTh62MLusPwwFVtXr8,4490
+xiaozhi_sdk/cli.py,sha256=9BiFlGy_2wFES_4dpan8iFEnwuwTyg4vjxwWitLpO5U,5621
 xiaozhi_sdk/config.py,sha256=h4mpMeBf2vT9qYAqCCbGVGmMemkgk98pcXP2Rh4TEFc,89
 xiaozhi_sdk/core.py,sha256=564SefCBus6qNRApWqwI113aIN1p4eYpci1mLeMExIs,10007
-xiaozhi_sdk/iot.py,sha256=IO3SfiuQxucYl_917BCNCwIAv1dajCJI-IFTWwHnSDE,2580
+xiaozhi_sdk/iot.py,sha256=VVAheynp1iV4GCaoPywQWpKtlyoACDLswH8yfV_JZgI,2699
 xiaozhi_sdk/mcp.py,sha256=Pbr3JEDqEnWsPwLHKzRdrDYZMg6krYZoz74lXUSh9iI,5544
 xiaozhi_sdk/opus.py,sha256=r3nnYg0ZKAJTreb_3nKgfHJh06MJiMvnNMPO1SWdoMM,2224
 xiaozhi_sdk/utils/__init__.py,sha256=XKSHWoFmuSkpwaIr308HybRzfFIXoT1Fd-eUKo_im6Y,1705
 xiaozhi_sdk/utils/mcp_data.py,sha256=r_GLjFpvUKeV8IhlkT7Pfb4c9bDO9q4k3f4iPZBBcKI,3892
 xiaozhi_sdk/utils/mcp_tool.py,sha256=imwehfUlENjelYmGbGYgb6C82-ijs53XCxrtCpqrJps,3152
-xiaozhi_sdk-0.0.9.dist-info/licenses/LICENSE,sha256=Vwgps1iODKl43cAtME_0dawTjAzNW-O2BWiN5BHggww,1085
-xiaozhi_sdk-0.0.9.dist-info/METADATA,sha256=kqk96A4tgE9-b4rn-eyFKLLoXYqky1v3tlnVscck14k,2091
-xiaozhi_sdk-0.0.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-xiaozhi_sdk-0.0.9.dist-info/top_level.txt,sha256=nBpue4hU5Ykm5CtYPsAdxSa_yqbtZsIT_gF_EkBaJPM,12
-xiaozhi_sdk-0.0.9.dist-info/RECORD,,
+xiaozhi_sdk-0.0.10.dist-info/licenses/LICENSE,sha256=Vwgps1iODKl43cAtME_0dawTjAzNW-O2BWiN5BHggww,1085
+xiaozhi_sdk-0.0.10.dist-info/METADATA,sha256=uaSfv6gxD9hEvaKFWPSLasJVecL-BTk8Dhv-7lniDzk,2092
+xiaozhi_sdk-0.0.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+xiaozhi_sdk-0.0.10.dist-info/top_level.txt,sha256=nBpue4hU5Ykm5CtYPsAdxSa_yqbtZsIT_gF_EkBaJPM,12
+xiaozhi_sdk-0.0.10.dist-info/RECORD,,