xiaozhi-sdk 0.2.2__tar.gz → 0.2.4__tar.gz

{xiaozhi_sdk-0.2.2/xiaozhi_sdk.egg-info → xiaozhi_sdk-0.2.4}/PKG-INFO

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

{xiaozhi_sdk-0.2.2 → xiaozhi_sdk-0.2.4}/tests/test_wake_word.py

@@ -25,7 +25,9 @@ async def test_main():
     xiaozhi = XiaoZhiWebsocket(message_handler_callback, url=URL, ota_url=ota_url)
     await xiaozhi.init_connection(MAC_ADDR)
 
-    await xiaozhi.send_wake_word("你是")
+    await xiaozhi.send_wake_word("退下，拜拜不聊了")
+    await asyncio.wait_for(is_end.wait(), timeout=20.0)
+    await xiaozhi.send_wake_word("你好")
 
     await asyncio.wait_for(is_end.wait(), timeout=20.0)
     await xiaozhi.close()

{xiaozhi_sdk-0.2.2 → xiaozhi_sdk-0.2.4}/xiaozhi_sdk/__init__.py

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

{xiaozhi_sdk-0.2.2 → xiaozhi_sdk-0.2.4}/xiaozhi_sdk/core.py

@@ -120,11 +120,20 @@ class XiaoZhiWebsocket(McpTool):
             await self.websocket.send(opus_data)
         await self.send_silence_audio()
 
-    async def send_wake_word(self, wake_word: str) -> None:
+    async def send_wake_word(self, wake_word: str) -> bool:
         """发送唤醒词"""
-        await self.websocket.send(
-            json.dumps({"session_id": self.session_id, "type": "listen", "state": "detect", "text": wake_word})
-        )
+        try:
+            await self.websocket.send(
+                json.dumps({"session_id": self.session_id, "type": "listen", "state": "detect", "text": wake_word})
+            )
+            return True
+        except websockets.ConnectionClosed:
+            if self.message_handler_callback:
+                await self.message_handler_callback(
+                    {"type": "websocket", "state": "close", "source": "sdk.send_wake_word"}
+                )
+            logger.debug("[websocket] close")
+            return False
 
     async def send_silence_audio(self, duration_seconds: float = 1.2) -> None:
         """发送静音音频"""
@@ -186,7 +195,7 @@ class XiaoZhiWebsocket(McpTool):
                 await self.message_handler_callback(
                     {"type": "websocket", "state": "close", "source": "sdk.message_handler"}
                 )
-        logger.info("[websocket] close")
+            logger.debug("[websocket] close")
 
     async def set_mcp_tool(self, mcp_tool_list) -> None:
         """设置MCP工具"""
@@ -246,15 +255,16 @@ class XiaoZhiWebsocket(McpTool):
         if self.wake_word:
             await self.send_wake_word(self.wake_word)
 
-    async def send_audio(self, pcm: bytes) -> None:
+    async def send_audio(self, pcm: bytes) -> bool:
         """发送音频数据"""
         if not self.websocket:
-            return
+            return False
 
         state = self.websocket.state
         if state == websockets.protocol.State.OPEN:
             opus_data = await self.audio_opus.pcm_to_opus(pcm)
             await self.websocket.send(opus_data)
+            return True
         elif state in [websockets.protocol.State.CLOSED, websockets.protocol.State.CLOSING]:
             if self.wait_device_activated:
                 logger.debug("[websocket] Server actively disconnected, reconnecting...")
@@ -265,8 +275,10 @@ class XiaoZhiWebsocket(McpTool):
                 logger.debug("[websocket] Server actively disconnected")
 
             await asyncio.sleep(0.5)
+            return False
         else:
             await asyncio.sleep(0.1)
+            return False
 
     async def close(self) -> None:
         """关闭连接"""

{xiaozhi_sdk-0.2.2 → xiaozhi_sdk-0.2.4/xiaozhi_sdk.egg-info}/PKG-INFO

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