KevinHuSh commited on
Commit
1daa4bd
·
1 Parent(s): 0957ae0

refine log format (#312)

Browse files

### What problem does this PR solve?

Issue link:#264
### Type of change


- [x] Documentation Update
- [x] Refactoring

README.md CHANGED
@@ -65,7 +65,7 @@
65
 
66
  - CPU >= 2 cores
67
  - RAM >= 8 GB
68
- - Docker >= 24.0.0
69
  > If you have not installed Docker on your local machine (Windows, Mac, or Linux), see [Install Docker Engine](https://docs.docker.com/engine/install/).
70
 
71
  ### 🚀 Start up the server
 
65
 
66
  - CPU >= 2 cores
67
  - RAM >= 8 GB
68
+ - Docker >= 24.0.0 & Docker Compose >= v2.26.1
69
  > If you have not installed Docker on your local machine (Windows, Mac, or Linux), see [Install Docker Engine](https://docs.docker.com/engine/install/).
70
 
71
  ### 🚀 Start up the server
README_ja.md CHANGED
@@ -65,7 +65,7 @@
65
 
66
  - CPU >= 2 cores
67
  - RAM >= 8 GB
68
- - Docker >= 24.0.0
69
  > ローカルマシン(Windows、Mac、または Linux)に Docker をインストールしていない場合は、[Docker Engine のインストール](https://docs.docker.com/engine/install/) を参照してください。
70
 
71
  ### 🚀 サーバーを起動
 
65
 
66
  - CPU >= 2 cores
67
  - RAM >= 8 GB
68
+ - Docker >= 24.0.0 & Docker Compose >= v2.26.1
69
  > ローカルマシン(Windows、Mac、または Linux)に Docker をインストールしていない場合は、[Docker Engine のインストール](https://docs.docker.com/engine/install/) を参照してください。
70
 
71
  ### 🚀 サーバーを起動
README_zh.md CHANGED
@@ -65,7 +65,7 @@
65
 
66
  - CPU >= 2 核
67
  - RAM >= 8 GB
68
- - Docker >= 24.0.0
69
  > 如果你并没有在本机安装 Docker(Windows、Mac,或者 Linux), 可以参考文档 [Install Docker Engine](https://docs.docker.com/engine/install/) 自行安装。
70
 
71
  ### 🚀 启动服务器
 
65
 
66
  - CPU >= 2 核
67
  - RAM >= 8 GB
68
+ - Docker >= 24.0.0 & Docker Compose >= v2.26.1
69
  > 如果你并没有在本机安装 Docker(Windows、Mac,或者 Linux), 可以参考文档 [Install Docker Engine](https://docs.docker.com/engine/install/) 自行安装。
70
 
71
  ### 🚀 启动服务器
api/db/db_models.py CHANGED
@@ -695,7 +695,7 @@ class Dialog(DataBaseModel):
695
  language = CharField(
696
  max_length=32,
697
  null=True,
698
- default="English",
699
  help_text="English|Chinese")
700
  llm_id = CharField(max_length=32, null=False, help_text="default llm ID")
701
  llm_setting = JSONField(null=False, default={"temperature": 0.1, "top_p": 0.3, "frequency_penalty": 0.7,
 
695
  language = CharField(
696
  max_length=32,
697
  null=True,
698
+ default="Chinese",
699
  help_text="English|Chinese")
700
  llm_id = CharField(max_length=32, null=False, help_text="default llm ID")
701
  llm_setting = JSONField(null=False, default={"temperature": 0.1, "top_p": 0.3, "frequency_penalty": 0.7,
api/settings.py CHANGED
@@ -13,15 +13,8 @@
13
  # See the License for the specific language governing permissions and
14
  # limitations under the License.
15
  #
16
- import logging
17
-
18
- from rag.utils import ELASTICSEARCH
19
- from rag.nlp import search
20
  import os
21
-
22
  from enum import IntEnum, Enum
23
-
24
- from api.utils import get_base_config, decrypt_database_config
25
  from api.utils.file_utils import get_project_base_directory
26
  from api.utils.log_utils import LoggerFactory, getLogger
27
 
@@ -32,13 +25,16 @@ LoggerFactory.set_directory(
32
  "logs",
33
  "api"))
34
  # {CRITICAL: 50, FATAL:50, ERROR:40, WARNING:30, WARN:30, INFO:20, DEBUG:10, NOTSET:0}
35
- LoggerFactory.LEVEL = 10
36
 
37
  stat_logger = getLogger("stat")
38
  access_logger = getLogger("access")
39
  database_logger = getLogger("database")
40
  chat_logger = getLogger("chat")
41
- database_logger.setLevel(logging.WARNING)
 
 
 
42
 
43
  API_VERSION = "v1"
44
  RAG_FLOW_SERVICE_NAME = "ragflow"
@@ -79,7 +75,7 @@ default_llm = {
79
  "image2text_model": "glm-4v",
80
  "asr_model": "",
81
  },
82
- "Local": {
83
  "chat_model": "qwen-14B-chat",
84
  "embedding_model": "flag-embedding",
85
  "image2text_model": "",
 
13
  # See the License for the specific language governing permissions and
14
  # limitations under the License.
15
  #
 
 
 
 
16
  import os
 
17
  from enum import IntEnum, Enum
 
 
18
  from api.utils.file_utils import get_project_base_directory
19
  from api.utils.log_utils import LoggerFactory, getLogger
20
 
 
25
  "logs",
26
  "api"))
27
  # {CRITICAL: 50, FATAL:50, ERROR:40, WARNING:30, WARN:30, INFO:20, DEBUG:10, NOTSET:0}
28
+ LoggerFactory.LEVEL = 30
29
 
30
  stat_logger = getLogger("stat")
31
  access_logger = getLogger("access")
32
  database_logger = getLogger("database")
33
  chat_logger = getLogger("chat")
34
+
35
+ from rag.utils import ELASTICSEARCH
36
+ from rag.nlp import search
37
+ from api.utils import get_base_config, decrypt_database_config
38
 
39
  API_VERSION = "v1"
40
  RAG_FLOW_SERVICE_NAME = "ragflow"
 
75
  "image2text_model": "glm-4v",
76
  "asr_model": "",
77
  },
78
+ "Ollama": {
79
  "chat_model": "qwen-14B-chat",
80
  "embedding_model": "flag-embedding",
81
  "image2text_model": "",
api/utils/log_utils.py CHANGED
@@ -26,7 +26,8 @@ from api.utils import file_utils
26
 
27
  class LoggerFactory(object):
28
  TYPE = "FILE"
29
- LOG_FORMAT = "[%(levelname)s] [%(asctime)s] [jobId] [%(process)s:%(thread)s] - [%(module)s.%(funcName)s] [line:%(lineno)d]: %(message)s"
 
30
  LEVEL = logging.DEBUG
31
  logger_dict = {}
32
  global_handler_dict = {}
 
26
 
27
  class LoggerFactory(object):
28
  TYPE = "FILE"
29
+ LOG_FORMAT = "[%(levelname)s] [%(asctime)s] [%(module)s.%(funcName)s] [line:%(lineno)d]: %(message)s"
30
+ logging.basicConfig(format=LOG_FORMAT)
31
  LEVEL = logging.DEBUG
32
  logger_dict = {}
33
  global_handler_dict = {}
rag/app/book.py CHANGED
@@ -110,7 +110,7 @@ def chunk(filename, binary=None, from_page=0, to_page=100000,
110
  [t for t in random_choices([t for t, _ in sections], k=100)])
111
  if bull >= 0:
112
  chunks = ["\n".join(ck)
113
- for ck in hierarchical_merge(bull, sections, 3)]
114
  else:
115
  sections = [s.split("@") for s, _ in sections]
116
  sections = [(pr[0], "@" + pr[1]) for pr in sections if len(pr) == 2]
 
110
  [t for t in random_choices([t for t, _ in sections], k=100)])
111
  if bull >= 0:
112
  chunks = ["\n".join(ck)
113
+ for ck in hierarchical_merge(bull, sections, 5)]
114
  else:
115
  sections = [s.split("@") for s, _ in sections]
116
  sections = [(pr[0], "@" + pr[1]) for pr in sections if len(pr) == 2]
rag/app/laws.py CHANGED
@@ -133,7 +133,7 @@ def chunk(filename, binary=None, from_page=0, to_page=100000,
133
 
134
  make_colon_as_title(sections)
135
  bull = bullets_category(sections)
136
- chunks = hierarchical_merge(bull, sections, 3)
137
  if not chunks:
138
  callback(0.99, "No chunk parsed out.")
139
 
 
133
 
134
  make_colon_as_title(sections)
135
  bull = bullets_category(sections)
136
+ chunks = hierarchical_merge(bull, sections, 5)
137
  if not chunks:
138
  callback(0.99, "No chunk parsed out.")
139
 
rag/llm/chat_model.py CHANGED
@@ -98,7 +98,7 @@ class QWenChat(Base):
98
  tk_count = 0
99
  if response.status_code == HTTPStatus.OK:
100
  ans += response.output.choices[0]['message']['content']
101
- tk_count += response.usage.output_tokens
102
  if response.output.choices[0].get("finish_reason", "") == "length":
103
  ans += "...\nFor the content length reason, it stopped, continue?" if is_english(
104
  [ans]) else "······\n由于长度的原因,回答被截断了,要继续吗?"
 
98
  tk_count = 0
99
  if response.status_code == HTTPStatus.OK:
100
  ans += response.output.choices[0]['message']['content']
101
+ tk_count += response.usage.total_tokens
102
  if response.output.choices[0].get("finish_reason", "") == "length":
103
  ans += "...\nFor the content length reason, it stopped, continue?" if is_english(
104
  [ans]) else "······\n由于长度的原因,回答被截断了,要继续吗?"