xiaozhi-sdk 0.2.0__tar.gz → 0.2.1__tar.gz

{xiaozhi_sdk-0.2.0/xiaozhi_sdk.egg-info → xiaozhi_sdk-0.2.1}/PKG-INFO

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

{xiaozhi_sdk-0.2.0 → xiaozhi_sdk-0.2.1}/tests/test_xiaozhi.py

@@ -54,13 +54,13 @@ def mcp_tool_func():
     def mcp_set_volume(data) -> tuple[dict, bool]:
         return {}, False
 
-    from xiaozhi_sdk.utils.mcp_tool import take_photo, get_device_status, set_volume
+    from xiaozhi_sdk.utils.mcp_tool import take_photo, get_device_status, set_volume, search_custom_music, play_custom_music
 
     take_photo["tool_func"] = mcp_take_photo
     get_device_status["tool_func"] = mcp_get_device_status
     set_volume["tool_func"] = mcp_set_volume
 
-    return [take_photo, get_device_status, set_volume]
+    return [take_photo, get_device_status, set_volume, search_custom_music, play_custom_music]
 
 
 async def message_handler_callback(message):

{xiaozhi_sdk-0.2.0 → xiaozhi_sdk-0.2.1}/xiaozhi_sdk/__init__.py

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

{xiaozhi_sdk-0.2.0 → xiaozhi_sdk-0.2.1}/xiaozhi_sdk/mcp.py

@@ -8,7 +8,7 @@ from typing import Any, Dict
 import numpy as np
 import requests
 
-from xiaozhi_sdk.utils.mcp_tool import _get_random_music_info
+from xiaozhi_sdk.utils.tool_func import _get_random_music_info
 
 logger = logging.getLogger("xiaozhi_sdk")
 

{xiaozhi_sdk-0.2.0 → xiaozhi_sdk-0.2.1}/xiaozhi_sdk/utils/mcp_tool.py

