mirror of
https://github.com/debpalash/VoiceStudio.git
synced 2026-10-02 01:26:35 +08:00
Palash Debnath
0834c8be28
Consolidate community engine, workflow, dictation, and setup fixes with the Electron composer, sidebar, and voice UI. Fix review findings in engine residency, remote exports, backup cleanup, bounded compressed-audio decoding, and reference preprocessing. Preserve contributor credits in CHANGELOG.md and leave the app version unchanged. Supersedes #2325, #2338, #2368, #2377, #2379, #2380, #2383, #2384, #2387, #2390, #2391, #2392, #2393, #2395, #2400, #2401, #2402, #2409, #2410, and #2412.
133 lines
4.4 KiB
Python
133 lines
4.4 KiB
Python
"""MCP describe_voice / design_voice: agents can design a voice profile.
|
|
|
|
The tools wrap ``POST /design/describe`` and ``POST /profiles`` (kind=design).
|
|
A stub app stands in for the backend: the real describe router (pure CPU, no
|
|
model) plus a ``/profiles`` endpoint that records the form it receives.
|
|
"""
|
|
import asyncio
|
|
import json
|
|
import os
|
|
import sys
|
|
|
|
os.environ.setdefault("OMNIVOICE_MODEL", "test")
|
|
os.environ.setdefault("OMNIVOICE_DISABLE_FILE_LOG", "1")
|
|
|
|
import pytest
|
|
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "backend"))
|
|
|
|
pytest.importorskip("mcp")
|
|
|
|
|
|
def _stub_app(received: list):
|
|
from fastapi import FastAPI, Form, Request, Response
|
|
|
|
from api.routers.describe_voice import router as describe_router
|
|
|
|
app = FastAPI()
|
|
app.include_router(describe_router)
|
|
|
|
@app.post("/profiles")
|
|
async def create_profile(
|
|
name: str = Form(...),
|
|
kind: str = Form("clone"),
|
|
vd_states: str = Form(""),
|
|
instruct: str = Form(""),
|
|
language: str = Form("Auto"),
|
|
):
|
|
received.append({
|
|
"name": name, "kind": kind, "vd_states": vd_states,
|
|
"instruct": instruct, "language": language,
|
|
})
|
|
return {"id": "abcd1234", "name": name, "kind": kind}
|
|
|
|
@app.post("/generate")
|
|
async def generate(request: Request):
|
|
form = await request.form()
|
|
received.append({"generate": dict(form)})
|
|
return Response(b"RIFF", media_type="audio/wav", headers={"X-Audio-Id": "a1"})
|
|
|
|
return app
|
|
|
|
|
|
def _call(mcp, tool: str, args: dict) -> dict:
|
|
result = asyncio.run(mcp.call_tool(tool, args))
|
|
content = result[0] if isinstance(result, tuple) else result
|
|
return json.loads(content[0].text)
|
|
|
|
|
|
@pytest.fixture
|
|
def server(monkeypatch):
|
|
monkeypatch.delenv("OMNIVOICE_API_URL", raising=False)
|
|
from mcp_server import create_mcp_server
|
|
|
|
received: list = []
|
|
return create_mcp_server(app=_stub_app(received)), received
|
|
|
|
|
|
def test_design_tools_are_registered(server):
|
|
mcp, _ = server
|
|
names = {t.name for t in asyncio.run(mcp.list_tools())}
|
|
assert {"describe_voice", "design_voice"} <= names
|
|
|
|
|
|
def test_describe_voice_previews_without_saving(server):
|
|
mcp, received = server
|
|
out = _call(mcp, "describe_voice", {"description": "an old man with a deep voice"})
|
|
assert out["attrs"]["Gender"] == "male"
|
|
assert out["attrs"]["Age"] == "elderly"
|
|
assert received == []
|
|
|
|
|
|
def test_design_voice_saves_a_design_profile(server):
|
|
mcp, received = server
|
|
out = _call(mcp, "design_voice", {
|
|
"name": "Innkeeper",
|
|
"description": "an elderly woman, british accent, gravelly",
|
|
})
|
|
assert out["profile_id"] == "abcd1234"
|
|
assert out["kind"] == "design"
|
|
assert "gravelly" in " ".join(out["unmatched"])
|
|
|
|
(form,) = received
|
|
assert form["kind"] == "design"
|
|
states = json.loads(form["vd_states"])
|
|
assert states["Gender"] == "female"
|
|
assert states["EnglishAccent"] == "british accent"
|
|
assert "female" in form["instruct"]
|
|
|
|
|
|
def test_design_voice_refuses_a_description_with_no_attributes(server):
|
|
mcp, received = server
|
|
out = _call(mcp, "design_voice", {"name": "Nobody", "description": "gravelly"})
|
|
assert "error" in out
|
|
assert out["unmatched"] == ["gravelly"]
|
|
assert received == []
|
|
|
|
|
|
def test_design_voice_save_waits_as_long_as_a_generation(server, monkeypatch):
|
|
# The save renders the identity sample through the GPU queue; the default
|
|
# 120 s POST deadline could give up while the profile still gets saved.
|
|
import mcp_server
|
|
|
|
kinds = []
|
|
real = mcp_server._post_timeout_s
|
|
monkeypatch.setattr(mcp_server, "_post_timeout_s",
|
|
lambda kind="", text="": kinds.append(kind) or real(kind, text))
|
|
mcp, _ = server
|
|
_call(mcp, "design_voice", {"name": "Guard", "description": "a young man"})
|
|
assert "generate" in kinds
|
|
|
|
|
|
@pytest.mark.parametrize("language, expected", [(None, None), ("Auto", "Auto"), ("fr", "fr")])
|
|
def test_generate_speech_leaves_language_to_the_profile_unless_given(server, language, expected):
|
|
# An explicit "Auto" makes the backend ignore a profile's saved language
|
|
# (#533), so an omitted language must stay omitted.
|
|
mcp, received = server
|
|
args = {"text": "Hello there.", "profile_id": "abcd1234"}
|
|
if language is not None:
|
|
args["language"] = language
|
|
asyncio.run(mcp.call_tool("generate_speech", args))
|
|
(entry,) = [r["generate"] for r in received if "generate" in r]
|
|
assert entry.get("language") == expected
|