From a3ee718ab5d99ea532ad7e8e32e7648e38526aa1 Mon Sep 17 00:00:00 2001
From: hzg0601 <hzg0601@163.com>
Date: Mon, 12 Jun 2023 12:57:41 +0800
Subject: [PATCH] temp

---
 cli.py                | 15 +++++++++++----
 models/loader/args.py |  2 +-
 2 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/cli.py b/cli.py
index cff822b..f97e9bc 100644
--- a/cli.py
+++ b/cli.py
@@ -1,6 +1,7 @@
 import click
 
 from api import api_start as api_start
+from cli_demo import main as cli_start
 from configs.model_config import llm_model_dict, embedding_model_dict
 
 
@@ -54,13 +55,19 @@ def start_api(ip, port):
 
 
 @start.command(name="cli", context_settings=dict(help_option_names=['-h', '--help']))
-def start_cli():
-    import cli_demo
-    cli_demo.main()
+@click.option('-i', '--info', default="start client", show_default=True, type=str)
+def start_cli(info):
+    print(info)
+
+    from models.loader.args import parser
+    cli_start()
+    
 
 
 @start.command(name="webui", context_settings=dict(help_option_names=['-h', '--help']))
-def start_webui():
+@click.option('-i', '--info', default="start client", show_default=True, type=str)
+def start_webui(info):
+    print(info)
     import webui
 
 
diff --git a/models/loader/args.py b/models/loader/args.py
index 8e05d20..b15ad5e 100644
--- a/models/loader/args.py
+++ b/models/loader/args.py
@@ -31,7 +31,7 @@ def dir_path(string):
     return s
 
 
-parser = argparse.ArgumentParser(prog='langchina-ChatGLM',
+parser = argparse.ArgumentParser(prog='langchain-ChatGLM',
                                  description='About langchain-ChatGLM, local knowledge based ChatGLM with langchain ｜ '
                                              '基于本地知识库的 ChatGLM 问答')