### 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: if not e:
return get_data_error_result( return get_data_error_result(
message="Database error (Knowledgebase rename)!") 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: except Exception as e:
return server_error_response(e) return server_error_response(e)

View File

@ -1051,11 +1051,6 @@ def migrate_db():
) )
except Exception: except Exception:
pass 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: try:
migrate( migrate(
migrator.add_column('task', 'retry_count', IntegerField(default=0)) 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(): def update_progress():
while True: while True:
time.sleep(3) time.sleep(6)
try: try:
DocumentService.update_progress() DocumentService.update_progress()
except Exception: except Exception:

View File

@ -44,7 +44,7 @@ def baike(cid, default_v=0):
def corpNorm(nm, add_region=True): def corpNorm(nm, add_region=True):
global CORP_TKS global CORP_TKS
if not nm or isinstance(nm, str): if not nm or not isinstance(nm, str):
return "" return ""
nm = rag_tokenizer.tradi2simp(rag_tokenizer.strQ2B(nm)).lower() nm = rag_tokenizer.tradi2simp(rag_tokenizer.strQ2B(nm)).lower()
nm = re.sub(r"&", "&", nm) nm = re.sub(r"&", "&", nm)

View File

@ -1,7 +1,6 @@
from abc import ABC, abstractmethod from abc import ABC, abstractmethod
from dataclasses import dataclass from dataclasses import dataclass
import numpy as np import numpy as np
import polars as pl
DEFAULT_MATCH_VECTOR_TOPN = 10 DEFAULT_MATCH_VECTOR_TOPN = 10
DEFAULT_MATCH_SPARSE_TOPN = 10 DEFAULT_MATCH_SPARSE_TOPN = 10
@ -187,7 +186,7 @@ class DocStoreConnection(ABC):
knowledgebaseIds: list[str], knowledgebaseIds: list[str],
aggFields: list[str] = [], aggFields: list[str] = [],
rank_feature: dict | None = None rank_feature: dict | None = None
) -> list[dict] | pl.DataFrame: ):
""" """
Search with given conjunctive equivalent filtering condition and return all fields of matched documents Search with given conjunctive equivalent filtering condition and return all fields of matched documents
""" """