35 lines
1.0 KiB
Python
35 lines
1.0 KiB
Python
# from src import Plugins
|
|
# from src import ExecutePipeline
|
|
# from src import EchoProcessor
|
|
import os
|
|
from viztracer import VizTracer
|
|
|
|
from dotenv import load_dotenv
|
|
load_dotenv()
|
|
|
|
from pathlib import Path
|
|
from src.dashscope.tongyiqianwen import CAITongYiQianWen as CAI
|
|
from src.dashscope.sambert import TTSSambert as TTS
|
|
from src.pyaudio.sounds_play_engine import SoundsPlayEngine as sounds
|
|
from aiostream import stream, pipe
|
|
|
|
async def main():
|
|
|
|
cai = CAI(api_key=os.environ.get("DASH_SCOPE_API_KEY", ""), use_stream_api=True)
|
|
tts = TTS(api_key=os.environ.get("DASH_SCOPE_API_KEY", ""))
|
|
exe_input = '给我一个100字的文章'
|
|
|
|
with VizTracer(log_async=True):
|
|
s = (
|
|
stream.iterate(await cai.execute_stream(cai.prepare_input(exe_input)))
|
|
| pipe.map(cai.transform_chunks)
|
|
| pipe.print()
|
|
| pipe.flatmap(tts.execute_stream)
|
|
| pipe.map(sounds().execute)
|
|
)
|
|
await s
|
|
|
|
if __name__ == '__main__':
|
|
import asyncio
|
|
asyncio.run(main())
|