feat(ChatKnowledge): Add custom text separators and refactor log configuration (#636)

### ChatKnowledge
- New custom text separators
- Modfy relations output format
- Read `KNOWLEDGE_CHAT_SHOW_RELATIONS` from environment
`KNOWLEDGE_CHAT_SHOW_RELATIONS`

### More knowledge command line capabilities
- Delete your knowledge space or document in space
- List knowledge space

### Refactor log configuration
- Get loggger with `logging.getLogger(__name__)`
- Setup logging with `setup_logging`
- Read logger directory from environment `DBGPT_LOG_DIR`

Close #636

.env.template

@@ -55,6 +55,8 @@ EMBEDDING_MODEL=text2vec
 #EMBEDDING_MODEL=bge-large-zh
 KNOWLEDGE_CHUNK_SIZE=500
 KNOWLEDGE_SEARCH_TOP_SIZE=5
+# Control whether to display the source document of knowledge on the front end.
+KNOWLEDGE_CHAT_SHOW_RELATIONS=False
 ## EMBEDDING_TOKENIZER   - Tokenizer to use for chunking large inputs
 ## EMBEDDING_TOKEN_LIMIT - Chunk size limit for large inputs
 # EMBEDDING_MODEL=all-MiniLM-L6-v2
@@ -155,3 +157,5 @@ SUMMARY_CONFIG=FAST
 #*******************************************************************#
 # FATAL, ERROR, WARNING, WARNING, INFO, DEBUG, NOTSET
 DBGPT_LOG_LEVEL=INFO
+# LOG dir, default: ./logs
+#DBGPT_LOG_DIR=

docs/getting_started/application/kbqa/kbqa.md

@@ -92,3 +92,206 @@ Prompt Argument
 #### WEAVIATE
 * WEAVIATE_URL=https://kt-region-m8hcy0wc.weaviate.network
 ```
+
+## KBQA command line
+
+### Load your local documents to DB-GPT
+
+```bash
+dbgpt knowledge load --space_name my_kbqa_space --local_doc_path ./pilot/datasets --vector_store_type Chroma
+```
+
+- `--space_name`: Your knowledge space name, default: `default`
+- `--local_doc_path`: Your document directory or document file path, default: `./pilot/datasets`
+- `--vector_store_type`: Vector store type, default: `Chroma`
+
+**View the `dbgpt knowledge load --help`help**
+
+```
+dbgpt knowledge load --help
+```
+
+Here you can see the parameters:
+
+```
+Usage: dbgpt knowledge load [OPTIONS]
+
+  Load your local knowledge to DB-GPT
+
+Options:
+  --space_name TEXT         Your knowledge space name  [default: default]
+  --vector_store_type TEXT  Vector store type.  [default: Chroma]
+  --local_doc_path TEXT     Your document directory or document file path.
+                            [default: ./pilot/datasets]
+  --skip_wrong_doc          Skip wrong document.
+  --overwrite               Overwrite existing document(they has same name).
+  --max_workers INTEGER     The maximum number of threads that can be used to
+                            upload document.
+  --pre_separator TEXT      Preseparator, this separator is used for pre-
+                            splitting before the document is actually split by
+                            the text splitter. Preseparator are not included
+                            in the vectorized text.
+  --separator TEXT          This is the document separator. Currently, only
+                            one separator is supported.
+  --chunk_size INTEGER      Maximum size of chunks to split.
+  --chunk_overlap INTEGER   Overlap in characters between chunks.
+  --help                    Show this message and exit.
+```
+
+### List knowledge space
+
+#### List knowledge space
+
+```
+dbgpt knowledge list
+```
+
+Output should look something like the following:
+```
++------------------------------------------------------------------+
+|                       All knowledge spaces                       |
++----------+-------------+-------------+-------------+-------------+
+| Space ID |  Space Name | Vector Type |    Owner    | Description |
++----------+-------------+-------------+-------------+-------------+
+|    6     |      n1     |    Chroma   |    DB-GPT   |  DB-GPT cli |
+|    5     |  default_2  |    Chroma   |    DB-GPT   |  DB-GPT cli |
+|    4     |  default_1  |    Chroma   |    DB-GPT   |  DB-GPT cli |
+|    3     |   default   |    Chroma   |    DB-GPT   |  DB-GPT cli |
++----------+-------------+-------------+-------------+-------------+
+```
+
+#### List documents in knowledge space
+
+```
+dbgpt knowledge list --space_name default
+```
+
+Output should look something like the following:
+```
++------------------------------------------------------------------------+
+|                       Space default description                        |
++------------+-----------------+--------------+--------------+-----------+
+| Space Name | Total Documents | Current Page | Current Size | Page Size |
++------------+-----------------+--------------+--------------+-----------+
+|  default   |        1        |      1       |      1       |     20    |
++------------+-----------------+--------------+--------------+-----------+
+
++-----------------------------------------------------------------------------------------------------------------------------------+
+|                                                     Documents of space default                                                    |
++------------+-------------+---------------+----------+--------+----------------------------+----------+----------------------------+
+| Space Name | Document ID | Document Name |   Type   | Chunks |         Last Sync          |  Status  |           Result           |
++------------+-------------+---------------+----------+--------+----------------------------+----------+----------------------------+
+|  default   |      61     | Knowledge.pdf | DOCUMENT |   745   | 2023-09-28T03:25:39.065762 | FINISHED | document embedding success |
++------------+-------------+---------------+----------+--------+----------------------------+----------+----------------------------+
+```
+
+#### List chunks of document in space `default`
+
+```
+dbgpt knowledge list --space_name default --doc_id 61 --page_size 5
+```
+
+
+```
++-----------------------------------------------------------------------------------+
+|                         Document 61 in default description                        |
++------------+-------------+--------------+--------------+--------------+-----------+
+| Space Name | Document ID | Total Chunks | Current Page | Current Size | Page Size |
++------------+-------------+--------------+--------------+--------------+-----------+
+|  default   |      61     |      745     |      1       |      5       |     5     |
++------------+-------------+--------------+--------------+--------------+-----------+
+
++-----------------------------------------------------------------------------------------------------------------------+
+|                                       chunks of document id 61 in space default                                       |
++------------+-------------+---------------+----------+-----------------------------------------------------------------+
+| Space Name | Document ID | Document Name | Content  |                            Meta Data                            |
++------------+-------------+---------------+----------+-----------------------------------------------------------------+
+|  default   |      61     | Knowledge.pdf | [Hidden] | {'source': '/app/pilot/data/default/Knowledge.pdf', 'page': 10} |
+|  default   |      61     | Knowledge.pdf | [Hidden] |  {'source': '/app/pilot/data/default/Knowledge.pdf', 'page': 9} |
+|  default   |      61     | Knowledge.pdf | [Hidden] |  {'source': '/app/pilot/data/default/Knowledge.pdf', 'page': 9} |
+|  default   |      61     | Knowledge.pdf | [Hidden] |  {'source': '/app/pilot/data/default/Knowledge.pdf', 'page': 8} |
+|  default   |      61     | Knowledge.pdf | [Hidden] |  {'source': '/app/pilot/data/default/Knowledge.pdf', 'page': 8} |
++------------+-------------+---------------+----------+-----------------------------------------------------------------+
+```
+
+#### More list usage
+
+```
+dbgpt knowledge list --help
+```
+
+```
+Usage: dbgpt knowledge list [OPTIONS]
+
+  List knowledge space
+
+Options:
+  --space_name TEXT               Your knowledge space name. If None, list all
+                                  spaces
+  --doc_id INTEGER                Your document id in knowledge space. If Not
+                                  None, list all chunks in current document
+  --page INTEGER                  The page for every query  [default: 1]
+  --page_size INTEGER             The page size for every query  [default: 20]
+  --show_content                  Query the document content of chunks
+  --output [text|html|csv|latex|json]
+                                  The output format
+  --help                          Show this message and exit.
+```
+
+
+### Delete your knowledge space or document in space
+
+#### Delete your knowledge space
+
+```
+dbgpt knowledge delete --space_name default
+```
+
+#### Delete your document in space
+
+```
+dbgpt knowledge delete --space_name default --doc_name Knowledge.pdf
+```
+
+
+#### More delete usage
+
+```
+dbgpt knowledge delete --help
+```
+
+```
+Usage: dbgpt knowledge delete [OPTIONS]
+
+  Delete your knowledge space or document in space
+
+Options:
+  --space_name TEXT  Your knowledge space name  [default: default]
+  --doc_name TEXT    The document name you want to delete. If doc_name is
+                     None, this command will delete the whole space.
+  -y                 Confirm your choice
+  --help             Show this message and exit.
+```
+
+#### More knowledge usage
+
+```
+dbgpt knowledge --help
+```
+
+```
+Usage: dbgpt knowledge [OPTIONS] COMMAND [ARGS]...
+
+  Knowledge command line tool
+
+Options:
+  --address TEXT  Address of the Api server(If not set, try to read from
+                  environment variable: API_ADDRESS).  [default:
+                  http://127.0.0.1:5000]
+  --help          Show this message and exit.
+
+Commands:
+  delete  Delete your knowledge space or document in space
+  list    List knowledge space
+  load    Load your local documents to DB-GPT
+```

docs/locales/zh_CN/LC_MESSAGES/getting_started/application/kbqa/kbqa.po

@@ -8,7 +8,7 @@ msgid ""
 msgstr ""
 "Project-Id-Version: DB-GPT 👏👏 0.3.5\n"
 "Report-Msgid-Bugs-To: \n"
-"POT-Creation-Date: 2023-08-29 21:14+0800\n"
+"POT-Creation-Date: 2023-09-28 12:43+0800\n"
 "PO-Revision-Date: YEAR-MO-DA HO:MI+ZONE\n"
 "Last-Translator: FULL NAME <EMAIL@ADDRESS>\n"
 "Language: zh_CN\n"
@@ -20,12 +20,12 @@ msgstr ""
 "Generated-By: Babel 2.12.1\n"
 
 #: ../../getting_started/application/kbqa/kbqa.md:1
-#: 4b44ec7d6b3b492489b84bab4471fe46
+#: 2f77b0e5795f4d3593b5f04c54c2c6c2
 msgid "KBQA"
 msgstr ""
 
 #: ../../getting_started/application/kbqa/kbqa.md:3
-#: c4217f8786e24f5190354d129b21dff5
+#: 1e4f61ecf9084777ae451d2259bf804d
 msgid ""
 "![chat_knowledge](https://github.com/eosphoros-ai/DB-"
 "GPT/assets/13723926/bc343c94-df3e-41e5-90d5-23b68c768c59)"
@@ -33,12 +33,12 @@ msgstr ""
 
 #: ../../getting_started/application/kbqa/kbqa.md:3
 #: ../../getting_started/application/kbqa/kbqa.md:7
-#: 2f02ab2494b54fff87e7d1e310f38119 dc0349e48d5e4d89b1f9f353813c9f06
+#: 7144258d06f3459d9404ef583b4f3696 e0aec0e78a9a40a2a6c6d496a5966fe8
 msgid "chat_knowledge"
 msgstr "chat_knowledge"
 
 #: ../../getting_started/application/kbqa/kbqa.md:5
-#: 149c4c9e15004aaf992c5896deb9e541
+#: 4eeebe69ee5b4055bb2ac48444b5f3cc
 msgid ""
 "DB-GPT supports a knowledge question-answering module, which aims to "
 "create an intelligent expert in the field of databases and provide "
@@ -46,7 +46,7 @@ msgid ""
 msgstr " DB-GPT支持知识问答模块，知识问答的初衷是打造DB领域的智能专家，为数据库从业人员解决专业的知识问题回答"
 
 #: ../../getting_started/application/kbqa/kbqa.md:7
-#: c7e103a20a2e4989aab8a750cdc4dbf4
+#: 99ec2e6a023744878a8e970de7588dfe
 msgid ""
 "![chat_knowledge](https://github.com/eosphoros-ai/DB-"
 "GPT/assets/13723926/6e55f2e5-94f7-4906-aed6-097db5c6c721)"
@@ -54,83 +54,83 @@ msgstr ""
 
 #: ../../getting_started/application/kbqa/kbqa.md:9
 #: ../../getting_started/application/kbqa/kbqa.md:12
-#: 3ddf012fb6d74eb1be1a0fd0ada9ddf6 e29bffef26fb44ac978d6cbe6584f48f
+#: 77f6d58b0c714a0cbb2510aa2a036560 f3837660bb674945904a2acb34fd816a
 msgid "KBQA abilities"
 msgstr "KBQA现有能力"
 
 #: ../../getting_started/application/kbqa/kbqa.md:13
-#: 87cf02966b574d6db8baf0839b13e1e7
+#: 96d691596d1f4309bd29e4d8b3a3bc61
 msgid "Knowledge Space."
 msgstr "知识空间"
 
 #: ../../getting_started/application/kbqa/kbqa.md:14
-#: 2777488125234408aa7156a69fcfacef
+#: 36fe78a666a0410380e338674bb76e96
 msgid "Multi Source Knowledge Source Embedding."
 msgstr "多数据源Embedding"
 
 #: ../../getting_started/application/kbqa/kbqa.md:15
-#: c50f2de4ea6f4aedb7d4fd674ee3f6f7
+#: 14659bb6964b402ba2aa912feffe0a28
 msgid "Embedding Argument Adjust"
 msgstr "Embedding参数自定义"
 
 #: ../../getting_started/application/kbqa/kbqa.md:16
-#: f8a7af83e94e45239bd0efdb06eb320b
+#: c72212e1a0e54aa9b0b5adee687fed65
 msgid "Chat Knowledge"
 msgstr "知识问答"
 
 #: ../../getting_started/application/kbqa/kbqa.md:17
-#: e6025e23178f4f58a6d4c75a6bc1d036
+#: d5fea791138240e8b26b4da0a7d3a4bf
 msgid "Multi Vector DB"
 msgstr "多向量数据库管理"
 
 #: ../../getting_started/application/kbqa/kbqa.md:21
-#: 0780e1d27af244429186caa866772c06
+#: d2460e9418c44064be16f8b8bd76bf7a
 msgid ""
 "If your DB type is Sqlite, there is nothing to do to build KBQA service "
 "database schema."
 msgstr ""
 
 #: ../../getting_started/application/kbqa/kbqa.md:23
-#: 38584aba054f4520a0b1d9b00d6abf06
+#: b0ebff37da05465fb5c91abb496647ee
 msgid ""
 "If your DB type is Mysql or other DBTYPE, you will build kbqa service "
 "database schema."
 msgstr ""
 
 #: ../../getting_started/application/kbqa/kbqa.md:25
-#: 9ef800860afe4728b2103286864ed3fb
+#: ac0e3e657e9c4dfaafdb8884bf0d915a
 msgid "Mysql"
 msgstr ""
 
 #: ../../getting_started/application/kbqa/kbqa.md:26
-#: 3b30239fd457457eb707821794e786db
+#: 4b899ec4f62c47a79e9cff79524ed069
 msgid ""
 "$ mysql -h127.0.0.1 -uroot -paa12345678 < "
 "./assets/schema/knowledge_management.sql"
 msgstr ""
 
 #: ../../getting_started/application/kbqa/kbqa.md:28
-#: 1cd533c2f0254f8d8d10bafe5811a279
+#: a74b6c512b4d4ed399d5ee25e4a4f44a
 msgid "or"
 msgstr ""
 
 #: ../../getting_started/application/kbqa/kbqa.md:30
-#: 2b57e7d9a70f427e81122fe8d7d3c50b
+#: 87216b282f664c6abfe736db5e3badd5
 msgid "execute DBGPT/assets/schema/knowledge_management.sql"
 msgstr ""
 
 #: ../../getting_started/application/kbqa/kbqa.md:33
-#: 45a0c8c8ff0a4b48ac2d66b4713c4108
+#: 67aaa574f941485c8ba55a9058750175
 msgid "Steps to KBQA In DB-GPT"
 msgstr "怎样一步一步使用KBQA"
 
 #: ../../getting_started/application/kbqa/kbqa.md:35
-#: 2ff844b9a29f4717909d091a57d58fe8
+#: 202ed073da4948a49d32b1beca69292d
 msgid "1.Create Knowledge Space"
 msgstr "1.首先创建知识空间"
 
 #: ../../getting_started/application/kbqa/kbqa.md:36
-#: 081e1d1ef5bb42ddbd7330dd3ac1d38e
+#: f1defb29643f4664a125cf127e63ef71
 #, fuzzy
 msgid ""
 "If you are using Knowledge Space for the first time, you need to create a"
@@ -140,17 +140,17 @@ msgid ""
 msgstr "如果你是第一次使用，先创建知识空间，指定名字,拥有者和描述信息"
 
 #: ../../getting_started/application/kbqa/kbqa.md:36
-#: 4a393bf0f50647d4b0cfc64db80847eb
+#: 378e6091c3eb4f0d8ca4d4144303dde1
 msgid "create_space"
 msgstr "create_space"
 
 #: ../../getting_started/application/kbqa/kbqa.md:41
-#: 96303597bc364952b7249e805486e73f
+#: 669504b548ce4b7b9ea4c463b446cf39
 msgid "2.Create Knowledge Document"
 msgstr "2.上传知识"
 
 #: ../../getting_started/application/kbqa/kbqa.md:42
-#: 765f93f2b668491cb6824ea4706cb449
+#: 731b15e4396b450589d619fccd78d542
 msgid ""
 "DB-GPT now support Multi Knowledge Source, including Text, WebUrl, and "
 "Document(PDF, Markdown, Word, PPT, HTML and CSV). After successfully "
@@ -165,29 +165,29 @@ msgstr ""
 "CSV)。上传文档成功后后台会自动将文档内容进行读取，切片，然后导入到向量数据库中，当然你也可以手动进行同步，你也可以点击详情查看具体的文档切片内容"
 
 #: ../../getting_started/application/kbqa/kbqa.md:44
-#: a0eddeecc620479483bf50857da39ffd
+#: 7c020f677a364693800952d558ba357e
 msgid "2.1 Choose Knowledge Type:"
 msgstr "2.1 选择知识类型"
 
 #: ../../getting_started/application/kbqa/kbqa.md:45
-#: 44de66f399324454b389d5c348af94e9
+#: c88460c498ff4fbca22f35e181c66723
 msgid ""
 "![document](https://github.com/eosphoros-ai/DB-"
 "GPT/assets/13723926/5b8173da-f444-4607-9d12-14bcab8179d0)"
 msgstr ""
 
 #: ../../getting_started/application/kbqa/kbqa.md:45
-#: 1ae65290b99d490bb72b060084ecc726
+#: 60fcd6db28c546728bf9c015b63e9434
 msgid "document"
 msgstr "document"
 
 #: ../../getting_started/application/kbqa/kbqa.md:47
-#: d8720434626444b593bb3b06b50dc70f
+#: ffb8436b6e97468296e5852530e3d9c8
 msgid "2.2 Upload Document:"
 msgstr "2.2上传文档"
 
 #: ../../getting_started/application/kbqa/kbqa.md:48
-#: 2d4bcbeb391a47b89454b06cb041dff2
+#: a652b17e64b040938cf4e09ef7ed470a
 msgid ""
 "![upload](https://github.com/eosphoros-ai/DB-GPT/assets/13723926"
 "/91b338fc-d3b2-476e-9396-3f6b4f16a890)"
@@ -197,30 +197,30 @@ msgstr ""
 #: ../../getting_started/application/kbqa/kbqa.md:52
 #: ../../getting_started/application/kbqa/kbqa.md:57
 #: ../../getting_started/application/kbqa/kbqa.md:70
-#: 58b32cb59a6242679f4a1e5fc7ca819f 81041a1f25b64b19a7f662ed55029224
+#: 3f1c6873d54c47fea96d91188f728af9 434f854aa1af4f4bbf9ec4060388c643
-#: 8891e61f67014355b74c17d013c09cca f0e9c548494f4045a3dc92e993f4cfe7
+#: c1404f398cc3480d9c59b4ce326501e7 f18624fc081d4276859d4a4c99e998ee
 msgid "upload"
 msgstr "upload"
 
 #: ../../getting_started/application/kbqa/kbqa.md:51
-#: 645bf91a4f6a428a9e99ca29599c0722
+#: 6695dbe6d16e43d7b88dbad61039ff7e
 msgid "3.Chat With Knowledge"
 msgstr "3.知识问答"
 
 #: ../../getting_started/application/kbqa/kbqa.md:52
-#: 85077af67bd740c1b3b02996dc287a80
+#: 30b3afb9cb4d4705833ba15580425b05
 msgid ""
 "![upload](https://github.com/eosphoros-ai/DB-"
 "GPT/assets/13723926/a8281be7-1454-467d-81c9-15ef108aac10)"
 msgstr ""
 
 #: ../../getting_started/application/kbqa/kbqa.md:54
-#: 42d5776030ad4109810a0cb18e19de37
+#: 05cc4bf0473f4f88aef63342b714d472
 msgid "4.Adjust Space arguments"
 msgstr "4.调整知识参数"
 
 #: ../../getting_started/application/kbqa/kbqa.md:55
-#: a8ea23895a4b4312b1c1e072865f8b90
+#: 9ec125362dbb4aec864777313b1a8327
 msgid ""
 "Each knowledge space supports argument customization, including the "
 "relevant arguments for vector retrieval and the arguments for knowledge "
@@ -228,78 +228,78 @@ msgid ""
 msgstr "每一个知识空间都支持参数自定义, 包括向量召回的相关参数以及知识问答Promp参数"
 
 #: ../../getting_started/application/kbqa/kbqa.md:56
-#: 2f5c087e4f7a49828aa797fafff237f0
+#: 54fd9461ced1436aa6c51984024caa74
 msgid "4.1 Embedding"
 msgstr "4.1 Embedding"
 
 #: ../../getting_started/application/kbqa/kbqa.md:57
-#: 47087bf391b642f3ba73e461d2d132a0
+#: 4b17fde7063e40e88dd121024f3e1c42
 msgid ""
 "Embedding Argument ![upload](https://github.com/eosphoros-ai/DB-"
 "GPT/assets/13723926/f1221bd5-d049-4ceb-96e6-8709e76e502e)"
 msgstr ""
 
 #: ../../getting_started/application/kbqa/kbqa.md:61
-#: 1ed384ba0501423184a4c977d86b8b3a
+#: 6715234540d74eb79da60cf5ec29e236
 msgid "Embedding arguments"
 msgstr "Embedding arguments"
 
 #: ../../getting_started/application/kbqa/kbqa.md:62
-#: 848962be69a348ffab3dd48839fb100a
+#: af7ed9e32c6846b6a75893a55e0d78a4
 msgid "topk:the top k vectors based on similarity score."
 msgstr "topk:相似性检索出tok条文档"
 
 #: ../../getting_started/application/kbqa/kbqa.md:63
-#: 0880963aeb9c426187809a7086c224a8
+#: 05290f955cd142a2857d20786edf0952
 msgid "recall_score:set a threshold score for the retrieval of similar vectors."
 msgstr "recall_score:向量检索相关度衡量指标分数"
 
 #: ../../getting_started/application/kbqa/kbqa.md:64
-#: a6ead8ae58164749b83e5c972537fe8b
+#: b6e088dda8f142438bc0baebd357fa0e
 msgid "recall_type:recall type."
 msgstr "recall_type:召回类型"
 
 #: ../../getting_started/application/kbqa/kbqa.md:65
-#: 70d900b0948849d080effcdfc79bb685
+#: 509bb3adefb74b65baea79bcbe94a240
 msgid "model:A model used to create vector representations of text or other data."
 msgstr "model:embdding模型"
 
 #: ../../getting_started/application/kbqa/kbqa.md:66
-#: 489a58835a5d4c98ad5f6f904f7af370
+#: 43ef5b6a16c0435b9b7416d3600556e5
 msgid "chunk_size:The size of the data chunks used in processing."
 msgstr "chunk_size:文档切片阈值大小"
 
 #: ../../getting_started/application/kbqa/kbqa.md:67
-#: b42781f2340c478f86137147fd4a6c91
+#: 800fb14f22304155bedc5dc666f56a25
 msgid "chunk_overlap:The amount of overlap between adjacent data chunks."
 msgstr "chunk_overlap:文本块之间的最大重叠量。保留一些重叠可以保持文本块之间的连续性（例如使用滑动窗口）"
 
 #: ../../getting_started/application/kbqa/kbqa.md:69
-#: 5f027d4a10394e7da3d4b50fc2663f82
+#: 5497d17bcf014ad5b61d5febd7a9fda5
 msgid "4.2 Prompt"
 msgstr "4.2 Prompt"
 
 #: ../../getting_started/application/kbqa/kbqa.md:70
-#: 9ad62d8626584e80a82fcef239c0f546
+#: dd9cdfac5aae44c8998b36ae0ed286b8
 msgid ""
 "Prompt Argument ![upload](https://github.com/eosphoros-ai/DB-"
 "GPT/assets/13723926/9918c9c3-ed64-4804-9e05-fa7d7d177bec)"
 msgstr ""
 
 #: ../../getting_started/application/kbqa/kbqa.md:74
-#: ec0717b8d210410f9894d2a4f51642e1
+#: fbede51cc73444aab74fa646834ea1c2
 msgid "Prompt arguments"
 msgstr "Prompt arguments"
 
 #: ../../getting_started/application/kbqa/kbqa.md:75
-#: 7cf68eacd5ae4151abdefe44feb239e1
+#: a85ab9b05ec04cd59d3f4539599a8e35
 msgid ""
 "scene:A contextual parameter used to define the setting or environment in"
 " which the prompt is being used."
 msgstr "scene:上下文环境的场景定义"
 
 #: ../../getting_started/application/kbqa/kbqa.md:76
-#: 2d9293acfb1a495c9b08261e957b2395
+#: 48258fcdbb134a0e820347185ea45ca0
 msgid ""
 "template:A pre-defined structure or format for the prompt, which can help"
 " ensure that the AI system generates responses that are consistent with "
@@ -307,75 +307,168 @@ msgid ""
 msgstr "template:预定义的提示结构或格式，可以帮助确保AI系统生成与所期望的风格或语气一致的回复。"
 
 #: ../../getting_started/application/kbqa/kbqa.md:77
-#: 307dd62261214e4e84beb7f19b3e2f26
+#: 2dab662fc6a945b3b2691831f16b52e7
 msgid "max_token:The maximum number of tokens or words allowed in a prompt."
 msgstr "max_token: prompt token最大值"
 
 #: ../../getting_started/application/kbqa/kbqa.md:79
-#: 07b797731fa74738acb3e1fb4c03deac
+#: a7c5a81457334f39a8956146fe4aa951
 msgid "5.Change Vector Database"
 msgstr "5.Change Vector Database"
 
 #: ../../getting_started/application/kbqa/kbqa.md:81
-#: 43fa40ced23842b48007b4264c1423c0
+#: dccbc6e611e64f79aadb849ae1cdf0d2
 msgid "Vector Store SETTINGS"
 msgstr "Vector Store SETTINGS"
 
 #: ../../getting_started/application/kbqa/kbqa.md:82
-#: 0ed4e0c6a81e4265b443a2c6d05d440b
+#: a3af9f960c3144b2884a5ff3dc4f8cfa
 msgid "Chroma"
 msgstr "Chroma"
 
 #: ../../getting_started/application/kbqa/kbqa.md:83
-#: 8ef2d15ffd0c4919bdbfdb52443021eb
+#: f195989635ec403d9f7ece263c451fa2
 msgid "VECTOR_STORE_TYPE=Chroma"
 msgstr "VECTOR_STORE_TYPE=Chroma"
 
 #: ../../getting_started/application/kbqa/kbqa.md:84
-#: 08504aaae0014bd9992345f036989198
+#: 9821aa494e514654bf33335c2db7700f
 msgid "MILVUS"
 msgstr "MILVUS"
 
 #: ../../getting_started/application/kbqa/kbqa.md:85
-#: aa30fe77dc3f46bfb153b81e0cbdfb97
+#: 1d7ef3150b6549fc834567f2647c32db
 msgid "VECTOR_STORE_TYPE=Milvus"
 msgstr "VECTOR_STORE_TYPE=Milvus"
 
 #: ../../getting_started/application/kbqa/kbqa.md:86
-#: e14c1646e5ca4677a471645c13dca835
+#: 702e9fc52c1d45cc9136ffd0a1bbcbc8
 msgid "MILVUS_URL=127.0.0.1"
 msgstr "MILVUS_URL=127.0.0.1"
 
 #: ../../getting_started/application/kbqa/kbqa.md:87
-#: 0f0614f951934fd6abfb0fd45e0b79e7
+#: 44ed55973e50490db566f5d65fa87207
 msgid "MILVUS_PORT=19530"
 msgstr "MILVUS_PORT=19530"
 
 #: ../../getting_started/application/kbqa/kbqa.md:88
-#: 4dafbbd894e4469e80414b54bde69193
+#: e1e13a389f404a3384e96b178b7887d0
 msgid "MILVUS_USERNAME"
 msgstr "MILVUS_USERNAME"
 
 #: ../../getting_started/application/kbqa/kbqa.md:89
-#: 1759ec4d744e42a5a928c6abcc2dd2ac
+#: 5755432971214b6dbe016a1556be759c
 msgid "MILVUS_PASSWORD"
 msgstr "MILVUS_PASSWORD"
 
 #: ../../getting_started/application/kbqa/kbqa.md:90
-#: d44bcea1db4f4388bc2ae7968e541761
+#: f92e8123fadc4f67a3a81ce76f642d67
 msgid "MILVUS_SECURE="
 msgstr "MILVUS_SECURE="
 
 #: ../../getting_started/application/kbqa/kbqa.md:92
-#: f33218e0880f4d888438d3333f4a0895
+#: 5ad47c8768484e67a0b919d290b0fcde
 msgid "WEAVIATE"
 msgstr "WEAVIATE"
 
 #: ../../getting_started/application/kbqa/kbqa.md:93
-#: 42a2102b236447ff83820d3a1602c3f2
+#: 139b2fab9a294215ba152710e464d425
 msgid "WEAVIATE_URL=https://kt-region-m8hcy0wc.weaviate.network"
 msgstr "WEAVIATE_URL=https://kt-region-m8hcy0wc.weaviate.networkc"
 
+#: ../../getting_started/application/kbqa/kbqa.md:96
+#: bd9c81cd547142fab5857dbf0df441d0
+msgid "KBQA command line"
+msgstr "KBQA 命令行工具"
+
+#: ../../getting_started/application/kbqa/kbqa.md:98
+#: fba52bfa428549e683c9cfb8e9684636
+msgid "Load your local documents to DB-GPT"
+msgstr "将你的本地文档导入到 DB-GPT"
+
+#: ../../getting_started/application/kbqa/kbqa.md:104
+#: 5c3e7fa9b2b04e6097781d3f406c2bae
+msgid "`--space_name`: Your knowledge space name, default: `default`"
+msgstr ""
+
+#: ../../getting_started/application/kbqa/kbqa.md:105
+#: 083bb6913da64f8a880b20e31d138d7b
+msgid ""
+"`--local_doc_path`: Your document directory or document file path, "
+"default: `./pilot/datasets`"
+msgstr ""
+
+#: ../../getting_started/application/kbqa/kbqa.md:106
+#: 7b17564ffc7a43ab8e5beb023f18a59a
+msgid "`--vector_store_type`: Vector store type, default: `Chroma`"
+msgstr ""
+
+#: ../../getting_started/application/kbqa/kbqa.md:108
+#: 814bdf5ecddd4490af57dcab73ae8c85
+msgid "**View the `dbgpt knowledge load --help`help**"
+msgstr ""
+
+#: ../../getting_started/application/kbqa/kbqa.md:114
+#: f1aaa308ef4b46f1adbb8ec64738f361
+msgid "Here you can see the parameters:"
+msgstr ""
+
+#: ../../getting_started/application/kbqa/kbqa.md:141
+#: ../../getting_started/application/kbqa/kbqa.md:143
+#: 163ff96e128641509933c1871b6bde64 fca3ea3c13bc4fa3bfa33845f4d88b52
+#, fuzzy
+msgid "List knowledge space"
+msgstr "查看知识空间"
+
+#: ../../getting_started/application/kbqa/kbqa.md:149
+#: ../../getting_started/application/kbqa/kbqa.md:169
+#: 57b89ee3fde745ff8502ede525fc4a8c d9852d13a9b444618a598ba4abe47493
+msgid "Output should look something like the following:"
+msgstr "输出如下："
+
+#: ../../getting_started/application/kbqa/kbqa.md:163
+#: 14300078a0a74aee9cf6e54c470e3862
+#, fuzzy
+msgid "List documents in knowledge space"
+msgstr "查看某个知识空间中的文档"
+
+#: ../../getting_started/application/kbqa/kbqa.md:188
+#: b9afd5b23fb947c691d66f82f86c36f2
+msgid "List chunks of document in space `default`"
+msgstr "查看某个知识空间文档中的块"
+
+#: ../../getting_started/application/kbqa/kbqa.md:217
+#: 294d9d963299400db39c5c51ab5d2e23
+msgid "More list usage"
+msgstr "更多 list 命令用户"
+
+#: ../../getting_started/application/kbqa/kbqa.md:242
+#: 5d962a6b5b2d4fb6809bc84cb7711120
+msgid "Delete your knowledge space or document in space"
+msgstr "删除你的知识空间或者知识空间中的文档"
+
+#: ../../getting_started/application/kbqa/kbqa.md:244
+#: f1deee3aaf8e41308f2f81974032e4d4
+#, fuzzy
+msgid "Delete your knowledge space"
+msgstr "删除知识空间"
+
+#: ../../getting_started/application/kbqa/kbqa.md:250
+#: ba2314a2d20944f7b64a60e67a5b8d94
+msgid "Delete your document in space"
+msgstr "删除知识空间中的文档"
+
+#: ../../getting_started/application/kbqa/kbqa.md:257
+#: 2400777f270d40e2adcdc38a2d3cb73a
+msgid "More delete usage"
+msgstr "更多删除用法"
+
+#: ../../getting_started/application/kbqa/kbqa.md:276
+#: 4e79bbda201347f2822b1742805923a6
+#, fuzzy
+msgid "More knowledge usage"
+msgstr "更知识库命令用法"
+
 #~ msgid "![chat_knowledge](../../../../assets/chat_knowledge.png)"
 #~ msgstr "![chat_knowledge](../../../../assets/chat_knowledge.png)"
 

pilot/agent/json_fix_llm.py

@@ -1,6 +1,7 @@
 import contextlib
 import json
 from typing import Any, Dict
+import logging
 
 from colorama import Fore
 from regex import regex
@@ -11,9 +12,9 @@ from pilot.json_utils.json_fix_general import (
     balance_braces,
     fix_invalid_escape,
 )
-from pilot.logs import logger
 
 
+logger = logging.getLogger(__name__)
 CFG = Config()
 
 

pilot/commands/built_in/image_gen.py

@@ -2,14 +2,15 @@
 import io
 import uuid
 from base64 import b64decode
+import logging
 
 import requests
 from PIL import Image
 
 from pilot.commands.command_mange import command
 from pilot.configs.config import Config
-from pilot.logs import logger
 
+logger = logging.getLogger(__name__)
 CFG = Config()
 
 

pilot/commands/disply_type/show_chart_gen.py

@@ -1,5 +1,5 @@
+import logging
 from pandas import DataFrame
-
 from pilot.commands.command_mange import command
 from pilot.configs.config import Config
 import pandas as pd
@@ -13,11 +13,10 @@ import matplotlib.pyplot as plt
 import matplotlib.ticker as mtick
 from matplotlib.font_manager import FontManager
 
-from pilot.configs.model_config import LOGDIR
-from pilot.utils import build_logger
-
 CFG = Config()
-logger = build_logger("show_chart_gen", LOGDIR + "show_chart_gen.log")
+
+logger = logging.getLogger(__name__)
+
 static_message_img_path = os.path.join(os.getcwd(), "message/img")
 
 

pilot/commands/disply_type/show_table_gen.py

@@ -1,14 +1,13 @@
+import logging
+
 import pandas as pd
 from pandas import DataFrame
 
 from pilot.commands.command_mange import command
 from pilot.configs.config import Config
 
-from pilot.configs.model_config import LOGDIR
-from pilot.utils import build_logger
-
 CFG = Config()
-logger = build_logger("show_table_gen", LOGDIR + "show_table_gen.log")
+logger = logging.getLogger(__name__)
 
 
 @command(

pilot/commands/disply_type/show_text_gen.py

@@ -1,13 +1,12 @@
+import logging
 import pandas as pd
 from pandas import DataFrame
 
 from pilot.commands.command_mange import command
 from pilot.configs.config import Config
-from pilot.configs.model_config import LOGDIR
-from pilot.utils import build_logger
 
 CFG = Config()
-logger = build_logger("show_table_gen", LOGDIR + "show_table_gen.log")
+logger = logging.getLogger(__name__)
 
 
 @command(

pilot/common/plugins.py

@@ -8,6 +8,7 @@ import zipfile
 import requests
 import threading
 import datetime
+import logging
 from pathlib import Path
 from typing import List, TYPE_CHECKING
 from urllib.parse import urlparse
@@ -17,7 +18,8 @@ import requests
 
 from pilot.configs.config import Config
 from pilot.configs.model_config import PLUGINS_DIR
-from pilot.logs import logger
+
+logger = logging.getLogger(__name__)
 
 if TYPE_CHECKING:
     from auto_gpt_plugin_template import AutoGPTPluginTemplate

pilot/configs/config.py

@@ -32,7 +32,7 @@ class Config(metaclass=Singleton):
         # self.NUM_GPUS = int(os.getenv("NUM_GPUS", 1))
 
         self.execute_local_commands = (
-            os.getenv("EXECUTE_LOCAL_COMMANDS", "False") == "True"
+            os.getenv("EXECUTE_LOCAL_COMMANDS", "False").lower() == "true"
         )
         # User agent header to use when making HTTP requests
         # Some websites might just completely deny request with an error code if
@@ -64,7 +64,7 @@ class Config(metaclass=Singleton):
         self.milvus_username = os.getenv("MILVUS_USERNAME")
         self.milvus_password = os.getenv("MILVUS_PASSWORD")
         self.milvus_collection = os.getenv("MILVUS_COLLECTION", "dbgpt")
-        self.milvus_secure = os.getenv("MILVUS_SECURE") == "True"
+        self.milvus_secure = os.getenv("MILVUS_SECURE", "False").lower() == "true"
 
         self.authorise_key = os.getenv("AUTHORISE_COMMAND_KEY", "y")
         self.exit_key = os.getenv("EXIT_KEY", "n")
@@ -98,7 +98,7 @@ class Config(metaclass=Singleton):
             self.disabled_command_categories = []
 
         self.execute_local_commands = (
-            os.getenv("EXECUTE_LOCAL_COMMANDS", "False") == "True"
+            os.getenv("EXECUTE_LOCAL_COMMANDS", "False").lower() == "true"
         )
         ### message stor file
         self.message_dir = os.getenv("MESSAGE_HISTORY_DIR", "../../message")
@@ -107,7 +107,7 @@ class Config(metaclass=Singleton):
 
         self.plugins: List["AutoGPTPluginTemplate"] = []
         self.plugins_openai = []
-        self.plugins_auto_load = os.getenv("AUTO_LOAD_PLUGIN", "True") == "True"
+        self.plugins_auto_load = os.getenv("AUTO_LOAD_PLUGIN", "True").lower() == "true"
 
         self.plugins_git_branch = os.getenv("PLUGINS_GIT_BRANCH", "plugin_dashboard")
 
@@ -124,10 +124,10 @@ class Config(metaclass=Singleton):
             self.plugins_denylist = []
         ### Native SQL Execution Capability Control Configuration
         self.NATIVE_SQL_CAN_RUN_DDL = (
-            os.getenv("NATIVE_SQL_CAN_RUN_DDL", "True") == "True"
+            os.getenv("NATIVE_SQL_CAN_RUN_DDL", "True").lower() == "true"
         )
         self.NATIVE_SQL_CAN_RUN_WRITE = (
-            os.getenv("NATIVE_SQL_CAN_RUN_WRITE", "True") == "True"
+            os.getenv("NATIVE_SQL_CAN_RUN_WRITE", "True").lower() == "true"
         )
 
         ### default Local database connection configuration
@@ -170,8 +170,8 @@ class Config(metaclass=Singleton):
 
         # QLoRA
         self.QLoRA = os.getenv("QUANTIZE_QLORA", "True")
-        self.IS_LOAD_8BIT = os.getenv("QUANTIZE_8bit", "True") == "True"
+        self.IS_LOAD_8BIT = os.getenv("QUANTIZE_8bit", "True").lower() == "true"
-        self.IS_LOAD_4BIT = os.getenv("QUANTIZE_4bit", "False") == "True"
+        self.IS_LOAD_4BIT = os.getenv("QUANTIZE_4bit", "False").lower() == "true"
         if self.IS_LOAD_8BIT and self.IS_LOAD_4BIT:
             self.IS_LOAD_8BIT = False
         # In order to be compatible with the new and old model parameter design
@@ -187,7 +187,9 @@ class Config(metaclass=Singleton):
             os.getenv("KNOWLEDGE_SEARCH_MAX_TOKEN", 2000)
         )
         ### Control whether to display the source document of knowledge on the front end.
-        self.KNOWLEDGE_CHAT_SHOW_RELATIONS = False
+        self.KNOWLEDGE_CHAT_SHOW_RELATIONS = (
+            os.getenv("KNOWLEDGE_CHAT_SHOW_RELATIONS", "False").lower() == "true"
+        )
 
         ### SUMMARY_CONFIG Configuration
         self.SUMMARY_CONFIG = os.getenv("SUMMARY_CONFIG", "FAST")

pilot/configs/model_config.py

@@ -3,13 +3,12 @@
 
 import os
 
-# import nltk
-
 ROOT_PATH = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 MODEL_PATH = os.path.join(ROOT_PATH, "models")
 PILOT_PATH = os.path.join(ROOT_PATH, "pilot")
 VECTORE_PATH = os.path.join(PILOT_PATH, "vector_store")
-LOGDIR = os.path.join(ROOT_PATH, "logs")
+LOGDIR = os.getenv("DBGPT_LOG_DIR", os.path.join(ROOT_PATH, "logs"))
+
 DATASETS_DIR = os.path.join(PILOT_PATH, "datasets")
 DATA_DIR = os.path.join(PILOT_PATH, "data")
 # nltk.data.path = [os.path.join(PILOT_PATH, "nltk_data")] + nltk.data.path

pilot/connections/rdbms/base_dao.py

@@ -1,10 +1,11 @@
+import logging
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker
 from pilot.configs.config import Config
 from pilot.common.schema import DBType
 from pilot.connections.rdbms.base import RDBMSDatabase
-from pilot.logs import logger
 
+logger = logging.getLogger(__name__)
 CFG = Config()
 
 

pilot/embedding_engine/__init__.py

@@ -1,5 +1,12 @@
 from pilot.embedding_engine.source_embedding import SourceEmbedding, register
 from pilot.embedding_engine.embedding_engine import EmbeddingEngine
 from pilot.embedding_engine.knowledge_type import KnowledgeType
+from pilot.embedding_engine.pre_text_splitter import PreTextSplitter
 
-__all__ = ["SourceEmbedding", "register", "EmbeddingEngine", "KnowledgeType"]
+__all__ = [
+    "SourceEmbedding",
+    "register",
+    "EmbeddingEngine",
+    "KnowledgeType",
+    "PreTextSplitter",
+]

pilot/embedding_engine/pre_text_splitter.py

@@ -0,0 +1,30 @@
+from typing import Iterable, List
+from langchain.schema import Document
+from langchain.text_splitter import TextSplitter
+
+
+def _single_document_split(
+    document: Document, pre_separator: str
+) -> Iterable[Document]:
+    page_content = document.page_content
+    for i, content in enumerate(page_content.split(pre_separator)):
+        metadata = document.metadata.copy()
+        if "source" in metadata:
+            metadata["source"] = metadata["source"] + "_pre_split_" + str(i)
+        yield Document(page_content=content, metadata=metadata)
+
+
+class PreTextSplitter(TextSplitter):
+    def __init__(self, pre_separator: str, text_splitter_impl: TextSplitter):
+        self.pre_separator = pre_separator
+        self._impl = text_splitter_impl
+
+    def split_text(self, text: str) -> List[str]:
+        return self._impl.split_text(text)
+
+    def split_documents(self, documents: Iterable[Document]) -> List[Document]:
+        def generator() -> Iterable[Document]:
+            for doc in documents:
+                yield from _single_document_split(doc, pre_separator=self.pre_separator)
+
+        return self._impl.split_documents(generator())

pilot/json_utils/json_fix_general.py

@@ -5,12 +5,13 @@ from __future__ import annotations
 import contextlib
 import json
 import re
+import logging
 from typing import Optional
 
 from pilot.configs.config import Config
 from pilot.json_utils.utilities import extract_char_position
-from pilot.logs import logger
 
+logger = logging.getLogger(__name__)
 CFG = Config()
 
 

pilot/json_utils/utilities.py

@@ -3,12 +3,14 @@ import json
 import os.path
 import re
 import json
+import logging
 from datetime import datetime
 
 from jsonschema import Draft7Validator
 
 from pilot.configs.config import Config
-from pilot.logs import logger
+
+logger = logging.getLogger(__name__)
 
 CFG = Config()
 LLM_DEFAULT_RESPONSE_FORMAT = "llm_response_format_1"

pilot/model/adapter.py

@@ -3,6 +3,7 @@
 
 import os
 import re
+import logging
 from pathlib import Path
 from typing import List, Tuple, Callable, Type
 from functools import cache
@@ -19,8 +20,8 @@ from pilot.model.parameter import (
 )
 from pilot.configs.model_config import get_device
 from pilot.configs.config import Config
-from pilot.logs import logger
 
+logger = logging.getLogger(__name__)
 
 CFG = Config()
 

pilot/model/cluster/controller/controller.py

@@ -13,6 +13,9 @@ from pilot.utils.api_utils import (
     _api_remote as api_remote,
     _sync_api_remote as sync_api_remote,
 )
+from pilot.utils.utils import setup_logging
+
+logger = logging.getLogger(__name__)
 
 
 class BaseModelController(BaseComponent, ABC):
@@ -59,7 +62,7 @@ class LocalModelController(BaseModelController):
     async def get_all_instances(
         self, model_name: str = None, healthy_only: bool = False
     ) -> List[ModelInstance]:
-        logging.info(
+        logger.info(
             f"Get all instances with {model_name}, healthy_only: {healthy_only}"
         )
         if not model_name:
@@ -178,6 +181,13 @@ def run_model_controller():
     controller_params: ModelControllerParameters = parser.parse_args_into_dataclass(
         ModelControllerParameters, env_prefix=env_prefix
     )
+
+    setup_logging(
+        "pilot",
+        logging_level=controller_params.log_level,
+        logger_filename="dbgpt_model_controller.log",
+    )
+
     initialize_controller(host=controller_params.host, port=controller_params.port)
 
 

pilot/model/cluster/worker/default_worker.py

@@ -120,7 +120,9 @@ class DefaultModelWorker(ModelWorker):
                     text=output, error_code=0, model_context=model_context
                 )
                 yield model_output
-            print(f"\n\nfull stream output:\n{previous_response}")
+            print(
+                f"\n\nfull stream output:\n{previous_response}\n\nmodel generate_stream params:\n{params}"
+            )
         except Exception as e:
             # Check if the exception is a torch.cuda.CudaError and if torch was imported.
             if torch_imported and isinstance(e, torch.cuda.CudaError):

pilot/model/cluster/worker/manager.py

@@ -36,6 +36,7 @@ from pilot.utils.parameter_utils import (
     ParameterDescription,
     _dict_to_command_args,
 )
+from pilot.utils.utils import setup_logging
 
 logger = logging.getLogger(__name__)
 
@@ -885,6 +886,12 @@ def run_worker_manager(
         model_name=model_name, model_path=model_path, standalone=standalone, port=port
     )
 
+    setup_logging(
+        "pilot",
+        logging_level=worker_params.log_level,
+        logger_filename="dbgpt_model_worker_manager.log",
+    )
+
     embedded_mod = True
     logger.info(f"Worker params: {worker_params}")
     if not app:

pilot/model/llm/llama_cpp/llama_cpp.py

@@ -3,11 +3,13 @@ Fork from text-generation-webui https://github.com/oobabooga/text-generation-web
 """
 import re
 from typing import Dict
+import logging
 import torch
 import llama_cpp
 
 from pilot.model.parameter import LlamaCppModelParameters
-from pilot.logs import logger
+
+logger = logging.getLogger(__name__)
 
 if torch.cuda.is_available() and not torch.version.hip:
     try:

pilot/model/loader.py

@@ -3,6 +3,7 @@
 
 from typing import Optional, Dict
 
+import logging
 from pilot.configs.model_config import get_device
 from pilot.model.adapter import get_llm_model_adapter, BaseLLMAdaper, ModelType
 from pilot.model.parameter import (
@@ -12,7 +13,8 @@ from pilot.model.parameter import (
 )
 from pilot.utils import get_gpu_memory
 from pilot.utils.parameter_utils import EnvArgumentParser, _genenv_ignoring_key_case
-from pilot.logs import logger
+
+logger = logging.getLogger(__name__)
 
 
 def _check_multi_gpu_or_4bit_quantization(model_params: ModelParameters):

pilot/model/parameter.py

@@ -31,6 +31,21 @@ class ModelControllerParameters(BaseParameters):
     daemon: Optional[bool] = field(
         default=False, metadata={"help": "Run Model Controller in background"}
     )
+    log_level: Optional[str] = field(
+        default=None,
+        metadata={
+            "help": "Logging level",
+            "valid_values": [
+                "FATAL",
+                "ERROR",
+                "WARNING",
+                "WARNING",
+                "INFO",
+                "DEBUG",
+                "NOTSET",
+            ],
+        },
+    )
 
 
 @dataclass
@@ -85,6 +100,22 @@ class ModelWorkerParameters(BaseModelParameters):
         default=20, metadata={"help": "The interval for sending heartbeats (seconds)"}
     )
 
+    log_level: Optional[str] = field(
+        default=None,
+        metadata={
+            "help": "Logging level",
+            "valid_values": [
+                "FATAL",
+                "ERROR",
+                "WARNING",
+                "WARNING",
+                "INFO",
+                "DEBUG",
+                "NOTSET",
+            ],
+        },
+    )
+
 
 @dataclass
 class BaseEmbeddingModelParameters(BaseModelParameters):

pilot/openapi/api_v1/api_v1.py

@@ -38,8 +38,6 @@ from pilot.server.knowledge.request.request import KnowledgeSpaceRequest
 from pilot.scene.base_chat import BaseChat
 from pilot.scene.base import ChatScene
 from pilot.scene.chat_factory import ChatFactory
-from pilot.configs.model_config import LOGDIR
-from pilot.utils import build_logger
 from pilot.common.schema import DBType
 from pilot.memory.chat_history.duckdb_history import DuckdbHistoryMemory
 from pilot.scene.message import OnceConversation
@@ -409,7 +407,7 @@ async def model_types(controller: BaseModelController = Depends(get_model_contro
 
 
 @router.get("/v1/model/supports")
-async def model_types(worker_manager: WorkerManager = Depends(get_worker_manager)):
+async def model_supports(worker_manager: WorkerManager = Depends(get_worker_manager)):
     logger.info(f"/controller/model/supports")
     try:
         models = await worker_manager.supported_models()

pilot/openapi/api_v1/editor/api_editor_v1.py

@@ -6,12 +6,11 @@ from fastapi import (
 )
 
 from typing import List
+import logging
 
 from pilot.configs.config import Config
 
 from pilot.scene.chat_factory import ChatFactory
-from pilot.configs.model_config import LOGDIR
-from pilot.utils import build_logger
 
 from pilot.openapi.api_view_model import (
     Result,
@@ -34,7 +33,8 @@ from pilot.scene.chat_db.data_loader import DbDataLoader
 router = APIRouter()
 CFG = Config()
 CHAT_FACTORY = ChatFactory()
-logger = build_logger("api_editor_v1", LOGDIR + "api_editor_v1.log")
+
+logger = logging.getLogger(__name__)
 
 
 @router.get("/v1/editor/db/tables", response_model=Result[DbTable])

pilot/out_parser/base.py

@@ -2,18 +2,17 @@ from __future__ import annotations
 
 import json
 from abc import ABC
+import logging
 from dataclasses import asdict
 from typing import Any, Dict, TypeVar, Union
 
 from pilot.configs.config import Config
-from pilot.configs.model_config import LOGDIR
 from pilot.model.base import ModelOutput
-from pilot.utils import build_logger
 
 T = TypeVar("T")
 ResponseTye = Union[str, bytes, ModelOutput]
 
-logger = build_logger("webserver", LOGDIR + "DbChatOutputParser.log")
+logger = logging.getLogger(__name__)
 
 CFG = Config()
 

pilot/scene/base_chat.py

@@ -1,11 +1,11 @@
 import datetime
 import traceback
 import warnings
+import logging
 from abc import ABC, abstractmethod
 from typing import Any, List, Dict
 
 from pilot.configs.config import Config
-from pilot.configs.model_config import LOGDIR
 from pilot.component import ComponentType
 from pilot.memory.chat_history.base import BaseChatHistoryMemory
 from pilot.memory.chat_history.duckdb_history import DuckdbHistoryMemory
@@ -14,10 +14,10 @@ from pilot.memory.chat_history.mem_history import MemHistoryMemory
 from pilot.prompts.prompt_new import PromptTemplate
 from pilot.scene.base_message import ModelMessage, ModelMessageRoleType
 from pilot.scene.message import OnceConversation
-from pilot.utils import build_logger, get_or_create_event_loop
+from pilot.utils import get_or_create_event_loop
 from pydantic import Extra
 
-logger = build_logger("BaseChat", LOGDIR + "BaseChat.log")
+logger = logging.getLogger(__name__)
 headers = {"User-Agent": "dbgpt Client"}
 CFG = Config()
 

pilot/scene/chat_dashboard/data_loader.py

@@ -1,13 +1,12 @@
 from typing import List
 from decimal import Decimal
+import logging
 
 from pilot.configs.config import Config
-from pilot.configs.model_config import LOGDIR
-from pilot.utils import build_logger
 from pilot.scene.chat_dashboard.data_preparation.report_schma import ValueItem
 
 CFG = Config()
-logger = build_logger("dashboard_data", LOGDIR + "dashboard_data.log")
+logger = logging.getLogger(__name__)
 
 
 class DashboardDataLoader:

pilot/scene/chat_dashboard/out_parser.py

@@ -1,8 +1,8 @@
 import json
+import logging
+
 from typing import NamedTuple, List
-from pilot.utils import build_logger
 from pilot.out_parser.base import BaseOutputParser, T
-from pilot.configs.model_config import LOGDIR
 from pilot.scene.base import ChatScene
 
 
@@ -13,7 +13,7 @@ class ChartItem(NamedTuple):
     showcase: str
 
 
-logger = build_logger("webserver", LOGDIR + "ChatDashboardOutputParser.log")
+logger = logging.getLogger(__name__)
 
 
 class ChatDashboardOutputParser(BaseOutputParser):

pilot/scene/chat_data/chat_excel/excel_analyze/out_parser.py

@@ -1,11 +1,7 @@
 import json
-import re
+import logging
-from abc import ABC, abstractmethod
 from typing import Dict, NamedTuple, List
-import pandas as pd
-from pilot.utils import build_logger
 from pilot.out_parser.base import BaseOutputParser, T
-from pilot.configs.model_config import LOGDIR
 from pilot.configs.config import Config
 
 CFG = Config()
@@ -17,7 +13,7 @@ class ExcelAnalyzeResponse(NamedTuple):
     display: str
 
 
-logger = build_logger("chat_excel", LOGDIR + "ChatExcel.log")
+logger = logging.getLogger(__name__)
 
 
 class ChatExcelOutputParser(BaseOutputParser):

pilot/scene/chat_data/chat_excel/excel_learning/out_parser.py

@@ -1,11 +1,7 @@
 import json
-import re
+import logging
-from abc import ABC, abstractmethod
 from typing import Dict, NamedTuple, List
-import pandas as pd
-from pilot.utils import build_logger
 from pilot.out_parser.base import BaseOutputParser, T
-from pilot.configs.model_config import LOGDIR
 from pilot.configs.config import Config
 
 CFG = Config()
@@ -17,7 +13,7 @@ class ExcelResponse(NamedTuple):
     plans: List
 
 
-logger = build_logger("chat_excel", LOGDIR + "ChatExcel.log")
+logger = logging.getLogger(__name__)
 
 
 class LearningExcelOutputParser(BaseOutputParser):

pilot/scene/chat_db/auto_execute/out_parser.py

@@ -1,8 +1,7 @@
 import json
 from typing import Dict, NamedTuple
-from pilot.utils import build_logger
+import logging
 from pilot.out_parser.base import BaseOutputParser, T
-from pilot.configs.model_config import LOGDIR
 from pilot.configs.config import Config
 from pilot.scene.chat_db.data_loader import DbDataLoader
 
@@ -14,7 +13,7 @@ class SqlAction(NamedTuple):
     thoughts: Dict
 
 
-logger = build_logger("webserver", LOGDIR + "DbChatOutputParser.log")
+logger = logging.getLogger(__name__)
 
 
 class DbChatOutputParser(BaseOutputParser):

pilot/scene/chat_db/professional_qa/out_parser.py

@@ -1,8 +1,4 @@
-from pilot.configs.model_config import LOGDIR
 from pilot.out_parser.base import BaseOutputParser, T
-from pilot.utils import build_logger
-
-logger = build_logger("webserver", LOGDIR + "DbChatOutputParser.log")
 
 
 class NormalChatOutputParser(BaseOutputParser):

pilot/scene/chat_execution/out_parser.py

@@ -1,11 +1,10 @@
 import json
+import logging
 from typing import Dict, NamedTuple
-from pilot.utils import build_logger
 from pilot.out_parser.base import BaseOutputParser, T
-from pilot.configs.model_config import LOGDIR
 
 
-logger = build_logger("webserver", LOGDIR + "DbChatOutputParser.log")
+logger = logging.getLogger(__name__)
 
 
 class PluginAction(NamedTuple):

pilot/scene/chat_knowledge/inner_db_summary/out_parser.py

@@ -1,9 +1,7 @@
-from pilot.utils import build_logger
+import logging
 from pilot.out_parser.base import BaseOutputParser, T
-from pilot.configs.model_config import LOGDIR
 
-
+logger = logging.getLogger(__name__)
-logger = build_logger("webserver", LOGDIR + "DbChatOutputParser.log")
 
 
 class NormalChatOutputParser(BaseOutputParser):

pilot/scene/chat_knowledge/v1/chat.py

@@ -60,17 +60,24 @@ class ChatKnowledge(BaseChat):
 
     async def stream_call(self):
         input_values = self.generate_input_values()
-        async for output in super().stream_call():
         # Source of knowledge file
         relations = input_values.get("relations")
+        last_output = None
+        async for output in super().stream_call():
+            last_output = output
+            yield output
+
         if (
             CFG.KNOWLEDGE_CHAT_SHOW_RELATIONS
+            and last_output
             and type(relations) == list
             and len(relations) > 0
-                and hasattr(output, "text")
+            and hasattr(last_output, "text")
         ):
-                output.text = output.text + "\trelations:" + ",".join(relations)
+            last_output.text = (
-            yield output
+                last_output.text + "\n\nrelations:\n\n" + ",".join(relations)
+            )
+            yield last_output
 
     def generate_input_values(self):
         if self.space_context:

pilot/scene/chat_knowledge/v1/out_parser.py

@@ -1,9 +1,8 @@
-from pilot.utils import build_logger
+import logging
 from pilot.out_parser.base import BaseOutputParser, T
-from pilot.configs.model_config import LOGDIR
 
 
-logger = build_logger("webserver", LOGDIR + "DbChatOutputParser.log")
+logger = logging.getLogger(__name__)
 
 
 class NormalChatOutputParser(BaseOutputParser):

pilot/scene/chat_normal/out_parser.py

@@ -1,9 +1,8 @@
-from pilot.utils import build_logger
+import logging
 from pilot.out_parser.base import BaseOutputParser, T
-from pilot.configs.model_config import LOGDIR
 
 
-logger = build_logger("webserver", LOGDIR + "DbChatOutputParser.log")
+logger = logging.getLogger(__name__)
 
 
 class NormalChatOutputParser(BaseOutputParser):

pilot/server/base.py

@@ -108,7 +108,7 @@ class WebWerverParameters(BaseParameters):
         },
     )
     log_level: Optional[str] = field(
-        default="INFO",
+        default=None,
         metadata={
             "help": "Logging level",
             "valid_values": [

pilot/server/dbgpt_server.py

@@ -33,7 +33,11 @@ from pilot.openapi.api_v1.editor.api_editor_v1 import router as api_editor_route
 from pilot.openapi.api_v1.feedback.api_fb_v1 import router as api_fb_v1
 from pilot.commands.disply_type.show_chart_gen import static_message_img_path
 from pilot.model.cluster import initialize_worker_manager_in_client
-from pilot.utils.utils import setup_logging, logging_str_to_uvicorn_level
+from pilot.utils.utils import (
+    setup_logging,
+    _get_logging_level,
+    logging_str_to_uvicorn_level,
+)
 
 static_file_path = os.path.join(os.getcwd(), "server/static")
 
@@ -111,7 +115,11 @@ def initialize_app(param: WebWerverParameters = None, args: List[str] = None):
         )
         param = WebWerverParameters(**vars(parser.parse_args(args=args)))
 
-    setup_logging(logging_level=param.log_level)
+    if not param.log_level:
+        param.log_level = _get_logging_level()
+    setup_logging(
+        "pilot", logging_level=param.log_level, logger_filename="dbgpt_webserver.log"
+    )
     # Before start
     system_app.before_start()
 

pilot/server/knowledge/_cli/knowledge_cli.py

@@ -1,9 +1,12 @@
 import click
 import logging
+import os
+import functools
 
 from pilot.configs.model_config import DATASETS_DIR
 
-API_ADDRESS: str = "http://127.0.0.1:5000"
+_DEFAULT_API_ADDRESS: str = "http://127.0.0.1:5000"
+API_ADDRESS: str = _DEFAULT_API_ADDRESS
 
 logger = logging.getLogger("dbgpt_cli")
 
@@ -15,38 +18,51 @@ logger = logging.getLogger("dbgpt_cli")
     default=API_ADDRESS,
     required=False,
     show_default=True,
-    help=("Address of the Api server."),
+    help=(
+        "Address of the Api server(If not set, try to read from environment variable: API_ADDRESS)."
+    ),
 )
 def knowledge_cli_group(address: str):
     """Knowledge command line tool"""
     global API_ADDRESS
+    if address == _DEFAULT_API_ADDRESS:
+        address = os.getenv("API_ADDRESS", _DEFAULT_API_ADDRESS)
     API_ADDRESS = address
 
 
-@knowledge_cli_group.command()
+def add_knowledge_options(func):
     @click.option(
-    "--vector_name",
+        "--space_name",
         required=False,
         type=str,
         default="default",
         show_default=True,
-    help="Your vector store name",
+        help="Your knowledge space name",
     )
+    @functools.wraps(func)
+    def wrapper(*args, **kwargs):
+        return func(*args, **kwargs)
+
+    return wrapper
+
+
+@knowledge_cli_group.command()
+@add_knowledge_options
 @click.option(
     "--vector_store_type",
     required=False,
     type=str,
     default="Chroma",
     show_default=True,
-    help="Vector store type",
+    help="Vector store type.",
 )
 @click.option(
-    "--local_doc_dir",
+    "--local_doc_path",
     required=False,
     type=str,
     default=DATASETS_DIR,
     show_default=True,
-    help="Your document directory",
+    help="Your document directory or document file path.",
 )
 @click.option(
     "--skip_wrong_doc",
@@ -54,31 +70,165 @@ def knowledge_cli_group(address: str):
     type=bool,
     default=False,
     is_flag=True,
-    show_default=True,
+    help="Skip wrong document.",
-    help="Skip wrong document",
+)
+@click.option(
+    "--overwrite",
+    required=False,
+    type=bool,
+    default=False,
+    is_flag=True,
+    help="Overwrite existing document(they has same name).",
 )
 @click.option(
     "--max_workers",
     required=False,
     type=int,
     default=None,
-    help="The maximum number of threads that can be used to upload document",
+    help="The maximum number of threads that can be used to upload document.",
+)
+@click.option(
+    "--pre_separator",
+    required=False,
+    type=str,
+    default=None,
+    help="Preseparator, this separator is used for pre-splitting before the document is "
+    "actually split by the text splitter. Preseparator are not included in the vectorized text. ",
+)
+@click.option(
+    "--separator",
+    required=False,
+    type=str,
+    default=None,
+    help="This is the document separator. Currently, only one separator is supported.",
+)
+@click.option(
+    "--chunk_size",
+    required=False,
+    type=int,
+    default=None,
+    help="Maximum size of chunks to split.",
+)
+@click.option(
+    "--chunk_overlap",
+    required=False,
+    type=int,
+    default=None,
+    help="Overlap in characters between chunks.",
 )
 def load(
-    vector_name: str,
+    space_name: str,
     vector_store_type: str,
-    local_doc_dir: str,
+    local_doc_path: str,
     skip_wrong_doc: bool,
+    overwrite: bool,
     max_workers: int,
+    pre_separator: str,
+    separator: str,
+    chunk_size: int,
+    chunk_overlap: int,
 ):
-    """Load your local knowledge to DB-GPT"""
+    """Load your local documents to DB-GPT"""
     from pilot.server.knowledge._cli.knowledge_client import knowledge_init
 
     knowledge_init(
         API_ADDRESS,
-        vector_name,
+        space_name,
         vector_store_type,
-        local_doc_dir,
+        local_doc_path,
         skip_wrong_doc,
+        overwrite,
         max_workers,
+        pre_separator,
+        separator,
+        chunk_size,
+        chunk_overlap,
+    )
+
+
+@knowledge_cli_group.command()
+@add_knowledge_options
+@click.option(
+    "--doc_name",
+    required=False,
+    type=str,
+    default=None,
+    help="The document name you want to delete. If doc_name is None, this command will delete the whole space.",
+)
+@click.option(
+    "-y",
+    required=False,
+    type=bool,
+    default=False,
+    is_flag=True,
+    help="Confirm your choice",
+)
+def delete(space_name: str, doc_name: str, y: bool):
+    """Delete your knowledge space or document in space"""
+    from pilot.server.knowledge._cli.knowledge_client import knowledge_delete
+
+    knowledge_delete(API_ADDRESS, space_name, doc_name, confirm=y)
+
+
+@knowledge_cli_group.command()
+@click.option(
+    "--space_name",
+    required=False,
+    type=str,
+    default=None,
+    show_default=True,
+    help="Your knowledge space name. If None, list all spaces",
+)
+@click.option(
+    "--doc_id",
+    required=False,
+    type=int,
+    default=None,
+    show_default=True,
+    help="Your document id in knowledge space. If Not None, list all chunks in current document",
+)
+@click.option(
+    "--page",
+    required=False,
+    type=int,
+    default=1,
+    show_default=True,
+    help="The page for every query",
+)
+@click.option(
+    "--page_size",
+    required=False,
+    type=int,
+    default=20,
+    show_default=True,
+    help="The page size for every query",
+)
+@click.option(
+    "--show_content",
+    required=False,
+    type=bool,
+    default=False,
+    is_flag=True,
+    help="Query the document content of chunks",
+)
+@click.option(
+    "--output",
+    required=False,
+    type=click.Choice(["text", "html", "csv", "latex", "json"]),
+    default="text",
+    help="The output format",
+)
+def list(
+    space_name: str,
+    doc_id: int,
+    page: int,
+    page_size: int,
+    show_content: bool,
+    output: str,
+):
+    """List knowledge space"""
+    from pilot.server.knowledge._cli.knowledge_client import knowledge_list
+
+    knowledge_list(
+        API_ADDRESS, space_name, page, page_size, doc_id, show_content, output
     )

pilot/server/knowledge/_cli/knowledge_client.py

@@ -62,6 +62,9 @@ class KnowledgeApiClient(ApiClient):
             else:
                 raise e
 
+    def space_delete(self, request: KnowledgeSpaceRequest):
+        return self._post("/knowledge/space/delete", data=request)
+
     def space_list(self, request: KnowledgeSpaceRequest):
         return self._post("/knowledge/space/list", data=request)
 
@@ -69,6 +72,10 @@ class KnowledgeApiClient(ApiClient):
         url = f"/knowledge/{space_name}/document/add"
         return self._post(url, data=request)
 
+    def document_delete(self, space_name: str, request: KnowledgeDocumentRequest):
+        url = f"/knowledge/{space_name}/document/delete"
+        return self._post(url, data=request)
+
     def document_list(self, space_name: str, query_request: DocumentQueryRequest):
         url = f"/knowledge/{space_name}/document/list"
         return self._post(url, data=query_request)
@@ -97,15 +104,20 @@ class KnowledgeApiClient(ApiClient):
 
 def knowledge_init(
     api_address: str,
-    vector_name: str,
+    space_name: str,
     vector_store_type: str,
-    local_doc_dir: str,
+    local_doc_path: str,
     skip_wrong_doc: bool,
-    max_workers: int = None,
+    overwrite: bool,
+    max_workers: int,
+    pre_separator: str,
+    separator: str,
+    chunk_size: int,
+    chunk_overlap: int,
 ):
     client = KnowledgeApiClient(api_address)
     space = KnowledgeSpaceRequest()
-    space.name = vector_name
+    space.name = space_name
     space.desc = "DB-GPT cli"
     space.vector_type = vector_store_type
     space.owner = "DB-GPT"
@@ -124,24 +136,260 @@ def knowledge_init(
     def upload(filename: str):
         try:
             logger.info(f"Begin upload document: {filename} to {space.name}")
+            doc_id = None
+            try:
                 doc_id = client.document_upload(
                     space.name, filename, KnowledgeType.DOCUMENT.value, filename
                 )
-            client.document_sync(space.name, DocumentSyncRequest(doc_ids=[doc_id]))
+            except Exception as ex:
+                if overwrite and "have already named" in str(ex):
+                    logger.warn(
+                        f"Document {filename} already exist in space {space.name}, overwrite it"
+                    )
+                    client.document_delete(
+                        space.name, KnowledgeDocumentRequest(doc_name=filename)
+                    )
+                    doc_id = client.document_upload(
+                        space.name, filename, KnowledgeType.DOCUMENT.value, filename
+                    )
+                else:
+                    raise ex
+            sync_req = DocumentSyncRequest(doc_ids=[doc_id])
+            if pre_separator:
+                sync_req.pre_separator = pre_separator
+            if separator:
+                sync_req.separators = [separator]
+            if chunk_size:
+                sync_req.chunk_size = chunk_size
+            if chunk_overlap:
+                sync_req.chunk_overlap = chunk_overlap
+
+            client.document_sync(space.name, sync_req)
+            return doc_id
         except Exception as e:
             if skip_wrong_doc:
                 logger.warn(f"Upload {filename} to {space.name} failed: {str(e)}")
             else:
                 raise e
 
+    if not os.path.exists(local_doc_path):
+        raise Exception(f"{local_doc_path} not exists")
+
     with ThreadPoolExecutor(max_workers=max_workers) as pool:
         tasks = []
-        for root, _, files in os.walk(local_doc_dir, topdown=False):
+        file_names = []
+        if os.path.isdir(local_doc_path):
+            for root, _, files in os.walk(local_doc_path, topdown=False):
                 for file in files:
-                filename = os.path.join(root, file)
+                    file_names.append(os.path.join(root, file))
-                tasks.append(pool.submit(upload, filename))
+        else:
+            # Single file
+            file_names.append(local_doc_path)
+
+        [tasks.append(pool.submit(upload, filename)) for filename in file_names]
+
         doc_ids = [r.result() for r in as_completed(tasks)]
         doc_ids = list(filter(lambda x: x, doc_ids))
         if not doc_ids:
             logger.warn("Warning: no document to sync")
             return
+
+
+from prettytable import PrettyTable
+
+
+class _KnowledgeVisualizer:
+    def __init__(self, api_address: str, out_format: str):
+        self.client = KnowledgeApiClient(api_address)
+        self.out_format = out_format
+        self.out_kwargs = {}
+        if out_format == "json":
+            self.out_kwargs["ensure_ascii"] = False
+
+    def print_table(self, table):
+        print(table.get_formatted_string(out_format=self.out_format, **self.out_kwargs))
+
+    def list_spaces(self):
+        spaces = self.client.space_list(KnowledgeSpaceRequest())
+        table = PrettyTable(
+            ["Space ID", "Space Name", "Vector Type", "Owner", "Description"],
+            title="All knowledge spaces",
+        )
+        for sp in spaces:
+            context = sp.get("context")
+            table.add_row(
+                [
+                    sp.get("id"),
+                    sp.get("name"),
+                    sp.get("vector_type"),
+                    sp.get("owner"),
+                    sp.get("desc"),
+                ]
+            )
+        self.print_table(table)
+
+    def list_documents(self, space_name: str, page: int, page_size: int):
+        space_data = self.client.document_list(
+            space_name, DocumentQueryRequest(page=page, page_size=page_size)
+        )
+
+        space_table = PrettyTable(
+            [
+                "Space Name",
+                "Total Documents",
+                "Current Page",
+                "Current Size",
+                "Page Size",
+            ],
+            title=f"Space {space_name} description",
+        )
+        space_table.add_row(
+            [space_name, space_data["total"], page, len(space_data["data"]), page_size]
+        )
+
+        table = PrettyTable(
+            [
+                "Space Name",
+                "Document ID",
+                "Document Name",
+                "Type",
+                "Chunks",
+                "Last Sync",
+                "Status",
+                "Result",
+            ],
+            title=f"Documents of space {space_name}",
+        )
+        for doc in space_data["data"]:
+            table.add_row(
+                [
+                    space_name,
+                    doc.get("id"),
+                    doc.get("doc_name"),
+                    doc.get("doc_type"),
+                    doc.get("chunk_size"),
+                    doc.get("last_sync"),
+                    doc.get("status"),
+                    doc.get("result"),
+                ]
+            )
+        if self.out_format == "text":
+            self.print_table(space_table)
+            print("")
+        self.print_table(table)
+
+    def list_chunks(
+        self,
+        space_name: str,
+        doc_id: int,
+        page: int,
+        page_size: int,
+        show_content: bool,
+    ):
+        doc_data = self.client.chunk_list(
+            space_name,
+            ChunkQueryRequest(document_id=doc_id, page=page, page_size=page_size),
+        )
+
+        doc_table = PrettyTable(
+            [
+                "Space Name",
+                "Document ID",
+                "Total Chunks",
+                "Current Page",
+                "Current Size",
+                "Page Size",
+            ],
+            title=f"Document {doc_id} in {space_name} description",
+        )
+        doc_table.add_row(
+            [
+                space_name,
+                doc_id,
+                doc_data["total"],
+                page,
+                len(doc_data["data"]),
+                page_size,
+            ]
+        )
+
+        table = PrettyTable(
+            ["Space Name", "Document ID", "Document Name", "Content", "Meta Data"],
+            title=f"chunks of document id {doc_id} in space {space_name}",
+        )
+        for chunk in doc_data["data"]:
+            table.add_row(
+                [
+                    space_name,
+                    doc_id,
+                    chunk.get("doc_name"),
+                    chunk.get("content") if show_content else "[Hidden]",
+                    chunk.get("meta_info"),
+                ]
+            )
+        if self.out_format == "text":
+            self.print_table(doc_table)
+            print("")
+        self.print_table(table)
+
+
+def knowledge_list(
+    api_address: str,
+    space_name: str,
+    page: int,
+    page_size: int,
+    doc_id: int,
+    show_content: bool,
+    out_format: str,
+):
+    visualizer = _KnowledgeVisualizer(api_address, out_format)
+    if not space_name:
+        visualizer.list_spaces()
+    elif not doc_id:
+        visualizer.list_documents(space_name, page, page_size)
+    else:
+        visualizer.list_chunks(space_name, doc_id, page, page_size, show_content)
+
+
+def knowledge_delete(
+    api_address: str, space_name: str, doc_name: str, confirm: bool = False
+):
+    client = KnowledgeApiClient(api_address)
+    space = KnowledgeSpaceRequest()
+    space.name = space_name
+    space_list = client.space_list(KnowledgeSpaceRequest(name=space.name))
+    if not space_list:
+        raise Exception(f"No knowledge space name {space_name}")
+
+    if not doc_name:
+        if not confirm:
+            # Confirm by user
+            user_input = (
+                input(
+                    f"Are you sure you want to delete the whole knowledge space {space_name}? Type 'yes' to confirm: "
+                )
+                .strip()
+                .lower()
+            )
+            if user_input != "yes":
+                logger.warn("Delete operation cancelled.")
+                return
+        client.space_delete(space)
+        logger.info("Delete the whole knowledge space successfully!")
+    else:
+        if not confirm:
+            # Confirm by user
+            user_input = (
+                input(
+                    f"Are you sure you want to delete the doucment {doc_name} in knowledge space {space_name}? Type 'yes' to confirm: "
+                )
+                .strip()
+                .lower()
+            )
+            if user_input != "yes":
+                logger.warn("Delete operation cancelled.")
+                return
+        client.document_delete(space_name, KnowledgeDocumentRequest(doc_name=doc_name))
+        logger.info(
+            f"Delete the doucment {doc_name} in knowledge space {space_name} successfully!"
+        )

pilot/server/knowledge/api.py

@@ -1,6 +1,7 @@
 import os
 import shutil
 import tempfile
+import logging
 
 from fastapi import APIRouter, File, UploadFile, Form
 
@@ -27,6 +28,8 @@ from pilot.server.knowledge.request.request import (
 
 from pilot.server.knowledge.request.request import KnowledgeSpaceRequest
 
+logger = logging.getLogger(__name__)
+
 CFG = Config()
 router = APIRouter()
 
@@ -159,10 +162,10 @@ async def document_upload(
 
 @router.post("/knowledge/{space_name}/document/sync")
 def document_sync(space_name: str, request: DocumentSyncRequest):
-    print(f"Received params: {space_name}, {request}")
+    logger.info(f"Received params: {space_name}, {request}")
     try:
         knowledge_space_service.sync_knowledge_document(
-            space_name=space_name, doc_ids=request.doc_ids
+            space_name=space_name, sync_request=request
         )
         return Result.succ([])
     except Exception as e:

pilot/server/knowledge/request/request.py

@@ -1,4 +1,4 @@
-from typing import List
+from typing import List, Optional
 
 from pydantic import BaseModel
 from fastapi import UploadFile
@@ -54,10 +54,25 @@ class DocumentQueryRequest(BaseModel):
 
 
 class DocumentSyncRequest(BaseModel):
-    """doc_ids: doc ids"""
+    """Sync request"""
 
+    """doc_ids: doc ids"""
     doc_ids: List
 
+    """Preseparator, this separator is used for pre-splitting before the document is actually split by the text splitter.
+    Preseparator are not included in the vectorized text. 
+    """
+    pre_separator: Optional[str] = None
+
+    """Custom separators"""
+    separators: Optional[List[str]] = None
+
+    """Custom chunk size"""
+    chunk_size: Optional[int] = None
+
+    """Custom chunk overlap"""
+    chunk_overlap: Optional[int] = None
+
 
 class ChunkQueryRequest(BaseModel):
     """id: id"""

pilot/server/knowledge/service.py

@@ -1,5 +1,5 @@
 import json
-import threading
+import logging
 from datetime import datetime
 
 from pilot.vector_store.connector import VectorStoreConnector
@@ -12,7 +12,6 @@ from pilot.configs.model_config import (
 from pilot.component import ComponentType
 from pilot.utils.executor_utils import ExecutorFactory
 
-from pilot.logs import logger
 from pilot.server.knowledge.chunk_db import (
     DocumentChunkEntity,
     DocumentChunkDao,
@@ -31,6 +30,7 @@ from pilot.server.knowledge.request.request import (
     DocumentQueryRequest,
     ChunkQueryRequest,
     SpaceArgumentRequest,
+    DocumentSyncRequest,
 )
 from enum import Enum
 
@@ -44,6 +44,7 @@ knowledge_space_dao = KnowledgeSpaceDao()
 knowledge_document_dao = KnowledgeDocumentDao()
 document_chunk_dao = DocumentChunkDao()
 
+logger = logging.getLogger(__name__)
 CFG = Config()
 
 
@@ -107,7 +108,6 @@ class KnowledgeService:
             res.owner = space.owner
             res.gmt_created = space.gmt_created
             res.gmt_modified = space.gmt_modified
-            res.owner = space.owner
             res.context = space.context
             query = KnowledgeDocumentEntity(space=space.name)
             doc_count = knowledge_document_dao.get_knowledge_documents_count(query)
@@ -155,9 +155,10 @@ class KnowledgeService:
 
     """sync knowledge document chunk into vector store"""
 
-    def sync_knowledge_document(self, space_name, doc_ids):
+    def sync_knowledge_document(self, space_name, sync_request: DocumentSyncRequest):
         from pilot.embedding_engine.embedding_engine import EmbeddingEngine
         from pilot.embedding_engine.embedding_factory import EmbeddingFactory
+        from pilot.embedding_engine.pre_text_splitter import PreTextSplitter
         from langchain.text_splitter import (
             RecursiveCharacterTextSplitter,
             SpacyTextSplitter,
@@ -165,6 +166,7 @@ class KnowledgeService:
 
         # import langchain is very very slow!!!
 
+        doc_ids = sync_request.doc_ids
         for doc_id in doc_ids:
             query = KnowledgeDocumentEntity(
                 id=doc_id,
@@ -190,24 +192,43 @@ class KnowledgeService:
                 if space_context is None
                 else int(space_context["embedding"]["chunk_overlap"])
             )
+            if sync_request.chunk_size:
+                chunk_size = sync_request.chunk_size
+            if sync_request.chunk_overlap:
+                chunk_overlap = sync_request.chunk_overlap
+            separators = sync_request.separators or None
             if CFG.LANGUAGE == "en":
                 text_splitter = RecursiveCharacterTextSplitter(
+                    separators=separators,
                     chunk_size=chunk_size,
                     chunk_overlap=chunk_overlap,
                     length_function=len,
                 )
             else:
+                if separators and len(separators) > 1:
+                    raise ValueError(
+                        "SpacyTextSplitter do not support multiple separators"
+                    )
                 try:
+                    separator = "\n\n" if not separators else separators[0]
                     text_splitter = SpacyTextSplitter(
+                        separator=separator,
                         pipeline="zh_core_web_sm",
                         chunk_size=chunk_size,
                         chunk_overlap=chunk_overlap,
                     )
                 except Exception:
                     text_splitter = RecursiveCharacterTextSplitter(
+                        separators=separators,
                         chunk_size=chunk_size,
                         chunk_overlap=chunk_overlap,
                     )
+            if sync_request.pre_separator:
+                logger.info(f"Use preseparator, {sync_request.pre_separator}")
+                text_splitter = PreTextSplitter(
+                    pre_separator=sync_request.pre_separator,
+                    text_splitter_impl=text_splitter,
+                )
             embedding_factory = CFG.SYSTEM_APP.get_component(
                 "embedding_factory", EmbeddingFactory
             )

pilot/speech/eleven_labs.py

@@ -1,6 +1,6 @@
 """ElevenLabs speech module"""
 import os
-
+import logging
 import requests
 
 from pilot.configs.config import Config
@@ -8,6 +8,8 @@ from pilot.speech.base import VoiceBase
 
 PLACEHOLDERS = {"your-voice-id"}
 
+logger = logging.getLogger(__name__)
+
 
 class ElevenLabsSpeech(VoiceBase):
     """ElevenLabs speech class"""
@@ -68,7 +70,6 @@ class ElevenLabsSpeech(VoiceBase):
         Returns:
             bool: True if the request was successful, False otherwise
         """
-        from pilot.logs import logger
         from playsound import playsound
 
         tts_url = (

pilot/summary/db_summary_client.py

@@ -1,5 +1,6 @@
 import json
 import uuid
+import logging
 
 from pilot.common.schema import DBType
 from pilot.component import SystemApp
@@ -7,16 +8,14 @@ from pilot.configs.config import Config
 from pilot.configs.model_config import (
     KNOWLEDGE_UPLOAD_ROOT_PATH,
     EMBEDDING_MODEL_CONFIG,
-    LOGDIR,
 )
+
 from pilot.scene.base import ChatScene
 from pilot.scene.base_chat import BaseChat
 from pilot.scene.chat_factory import ChatFactory
 from pilot.summary.rdbms_db_summary import RdbmsSummary
-from pilot.utils import build_logger
-
-logger = build_logger("db_summary", LOGDIR + "db_summary.log")
 
+logger = logging.getLogger(__name__)
 
 CFG = Config()
 chat_factory = ChatFactory()

pilot/utils/__init__.py

@@ -1,6 +1,5 @@
 from .utils import (
     get_gpu_memory,
-    build_logger,
     StreamToLogger,
     disable_torch_init,
     pretty_print_semaphore,

pilot/utils/api_utils.py

@@ -5,6 +5,8 @@ from dataclasses import is_dataclass, asdict
 
 T = TypeVar("T")
 
+logger = logging.getLogger(__name__)
+
 
 def _extract_dataclass_from_generic(type_hint: Type[T]) -> Union[Type[T], None]:
     import typing_inspect
@@ -21,7 +23,7 @@ def _build_request(self, func, path, method, *args, **kwargs):
         raise TypeError("Return type must be annotated in the decorated function.")
 
     actual_dataclass = _extract_dataclass_from_generic(return_type)
-    logging.debug(f"return_type: {return_type}, actual_dataclass: {actual_dataclass}")
+    logger.debug(f"return_type: {return_type}, actual_dataclass: {actual_dataclass}")
     if not actual_dataclass:
         actual_dataclass = return_type
     sig = signature(func)
@@ -55,7 +57,7 @@ def _build_request(self, func, path, method, *args, **kwargs):
     else:  # For GET, DELETE, etc.
         request_params["params"] = request_data
 
-    logging.debug(f"request_params: {request_params}, args: {args}, kwargs: {kwargs}")
+    logger.debug(f"request_params: {request_params}, args: {args}, kwargs: {kwargs}")
     return return_type, actual_dataclass, request_params
 
 

pilot/utils/utils.py

@@ -20,7 +20,7 @@ def _get_logging_level() -> str:
     return os.getenv("DBGPT_LOG_LEVEL", "INFO")
 
 
-def setup_logging(logging_level=None, logger_name: str = None):
+def setup_logging_level(logging_level=None, logger_name: str = None):
     if not logging_level:
         logging_level = _get_logging_level()
     if type(logging_level) is str:
@@ -32,6 +32,19 @@ def setup_logging(logging_level=None, logger_name: str = None):
         logging.basicConfig(level=logging_level, encoding="utf-8")
 
 
+def setup_logging(logger_name: str, logging_level=None, logger_filename: str = None):
+    if not logging_level:
+        logging_level = _get_logging_level()
+    logger = _build_logger(logger_name, logging_level, logger_filename)
+    try:
+        import coloredlogs
+
+        color_level = logging_level if logging_level else "INFO"
+        coloredlogs.install(level=color_level, logger=logger)
+    except ImportError:
+        pass
+
+
 def get_gpu_memory(max_gpus=None):
     import torch
 
@@ -53,7 +66,7 @@ def get_gpu_memory(max_gpus=None):
     return gpu_memory
 
 
-def build_logger(logger_name, logger_filename):
+def _build_logger(logger_name, logging_level=None, logger_filename: str = None):
     global handler
 
     formatter = logging.Formatter(
@@ -63,7 +76,7 @@ def build_logger(logger_name, logger_filename):
 
     # Set the format of root handlers
     if not logging.getLogger().handlers:
-        setup_logging()
+        setup_logging_level(logging_level=logging_level)
     logging.getLogger().handlers[0].setFormatter(formatter)
 
     # Redirect stdout and stderr to loggers
@@ -78,7 +91,7 @@ def build_logger(logger_name, logger_filename):
     # sys.stderr = sl
 
     # Add a file handler for all loggers
-    if handler is None:
+    if handler is None and logger_filename:
         os.makedirs(LOGDIR, exist_ok=True)
         filename = os.path.join(LOGDIR, logger_filename)
         handler = logging.handlers.TimedRotatingFileHandler(
@@ -89,11 +102,9 @@ def build_logger(logger_name, logger_filename):
         for name, item in logging.root.manager.loggerDict.items():
             if isinstance(item, logging.Logger):
                 item.addHandler(handler)
-    setup_logging()
-
     # Get logger
     logger = logging.getLogger(logger_name)
-    setup_logging(logger_name=logger_name)
+    setup_logging_level(logging_level=logging_level, logger_name=logger_name)
 
     return logger
 

pilot/vector_store/chroma_store.py

@@ -1,11 +1,13 @@
 import os
+import logging
 from typing import Any
 
 from chromadb.config import Settings
 from chromadb import PersistentClient
-from pilot.logs import logger
 from pilot.vector_store.base import VectorStoreBase
 
+logger = logging.getLogger(__name__)
+
 
 class ChromaStore(VectorStoreBase):
     """chroma database"""

pilot/vector_store/milvus_store.py

@@ -1,12 +1,13 @@
 from __future__ import annotations
-
+import logging
 from typing import Any, Iterable, List, Optional, Tuple
 
 from pymilvus import Collection, DataType, connections, utility
 
-from pilot.logs import logger
 from pilot.vector_store.base import VectorStoreBase
 
+logger = logging.getLogger(__name__)
+
 
 class MilvusStore(VectorStoreBase):
     """Milvus database"""

pilot/vector_store/weaviate_store.py

@@ -1,5 +1,6 @@
 import os
 import json
+import logging
 import weaviate
 from langchain.schema import Document
 from langchain.vectorstores import Weaviate
@@ -7,9 +8,9 @@ from weaviate.exceptions import WeaviateBaseError
 
 from pilot.configs.config import Config
 from pilot.configs.model_config import KNOWLEDGE_UPLOAD_ROOT_PATH
-from pilot.logs import logger
 from pilot.vector_store.base import VectorStoreBase
 
+logger = logging.getLogger(__name__)
 CFG = Config()
 
 

setup.py

@@ -287,6 +287,7 @@ def core_requires():
     ]
 
     setup_spec.extras["framework"] = [
+        "coloredlogs",
         "httpx",
         "sqlparse==0.4.4",
         "seaborn",