瀏覽代碼

Refactor. (#4487)

### What problem does this PR solve?

### Type of change

- [x] Refactoring
tags/v0.16.0
Kevin Hu 9 月之前
父節點
當前提交
e478586a8e
沒有連結到貢獻者的電子郵件帳戶。
共有 5 個檔案被更改,包括 6 行新增10 行删除
  1. 3
    1
      api/apps/kb_app.py
  2. 0
    5
      api/db/db_models.py
  3. 1
    1
      api/ragflow_server.py
  4. 1
    1
      deepdoc/parser/resume/entities/corporations.py
  5. 1
    2
      rag/utils/doc_store_conn.py

+ 3
- 1
api/apps/kb_app.py 查看文件

@@ -116,8 +116,10 @@ def update():
if not e:
return get_data_error_result(
message="Database error (Knowledgebase rename)!")
kb = kb.to_dict()
kb.update(req)

return get_json_result(data=kb.to_json())
return get_json_result(data=kb)
except Exception as e:
return server_error_response(e)


+ 0
- 5
api/db/db_models.py 查看文件

@@ -1051,11 +1051,6 @@ def migrate_db():
)
except Exception:
pass
try:
DB.execute_sql('ALTER TABLE llm DROP PRIMARY KEY;')
DB.execute_sql('ALTER TABLE llm ADD PRIMARY KEY (llm_name,fid);')
except Exception:
pass
try:
migrate(
migrator.add_column('task', 'retry_count', IntegerField(default=0))

+ 1
- 1
api/ragflow_server.py 查看文件

@@ -45,7 +45,7 @@ from rag.settings import print_rag_settings

def update_progress():
while True:
time.sleep(3)
time.sleep(6)
try:
DocumentService.update_progress()
except Exception:

+ 1
- 1
deepdoc/parser/resume/entities/corporations.py 查看文件

@@ -44,7 +44,7 @@ def baike(cid, default_v=0):

def corpNorm(nm, add_region=True):
global CORP_TKS
if not nm or isinstance(nm, str):
if not nm or not isinstance(nm, str):
return ""
nm = rag_tokenizer.tradi2simp(rag_tokenizer.strQ2B(nm)).lower()
nm = re.sub(r"&", "&", nm)

+ 1
- 2
rag/utils/doc_store_conn.py 查看文件

@@ -1,7 +1,6 @@
from abc import ABC, abstractmethod
from dataclasses import dataclass
import numpy as np
import polars as pl

DEFAULT_MATCH_VECTOR_TOPN = 10
DEFAULT_MATCH_SPARSE_TOPN = 10
@@ -187,7 +186,7 @@ class DocStoreConnection(ABC):
knowledgebaseIds: list[str],
aggFields: list[str] = [],
rank_feature: dict | None = None
) -> list[dict] | pl.DataFrame:
):
"""
Search with given conjunctive equivalent filtering condition and return all fields of matched documents
"""

Loading…
取消
儲存