### What problem does this PR solve? ### Type of change - [x] Refactoringtags/v0.16.0
| @@ -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) | |||
| @@ -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)) | |||
| @@ -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: | |||
| @@ -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,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 | |||
| """ | |||