move chat to demo chat
This commit is contained in:
23
scripts/demo_chat.py
Normal file
23
scripts/demo_chat.py
Normal 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))
|
||||
Reference in New Issue
Block a user