Преглед на файлове

set DLA active for KG (#3386)

### What problem does this PR solve?

### Type of change


- [x] Refactoring
tags/v0.14.0
Kevin Hu преди 11 месеца
родител
ревизия
83c6b1f308
No account linked to committer's email address
променени са 2 файла, в които са добавени 3 реда и са изтрити 3 реда
  1. 1
    1
      api/apps/document_app.py
  2. 2
    2
      rag/app/knowledge_graph.py

+ 1
- 1
api/apps/document_app.py Целия файл

@@ -537,7 +537,7 @@ def parse():
options.add_argument('--disable-dev-shm-usage')
driver = Chrome(options=options)
driver.get(url)
sections = RAGFlowHtmlParser()(driver.page_source)
sections = RAGFlowHtmlParser()("", binary=driver.page_source)
return get_json_result(data="\n".join(sections))

if 'file' not in request.files:

+ 2
- 2
rag/app/knowledge_graph.py Целия файл

@@ -9,10 +9,10 @@ def chunk(filename, binary, tenant_id, from_page=0, to_page=100000,
lang="Chinese", callback=None, **kwargs):
parser_config = kwargs.get(
"parser_config", {
"chunk_token_num": 512, "delimiter": "\n!?。;!?", "layout_recognize": False})
"chunk_token_num": 512, "delimiter": "\n!?。;!?", "layout_recognize": True})
eng = lang.lower() == "english"

parser_config["layout_recognize"] = False
parser_config["layout_recognize"] = True
sections = naive.chunk(filename, binary, from_page=from_page, to_page=to_page, section_only=True,
parser_config=parser_config, callback=callback)
chunks = build_knowledge_graph_chunks(tenant_id, sections, callback,

Loading…
Отказ
Запис