mirror of
https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git
synced 2025-08-15 08:55:57 +08:00
Refactor. (#4487)
### What problem does this PR solve? ### Type of change - [x] Refactoring
This commit is contained in:
parent
713f38090b
commit
e478586a8e
@ -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)
|
||||||
|
|
||||||
|
@ -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))
|
||||||
|
@ -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:
|
||||||
|
@ -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)
|
||||||
|
@ -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
|
||||||
"""
|
"""
|
||||||
|
Loading…
x
Reference in New Issue
Block a user