move chat to demo chat

This commit is contained in:
2025-10-22 14:14:51 +08:00
parent 0d91fc7b79
commit 759fa85dd1
2 changed files with 27 additions and 31 deletions

23
scripts/demo_chat.py Normal file
View File

@@ -0,0 +1,23 @@
import tyro
import asyncio
from lang_agent.pipeline import Pipeline, PipelineConfig
from lang_agent.config import load_tyro_conf
def main(conf:PipelineConfig):
if conf.config_f is not None:
conf = load_tyro_conf(conf.config_f)
pipeline:Pipeline = conf.setup()
while True:
user_input = input("请讲:")
if user_input.lower() == "exit":
break
response = pipeline.chat(user_input, as_stream=True)
print(f"回答: {response}")
if __name__ == "__main__":
main(tyro.cli(PipelineConfig))