Вы не можете выбрать более 25 тем Темы должны начинаться с буквы или цифры, могут содержать дефисы(-) и должны содержать не более 35 символов.

document_app.py 28KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670
  1. #
  2. # Copyright 2024 The InfiniFlow Authors. All Rights Reserved.
  3. #
  4. # Licensed under the Apache License, Version 2.0 (the "License");
  5. # you may not use this file except in compliance with the License.
  6. # You may obtain a copy of the License at
  7. #
  8. # http://www.apache.org/licenses/LICENSE-2.0
  9. #
  10. # Unless required by applicable law or agreed to in writing, software
  11. # distributed under the License is distributed on an "AS IS" BASIS,
  12. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13. # See the License for the specific language governing permissions and
  14. # limitations under the License
  15. #
  16. import json
  17. import os.path
  18. import pathlib
  19. import re
  20. from pathlib import Path
  21. import flask
  22. from flask import request
  23. from flask_login import current_user, login_required
  24. from api import settings
  25. from api.constants import FILE_NAME_LEN_LIMIT, IMG_BASE64_PREFIX
  26. from api.db import VALID_FILE_TYPES, VALID_TASK_STATUS, FileSource, FileType, ParserType, TaskStatus
  27. from api.db.db_models import File, Task
  28. from api.db.services import duplicate_name
  29. from api.db.services.document_service import DocumentService, doc_upload_and_parse
  30. from api.db.services.file2document_service import File2DocumentService
  31. from api.db.services.file_service import FileService
  32. from api.db.services.knowledgebase_service import KnowledgebaseService
  33. from api.db.services.task_service import TaskService, queue_tasks
  34. from api.db.services.user_service import UserTenantService
  35. from api.utils import get_uuid
  36. from api.utils.api_utils import (
  37. get_data_error_result,
  38. get_json_result,
  39. server_error_response,
  40. validate_request,
  41. )
  42. from api.utils.file_utils import filename_type, get_project_base_directory, thumbnail
  43. from api.utils.web_utils import html2pdf, is_valid_url
  44. from deepdoc.parser.html_parser import RAGFlowHtmlParser
  45. from rag.nlp import search
  46. from rag.utils.storage_factory import STORAGE_IMPL
  47. @manager.route("/upload", methods=["POST"]) # noqa: F821
  48. @login_required
  49. @validate_request("kb_id")
  50. def upload():
  51. kb_id = request.form.get("kb_id")
  52. if not kb_id:
  53. return get_json_result(data=False, message='Lack of "KB ID"', code=settings.RetCode.ARGUMENT_ERROR)
  54. if "file" not in request.files:
  55. return get_json_result(data=False, message="No file part!", code=settings.RetCode.ARGUMENT_ERROR)
  56. file_objs = request.files.getlist("file")
  57. for file_obj in file_objs:
  58. if file_obj.filename == "":
  59. return get_json_result(data=False, message="No file selected!", code=settings.RetCode.ARGUMENT_ERROR)
  60. if len(file_obj.filename.encode("utf-8")) > FILE_NAME_LEN_LIMIT:
  61. return get_json_result(data=False, message=f"File name must be {FILE_NAME_LEN_LIMIT} bytes or less.", code=settings.RetCode.ARGUMENT_ERROR)
  62. e, kb = KnowledgebaseService.get_by_id(kb_id)
  63. if not e:
  64. raise LookupError("Can't find this knowledgebase!")
  65. err, files = FileService.upload_document(kb, file_objs, current_user.id)
  66. if err:
  67. return get_json_result(data=files, message="\n".join(err), code=settings.RetCode.SERVER_ERROR)
  68. if not files:
  69. return get_json_result(data=files, message="There seems to be an issue with your file format. Please verify it is correct and not corrupted.", code=settings.RetCode.DATA_ERROR)
  70. files = [f[0] for f in files] # remove the blob
  71. return get_json_result(data=files)
  72. @manager.route("/web_crawl", methods=["POST"]) # noqa: F821
  73. @login_required
  74. @validate_request("kb_id", "name", "url")
  75. def web_crawl():
  76. kb_id = request.form.get("kb_id")
  77. if not kb_id:
  78. return get_json_result(data=False, message='Lack of "KB ID"', code=settings.RetCode.ARGUMENT_ERROR)
  79. name = request.form.get("name")
  80. url = request.form.get("url")
  81. if not is_valid_url(url):
  82. return get_json_result(data=False, message="The URL format is invalid", code=settings.RetCode.ARGUMENT_ERROR)
  83. e, kb = KnowledgebaseService.get_by_id(kb_id)
  84. if not e:
  85. raise LookupError("Can't find this knowledgebase!")
  86. blob = html2pdf(url)
  87. if not blob:
  88. return server_error_response(ValueError("Download failure."))
  89. root_folder = FileService.get_root_folder(current_user.id)
  90. pf_id = root_folder["id"]
  91. FileService.init_knowledgebase_docs(pf_id, current_user.id)
  92. kb_root_folder = FileService.get_kb_folder(current_user.id)
  93. kb_folder = FileService.new_a_file_from_kb(kb.tenant_id, kb.name, kb_root_folder["id"])
  94. try:
  95. filename = duplicate_name(DocumentService.query, name=name + ".pdf", kb_id=kb.id)
  96. filetype = filename_type(filename)
  97. if filetype == FileType.OTHER.value:
  98. raise RuntimeError("This type of file has not been supported yet!")
  99. location = filename
  100. while STORAGE_IMPL.obj_exist(kb_id, location):
  101. location += "_"
  102. STORAGE_IMPL.put(kb_id, location, blob)
  103. doc = {
  104. "id": get_uuid(),
  105. "kb_id": kb.id,
  106. "parser_id": kb.parser_id,
  107. "parser_config": kb.parser_config,
  108. "created_by": current_user.id,
  109. "type": filetype,
  110. "name": filename,
  111. "location": location,
  112. "size": len(blob),
  113. "thumbnail": thumbnail(filename, blob),
  114. "suffix": Path(filename).suffix.lstrip("."),
  115. }
  116. if doc["type"] == FileType.VISUAL:
  117. doc["parser_id"] = ParserType.PICTURE.value
  118. if doc["type"] == FileType.AURAL:
  119. doc["parser_id"] = ParserType.AUDIO.value
  120. if re.search(r"\.(ppt|pptx|pages)$", filename):
  121. doc["parser_id"] = ParserType.PRESENTATION.value
  122. if re.search(r"\.(eml)$", filename):
  123. doc["parser_id"] = ParserType.EMAIL.value
  124. DocumentService.insert(doc)
  125. FileService.add_file_from_kb(doc, kb_folder["id"], kb.tenant_id)
  126. except Exception as e:
  127. return server_error_response(e)
  128. return get_json_result(data=True)
  129. @manager.route("/create", methods=["POST"]) # noqa: F821
  130. @login_required
  131. @validate_request("name", "kb_id")
  132. def create():
  133. req = request.json
  134. kb_id = req["kb_id"]
  135. if not kb_id:
  136. return get_json_result(data=False, message='Lack of "KB ID"', code=settings.RetCode.ARGUMENT_ERROR)
  137. if len(req["name"].encode("utf-8")) > FILE_NAME_LEN_LIMIT:
  138. return get_json_result(data=False, message=f"File name must be {FILE_NAME_LEN_LIMIT} bytes or less.", code=settings.RetCode.ARGUMENT_ERROR)
  139. if req["name"].strip() == "":
  140. return get_json_result(data=False, message="File name can't be empty.", code=settings.RetCode.ARGUMENT_ERROR)
  141. req["name"] = req["name"].strip()
  142. try:
  143. e, kb = KnowledgebaseService.get_by_id(kb_id)
  144. if not e:
  145. return get_data_error_result(message="Can't find this knowledgebase!")
  146. if DocumentService.query(name=req["name"], kb_id=kb_id):
  147. return get_data_error_result(message="Duplicated document name in the same knowledgebase.")
  148. doc = DocumentService.insert(
  149. {
  150. "id": get_uuid(),
  151. "kb_id": kb.id,
  152. "parser_id": kb.parser_id,
  153. "parser_config": kb.parser_config,
  154. "created_by": current_user.id,
  155. "type": FileType.VIRTUAL,
  156. "name": req["name"],
  157. "suffix": Path(req["name"]).suffix.lstrip("."),
  158. "location": "",
  159. "size": 0,
  160. }
  161. )
  162. return get_json_result(data=doc.to_json())
  163. except Exception as e:
  164. return server_error_response(e)
  165. @manager.route("/list", methods=["POST"]) # noqa: F821
  166. @login_required
  167. def list_docs():
  168. kb_id = request.args.get("kb_id")
  169. if not kb_id:
  170. return get_json_result(data=False, message='Lack of "KB ID"', code=settings.RetCode.ARGUMENT_ERROR)
  171. tenants = UserTenantService.query(user_id=current_user.id)
  172. for tenant in tenants:
  173. if KnowledgebaseService.query(tenant_id=tenant.tenant_id, id=kb_id):
  174. break
  175. else:
  176. return get_json_result(data=False, message="Only owner of knowledgebase authorized for this operation.", code=settings.RetCode.OPERATING_ERROR)
  177. keywords = request.args.get("keywords", "")
  178. page_number = int(request.args.get("page", 0))
  179. items_per_page = int(request.args.get("page_size", 0))
  180. orderby = request.args.get("orderby", "create_time")
  181. if request.args.get("desc", "true").lower() == "false":
  182. desc = False
  183. else:
  184. desc = True
  185. req = request.get_json()
  186. run_status = req.get("run_status", [])
  187. if run_status:
  188. invalid_status = {s for s in run_status if s not in VALID_TASK_STATUS}
  189. if invalid_status:
  190. return get_data_error_result(message=f"Invalid filter run status conditions: {', '.join(invalid_status)}")
  191. types = req.get("types", [])
  192. if types:
  193. invalid_types = {t for t in types if t not in VALID_FILE_TYPES}
  194. if invalid_types:
  195. return get_data_error_result(message=f"Invalid filter conditions: {', '.join(invalid_types)} type{'s' if len(invalid_types) > 1 else ''}")
  196. suffix = req.get("suffix", [])
  197. try:
  198. docs, tol = DocumentService.get_by_kb_id(kb_id, page_number, items_per_page, orderby, desc, keywords, run_status, types, suffix)
  199. for doc_item in docs:
  200. if doc_item["thumbnail"] and not doc_item["thumbnail"].startswith(IMG_BASE64_PREFIX):
  201. doc_item["thumbnail"] = f"/v1/document/image/{kb_id}-{doc_item['thumbnail']}"
  202. return get_json_result(data={"total": tol, "docs": docs})
  203. except Exception as e:
  204. return server_error_response(e)
  205. @manager.route("/filter", methods=["POST"]) # noqa: F821
  206. @login_required
  207. def get_filter():
  208. req = request.get_json()
  209. kb_id = req.get("kb_id")
  210. if not kb_id:
  211. return get_json_result(data=False, message='Lack of "KB ID"', code=settings.RetCode.ARGUMENT_ERROR)
  212. tenants = UserTenantService.query(user_id=current_user.id)
  213. for tenant in tenants:
  214. if KnowledgebaseService.query(tenant_id=tenant.tenant_id, id=kb_id):
  215. break
  216. else:
  217. return get_json_result(data=False, message="Only owner of knowledgebase authorized for this operation.", code=settings.RetCode.OPERATING_ERROR)
  218. keywords = req.get("keywords", "")
  219. suffix = req.get("suffix", [])
  220. run_status = req.get("run_status", [])
  221. if run_status:
  222. invalid_status = {s for s in run_status if s not in VALID_TASK_STATUS}
  223. if invalid_status:
  224. return get_data_error_result(message=f"Invalid filter run status conditions: {', '.join(invalid_status)}")
  225. types = req.get("types", [])
  226. if types:
  227. invalid_types = {t for t in types if t not in VALID_FILE_TYPES}
  228. if invalid_types:
  229. return get_data_error_result(message=f"Invalid filter conditions: {', '.join(invalid_types)} type{'s' if len(invalid_types) > 1 else ''}")
  230. try:
  231. filter, total = DocumentService.get_filter_by_kb_id(kb_id, keywords, run_status, types, suffix)
  232. return get_json_result(data={"total": total, "filter": filter})
  233. except Exception as e:
  234. return server_error_response(e)
  235. @manager.route("/infos", methods=["POST"]) # noqa: F821
  236. @login_required
  237. def docinfos():
  238. req = request.json
  239. doc_ids = req["doc_ids"]
  240. for doc_id in doc_ids:
  241. if not DocumentService.accessible(doc_id, current_user.id):
  242. return get_json_result(data=False, message="No authorization.", code=settings.RetCode.AUTHENTICATION_ERROR)
  243. docs = DocumentService.get_by_ids(doc_ids)
  244. return get_json_result(data=list(docs.dicts()))
  245. @manager.route("/thumbnails", methods=["GET"]) # noqa: F821
  246. # @login_required
  247. def thumbnails():
  248. doc_ids = request.args.get("doc_ids").split(",")
  249. if not doc_ids:
  250. return get_json_result(data=False, message='Lack of "Document ID"', code=settings.RetCode.ARGUMENT_ERROR)
  251. try:
  252. docs = DocumentService.get_thumbnails(doc_ids)
  253. for doc_item in docs:
  254. if doc_item["thumbnail"] and not doc_item["thumbnail"].startswith(IMG_BASE64_PREFIX):
  255. doc_item["thumbnail"] = f"/v1/document/image/{doc_item['kb_id']}-{doc_item['thumbnail']}"
  256. return get_json_result(data={d["id"]: d["thumbnail"] for d in docs})
  257. except Exception as e:
  258. return server_error_response(e)
  259. @manager.route("/change_status", methods=["POST"]) # noqa: F821
  260. @login_required
  261. @validate_request("doc_ids", "status")
  262. def change_status():
  263. req = request.get_json()
  264. doc_ids = req.get("doc_ids", [])
  265. status = str(req.get("status", ""))
  266. if status not in ["0", "1"]:
  267. return get_json_result(data=False, message='"Status" must be either 0 or 1!', code=settings.RetCode.ARGUMENT_ERROR)
  268. result = {}
  269. for doc_id in doc_ids:
  270. if not DocumentService.accessible(doc_id, current_user.id):
  271. result[doc_id] = {"error": "No authorization."}
  272. continue
  273. try:
  274. e, doc = DocumentService.get_by_id(doc_id)
  275. if not e:
  276. result[doc_id] = {"error": "No authorization."}
  277. continue
  278. e, kb = KnowledgebaseService.get_by_id(doc.kb_id)
  279. if not e:
  280. result[doc_id] = {"error": "Can't find this knowledgebase!"}
  281. continue
  282. if not DocumentService.update_by_id(doc_id, {"status": str(status)}):
  283. result[doc_id] = {"error": "Database error (Document update)!"}
  284. continue
  285. status_int = int(status)
  286. if not settings.docStoreConn.update({"doc_id": doc_id}, {"available_int": status_int}, search.index_name(kb.tenant_id), doc.kb_id):
  287. result[doc_id] = {"error": "Database error (docStore update)!"}
  288. result[doc_id] = {"status": status}
  289. except Exception as e:
  290. result[doc_id] = {"error": f"Internal server error: {str(e)}"}
  291. return get_json_result(data=result)
  292. @manager.route("/rm", methods=["POST"]) # noqa: F821
  293. @login_required
  294. @validate_request("doc_id")
  295. def rm():
  296. req = request.json
  297. doc_ids = req["doc_id"]
  298. if isinstance(doc_ids, str):
  299. doc_ids = [doc_ids]
  300. for doc_id in doc_ids:
  301. if not DocumentService.accessible4deletion(doc_id, current_user.id):
  302. return get_json_result(data=False, message="No authorization.", code=settings.RetCode.AUTHENTICATION_ERROR)
  303. root_folder = FileService.get_root_folder(current_user.id)
  304. pf_id = root_folder["id"]
  305. FileService.init_knowledgebase_docs(pf_id, current_user.id)
  306. errors = ""
  307. kb_table_num_map = {}
  308. for doc_id in doc_ids:
  309. try:
  310. e, doc = DocumentService.get_by_id(doc_id)
  311. if not e:
  312. return get_data_error_result(message="Document not found!")
  313. tenant_id = DocumentService.get_tenant_id(doc_id)
  314. if not tenant_id:
  315. return get_data_error_result(message="Tenant not found!")
  316. b, n = File2DocumentService.get_storage_address(doc_id=doc_id)
  317. TaskService.filter_delete([Task.doc_id == doc_id])
  318. if not DocumentService.remove_document(doc, tenant_id):
  319. return get_data_error_result(message="Database error (Document removal)!")
  320. f2d = File2DocumentService.get_by_document_id(doc_id)
  321. deleted_file_count = 0
  322. if f2d:
  323. deleted_file_count = FileService.filter_delete([File.source_type == FileSource.KNOWLEDGEBASE, File.id == f2d[0].file_id])
  324. File2DocumentService.delete_by_document_id(doc_id)
  325. if deleted_file_count > 0:
  326. STORAGE_IMPL.rm(b, n)
  327. doc_parser = doc.parser_id
  328. if doc_parser == ParserType.TABLE:
  329. kb_id = doc.kb_id
  330. if kb_id not in kb_table_num_map:
  331. counts = DocumentService.count_by_kb_id(kb_id=kb_id, keywords="", run_status=[TaskStatus.DONE], types=[])
  332. kb_table_num_map[kb_id] = counts
  333. kb_table_num_map[kb_id] -= 1
  334. if kb_table_num_map[kb_id] <= 0:
  335. KnowledgebaseService.delete_field_map(kb_id)
  336. except Exception as e:
  337. errors += str(e)
  338. if errors:
  339. return get_json_result(data=False, message=errors, code=settings.RetCode.SERVER_ERROR)
  340. return get_json_result(data=True)
  341. @manager.route("/run", methods=["POST"]) # noqa: F821
  342. @login_required
  343. @validate_request("doc_ids", "run")
  344. def run():
  345. req = request.json
  346. for doc_id in req["doc_ids"]:
  347. if not DocumentService.accessible(doc_id, current_user.id):
  348. return get_json_result(data=False, message="No authorization.", code=settings.RetCode.AUTHENTICATION_ERROR)
  349. try:
  350. kb_table_num_map = {}
  351. for id in req["doc_ids"]:
  352. info = {"run": str(req["run"]), "progress": 0}
  353. if str(req["run"]) == TaskStatus.RUNNING.value and req.get("delete", False):
  354. info["progress_msg"] = ""
  355. info["chunk_num"] = 0
  356. info["token_num"] = 0
  357. e, doc = DocumentService.get_by_id(id)
  358. if not e:
  359. return get_data_error_result(message="Document not found!")
  360. if doc.run == TaskStatus.DONE.value:
  361. DocumentService.clear_chunk_num_when_rerun(doc.id)
  362. DocumentService.update_by_id(id, info)
  363. tenant_id = DocumentService.get_tenant_id(id)
  364. if not tenant_id:
  365. return get_data_error_result(message="Tenant not found!")
  366. e, doc = DocumentService.get_by_id(id)
  367. if not e:
  368. return get_data_error_result(message="Document not found!")
  369. if req.get("delete", False):
  370. TaskService.filter_delete([Task.doc_id == id])
  371. if settings.docStoreConn.indexExist(search.index_name(tenant_id), doc.kb_id):
  372. settings.docStoreConn.delete({"doc_id": id}, search.index_name(tenant_id), doc.kb_id)
  373. if str(req["run"]) == TaskStatus.RUNNING.value:
  374. e, doc = DocumentService.get_by_id(id)
  375. doc = doc.to_dict()
  376. doc["tenant_id"] = tenant_id
  377. doc_parser = doc.get("parser_id", ParserType.NAIVE)
  378. if doc_parser == ParserType.TABLE:
  379. kb_id = doc.get("kb_id")
  380. if not kb_id:
  381. continue
  382. if kb_id not in kb_table_num_map:
  383. count = DocumentService.count_by_kb_id(kb_id=kb_id, keywords="", run_status=[TaskStatus.DONE], types=[])
  384. kb_table_num_map[kb_id] = count
  385. if kb_table_num_map[kb_id] <= 0:
  386. KnowledgebaseService.delete_field_map(kb_id)
  387. bucket, name = File2DocumentService.get_storage_address(doc_id=doc["id"])
  388. queue_tasks(doc, bucket, name, 0)
  389. return get_json_result(data=True)
  390. except Exception as e:
  391. return server_error_response(e)
  392. @manager.route("/rename", methods=["POST"]) # noqa: F821
  393. @login_required
  394. @validate_request("doc_id", "name")
  395. def rename():
  396. req = request.json
  397. if not DocumentService.accessible(req["doc_id"], current_user.id):
  398. return get_json_result(data=False, message="No authorization.", code=settings.RetCode.AUTHENTICATION_ERROR)
  399. try:
  400. e, doc = DocumentService.get_by_id(req["doc_id"])
  401. if not e:
  402. return get_data_error_result(message="Document not found!")
  403. if pathlib.Path(req["name"].lower()).suffix != pathlib.Path(doc.name.lower()).suffix:
  404. return get_json_result(data=False, message="The extension of file can't be changed", code=settings.RetCode.ARGUMENT_ERROR)
  405. if len(req["name"].encode("utf-8")) > FILE_NAME_LEN_LIMIT:
  406. return get_json_result(data=False, message=f"File name must be {FILE_NAME_LEN_LIMIT} bytes or less.", code=settings.RetCode.ARGUMENT_ERROR)
  407. for d in DocumentService.query(name=req["name"], kb_id=doc.kb_id):
  408. if d.name == req["name"]:
  409. return get_data_error_result(message="Duplicated document name in the same knowledgebase.")
  410. if not DocumentService.update_by_id(req["doc_id"], {"name": req["name"]}):
  411. return get_data_error_result(message="Database error (Document rename)!")
  412. informs = File2DocumentService.get_by_document_id(req["doc_id"])
  413. if informs:
  414. e, file = FileService.get_by_id(informs[0].file_id)
  415. FileService.update_by_id(file.id, {"name": req["name"]})
  416. return get_json_result(data=True)
  417. except Exception as e:
  418. return server_error_response(e)
  419. @manager.route("/get/<doc_id>", methods=["GET"]) # noqa: F821
  420. # @login_required
  421. def get(doc_id):
  422. try:
  423. e, doc = DocumentService.get_by_id(doc_id)
  424. if not e:
  425. return get_data_error_result(message="Document not found!")
  426. b, n = File2DocumentService.get_storage_address(doc_id=doc_id)
  427. response = flask.make_response(STORAGE_IMPL.get(b, n))
  428. ext = re.search(r"\.([^.]+)$", doc.name)
  429. if ext:
  430. if doc.type == FileType.VISUAL.value:
  431. response.headers.set("Content-Type", "image/%s" % ext.group(1))
  432. else:
  433. response.headers.set("Content-Type", "application/%s" % ext.group(1))
  434. return response
  435. except Exception as e:
  436. return server_error_response(e)
  437. @manager.route("/change_parser", methods=["POST"]) # noqa: F821
  438. @login_required
  439. @validate_request("doc_id", "parser_id")
  440. def change_parser():
  441. req = request.json
  442. if not DocumentService.accessible(req["doc_id"], current_user.id):
  443. return get_json_result(data=False, message="No authorization.", code=settings.RetCode.AUTHENTICATION_ERROR)
  444. try:
  445. e, doc = DocumentService.get_by_id(req["doc_id"])
  446. if not e:
  447. return get_data_error_result(message="Document not found!")
  448. if doc.parser_id.lower() == req["parser_id"].lower():
  449. if "parser_config" in req:
  450. if req["parser_config"] == doc.parser_config:
  451. return get_json_result(data=True)
  452. else:
  453. return get_json_result(data=True)
  454. if (doc.type == FileType.VISUAL and req["parser_id"] != "picture") or (re.search(r"\.(ppt|pptx|pages)$", doc.name) and req["parser_id"] != "presentation"):
  455. return get_data_error_result(message="Not supported yet!")
  456. e = DocumentService.update_by_id(doc.id, {"parser_id": req["parser_id"], "progress": 0, "progress_msg": "", "run": TaskStatus.UNSTART.value})
  457. if not e:
  458. return get_data_error_result(message="Document not found!")
  459. if "parser_config" in req:
  460. DocumentService.update_parser_config(doc.id, req["parser_config"])
  461. if doc.token_num > 0:
  462. e = DocumentService.increment_chunk_num(doc.id, doc.kb_id, doc.token_num * -1, doc.chunk_num * -1, doc.process_duration * -1)
  463. if not e:
  464. return get_data_error_result(message="Document not found!")
  465. tenant_id = DocumentService.get_tenant_id(req["doc_id"])
  466. if not tenant_id:
  467. return get_data_error_result(message="Tenant not found!")
  468. if settings.docStoreConn.indexExist(search.index_name(tenant_id), doc.kb_id):
  469. settings.docStoreConn.delete({"doc_id": doc.id}, search.index_name(tenant_id), doc.kb_id)
  470. return get_json_result(data=True)
  471. except Exception as e:
  472. return server_error_response(e)
  473. @manager.route("/image/<image_id>", methods=["GET"]) # noqa: F821
  474. # @login_required
  475. def get_image(image_id):
  476. try:
  477. arr = image_id.split("-")
  478. if len(arr) != 2:
  479. return get_data_error_result(message="Image not found.")
  480. bkt, nm = image_id.split("-")
  481. response = flask.make_response(STORAGE_IMPL.get(bkt, nm))
  482. response.headers.set("Content-Type", "image/JPEG")
  483. return response
  484. except Exception as e:
  485. return server_error_response(e)
  486. @manager.route("/upload_and_parse", methods=["POST"]) # noqa: F821
  487. @login_required
  488. @validate_request("conversation_id")
  489. def upload_and_parse():
  490. if "file" not in request.files:
  491. return get_json_result(data=False, message="No file part!", code=settings.RetCode.ARGUMENT_ERROR)
  492. file_objs = request.files.getlist("file")
  493. for file_obj in file_objs:
  494. if file_obj.filename == "":
  495. return get_json_result(data=False, message="No file selected!", code=settings.RetCode.ARGUMENT_ERROR)
  496. doc_ids = doc_upload_and_parse(request.form.get("conversation_id"), file_objs, current_user.id)
  497. return get_json_result(data=doc_ids)
  498. @manager.route("/parse", methods=["POST"]) # noqa: F821
  499. @login_required
  500. def parse():
  501. url = request.json.get("url") if request.json else ""
  502. if url:
  503. if not is_valid_url(url):
  504. return get_json_result(data=False, message="The URL format is invalid", code=settings.RetCode.ARGUMENT_ERROR)
  505. download_path = os.path.join(get_project_base_directory(), "logs/downloads")
  506. os.makedirs(download_path, exist_ok=True)
  507. from seleniumwire.webdriver import Chrome, ChromeOptions
  508. options = ChromeOptions()
  509. options.add_argument("--headless")
  510. options.add_argument("--disable-gpu")
  511. options.add_argument("--no-sandbox")
  512. options.add_argument("--disable-dev-shm-usage")
  513. options.add_experimental_option("prefs", {"download.default_directory": download_path, "download.prompt_for_download": False, "download.directory_upgrade": True, "safebrowsing.enabled": True})
  514. driver = Chrome(options=options)
  515. driver.get(url)
  516. res_headers = [r.response.headers for r in driver.requests if r and r.response]
  517. if len(res_headers) > 1:
  518. sections = RAGFlowHtmlParser().parser_txt(driver.page_source)
  519. driver.quit()
  520. return get_json_result(data="\n".join(sections))
  521. class File:
  522. filename: str
  523. filepath: str
  524. def __init__(self, filename, filepath):
  525. self.filename = filename
  526. self.filepath = filepath
  527. def read(self):
  528. with open(self.filepath, "rb") as f:
  529. return f.read()
  530. r = re.search(r"filename=\"([^\"]+)\"", str(res_headers))
  531. if not r or not r.group(1):
  532. return get_json_result(data=False, message="Can't not identify downloaded file", code=settings.RetCode.ARGUMENT_ERROR)
  533. f = File(r.group(1), os.path.join(download_path, r.group(1)))
  534. txt = FileService.parse_docs([f], current_user.id)
  535. return get_json_result(data=txt)
  536. if "file" not in request.files:
  537. return get_json_result(data=False, message="No file part!", code=settings.RetCode.ARGUMENT_ERROR)
  538. file_objs = request.files.getlist("file")
  539. txt = FileService.parse_docs(file_objs, current_user.id)
  540. return get_json_result(data=txt)
  541. @manager.route("/set_meta", methods=["POST"]) # noqa: F821
  542. @login_required
  543. @validate_request("doc_id", "meta")
  544. def set_meta():
  545. req = request.json
  546. if not DocumentService.accessible(req["doc_id"], current_user.id):
  547. return get_json_result(data=False, message="No authorization.", code=settings.RetCode.AUTHENTICATION_ERROR)
  548. try:
  549. meta = json.loads(req["meta"])
  550. except Exception as e:
  551. return get_json_result(data=False, message=f"Json syntax error: {e}", code=settings.RetCode.ARGUMENT_ERROR)
  552. if not isinstance(meta, dict):
  553. return get_json_result(data=False, message='Meta data should be in Json map format, like {"key": "value"}', code=settings.RetCode.ARGUMENT_ERROR)
  554. try:
  555. e, doc = DocumentService.get_by_id(req["doc_id"])
  556. if not e:
  557. return get_data_error_result(message="Document not found!")
  558. if not DocumentService.update_by_id(req["doc_id"], {"meta_fields": meta}):
  559. return get_data_error_result(message="Database error (meta updates)!")
  560. return get_json_result(data=True)
  561. except Exception as e:
  562. return server_error_response(e)