change to printer in demo chat
This commit is contained in:
@@ -1,10 +1,11 @@
|
|||||||
import tyro
|
import tyro
|
||||||
from typing import Annotated
|
from typing import Annotated
|
||||||
import uuid
|
import uuid
|
||||||
|
from loguru import logger
|
||||||
|
|
||||||
from lang_agent.pipeline import Pipeline, PipelineConfig
|
from lang_agent.pipeline import Pipeline, PipelineConfig
|
||||||
from lang_agent.config import load_tyro_conf
|
from lang_agent.config import load_tyro_conf
|
||||||
|
from lang_agent.components.conv_store import use_printer
|
||||||
|
|
||||||
|
|
||||||
def main(
|
def main(
|
||||||
@@ -17,9 +18,11 @@ def main(
|
|||||||
conf: Pipeline configuration
|
conf: Pipeline configuration
|
||||||
stream: Enable streaming mode for chat responses
|
stream: Enable streaming mode for chat responses
|
||||||
"""
|
"""
|
||||||
|
use_printer()
|
||||||
if conf.config_f is not None:
|
if conf.config_f is not None:
|
||||||
conf = load_tyro_conf(conf.config_f)
|
conf = load_tyro_conf(conf.config_f)
|
||||||
|
|
||||||
|
logger.info(conf)
|
||||||
pipeline: Pipeline = conf.setup()
|
pipeline: Pipeline = conf.setup()
|
||||||
thread_id = str(uuid.uuid4())
|
thread_id = str(uuid.uuid4())
|
||||||
while True:
|
while True:
|
||||||
|
|||||||
Reference in New Issue
Block a user