### What problem does this PR solve?

### Type of change

- [x] Refactoring
This commit is contained in:
Kevin Hu 2025-01-15 14:06:46 +08:00 committed by GitHub
parent 713f38090b
commit e478586a8e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
5 changed files with 6 additions and 10 deletions

View File

@ -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)

View File

@ -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))

View File

@ -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:

View File

@ -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)

View File

@ -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
"""