🐛 fix refine api #116

- add tests

modules/core/models/TTSModel.py

@@ -38,7 +38,7 @@ def interrupt(self, context: TTSPipelineContext = None) -> None:
         pass
 
     def encode(self, text: str) -> list[int]:
-        return [ord(char) for char in text.split("")]
+        return [ord(char) for char in text]
 
     def decode(self, ids: list[int]) -> str:
         return "".join([chr(id) for id in ids])

modules/core/tools/SentenceSplitter.py

@@ -6,7 +6,7 @@
 
 
 def char_tokenizer(text: str):
-    return [ord(char) for char in text.split("")]
+    return [ord(char) for char in text]
 
 
 # 解析文本 并根据停止符号分割成句子

tests/api/test_prompt.py

@@ -0,0 +1,26 @@
+from pytest import mark
+
+
+@mark.refine_api
+def test_refine_success(client):
+    response = client.post(
+        "/v1/prompt/refine",
+        json={
+            "text": "你好，世界",
+            "prompt": "[oral_2][laugh_0][break_6]",
+            "seed": -1,
+            "top_P": 0.7,
+            "top_K": 20,
+            "temperature": 0.7,
+            "repetition_penalty": 1,
+            "max_new_token": 384,
+            "spliter_threshold": 300,
+            "normalize": True,
+        },
+    )
+    data = response.json()
+    assert response.status_code == 200
+    assert "message" in data
+    assert data["message"] == "ok"
+    assert "data" in data
+    assert isinstance(data["data"], str)