@@ -1,96 +1,4 @@
-import io
-import random
-
-import aiohttp
-import numpy as np
-
-
-async def async_search_custom_music(data) -> tuple[dict, bool]:
-    search_url = f"https://music-api.gdstudio.xyz/api.php?types=search&name={data['music_name']}&count=100&pages=1"
-
-    # 为搜索请求设置 10 秒超时
-    timeout = aiohttp.ClientTimeout(total=10)
-    async with aiohttp.ClientSession(timeout=timeout) as session:
-        async with session.get(search_url) as response:
-            response_json = await response.json()
-
-    music_list = []
-    first_music_list = []
-    other_music_list1 = []
-    other_music_list2 = []
-    for line in response_json:
-        if data.get("author_name") and data["author_name"] in line["artist"][0]:
-            first_music_list.append(line)
-        elif data.get("author_name") and (data["author_name"] in line["artist"] or data["author_name"] in line["name"]):
-            other_music_list1.append(line)
-        else:
-            other_music_list2.append(line)
-
-    if len(first_music_list) <= 10:
-        music_list = first_music_list
-        random.shuffle(other_music_list2)
-        music_list = music_list + other_music_list1[: 20 - len(music_list)]
-        music_list = music_list + other_music_list2[: 20 - len(music_list)]
-
-    # print(data)
-    # print("找到音乐，数量：", len(first_music_list), len(music_list))
-
-    if not music_list:
-        return {}, False
-    return {"message": "已找到歌曲", "music_list": music_list}, False
-
-
-async def _get_random_music_info(id_list: list) -> dict:
-    timeout = aiohttp.ClientTimeout(total=10)
-    async with aiohttp.ClientSession(timeout=timeout) as session:
-        random.shuffle(id_list)
-
-        for music_id in id_list:
-            url = f"https://music-api.gdstudio.xyz/api.php?types=url&id={music_id}&br=128"
-            async with session.get(url) as response:
-                res_json = await response.json()
-            if res_json.get("url"):
-                break
-
-    return res_json
-
-
-async def async_mcp_play_music(data) -> tuple[list, bool]:
-    try:
-        from pydub import AudioSegment
-    except ImportError:
-        return [], True
-
-    id_list = data["id_list"]
-    res_json = await _get_random_music_info(id_list)
-
-    if not res_json:
-        return [], False
-
-    pcm_list = []
-    buffer = io.BytesIO()
-    # 为下载音乐文件设置 60 秒超时（音乐文件可能比较大）
-    download_timeout = aiohttp.ClientTimeout(total=60)
-    async with aiohttp.ClientSession(timeout=download_timeout) as session:
-        async with session.get(res_json["url"]) as resp:
-            async for chunk in resp.content.iter_chunked(1024):
-                buffer.write(chunk)
-
-    buffer.seek(0)
-    audio = AudioSegment.from_mp3(buffer)
-    audio = audio.set_frame_rate(16000).set_channels(1).set_sample_width(2)  # 2 bytes = 16 bits
-    pcm_data = audio.raw_data
-
-    chunk_size = 960 * 2
-    for i in range(0, len(pcm_data), chunk_size):
-        chunk = pcm_data[i : i + chunk_size]
-
-        if chunk:  # 确保不添加空块
-            chunk = np.frombuffer(chunk, dtype=np.int16)
-            pcm_list.extend(chunk)
-
-    return pcm_list, False
-
+from xiaozhi_sdk.utils.tool_func import async_mcp_play_music, async_search_custom_music
 
 search_custom_music = {
     "name": "search_custom_music",
@@ -164,7 +72,7 @@ set_theme = {
 
 take_photo = {
     "name": "take_photo",
-    "description": "Use this tool when the user asks you to look at something, take a picture, or solve a problem based on what is captured.\nArgs:\n`question`: A clear question or task you want to ask about the captured photo (e.g., identify objects, read text, explain content, or solve a math/logic problem).\nReturn:\n  A JSON object that provides the photo information, including answers, explanations, or problem-solving results if applicable.",
+    "description": "Have visual ability, Use this tool when the user asks you to look at something, take a picture, or solve a problem based on what is captured.\nArgs:\n`question`: A clear question or task you want to ask about the captured photo (e.g., identify objects, read text, explain content, or solve a math/logic problem).\nReturn:\n  A JSON object that provides the photo information, including answers, explanations, or problem-solving results if applicable.",
     "inputSchema": {
         "type": "object",
         "properties": {"question": {"type": "string"}},

xiaozhi_sdk-0.2.1/xiaozhi_sdk/utils/tool_func.py

@@ -0,0 +1,92 @@
+import io
+import random
+
+import aiohttp
+import numpy as np
+
+
+async def async_search_custom_music(data) -> tuple[dict, bool]:
+    search_url = f"https://music-api.gdstudio.xyz/api.php?types=search&name={data['music_name']}&count=100&pages=1"
+
+    # 为搜索请求设置 10 秒超时
+    timeout = aiohttp.ClientTimeout(total=10)
+    async with aiohttp.ClientSession(timeout=timeout) as session:
+        async with session.get(search_url) as response:
+            response_json = await response.json()
+
+    music_list = []
+    first_music_list = []
+    other_music_list1 = []
+    other_music_list2 = []
+    for line in response_json:
+        if data.get("author_name") and data["author_name"] in line["artist"][0]:
+            first_music_list.append(line)
+        elif data.get("author_name") and (data["author_name"] in line["artist"] or data["author_name"] in line["name"]):
+            other_music_list1.append(line)
+        else:
+            other_music_list2.append(line)
+
+    if len(first_music_list) <= 10:
+        music_list = first_music_list
+        random.shuffle(other_music_list2)
+        music_list = music_list + other_music_list1[: 20 - len(music_list)]
+        music_list = music_list + other_music_list2[: 20 - len(music_list)]
+
+    # print(data)
+    # print("找到音乐，数量：", len(first_music_list), len(music_list))
+
+    if not music_list:
+        return {}, False
+    return {"message": "已找到歌曲", "music_list": music_list}, False
+
+
+async def _get_random_music_info(id_list: list) -> dict:
+    timeout = aiohttp.ClientTimeout(total=10)
+    async with aiohttp.ClientSession(timeout=timeout) as session:
+        random.shuffle(id_list)
+
+        for music_id in id_list:
+            url = f"https://music-api.gdstudio.xyz/api.php?types=url&id={music_id}&br=128"
+            async with session.get(url) as response:
+                res_json = await response.json()
+            if res_json.get("url"):
+                break
+
+    return res_json
+
+
+async def async_mcp_play_music(data) -> tuple[list, bool]:
+    try:
+        from pydub import AudioSegment
+    except ImportError:
+        return [], True
+
+    id_list = data["id_list"]
+    res_json = await _get_random_music_info(id_list)
+
+    if not res_json:
+        return [], False
+
+    pcm_list = []
+    buffer = io.BytesIO()
+    # 为下载音乐文件设置 60 秒超时（音乐文件可能比较大）
+    download_timeout = aiohttp.ClientTimeout(total=60)
+    async with aiohttp.ClientSession(timeout=download_timeout) as session:
+        async with session.get(res_json["url"]) as resp:
+            async for chunk in resp.content.iter_chunked(1024):
+                buffer.write(chunk)
+
+    buffer.seek(0)
+    audio = AudioSegment.from_mp3(buffer)
+    audio = audio.set_frame_rate(16000).set_channels(1).set_sample_width(2)  # 2 bytes = 16 bits
+    pcm_data = audio.raw_data
+
+    chunk_size = 960 * 2
+    for i in range(0, len(pcm_data), chunk_size):
+        chunk = pcm_data[i : i + chunk_size]
+
+        if chunk:  # 确保不添加空块
+            chunk = np.frombuffer(chunk, dtype=np.int16)
+            pcm_list.extend(chunk)
+
+    return pcm_list, False

{xiaozhi_sdk-0.2.0 → xiaozhi_sdk-0.2.1/xiaozhi_sdk.egg-info}/PKG-INFO

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

{xiaozhi_sdk-0.2.0 → xiaozhi_sdk-0.2.1}/xiaozhi_sdk.egg-info/SOURCES.txt

@@ -40,4 +40,5 @@ xiaozhi_sdk/../file/opus/macos-arm64-libopus.dylib
 xiaozhi_sdk/../file/opus/macos-x64-libopus.dylib
 xiaozhi_sdk/../file/opus/windows-opus.dll
 xiaozhi_sdk/utils/__init__.py
-xiaozhi_sdk/utils/mcp_tool.py
+xiaozhi_sdk/utils/mcp_tool.py
+xiaozhi_sdk/utils/tool_func.py