mirror of
https://github.com/hanzoai/go-cosyvoice.git
synced 2026-09-30 21:53:53 +00:00
zeekayandHanzo Dev
d6145d89ea
Hard-fork of WqyJh/go-cosyvoice into the hanzoai org so the OpenAI/audio-LLM Go stack is fully owned. Renames the module path and repoints the realtime deps to the hanzoai fork required directly (no replace directives): github.com/hanzoai/go-openai-realtime v1.0.0 github.com/hanzoai/go-openai-realtime/contrib/ws-gorilla v1.0.0 Co-authored-by: Hanzo Dev <[email protected]>
75 lines
1.8 KiB
Go
75 lines
1.8 KiB
Go
package cosyvoice_test
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hanzoai/go-cosyvoice"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestAsyncCall(t *testing.T) {
|
|
key := os.Getenv("COSYVOICE_API_KEY")
|
|
if key == "" {
|
|
t.Skip("COSYVOICE_API_KEY is not set")
|
|
}
|
|
ctx := context.Background()
|
|
client := cosyvoice.NewClient(key)
|
|
asyncSynthesizer, err := client.AsyncSynthesizer(ctx)
|
|
require.NoError(t, err)
|
|
|
|
outputCh, err := asyncSynthesizer.RunTask(ctx)
|
|
require.NoError(t, err)
|
|
err = asyncSynthesizer.SendText(ctx, "你好，世界！")
|
|
require.NoError(t, err)
|
|
err = asyncSynthesizer.SendText(ctx, "hello, world!")
|
|
require.NoError(t, err)
|
|
err = asyncSynthesizer.FinishTask(ctx)
|
|
require.NoError(t, err)
|
|
|
|
require.Greater(t, len(outputCh), 0)
|
|
for result := range outputCh {
|
|
require.NoError(t, result.Err)
|
|
require.Greater(t, len(result.Data), 0)
|
|
}
|
|
|
|
err = asyncSynthesizer.Close()
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestAsyncPing(t *testing.T) {
|
|
key := os.Getenv("COSYVOICE_API_KEY")
|
|
if key == "" {
|
|
t.Skip("COSYVOICE_API_KEY is not set")
|
|
}
|
|
ctx := context.Background()
|
|
client := cosyvoice.NewClient(key)
|
|
|
|
asyncSynthesizer, err := client.AsyncSynthesizer(ctx, cosyvoice.WithPingInterval(30))
|
|
require.NoError(t, err)
|
|
|
|
time.Sleep(61 * time.Second)
|
|
|
|
outputCh, err := asyncSynthesizer.RunTask(ctx)
|
|
require.NoError(t, err)
|
|
err = asyncSynthesizer.SendText(ctx, "你好，世界！")
|
|
require.NoError(t, err)
|
|
|
|
err = asyncSynthesizer.SendText(ctx, "hello, world!")
|
|
require.NoError(t, err)
|
|
|
|
err = asyncSynthesizer.FinishTask(ctx)
|
|
require.NoError(t, err)
|
|
|
|
require.Greater(t, len(outputCh), 0)
|
|
for result := range outputCh {
|
|
require.NoError(t, result.Err)
|
|
require.Greater(t, len(result.Data), 0)
|
|
}
|
|
|
|
err = asyncSynthesizer.Close()
|
|
require.NoError(t, err)
|
|
}
|