pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix api argument error (#1920)

Browse Source 
### What problem does this PR solve?

#1918 

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu
2024-08-13 10:15:55 +08:00
committed by GitHub
parent eb8feaf20a
commit 8badf3f423
3 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/apps/api_app.py
View File

@ -87,7 +87,7 @@ def token_list():
if not tenants: if not tenants:
return get_data_error_result(retmsg="Tenant not found!") return get_data_error_result(retmsg="Tenant not found!")
id = request.args.get("dialog_id", request.args["canvas_id"]) id = request.args["dialog_id"] if "dialog_id" in request.args else request.args["canvas_id"]
objs = APITokenService.query(tenant_id=tenants[0].tenant_id, dialog_id=id) objs = APITokenService.query(tenant_id=tenants[0].tenant_id, dialog_id=id)
return get_json_result(data=[o.to_dict() for o in objs]) return get_json_result(data=[o.to_dict() for o in objs])
except Exception as e: except Exception as e:
@ -121,11 +121,11 @@ def stats():
"from_date", "from_date",
(datetime.now() - (datetime.now() -
timedelta( timedelta(
days=7)).strftime("%Y-%m-%d 24:00:00")), days=7)).strftime("%Y-%m-%d 00:00:00")),
request.args.get( request.args.get(
"to_date", "to_date",
datetime.now().strftime("%Y-%m-%d %H:%M:%S")), datetime.now().strftime("%Y-%m-%d %H:%M:%S")),
"agent" if request.args.get("canvas_id") else None) "agent" if "canvas_id" in request.args else None)
res = { res = {
"pv": [(o["dt"], o["pv"]) for o in objs], "pv": [(o["dt"], o["pv"]) for o in objs],
"uv": [(o["dt"], o["uv"]) for o in objs], "uv": [(o["dt"], o["uv"]) for o in objs],

2
api/apps/document_app.py
View File

@ -487,7 +487,7 @@ def upload_and_parse():
def dummy(prog=None, msg=""): def dummy(prog=None, msg=""):
pass pass
parser_config = {"chunk_token_num": 4096, "delimiter": "\n!?。；！？", "layout_recognize": False} parser_config = {"chunk_token_num": 4096, "delimiter": "\n!?;。；！？", "layout_recognize": False}
exe = ThreadPoolExecutor(max_workers=12) exe = ThreadPoolExecutor(max_workers=12)
threads = [] threads = []
for d, blob in files: for d, blob in files:

2
api/db/services/file_service.py
View File

@ -372,6 +372,8 @@ class FileService(CommonService):
doc["parser_id"] = ParserType.AUDIO.value doc["parser_id"] = ParserType.AUDIO.value
if re.search(r"\.(ppt|pptx|pages)$", filename): if re.search(r"\.(ppt|pptx|pages)$", filename):
doc["parser_id"] = ParserType.PRESENTATION.value doc["parser_id"] = ParserType.PRESENTATION.value
if re.search(r"\.(eml)$", filename):
doc["parser_id"] = ParserType.EMAIL.value
DocumentService.insert(doc) DocumentService.insert(doc)
FileService.add_file_from_kb(doc, kb_folder["id"], kb.tenant_id) FileService.add_file_from_kb(doc, kb_folder["id"], kb.tenant_id)