瀏覽代碼

fix special code (#473)

tags/0.3.6
Jyong 2 年之前
父節點
當前提交
2eea114ac0
No account linked to committer's email address
共有 1 個文件被更改,包括 5 次插入2 次删除
  1. 5
    2
      api/core/indexing_runner.py

+ 5
- 2
api/core/indexing_runner.py 查看文件

@@ -235,7 +235,8 @@ class IndexingRunner:
if len(preview_texts) < 5:
preview_texts.append(document.page_content)

tokens += TokenCalculator.get_num_tokens(self.embedding_model_name, document.page_content)
tokens += TokenCalculator.get_num_tokens(self.embedding_model_name,
self.filter_string(document.page_content))

return {
"total_segments": total_segments,
@@ -345,6 +346,8 @@ class IndexingRunner:
return text_docs

def filter_string(self, text):
text = text.replace('<|', '<')
text = text.replace('|>', '>')
pattern = re.compile('[\x00-\x08\x0B\x0C\x0E-\x1F\x7F\x80-\xFF]')
return pattern.sub('', text)

@@ -425,7 +428,7 @@ class IndexingRunner:
return documents

def _split_to_documents(self, text_docs: List[Document], splitter: TextSplitter,
processing_rule: DatasetProcessRule) -> List[Document]:
processing_rule: DatasetProcessRule) -> List[Document]:
"""
Split the text documents into nodes.
"""

Loading…
取消
儲存