Added static check at PR CI (#3921)
Browse files### What problem does this PR solve?
Added static check at PR CI
### Type of change
- [x] Bug Fix (non-breaking change which fixes an issue)
- [x] Refactoring
- .github/workflows/tests.yml +6 -0
- api/apps/api_app.py +15 -15
- api/apps/canvas_app.py +8 -8
- api/apps/chunk_app.py +8 -8
- api/apps/conversation_app.py +11 -11
- api/apps/dialog_app.py +4 -4
- api/apps/document_app.py +15 -15
- api/apps/file2document_app.py +2 -2
- api/apps/file_app.py +10 -10
- api/apps/kb_app.py +5 -5
- api/apps/llm_app.py +7 -7
- api/apps/sdk/agent.py +1 -1
- api/apps/sdk/chat.py +4 -4
- api/apps/sdk/dataset.py +4 -4
- api/apps/sdk/dify_retrieval.py +1 -1
- api/apps/sdk/doc.py +12 -12
- api/apps/sdk/session.py +10 -10
- api/apps/system_app.py +5 -5
- api/apps/tenant_app.py +5 -5
- api/apps/user_app.py +9 -9
- deepdoc/parser/pdf_parser.py +0 -2
- deepdoc/vision/ocr.py +1 -1
- deepdoc/vision/recognizer.py +2 -1
- graphrag/entity_embedding.py +1 -0
.github/workflows/tests.yml
CHANGED
@@ -49,6 +49,12 @@ jobs:
|
|
49 |
fetch-depth: 0
|
50 |
fetch-tags: true
|
51 |
|
|
|
|
|
|
|
|
|
|
|
|
|
52 |
- name: Build ragflow:dev-slim
|
53 |
run: |
|
54 |
RUNNER_WORKSPACE_PREFIX=${RUNNER_WORKSPACE_PREFIX:-$HOME}
|
|
|
49 |
fetch-depth: 0
|
50 |
fetch-tags: true
|
51 |
|
52 |
+
# https://github.com/astral-sh/ruff-action
|
53 |
+
- name: Static check with Ruff
|
54 |
+
uses: astral-sh/ruff-action@v2
|
55 |
+
with:
|
56 |
+
version: ">=0.8.2"
|
57 |
+
|
58 |
- name: Build ragflow:dev-slim
|
59 |
run: |
|
60 |
RUNNER_WORKSPACE_PREFIX=${RUNNER_WORKSPACE_PREFIX:-$HOME}
|
api/apps/api_app.py
CHANGED
@@ -45,7 +45,7 @@ from agent.canvas import Canvas
|
|
45 |
from functools import partial
|
46 |
|
47 |
|
48 |
-
@manager.route('/new_token', methods=['POST'])
|
49 |
@login_required
|
50 |
def new_token():
|
51 |
req = request.json
|
@@ -75,7 +75,7 @@ def new_token():
|
|
75 |
return server_error_response(e)
|
76 |
|
77 |
|
78 |
-
@manager.route('/token_list', methods=['GET'])
|
79 |
@login_required
|
80 |
def token_list():
|
81 |
try:
|
@@ -90,7 +90,7 @@ def token_list():
|
|
90 |
return server_error_response(e)
|
91 |
|
92 |
|
93 |
-
@manager.route('/rm', methods=['POST'])
|
94 |
@validate_request("tokens", "tenant_id")
|
95 |
@login_required
|
96 |
def rm():
|
@@ -104,7 +104,7 @@ def rm():
|
|
104 |
return server_error_response(e)
|
105 |
|
106 |
|
107 |
-
@manager.route('/stats', methods=['GET'])
|
108 |
@login_required
|
109 |
def stats():
|
110 |
try:
|
@@ -135,7 +135,7 @@ def stats():
|
|
135 |
return server_error_response(e)
|
136 |
|
137 |
|
138 |
-
@manager.route('/new_conversation', methods=['GET'])
|
139 |
def set_conversation():
|
140 |
token = request.headers.get('Authorization').split()[1]
|
141 |
objs = APIToken.query(token=token)
|
@@ -175,7 +175,7 @@ def set_conversation():
|
|
175 |
return server_error_response(e)
|
176 |
|
177 |
|
178 |
-
@manager.route('/completion', methods=['POST'])
|
179 |
@validate_request("conversation_id", "messages")
|
180 |
def completion():
|
181 |
token = request.headers.get('Authorization').split()[1]
|
@@ -341,7 +341,7 @@ def completion():
|
|
341 |
return server_error_response(e)
|
342 |
|
343 |
|
344 |
-
@manager.route('/conversation/<conversation_id>', methods=['GET'])
|
345 |
# @login_required
|
346 |
def get(conversation_id):
|
347 |
token = request.headers.get('Authorization').split()[1]
|
@@ -372,7 +372,7 @@ def get(conversation_id):
|
|
372 |
return server_error_response(e)
|
373 |
|
374 |
|
375 |
-
@manager.route('/document/upload', methods=['POST'])
|
376 |
@validate_request("kb_name")
|
377 |
def upload():
|
378 |
token = request.headers.get('Authorization').split()[1]
|
@@ -484,7 +484,7 @@ def upload():
|
|
484 |
return get_json_result(data=doc_result.to_json())
|
485 |
|
486 |
|
487 |
-
@manager.route('/document/upload_and_parse', methods=['POST'])
|
488 |
@validate_request("conversation_id")
|
489 |
def upload_parse():
|
490 |
token = request.headers.get('Authorization').split()[1]
|
@@ -507,7 +507,7 @@ def upload_parse():
|
|
507 |
return get_json_result(data=doc_ids)
|
508 |
|
509 |
|
510 |
-
@manager.route('/list_chunks', methods=['POST'])
|
511 |
# @login_required
|
512 |
def list_chunks():
|
513 |
token = request.headers.get('Authorization').split()[1]
|
@@ -547,7 +547,7 @@ def list_chunks():
|
|
547 |
return get_json_result(data=res)
|
548 |
|
549 |
|
550 |
-
@manager.route('/list_kb_docs', methods=['POST'])
|
551 |
# @login_required
|
552 |
def list_kb_docs():
|
553 |
token = request.headers.get('Authorization').split()[1]
|
@@ -587,7 +587,7 @@ def list_kb_docs():
|
|
587 |
return server_error_response(e)
|
588 |
|
589 |
|
590 |
-
@manager.route('/document/infos', methods=['POST'])
|
591 |
@validate_request("doc_ids")
|
592 |
def docinfos():
|
593 |
token = request.headers.get('Authorization').split()[1]
|
@@ -601,7 +601,7 @@ def docinfos():
|
|
601 |
return get_json_result(data=list(docs.dicts()))
|
602 |
|
603 |
|
604 |
-
@manager.route('/document', methods=['DELETE'])
|
605 |
# @login_required
|
606 |
def document_rm():
|
607 |
token = request.headers.get('Authorization').split()[1]
|
@@ -660,7 +660,7 @@ def document_rm():
|
|
660 |
return get_json_result(data=True)
|
661 |
|
662 |
|
663 |
-
@manager.route('/completion_aibotk', methods=['POST'])
|
664 |
@validate_request("Authorization", "conversation_id", "word")
|
665 |
def completion_faq():
|
666 |
import base64
|
@@ -802,7 +802,7 @@ def completion_faq():
|
|
802 |
return server_error_response(e)
|
803 |
|
804 |
|
805 |
-
@manager.route('/retrieval', methods=['POST'])
|
806 |
@validate_request("kb_id", "question")
|
807 |
def retrieval():
|
808 |
token = request.headers.get('Authorization').split()[1]
|
|
|
45 |
from functools import partial
|
46 |
|
47 |
|
48 |
+
@manager.route('/new_token', methods=['POST']) # noqa: F821
|
49 |
@login_required
|
50 |
def new_token():
|
51 |
req = request.json
|
|
|
75 |
return server_error_response(e)
|
76 |
|
77 |
|
78 |
+
@manager.route('/token_list', methods=['GET']) # noqa: F821
|
79 |
@login_required
|
80 |
def token_list():
|
81 |
try:
|
|
|
90 |
return server_error_response(e)
|
91 |
|
92 |
|
93 |
+
@manager.route('/rm', methods=['POST']) # noqa: F821
|
94 |
@validate_request("tokens", "tenant_id")
|
95 |
@login_required
|
96 |
def rm():
|
|
|
104 |
return server_error_response(e)
|
105 |
|
106 |
|
107 |
+
@manager.route('/stats', methods=['GET']) # noqa: F821
|
108 |
@login_required
|
109 |
def stats():
|
110 |
try:
|
|
|
135 |
return server_error_response(e)
|
136 |
|
137 |
|
138 |
+
@manager.route('/new_conversation', methods=['GET']) # noqa: F821
|
139 |
def set_conversation():
|
140 |
token = request.headers.get('Authorization').split()[1]
|
141 |
objs = APIToken.query(token=token)
|
|
|
175 |
return server_error_response(e)
|
176 |
|
177 |
|
178 |
+
@manager.route('/completion', methods=['POST']) # noqa: F821
|
179 |
@validate_request("conversation_id", "messages")
|
180 |
def completion():
|
181 |
token = request.headers.get('Authorization').split()[1]
|
|
|
341 |
return server_error_response(e)
|
342 |
|
343 |
|
344 |
+
@manager.route('/conversation/<conversation_id>', methods=['GET']) # noqa: F821
|
345 |
# @login_required
|
346 |
def get(conversation_id):
|
347 |
token = request.headers.get('Authorization').split()[1]
|
|
|
372 |
return server_error_response(e)
|
373 |
|
374 |
|
375 |
+
@manager.route('/document/upload', methods=['POST']) # noqa: F821
|
376 |
@validate_request("kb_name")
|
377 |
def upload():
|
378 |
token = request.headers.get('Authorization').split()[1]
|
|
|
484 |
return get_json_result(data=doc_result.to_json())
|
485 |
|
486 |
|
487 |
+
@manager.route('/document/upload_and_parse', methods=['POST']) # noqa: F821
|
488 |
@validate_request("conversation_id")
|
489 |
def upload_parse():
|
490 |
token = request.headers.get('Authorization').split()[1]
|
|
|
507 |
return get_json_result(data=doc_ids)
|
508 |
|
509 |
|
510 |
+
@manager.route('/list_chunks', methods=['POST']) # noqa: F821
|
511 |
# @login_required
|
512 |
def list_chunks():
|
513 |
token = request.headers.get('Authorization').split()[1]
|
|
|
547 |
return get_json_result(data=res)
|
548 |
|
549 |
|
550 |
+
@manager.route('/list_kb_docs', methods=['POST']) # noqa: F821
|
551 |
# @login_required
|
552 |
def list_kb_docs():
|
553 |
token = request.headers.get('Authorization').split()[1]
|
|
|
587 |
return server_error_response(e)
|
588 |
|
589 |
|
590 |
+
@manager.route('/document/infos', methods=['POST']) # noqa: F821
|
591 |
@validate_request("doc_ids")
|
592 |
def docinfos():
|
593 |
token = request.headers.get('Authorization').split()[1]
|
|
|
601 |
return get_json_result(data=list(docs.dicts()))
|
602 |
|
603 |
|
604 |
+
@manager.route('/document', methods=['DELETE']) # noqa: F821
|
605 |
# @login_required
|
606 |
def document_rm():
|
607 |
token = request.headers.get('Authorization').split()[1]
|
|
|
660 |
return get_json_result(data=True)
|
661 |
|
662 |
|
663 |
+
@manager.route('/completion_aibotk', methods=['POST']) # noqa: F821
|
664 |
@validate_request("Authorization", "conversation_id", "word")
|
665 |
def completion_faq():
|
666 |
import base64
|
|
|
802 |
return server_error_response(e)
|
803 |
|
804 |
|
805 |
+
@manager.route('/retrieval', methods=['POST']) # noqa: F821
|
806 |
@validate_request("kb_id", "question")
|
807 |
def retrieval():
|
808 |
token = request.headers.get('Authorization').split()[1]
|
api/apps/canvas_app.py
CHANGED
@@ -25,13 +25,13 @@ from agent.canvas import Canvas
|
|
25 |
from peewee import MySQLDatabase, PostgresqlDatabase
|
26 |
|
27 |
|
28 |
-
@manager.route('/templates', methods=['GET'])
|
29 |
@login_required
|
30 |
def templates():
|
31 |
return get_json_result(data=[c.to_dict() for c in CanvasTemplateService.get_all()])
|
32 |
|
33 |
|
34 |
-
@manager.route('/list', methods=['GET'])
|
35 |
@login_required
|
36 |
def canvas_list():
|
37 |
return get_json_result(data=sorted([c.to_dict() for c in \
|
@@ -39,7 +39,7 @@ def canvas_list():
|
|
39 |
)
|
40 |
|
41 |
|
42 |
-
@manager.route('/rm', methods=['POST'])
|
43 |
@validate_request("canvas_ids")
|
44 |
@login_required
|
45 |
def rm():
|
@@ -52,7 +52,7 @@ def rm():
|
|
52 |
return get_json_result(data=True)
|
53 |
|
54 |
|
55 |
-
@manager.route('/set', methods=['POST'])
|
56 |
@validate_request("dsl", "title")
|
57 |
@login_required
|
58 |
def save():
|
@@ -77,7 +77,7 @@ def save():
|
|
77 |
return get_json_result(data=req)
|
78 |
|
79 |
|
80 |
-
@manager.route('/get/<canvas_id>', methods=['GET'])
|
81 |
@login_required
|
82 |
def get(canvas_id):
|
83 |
e, c = UserCanvasService.get_by_id(canvas_id)
|
@@ -86,7 +86,7 @@ def get(canvas_id):
|
|
86 |
return get_json_result(data=c.to_dict())
|
87 |
|
88 |
|
89 |
-
@manager.route('/completion', methods=['POST'])
|
90 |
@validate_request("id")
|
91 |
@login_required
|
92 |
def run():
|
@@ -163,7 +163,7 @@ def run():
|
|
163 |
return get_json_result(data={"answer": final_ans["content"], "reference": final_ans.get("reference", [])})
|
164 |
|
165 |
|
166 |
-
@manager.route('/reset', methods=['POST'])
|
167 |
@validate_request("id")
|
168 |
@login_required
|
169 |
def reset():
|
@@ -186,7 +186,7 @@ def reset():
|
|
186 |
return server_error_response(e)
|
187 |
|
188 |
|
189 |
-
@manager.route('/test_db_connect', methods=['POST'])
|
190 |
@validate_request("db_type", "database", "username", "host", "port", "password")
|
191 |
@login_required
|
192 |
def test_db_connect():
|
|
|
25 |
from peewee import MySQLDatabase, PostgresqlDatabase
|
26 |
|
27 |
|
28 |
+
@manager.route('/templates', methods=['GET']) # noqa: F821
|
29 |
@login_required
|
30 |
def templates():
|
31 |
return get_json_result(data=[c.to_dict() for c in CanvasTemplateService.get_all()])
|
32 |
|
33 |
|
34 |
+
@manager.route('/list', methods=['GET']) # noqa: F821
|
35 |
@login_required
|
36 |
def canvas_list():
|
37 |
return get_json_result(data=sorted([c.to_dict() for c in \
|
|
|
39 |
)
|
40 |
|
41 |
|
42 |
+
@manager.route('/rm', methods=['POST']) # noqa: F821
|
43 |
@validate_request("canvas_ids")
|
44 |
@login_required
|
45 |
def rm():
|
|
|
52 |
return get_json_result(data=True)
|
53 |
|
54 |
|
55 |
+
@manager.route('/set', methods=['POST']) # noqa: F821
|
56 |
@validate_request("dsl", "title")
|
57 |
@login_required
|
58 |
def save():
|
|
|
77 |
return get_json_result(data=req)
|
78 |
|
79 |
|
80 |
+
@manager.route('/get/<canvas_id>', methods=['GET']) # noqa: F821
|
81 |
@login_required
|
82 |
def get(canvas_id):
|
83 |
e, c = UserCanvasService.get_by_id(canvas_id)
|
|
|
86 |
return get_json_result(data=c.to_dict())
|
87 |
|
88 |
|
89 |
+
@manager.route('/completion', methods=['POST']) # noqa: F821
|
90 |
@validate_request("id")
|
91 |
@login_required
|
92 |
def run():
|
|
|
163 |
return get_json_result(data={"answer": final_ans["content"], "reference": final_ans.get("reference", [])})
|
164 |
|
165 |
|
166 |
+
@manager.route('/reset', methods=['POST']) # noqa: F821
|
167 |
@validate_request("id")
|
168 |
@login_required
|
169 |
def reset():
|
|
|
186 |
return server_error_response(e)
|
187 |
|
188 |
|
189 |
+
@manager.route('/test_db_connect', methods=['POST']) # noqa: F821
|
190 |
@validate_request("db_type", "database", "username", "host", "port", "password")
|
191 |
@login_required
|
192 |
def test_db_connect():
|
api/apps/chunk_app.py
CHANGED
@@ -35,7 +35,7 @@ import hashlib
|
|
35 |
import re
|
36 |
|
37 |
|
38 |
-
@manager.route('/list', methods=['POST'])
|
39 |
@login_required
|
40 |
@validate_request("doc_id")
|
41 |
def list_chunk():
|
@@ -84,7 +84,7 @@ def list_chunk():
|
|
84 |
return server_error_response(e)
|
85 |
|
86 |
|
87 |
-
@manager.route('/get', methods=['GET'])
|
88 |
@login_required
|
89 |
def get():
|
90 |
chunk_id = request.args["chunk_id"]
|
@@ -113,7 +113,7 @@ def get():
|
|
113 |
return server_error_response(e)
|
114 |
|
115 |
|
116 |
-
@manager.route('/set', methods=['POST'])
|
117 |
@login_required
|
118 |
@validate_request("doc_id", "chunk_id", "content_with_weight",
|
119 |
"important_kwd", "question_kwd")
|
@@ -164,7 +164,7 @@ def set():
|
|
164 |
return server_error_response(e)
|
165 |
|
166 |
|
167 |
-
@manager.route('/switch', methods=['POST'])
|
168 |
@login_required
|
169 |
@validate_request("chunk_ids", "available_int", "doc_id")
|
170 |
def switch():
|
@@ -184,7 +184,7 @@ def switch():
|
|
184 |
return server_error_response(e)
|
185 |
|
186 |
|
187 |
-
@manager.route('/rm', methods=['POST'])
|
188 |
@login_required
|
189 |
@validate_request("chunk_ids", "doc_id")
|
190 |
def rm():
|
@@ -203,7 +203,7 @@ def rm():
|
|
203 |
return server_error_response(e)
|
204 |
|
205 |
|
206 |
-
@manager.route('/create', methods=['POST'])
|
207 |
@login_required
|
208 |
@validate_request("doc_id", "content_with_weight")
|
209 |
def create():
|
@@ -255,7 +255,7 @@ def create():
|
|
255 |
return server_error_response(e)
|
256 |
|
257 |
|
258 |
-
@manager.route('/retrieval_test', methods=['POST'])
|
259 |
@login_required
|
260 |
@validate_request("kb_id", "question")
|
261 |
def retrieval_test():
|
@@ -314,7 +314,7 @@ def retrieval_test():
|
|
314 |
return server_error_response(e)
|
315 |
|
316 |
|
317 |
-
@manager.route('/knowledge_graph', methods=['GET'])
|
318 |
@login_required
|
319 |
def knowledge_graph():
|
320 |
doc_id = request.args["doc_id"]
|
|
|
35 |
import re
|
36 |
|
37 |
|
38 |
+
@manager.route('/list', methods=['POST']) # noqa: F821
|
39 |
@login_required
|
40 |
@validate_request("doc_id")
|
41 |
def list_chunk():
|
|
|
84 |
return server_error_response(e)
|
85 |
|
86 |
|
87 |
+
@manager.route('/get', methods=['GET']) # noqa: F821
|
88 |
@login_required
|
89 |
def get():
|
90 |
chunk_id = request.args["chunk_id"]
|
|
|
113 |
return server_error_response(e)
|
114 |
|
115 |
|
116 |
+
@manager.route('/set', methods=['POST']) # noqa: F821
|
117 |
@login_required
|
118 |
@validate_request("doc_id", "chunk_id", "content_with_weight",
|
119 |
"important_kwd", "question_kwd")
|
|
|
164 |
return server_error_response(e)
|
165 |
|
166 |
|
167 |
+
@manager.route('/switch', methods=['POST']) # noqa: F821
|
168 |
@login_required
|
169 |
@validate_request("chunk_ids", "available_int", "doc_id")
|
170 |
def switch():
|
|
|
184 |
return server_error_response(e)
|
185 |
|
186 |
|
187 |
+
@manager.route('/rm', methods=['POST']) # noqa: F821
|
188 |
@login_required
|
189 |
@validate_request("chunk_ids", "doc_id")
|
190 |
def rm():
|
|
|
203 |
return server_error_response(e)
|
204 |
|
205 |
|
206 |
+
@manager.route('/create', methods=['POST']) # noqa: F821
|
207 |
@login_required
|
208 |
@validate_request("doc_id", "content_with_weight")
|
209 |
def create():
|
|
|
255 |
return server_error_response(e)
|
256 |
|
257 |
|
258 |
+
@manager.route('/retrieval_test', methods=['POST']) # noqa: F821
|
259 |
@login_required
|
260 |
@validate_request("kb_id", "question")
|
261 |
def retrieval_test():
|
|
|
314 |
return server_error_response(e)
|
315 |
|
316 |
|
317 |
+
@manager.route('/knowledge_graph', methods=['GET']) # noqa: F821
|
318 |
@login_required
|
319 |
def knowledge_graph():
|
320 |
doc_id = request.args["doc_id"]
|
api/apps/conversation_app.py
CHANGED
@@ -31,7 +31,7 @@ from api.utils.api_utils import server_error_response, get_data_error_result, va
|
|
31 |
from graphrag.mind_map_extractor import MindMapExtractor
|
32 |
|
33 |
|
34 |
-
@manager.route('/set', methods=['POST'])
|
35 |
@login_required
|
36 |
def set_conversation():
|
37 |
req = request.json
|
@@ -72,7 +72,7 @@ def set_conversation():
|
|
72 |
return server_error_response(e)
|
73 |
|
74 |
|
75 |
-
@manager.route('/get', methods=['GET'])
|
76 |
@login_required
|
77 |
def get():
|
78 |
conv_id = request.args["conversation_id"]
|
@@ -94,7 +94,7 @@ def get():
|
|
94 |
return server_error_response(e)
|
95 |
|
96 |
|
97 |
-
@manager.route('/rm', methods=['POST'])
|
98 |
@login_required
|
99 |
def rm():
|
100 |
conv_ids = request.json["conversation_ids"]
|
@@ -117,7 +117,7 @@ def rm():
|
|
117 |
return server_error_response(e)
|
118 |
|
119 |
|
120 |
-
@manager.route('/list', methods=['GET'])
|
121 |
@login_required
|
122 |
def list_convsersation():
|
123 |
dialog_id = request.args["dialog_id"]
|
@@ -136,7 +136,7 @@ def list_convsersation():
|
|
136 |
return server_error_response(e)
|
137 |
|
138 |
|
139 |
-
@manager.route('/completion', methods=['POST'])
|
140 |
@login_required
|
141 |
@validate_request("conversation_id", "messages")
|
142 |
def completion():
|
@@ -209,7 +209,7 @@ def completion():
|
|
209 |
return server_error_response(e)
|
210 |
|
211 |
|
212 |
-
@manager.route('/tts', methods=['POST'])
|
213 |
@login_required
|
214 |
def tts():
|
215 |
req = request.json
|
@@ -243,7 +243,7 @@ def tts():
|
|
243 |
return resp
|
244 |
|
245 |
|
246 |
-
@manager.route('/delete_msg', methods=['POST'])
|
247 |
@login_required
|
248 |
@validate_request("conversation_id", "message_id")
|
249 |
def delete_msg():
|
@@ -266,7 +266,7 @@ def delete_msg():
|
|
266 |
return get_json_result(data=conv)
|
267 |
|
268 |
|
269 |
-
@manager.route('/thumbup', methods=['POST'])
|
270 |
@login_required
|
271 |
@validate_request("conversation_id", "message_id")
|
272 |
def thumbup():
|
@@ -293,7 +293,7 @@ def thumbup():
|
|
293 |
return get_json_result(data=conv)
|
294 |
|
295 |
|
296 |
-
@manager.route('/ask', methods=['POST'])
|
297 |
@login_required
|
298 |
@validate_request("question", "kb_ids")
|
299 |
def ask_about():
|
@@ -319,7 +319,7 @@ def ask_about():
|
|
319 |
return resp
|
320 |
|
321 |
|
322 |
-
@manager.route('/mindmap', methods=['POST'])
|
323 |
@login_required
|
324 |
@validate_request("question", "kb_ids")
|
325 |
def mindmap():
|
@@ -341,7 +341,7 @@ def mindmap():
|
|
341 |
return get_json_result(data=mind_map)
|
342 |
|
343 |
|
344 |
-
@manager.route('/related_questions', methods=['POST'])
|
345 |
@login_required
|
346 |
@validate_request("question")
|
347 |
def related_questions():
|
|
|
31 |
from graphrag.mind_map_extractor import MindMapExtractor
|
32 |
|
33 |
|
34 |
+
@manager.route('/set', methods=['POST']) # noqa: F821
|
35 |
@login_required
|
36 |
def set_conversation():
|
37 |
req = request.json
|
|
|
72 |
return server_error_response(e)
|
73 |
|
74 |
|
75 |
+
@manager.route('/get', methods=['GET']) # noqa: F821
|
76 |
@login_required
|
77 |
def get():
|
78 |
conv_id = request.args["conversation_id"]
|
|
|
94 |
return server_error_response(e)
|
95 |
|
96 |
|
97 |
+
@manager.route('/rm', methods=['POST']) # noqa: F821
|
98 |
@login_required
|
99 |
def rm():
|
100 |
conv_ids = request.json["conversation_ids"]
|
|
|
117 |
return server_error_response(e)
|
118 |
|
119 |
|
120 |
+
@manager.route('/list', methods=['GET']) # noqa: F821
|
121 |
@login_required
|
122 |
def list_convsersation():
|
123 |
dialog_id = request.args["dialog_id"]
|
|
|
136 |
return server_error_response(e)
|
137 |
|
138 |
|
139 |
+
@manager.route('/completion', methods=['POST']) # noqa: F821
|
140 |
@login_required
|
141 |
@validate_request("conversation_id", "messages")
|
142 |
def completion():
|
|
|
209 |
return server_error_response(e)
|
210 |
|
211 |
|
212 |
+
@manager.route('/tts', methods=['POST']) # noqa: F821
|
213 |
@login_required
|
214 |
def tts():
|
215 |
req = request.json
|
|
|
243 |
return resp
|
244 |
|
245 |
|
246 |
+
@manager.route('/delete_msg', methods=['POST']) # noqa: F821
|
247 |
@login_required
|
248 |
@validate_request("conversation_id", "message_id")
|
249 |
def delete_msg():
|
|
|
266 |
return get_json_result(data=conv)
|
267 |
|
268 |
|
269 |
+
@manager.route('/thumbup', methods=['POST']) # noqa: F821
|
270 |
@login_required
|
271 |
@validate_request("conversation_id", "message_id")
|
272 |
def thumbup():
|
|
|
293 |
return get_json_result(data=conv)
|
294 |
|
295 |
|
296 |
+
@manager.route('/ask', methods=['POST']) # noqa: F821
|
297 |
@login_required
|
298 |
@validate_request("question", "kb_ids")
|
299 |
def ask_about():
|
|
|
319 |
return resp
|
320 |
|
321 |
|
322 |
+
@manager.route('/mindmap', methods=['POST']) # noqa: F821
|
323 |
@login_required
|
324 |
@validate_request("question", "kb_ids")
|
325 |
def mindmap():
|
|
|
341 |
return get_json_result(data=mind_map)
|
342 |
|
343 |
|
344 |
+
@manager.route('/related_questions', methods=['POST']) # noqa: F821
|
345 |
@login_required
|
346 |
@validate_request("question")
|
347 |
def related_questions():
|
api/apps/dialog_app.py
CHANGED
@@ -26,7 +26,7 @@ from api.utils import get_uuid
|
|
26 |
from api.utils.api_utils import get_json_result
|
27 |
|
28 |
|
29 |
-
@manager.route('/set', methods=['POST'])
|
30 |
@login_required
|
31 |
def set_dialog():
|
32 |
req = request.json
|
@@ -125,7 +125,7 @@ def set_dialog():
|
|
125 |
return server_error_response(e)
|
126 |
|
127 |
|
128 |
-
@manager.route('/get', methods=['GET'])
|
129 |
@login_required
|
130 |
def get():
|
131 |
dialog_id = request.args["dialog_id"]
|
@@ -151,7 +151,7 @@ def get_kb_names(kb_ids):
|
|
151 |
return ids, nms
|
152 |
|
153 |
|
154 |
-
@manager.route('/list', methods=['GET'])
|
155 |
@login_required
|
156 |
def list_dialogs():
|
157 |
try:
|
@@ -168,7 +168,7 @@ def list_dialogs():
|
|
168 |
return server_error_response(e)
|
169 |
|
170 |
|
171 |
-
@manager.route('/rm', methods=['POST'])
|
172 |
@login_required
|
173 |
@validate_request("dialog_ids")
|
174 |
def rm():
|
|
|
26 |
from api.utils.api_utils import get_json_result
|
27 |
|
28 |
|
29 |
+
@manager.route('/set', methods=['POST']) # noqa: F821
|
30 |
@login_required
|
31 |
def set_dialog():
|
32 |
req = request.json
|
|
|
125 |
return server_error_response(e)
|
126 |
|
127 |
|
128 |
+
@manager.route('/get', methods=['GET']) # noqa: F821
|
129 |
@login_required
|
130 |
def get():
|
131 |
dialog_id = request.args["dialog_id"]
|
|
|
151 |
return ids, nms
|
152 |
|
153 |
|
154 |
+
@manager.route('/list', methods=['GET']) # noqa: F821
|
155 |
@login_required
|
156 |
def list_dialogs():
|
157 |
try:
|
|
|
168 |
return server_error_response(e)
|
169 |
|
170 |
|
171 |
+
@manager.route('/rm', methods=['POST']) # noqa: F821
|
172 |
@login_required
|
173 |
@validate_request("dialog_ids")
|
174 |
def rm():
|
api/apps/document_app.py
CHANGED
@@ -42,7 +42,7 @@ from api.utils.web_utils import html2pdf, is_valid_url
|
|
42 |
from api.constants import IMG_BASE64_PREFIX
|
43 |
|
44 |
|
45 |
-
@manager.route('/upload', methods=['POST'])
|
46 |
@login_required
|
47 |
@validate_request("kb_id")
|
48 |
def upload():
|
@@ -71,7 +71,7 @@ def upload():
|
|
71 |
return get_json_result(data=True)
|
72 |
|
73 |
|
74 |
-
@manager.route('/web_crawl', methods=['POST'])
|
75 |
@login_required
|
76 |
@validate_request("kb_id", "name", "url")
|
77 |
def web_crawl():
|
@@ -138,7 +138,7 @@ def web_crawl():
|
|
138 |
return get_json_result(data=True)
|
139 |
|
140 |
|
141 |
-
@manager.route('/create', methods=['POST'])
|
142 |
@login_required
|
143 |
@validate_request("name", "kb_id")
|
144 |
def create():
|
@@ -174,7 +174,7 @@ def create():
|
|
174 |
return server_error_response(e)
|
175 |
|
176 |
|
177 |
-
@manager.route('/list', methods=['GET'])
|
178 |
@login_required
|
179 |
def list_docs():
|
180 |
kb_id = request.args.get("kb_id")
|
@@ -209,7 +209,7 @@ def list_docs():
|
|
209 |
return server_error_response(e)
|
210 |
|
211 |
|
212 |
-
@manager.route('/infos', methods=['POST'])
|
213 |
@login_required
|
214 |
def docinfos():
|
215 |
req = request.json
|
@@ -225,7 +225,7 @@ def docinfos():
|
|
225 |
return get_json_result(data=list(docs.dicts()))
|
226 |
|
227 |
|
228 |
-
@manager.route('/thumbnails', methods=['GET'])
|
229 |
# @login_required
|
230 |
def thumbnails():
|
231 |
doc_ids = request.args.get("doc_ids").split(",")
|
@@ -245,7 +245,7 @@ def thumbnails():
|
|
245 |
return server_error_response(e)
|
246 |
|
247 |
|
248 |
-
@manager.route('/change_status', methods=['POST'])
|
249 |
@login_required
|
250 |
@validate_request("doc_id", "status")
|
251 |
def change_status():
|
@@ -284,7 +284,7 @@ def change_status():
|
|
284 |
return server_error_response(e)
|
285 |
|
286 |
|
287 |
-
@manager.route('/rm', methods=['POST'])
|
288 |
@login_required
|
289 |
@validate_request("doc_id")
|
290 |
def rm():
|
@@ -334,7 +334,7 @@ def rm():
|
|
334 |
return get_json_result(data=True)
|
335 |
|
336 |
|
337 |
-
@manager.route('/run', methods=['POST'])
|
338 |
@login_required
|
339 |
@validate_request("doc_ids", "run")
|
340 |
def run():
|
@@ -377,7 +377,7 @@ def run():
|
|
377 |
return server_error_response(e)
|
378 |
|
379 |
|
380 |
-
@manager.route('/rename', methods=['POST'])
|
381 |
@login_required
|
382 |
@validate_request("doc_id", "name")
|
383 |
def rename():
|
@@ -418,7 +418,7 @@ def rename():
|
|
418 |
return server_error_response(e)
|
419 |
|
420 |
|
421 |
-
@manager.route('/get/<doc_id>', methods=['GET'])
|
422 |
# @login_required
|
423 |
def get(doc_id):
|
424 |
try:
|
@@ -443,7 +443,7 @@ def get(doc_id):
|
|
443 |
return server_error_response(e)
|
444 |
|
445 |
|
446 |
-
@manager.route('/change_parser', methods=['POST'])
|
447 |
@login_required
|
448 |
@validate_request("doc_id", "parser_id")
|
449 |
def change_parser():
|
@@ -494,7 +494,7 @@ def change_parser():
|
|
494 |
return server_error_response(e)
|
495 |
|
496 |
|
497 |
-
@manager.route('/image/<image_id>', methods=['GET'])
|
498 |
# @login_required
|
499 |
def get_image(image_id):
|
500 |
try:
|
@@ -506,7 +506,7 @@ def get_image(image_id):
|
|
506 |
return server_error_response(e)
|
507 |
|
508 |
|
509 |
-
@manager.route('/upload_and_parse', methods=['POST'])
|
510 |
@login_required
|
511 |
@validate_request("conversation_id")
|
512 |
def upload_and_parse():
|
@@ -525,7 +525,7 @@ def upload_and_parse():
|
|
525 |
return get_json_result(data=doc_ids)
|
526 |
|
527 |
|
528 |
-
@manager.route('/parse', methods=['POST'])
|
529 |
@login_required
|
530 |
def parse():
|
531 |
url = request.json.get("url") if request.json else ""
|
|
|
42 |
from api.constants import IMG_BASE64_PREFIX
|
43 |
|
44 |
|
45 |
+
@manager.route('/upload', methods=['POST']) # noqa: F821
|
46 |
@login_required
|
47 |
@validate_request("kb_id")
|
48 |
def upload():
|
|
|
71 |
return get_json_result(data=True)
|
72 |
|
73 |
|
74 |
+
@manager.route('/web_crawl', methods=['POST']) # noqa: F821
|
75 |
@login_required
|
76 |
@validate_request("kb_id", "name", "url")
|
77 |
def web_crawl():
|
|
|
138 |
return get_json_result(data=True)
|
139 |
|
140 |
|
141 |
+
@manager.route('/create', methods=['POST']) # noqa: F821
|
142 |
@login_required
|
143 |
@validate_request("name", "kb_id")
|
144 |
def create():
|
|
|
174 |
return server_error_response(e)
|
175 |
|
176 |
|
177 |
+
@manager.route('/list', methods=['GET']) # noqa: F821
|
178 |
@login_required
|
179 |
def list_docs():
|
180 |
kb_id = request.args.get("kb_id")
|
|
|
209 |
return server_error_response(e)
|
210 |
|
211 |
|
212 |
+
@manager.route('/infos', methods=['POST']) # noqa: F821
|
213 |
@login_required
|
214 |
def docinfos():
|
215 |
req = request.json
|
|
|
225 |
return get_json_result(data=list(docs.dicts()))
|
226 |
|
227 |
|
228 |
+
@manager.route('/thumbnails', methods=['GET']) # noqa: F821
|
229 |
# @login_required
|
230 |
def thumbnails():
|
231 |
doc_ids = request.args.get("doc_ids").split(",")
|
|
|
245 |
return server_error_response(e)
|
246 |
|
247 |
|
248 |
+
@manager.route('/change_status', methods=['POST']) # noqa: F821
|
249 |
@login_required
|
250 |
@validate_request("doc_id", "status")
|
251 |
def change_status():
|
|
|
284 |
return server_error_response(e)
|
285 |
|
286 |
|
287 |
+
@manager.route('/rm', methods=['POST']) # noqa: F821
|
288 |
@login_required
|
289 |
@validate_request("doc_id")
|
290 |
def rm():
|
|
|
334 |
return get_json_result(data=True)
|
335 |
|
336 |
|
337 |
+
@manager.route('/run', methods=['POST']) # noqa: F821
|
338 |
@login_required
|
339 |
@validate_request("doc_ids", "run")
|
340 |
def run():
|
|
|
377 |
return server_error_response(e)
|
378 |
|
379 |
|
380 |
+
@manager.route('/rename', methods=['POST']) # noqa: F821
|
381 |
@login_required
|
382 |
@validate_request("doc_id", "name")
|
383 |
def rename():
|
|
|
418 |
return server_error_response(e)
|
419 |
|
420 |
|
421 |
+
@manager.route('/get/<doc_id>', methods=['GET']) # noqa: F821
|
422 |
# @login_required
|
423 |
def get(doc_id):
|
424 |
try:
|
|
|
443 |
return server_error_response(e)
|
444 |
|
445 |
|
446 |
+
@manager.route('/change_parser', methods=['POST']) # noqa: F821
|
447 |
@login_required
|
448 |
@validate_request("doc_id", "parser_id")
|
449 |
def change_parser():
|
|
|
494 |
return server_error_response(e)
|
495 |
|
496 |
|
497 |
+
@manager.route('/image/<image_id>', methods=['GET']) # noqa: F821
|
498 |
# @login_required
|
499 |
def get_image(image_id):
|
500 |
try:
|
|
|
506 |
return server_error_response(e)
|
507 |
|
508 |
|
509 |
+
@manager.route('/upload_and_parse', methods=['POST']) # noqa: F821
|
510 |
@login_required
|
511 |
@validate_request("conversation_id")
|
512 |
def upload_and_parse():
|
|
|
525 |
return get_json_result(data=doc_ids)
|
526 |
|
527 |
|
528 |
+
@manager.route('/parse', methods=['POST']) # noqa: F821
|
529 |
@login_required
|
530 |
def parse():
|
531 |
url = request.json.get("url") if request.json else ""
|
api/apps/file2document_app.py
CHANGED
@@ -28,7 +28,7 @@ from api import settings
|
|
28 |
from api.utils.api_utils import get_json_result
|
29 |
|
30 |
|
31 |
-
@manager.route('/convert', methods=['POST'])
|
32 |
@login_required
|
33 |
@validate_request("file_ids", "kb_ids")
|
34 |
def convert():
|
@@ -92,7 +92,7 @@ def convert():
|
|
92 |
return server_error_response(e)
|
93 |
|
94 |
|
95 |
-
@manager.route('/rm', methods=['POST'])
|
96 |
@login_required
|
97 |
@validate_request("file_ids")
|
98 |
def rm():
|
|
|
28 |
from api.utils.api_utils import get_json_result
|
29 |
|
30 |
|
31 |
+
@manager.route('/convert', methods=['POST']) # noqa: F821
|
32 |
@login_required
|
33 |
@validate_request("file_ids", "kb_ids")
|
34 |
def convert():
|
|
|
92 |
return server_error_response(e)
|
93 |
|
94 |
|
95 |
+
@manager.route('/rm', methods=['POST']) # noqa: F821
|
96 |
@login_required
|
97 |
@validate_request("file_ids")
|
98 |
def rm():
|
api/apps/file_app.py
CHANGED
@@ -34,7 +34,7 @@ from api.utils.file_utils import filename_type
|
|
34 |
from rag.utils.storage_factory import STORAGE_IMPL
|
35 |
|
36 |
|
37 |
-
@manager.route('/upload', methods=['POST'])
|
38 |
@login_required
|
39 |
# @validate_request("parent_id")
|
40 |
def upload():
|
@@ -120,7 +120,7 @@ def upload():
|
|
120 |
return server_error_response(e)
|
121 |
|
122 |
|
123 |
-
@manager.route('/create', methods=['POST'])
|
124 |
@login_required
|
125 |
@validate_request("name")
|
126 |
def create():
|
@@ -160,7 +160,7 @@ def create():
|
|
160 |
return server_error_response(e)
|
161 |
|
162 |
|
163 |
-
@manager.route('/list', methods=['GET'])
|
164 |
@login_required
|
165 |
def list_files():
|
166 |
pf_id = request.args.get("parent_id")
|
@@ -192,7 +192,7 @@ def list_files():
|
|
192 |
return server_error_response(e)
|
193 |
|
194 |
|
195 |
-
@manager.route('/root_folder', methods=['GET'])
|
196 |
@login_required
|
197 |
def get_root_folder():
|
198 |
try:
|
@@ -202,7 +202,7 @@ def get_root_folder():
|
|
202 |
return server_error_response(e)
|
203 |
|
204 |
|
205 |
-
@manager.route('/parent_folder', methods=['GET'])
|
206 |
@login_required
|
207 |
def get_parent_folder():
|
208 |
file_id = request.args.get("file_id")
|
@@ -217,7 +217,7 @@ def get_parent_folder():
|
|
217 |
return server_error_response(e)
|
218 |
|
219 |
|
220 |
-
@manager.route('/all_parent_folder', methods=['GET'])
|
221 |
@login_required
|
222 |
def get_all_parent_folders():
|
223 |
file_id = request.args.get("file_id")
|
@@ -235,7 +235,7 @@ def get_all_parent_folders():
|
|
235 |
return server_error_response(e)
|
236 |
|
237 |
|
238 |
-
@manager.route('/rm', methods=['POST'])
|
239 |
@login_required
|
240 |
@validate_request("file_ids")
|
241 |
def rm():
|
@@ -284,7 +284,7 @@ def rm():
|
|
284 |
return server_error_response(e)
|
285 |
|
286 |
|
287 |
-
@manager.route('/rename', methods=['POST'])
|
288 |
@login_required
|
289 |
@validate_request("file_id", "name")
|
290 |
def rename():
|
@@ -322,7 +322,7 @@ def rename():
|
|
322 |
return server_error_response(e)
|
323 |
|
324 |
|
325 |
-
@manager.route('/get/<file_id>', methods=['GET'])
|
326 |
@login_required
|
327 |
def get(file_id):
|
328 |
try:
|
@@ -350,7 +350,7 @@ def get(file_id):
|
|
350 |
return server_error_response(e)
|
351 |
|
352 |
|
353 |
-
@manager.route('/mv', methods=['POST'])
|
354 |
@login_required
|
355 |
@validate_request("src_file_ids", "dest_file_id")
|
356 |
def move():
|
|
|
34 |
from rag.utils.storage_factory import STORAGE_IMPL
|
35 |
|
36 |
|
37 |
+
@manager.route('/upload', methods=['POST']) # noqa: F821
|
38 |
@login_required
|
39 |
# @validate_request("parent_id")
|
40 |
def upload():
|
|
|
120 |
return server_error_response(e)
|
121 |
|
122 |
|
123 |
+
@manager.route('/create', methods=['POST']) # noqa: F821
|
124 |
@login_required
|
125 |
@validate_request("name")
|
126 |
def create():
|
|
|
160 |
return server_error_response(e)
|
161 |
|
162 |
|
163 |
+
@manager.route('/list', methods=['GET']) # noqa: F821
|
164 |
@login_required
|
165 |
def list_files():
|
166 |
pf_id = request.args.get("parent_id")
|
|
|
192 |
return server_error_response(e)
|
193 |
|
194 |
|
195 |
+
@manager.route('/root_folder', methods=['GET']) # noqa: F821
|
196 |
@login_required
|
197 |
def get_root_folder():
|
198 |
try:
|
|
|
202 |
return server_error_response(e)
|
203 |
|
204 |
|
205 |
+
@manager.route('/parent_folder', methods=['GET']) # noqa: F821
|
206 |
@login_required
|
207 |
def get_parent_folder():
|
208 |
file_id = request.args.get("file_id")
|
|
|
217 |
return server_error_response(e)
|
218 |
|
219 |
|
220 |
+
@manager.route('/all_parent_folder', methods=['GET']) # noqa: F821
|
221 |
@login_required
|
222 |
def get_all_parent_folders():
|
223 |
file_id = request.args.get("file_id")
|
|
|
235 |
return server_error_response(e)
|
236 |
|
237 |
|
238 |
+
@manager.route('/rm', methods=['POST']) # noqa: F821
|
239 |
@login_required
|
240 |
@validate_request("file_ids")
|
241 |
def rm():
|
|
|
284 |
return server_error_response(e)
|
285 |
|
286 |
|
287 |
+
@manager.route('/rename', methods=['POST']) # noqa: F821
|
288 |
@login_required
|
289 |
@validate_request("file_id", "name")
|
290 |
def rename():
|
|
|
322 |
return server_error_response(e)
|
323 |
|
324 |
|
325 |
+
@manager.route('/get/<file_id>', methods=['GET']) # noqa: F821
|
326 |
@login_required
|
327 |
def get(file_id):
|
328 |
try:
|
|
|
350 |
return server_error_response(e)
|
351 |
|
352 |
|
353 |
+
@manager.route('/mv', methods=['POST']) # noqa: F821
|
354 |
@login_required
|
355 |
@validate_request("src_file_ids", "dest_file_id")
|
356 |
def move():
|
api/apps/kb_app.py
CHANGED
@@ -32,7 +32,7 @@ from rag.nlp import search
|
|
32 |
from api.constants import DATASET_NAME_LIMIT
|
33 |
|
34 |
|
35 |
-
@manager.route('/create', methods=['post'])
|
36 |
@login_required
|
37 |
@validate_request("name")
|
38 |
def create():
|
@@ -67,7 +67,7 @@ def create():
|
|
67 |
return server_error_response(e)
|
68 |
|
69 |
|
70 |
-
@manager.route('/update', methods=['post'])
|
71 |
@login_required
|
72 |
@validate_request("kb_id", "name", "description", "permission", "parser_id")
|
73 |
@not_allowed_parameters("id", "tenant_id", "created_by", "create_time", "update_time", "create_date", "update_date", "created_by")
|
@@ -120,7 +120,7 @@ def update():
|
|
120 |
return server_error_response(e)
|
121 |
|
122 |
|
123 |
-
@manager.route('/detail', methods=['GET'])
|
124 |
@login_required
|
125 |
def detail():
|
126 |
kb_id = request.args["kb_id"]
|
@@ -143,7 +143,7 @@ def detail():
|
|
143 |
return server_error_response(e)
|
144 |
|
145 |
|
146 |
-
@manager.route('/list', methods=['GET'])
|
147 |
@login_required
|
148 |
def list_kbs():
|
149 |
keywords = request.args.get("keywords", "")
|
@@ -160,7 +160,7 @@ def list_kbs():
|
|
160 |
return server_error_response(e)
|
161 |
|
162 |
|
163 |
-
@manager.route('/rm', methods=['post'])
|
164 |
@login_required
|
165 |
@validate_request("kb_id")
|
166 |
def rm():
|
|
|
32 |
from api.constants import DATASET_NAME_LIMIT
|
33 |
|
34 |
|
35 |
+
@manager.route('/create', methods=['post']) # noqa: F821
|
36 |
@login_required
|
37 |
@validate_request("name")
|
38 |
def create():
|
|
|
67 |
return server_error_response(e)
|
68 |
|
69 |
|
70 |
+
@manager.route('/update', methods=['post']) # noqa: F821
|
71 |
@login_required
|
72 |
@validate_request("kb_id", "name", "description", "permission", "parser_id")
|
73 |
@not_allowed_parameters("id", "tenant_id", "created_by", "create_time", "update_time", "create_date", "update_date", "created_by")
|
|
|
120 |
return server_error_response(e)
|
121 |
|
122 |
|
123 |
+
@manager.route('/detail', methods=['GET']) # noqa: F821
|
124 |
@login_required
|
125 |
def detail():
|
126 |
kb_id = request.args["kb_id"]
|
|
|
143 |
return server_error_response(e)
|
144 |
|
145 |
|
146 |
+
@manager.route('/list', methods=['GET']) # noqa: F821
|
147 |
@login_required
|
148 |
def list_kbs():
|
149 |
keywords = request.args.get("keywords", "")
|
|
|
160 |
return server_error_response(e)
|
161 |
|
162 |
|
163 |
+
@manager.route('/rm', methods=['post']) # noqa: F821
|
164 |
@login_required
|
165 |
@validate_request("kb_id")
|
166 |
def rm():
|
api/apps/llm_app.py
CHANGED
@@ -28,7 +28,7 @@ from rag.llm import EmbeddingModel, ChatModel, RerankModel, CvModel, TTSModel
|
|
28 |
import requests
|
29 |
|
30 |
|
31 |
-
@manager.route('/factories', methods=['GET'])
|
32 |
@login_required
|
33 |
def factories():
|
34 |
try:
|
@@ -50,7 +50,7 @@ def factories():
|
|
50 |
return server_error_response(e)
|
51 |
|
52 |
|
53 |
-
@manager.route('/set_api_key', methods=['POST'])
|
54 |
@login_required
|
55 |
@validate_request("llm_factory", "api_key")
|
56 |
def set_api_key():
|
@@ -129,7 +129,7 @@ def set_api_key():
|
|
129 |
return get_json_result(data=True)
|
130 |
|
131 |
|
132 |
-
@manager.route('/add_llm', methods=['POST'])
|
133 |
@login_required
|
134 |
@validate_request("llm_factory")
|
135 |
def add_llm():
|
@@ -292,7 +292,7 @@ def add_llm():
|
|
292 |
return get_json_result(data=True)
|
293 |
|
294 |
|
295 |
-
@manager.route('/delete_llm', methods=['POST'])
|
296 |
@login_required
|
297 |
@validate_request("llm_factory", "llm_name")
|
298 |
def delete_llm():
|
@@ -303,7 +303,7 @@ def delete_llm():
|
|
303 |
return get_json_result(data=True)
|
304 |
|
305 |
|
306 |
-
@manager.route('/delete_factory', methods=['POST'])
|
307 |
@login_required
|
308 |
@validate_request("llm_factory")
|
309 |
def delete_factory():
|
@@ -313,7 +313,7 @@ def delete_factory():
|
|
313 |
return get_json_result(data=True)
|
314 |
|
315 |
|
316 |
-
@manager.route('/my_llms', methods=['GET'])
|
317 |
@login_required
|
318 |
def my_llms():
|
319 |
try:
|
@@ -334,7 +334,7 @@ def my_llms():
|
|
334 |
return server_error_response(e)
|
335 |
|
336 |
|
337 |
-
@manager.route('/list', methods=['GET'])
|
338 |
@login_required
|
339 |
def list_app():
|
340 |
self_deploied = ["Youdao", "FastEmbed", "BAAI", "Ollama", "Xinference", "LocalAI", "LM-Studio"]
|
|
|
28 |
import requests
|
29 |
|
30 |
|
31 |
+
@manager.route('/factories', methods=['GET']) # noqa: F821
|
32 |
@login_required
|
33 |
def factories():
|
34 |
try:
|
|
|
50 |
return server_error_response(e)
|
51 |
|
52 |
|
53 |
+
@manager.route('/set_api_key', methods=['POST']) # noqa: F821
|
54 |
@login_required
|
55 |
@validate_request("llm_factory", "api_key")
|
56 |
def set_api_key():
|
|
|
129 |
return get_json_result(data=True)
|
130 |
|
131 |
|
132 |
+
@manager.route('/add_llm', methods=['POST']) # noqa: F821
|
133 |
@login_required
|
134 |
@validate_request("llm_factory")
|
135 |
def add_llm():
|
|
|
292 |
return get_json_result(data=True)
|
293 |
|
294 |
|
295 |
+
@manager.route('/delete_llm', methods=['POST']) # noqa: F821
|
296 |
@login_required
|
297 |
@validate_request("llm_factory", "llm_name")
|
298 |
def delete_llm():
|
|
|
303 |
return get_json_result(data=True)
|
304 |
|
305 |
|
306 |
+
@manager.route('/delete_factory', methods=['POST']) # noqa: F821
|
307 |
@login_required
|
308 |
@validate_request("llm_factory")
|
309 |
def delete_factory():
|
|
|
313 |
return get_json_result(data=True)
|
314 |
|
315 |
|
316 |
+
@manager.route('/my_llms', methods=['GET']) # noqa: F821
|
317 |
@login_required
|
318 |
def my_llms():
|
319 |
try:
|
|
|
334 |
return server_error_response(e)
|
335 |
|
336 |
|
337 |
+
@manager.route('/list', methods=['GET']) # noqa: F821
|
338 |
@login_required
|
339 |
def list_app():
|
340 |
self_deploied = ["Youdao", "FastEmbed", "BAAI", "Ollama", "Xinference", "LocalAI", "LM-Studio"]
|
api/apps/sdk/agent.py
CHANGED
@@ -19,7 +19,7 @@ from api.utils.api_utils import get_error_data_result, token_required
|
|
19 |
from api.utils.api_utils import get_result
|
20 |
from flask import request
|
21 |
|
22 |
-
@manager.route('/agents', methods=['GET'])
|
23 |
@token_required
|
24 |
def list_agents(tenant_id):
|
25 |
id = request.args.get("id")
|
|
|
19 |
from api.utils.api_utils import get_result
|
20 |
from flask import request
|
21 |
|
22 |
+
@manager.route('/agents', methods=['GET']) # noqa: F821
|
23 |
@token_required
|
24 |
def list_agents(tenant_id):
|
25 |
id = request.args.get("id")
|
api/apps/sdk/chat.py
CHANGED
@@ -26,7 +26,7 @@ from api.utils.api_utils import get_result
|
|
26 |
|
27 |
|
28 |
|
29 |
-
@manager.route('/chats', methods=['POST'])
|
30 |
@token_required
|
31 |
def create(tenant_id):
|
32 |
req=request.json
|
@@ -150,7 +150,7 @@ def create(tenant_id):
|
|
150 |
res["avatar"] = res.pop("icon")
|
151 |
return get_result(data=res)
|
152 |
|
153 |
-
@manager.route('/chats/<chat_id>', methods=['PUT'])
|
154 |
@token_required
|
155 |
def update(tenant_id,chat_id):
|
156 |
if not DialogService.query(tenant_id=tenant_id, id=chat_id, status=StatusEnum.VALID.value):
|
@@ -238,7 +238,7 @@ def update(tenant_id,chat_id):
|
|
238 |
return get_result()
|
239 |
|
240 |
|
241 |
-
@manager.route('/chats', methods=['DELETE'])
|
242 |
@token_required
|
243 |
def delete(tenant_id):
|
244 |
req = request.json
|
@@ -260,7 +260,7 @@ def delete(tenant_id):
|
|
260 |
DialogService.update_by_id(id, temp_dict)
|
261 |
return get_result()
|
262 |
|
263 |
-
@manager.route('/chats', methods=['GET'])
|
264 |
@token_required
|
265 |
def list_chat(tenant_id):
|
266 |
id = request.args.get("id")
|
|
|
26 |
|
27 |
|
28 |
|
29 |
+
@manager.route('/chats', methods=['POST']) # noqa: F821
|
30 |
@token_required
|
31 |
def create(tenant_id):
|
32 |
req=request.json
|
|
|
150 |
res["avatar"] = res.pop("icon")
|
151 |
return get_result(data=res)
|
152 |
|
153 |
+
@manager.route('/chats/<chat_id>', methods=['PUT']) # noqa: F821
|
154 |
@token_required
|
155 |
def update(tenant_id,chat_id):
|
156 |
if not DialogService.query(tenant_id=tenant_id, id=chat_id, status=StatusEnum.VALID.value):
|
|
|
238 |
return get_result()
|
239 |
|
240 |
|
241 |
+
@manager.route('/chats', methods=['DELETE']) # noqa: F821
|
242 |
@token_required
|
243 |
def delete(tenant_id):
|
244 |
req = request.json
|
|
|
260 |
DialogService.update_by_id(id, temp_dict)
|
261 |
return get_result()
|
262 |
|
263 |
+
@manager.route('/chats', methods=['GET']) # noqa: F821
|
264 |
@token_required
|
265 |
def list_chat(tenant_id):
|
266 |
id = request.args.get("id")
|
api/apps/sdk/dataset.py
CHANGED
@@ -34,7 +34,7 @@ from api.utils.api_utils import (
|
|
34 |
)
|
35 |
|
36 |
|
37 |
-
@manager.route("/datasets", methods=["POST"])
|
38 |
@token_required
|
39 |
def create(tenant_id):
|
40 |
"""
|
@@ -190,7 +190,7 @@ def create(tenant_id):
|
|
190 |
return get_result(data=renamed_data)
|
191 |
|
192 |
|
193 |
-
@manager.route("/datasets", methods=["DELETE"])
|
194 |
@token_required
|
195 |
def delete(tenant_id):
|
196 |
"""
|
@@ -260,7 +260,7 @@ def delete(tenant_id):
|
|
260 |
return get_result(code=settings.RetCode.SUCCESS)
|
261 |
|
262 |
|
263 |
-
@manager.route("/datasets/<dataset_id>", methods=["PUT"])
|
264 |
@token_required
|
265 |
def update(tenant_id, dataset_id):
|
266 |
"""
|
@@ -429,7 +429,7 @@ def update(tenant_id, dataset_id):
|
|
429 |
return get_result(code=settings.RetCode.SUCCESS)
|
430 |
|
431 |
|
432 |
-
@manager.route("/datasets", methods=["GET"])
|
433 |
@token_required
|
434 |
def list(tenant_id):
|
435 |
"""
|
|
|
34 |
)
|
35 |
|
36 |
|
37 |
+
@manager.route("/datasets", methods=["POST"]) # noqa: F821
|
38 |
@token_required
|
39 |
def create(tenant_id):
|
40 |
"""
|
|
|
190 |
return get_result(data=renamed_data)
|
191 |
|
192 |
|
193 |
+
@manager.route("/datasets", methods=["DELETE"]) # noqa: F821
|
194 |
@token_required
|
195 |
def delete(tenant_id):
|
196 |
"""
|
|
|
260 |
return get_result(code=settings.RetCode.SUCCESS)
|
261 |
|
262 |
|
263 |
+
@manager.route("/datasets/<dataset_id>", methods=["PUT"]) # noqa: F821
|
264 |
@token_required
|
265 |
def update(tenant_id, dataset_id):
|
266 |
"""
|
|
|
429 |
return get_result(code=settings.RetCode.SUCCESS)
|
430 |
|
431 |
|
432 |
+
@manager.route("/datasets", methods=["GET"]) # noqa: F821
|
433 |
@token_required
|
434 |
def list(tenant_id):
|
435 |
"""
|
api/apps/sdk/dify_retrieval.py
CHANGED
@@ -22,7 +22,7 @@ from api import settings
|
|
22 |
from api.utils.api_utils import validate_request, build_error_result, apikey_required
|
23 |
|
24 |
|
25 |
-
@manager.route('/dify/retrieval', methods=['POST'])
|
26 |
@apikey_required
|
27 |
@validate_request("knowledge_id", "query")
|
28 |
def retrieval(tenant_id):
|
|
|
22 |
from api.utils.api_utils import validate_request, build_error_result, apikey_required
|
23 |
|
24 |
|
25 |
+
@manager.route('/dify/retrieval', methods=['POST']) # noqa: F821
|
26 |
@apikey_required
|
27 |
@validate_request("knowledge_id", "query")
|
28 |
def retrieval(tenant_id):
|
api/apps/sdk/doc.py
CHANGED
@@ -45,7 +45,7 @@ from rag.utils.storage_factory import STORAGE_IMPL
|
|
45 |
MAXIMUM_OF_UPLOADING_FILES = 256
|
46 |
|
47 |
|
48 |
-
@manager.route("/datasets/<dataset_id>/documents", methods=["POST"])
|
49 |
@token_required
|
50 |
def upload(dataset_id, tenant_id):
|
51 |
"""
|
@@ -153,7 +153,7 @@ def upload(dataset_id, tenant_id):
|
|
153 |
return get_result(data=renamed_doc_list)
|
154 |
|
155 |
|
156 |
-
@manager.route("/datasets/<dataset_id>/documents/<document_id>", methods=["PUT"])
|
157 |
@token_required
|
158 |
def update_doc(tenant_id, dataset_id, document_id):
|
159 |
"""
|
@@ -296,7 +296,7 @@ def update_doc(tenant_id, dataset_id, document_id):
|
|
296 |
return get_result()
|
297 |
|
298 |
|
299 |
-
@manager.route("/datasets/<dataset_id>/documents/<document_id>", methods=["GET"])
|
300 |
@token_required
|
301 |
def download(tenant_id, dataset_id, document_id):
|
302 |
"""
|
@@ -360,7 +360,7 @@ def download(tenant_id, dataset_id, document_id):
|
|
360 |
)
|
361 |
|
362 |
|
363 |
-
@manager.route("/datasets/<dataset_id>/documents", methods=["GET"])
|
364 |
@token_required
|
365 |
def list_docs(dataset_id, tenant_id):
|
366 |
"""
|
@@ -494,7 +494,7 @@ def list_docs(dataset_id, tenant_id):
|
|
494 |
return get_result(data={"total": tol, "docs": renamed_doc_list})
|
495 |
|
496 |
|
497 |
-
@manager.route("/datasets/<dataset_id>/documents", methods=["DELETE"])
|
498 |
@token_required
|
499 |
def delete(tenant_id, dataset_id):
|
500 |
"""
|
@@ -586,7 +586,7 @@ def delete(tenant_id, dataset_id):
|
|
586 |
return get_result()
|
587 |
|
588 |
|
589 |
-
@manager.route("/datasets/<dataset_id>/chunks", methods=["POST"])
|
590 |
@token_required
|
591 |
def parse(tenant_id, dataset_id):
|
592 |
"""
|
@@ -653,7 +653,7 @@ def parse(tenant_id, dataset_id):
|
|
653 |
return get_result()
|
654 |
|
655 |
|
656 |
-
@manager.route("/datasets/<dataset_id>/chunks", methods=["DELETE"])
|
657 |
@token_required
|
658 |
def stop_parsing(tenant_id, dataset_id):
|
659 |
"""
|
@@ -711,7 +711,7 @@ def stop_parsing(tenant_id, dataset_id):
|
|
711 |
return get_result()
|
712 |
|
713 |
|
714 |
-
@manager.route("/datasets/<dataset_id>/documents/<document_id>/chunks", methods=["GET"])
|
715 |
@token_required
|
716 |
def list_chunks(tenant_id, dataset_id, document_id):
|
717 |
"""
|
@@ -895,7 +895,7 @@ def list_chunks(tenant_id, dataset_id, document_id):
|
|
895 |
return get_result(data=res)
|
896 |
|
897 |
|
898 |
-
@manager.route(
|
899 |
"/datasets/<dataset_id>/documents/<document_id>/chunks", methods=["POST"]
|
900 |
)
|
901 |
@token_required
|
@@ -1038,7 +1038,7 @@ def add_chunk(tenant_id, dataset_id, document_id):
|
|
1038 |
# return get_result(data={"chunk_id": chunk_id})
|
1039 |
|
1040 |
|
1041 |
-
@manager.route(
|
1042 |
"datasets/<dataset_id>/documents/<document_id>/chunks", methods=["DELETE"]
|
1043 |
)
|
1044 |
@token_required
|
@@ -1098,7 +1098,7 @@ def rm_chunk(tenant_id, dataset_id, document_id):
|
|
1098 |
return get_result(message=f"deleted {chunk_number} chunks")
|
1099 |
|
1100 |
|
1101 |
-
@manager.route(
|
1102 |
"/datasets/<dataset_id>/documents/<document_id>/chunks/<chunk_id>", methods=["PUT"]
|
1103 |
)
|
1104 |
@token_required
|
@@ -1208,7 +1208,7 @@ def update_chunk(tenant_id, dataset_id, document_id, chunk_id):
|
|
1208 |
return get_result()
|
1209 |
|
1210 |
|
1211 |
-
@manager.route("/retrieval", methods=["POST"])
|
1212 |
@token_required
|
1213 |
def retrieval_test(tenant_id):
|
1214 |
"""
|
|
|
45 |
MAXIMUM_OF_UPLOADING_FILES = 256
|
46 |
|
47 |
|
48 |
+
@manager.route("/datasets/<dataset_id>/documents", methods=["POST"]) # noqa: F821
|
49 |
@token_required
|
50 |
def upload(dataset_id, tenant_id):
|
51 |
"""
|
|
|
153 |
return get_result(data=renamed_doc_list)
|
154 |
|
155 |
|
156 |
+
@manager.route("/datasets/<dataset_id>/documents/<document_id>", methods=["PUT"]) # noqa: F821
|
157 |
@token_required
|
158 |
def update_doc(tenant_id, dataset_id, document_id):
|
159 |
"""
|
|
|
296 |
return get_result()
|
297 |
|
298 |
|
299 |
+
@manager.route("/datasets/<dataset_id>/documents/<document_id>", methods=["GET"]) # noqa: F821
|
300 |
@token_required
|
301 |
def download(tenant_id, dataset_id, document_id):
|
302 |
"""
|
|
|
360 |
)
|
361 |
|
362 |
|
363 |
+
@manager.route("/datasets/<dataset_id>/documents", methods=["GET"]) # noqa: F821
|
364 |
@token_required
|
365 |
def list_docs(dataset_id, tenant_id):
|
366 |
"""
|
|
|
494 |
return get_result(data={"total": tol, "docs": renamed_doc_list})
|
495 |
|
496 |
|
497 |
+
@manager.route("/datasets/<dataset_id>/documents", methods=["DELETE"]) # noqa: F821
|
498 |
@token_required
|
499 |
def delete(tenant_id, dataset_id):
|
500 |
"""
|
|
|
586 |
return get_result()
|
587 |
|
588 |
|
589 |
+
@manager.route("/datasets/<dataset_id>/chunks", methods=["POST"]) # noqa: F821
|
590 |
@token_required
|
591 |
def parse(tenant_id, dataset_id):
|
592 |
"""
|
|
|
653 |
return get_result()
|
654 |
|
655 |
|
656 |
+
@manager.route("/datasets/<dataset_id>/chunks", methods=["DELETE"]) # noqa: F821
|
657 |
@token_required
|
658 |
def stop_parsing(tenant_id, dataset_id):
|
659 |
"""
|
|
|
711 |
return get_result()
|
712 |
|
713 |
|
714 |
+
@manager.route("/datasets/<dataset_id>/documents/<document_id>/chunks", methods=["GET"]) # noqa: F821
|
715 |
@token_required
|
716 |
def list_chunks(tenant_id, dataset_id, document_id):
|
717 |
"""
|
|
|
895 |
return get_result(data=res)
|
896 |
|
897 |
|
898 |
+
@manager.route( # noqa: F821
|
899 |
"/datasets/<dataset_id>/documents/<document_id>/chunks", methods=["POST"]
|
900 |
)
|
901 |
@token_required
|
|
|
1038 |
# return get_result(data={"chunk_id": chunk_id})
|
1039 |
|
1040 |
|
1041 |
+
@manager.route( # noqa: F821
|
1042 |
"datasets/<dataset_id>/documents/<document_id>/chunks", methods=["DELETE"]
|
1043 |
)
|
1044 |
@token_required
|
|
|
1098 |
return get_result(message=f"deleted {chunk_number} chunks")
|
1099 |
|
1100 |
|
1101 |
+
@manager.route( # noqa: F821
|
1102 |
"/datasets/<dataset_id>/documents/<document_id>/chunks/<chunk_id>", methods=["PUT"]
|
1103 |
)
|
1104 |
@token_required
|
|
|
1208 |
return get_result()
|
1209 |
|
1210 |
|
1211 |
+
@manager.route("/retrieval", methods=["POST"]) # noqa: F821
|
1212 |
@token_required
|
1213 |
def retrieval_test(tenant_id):
|
1214 |
"""
|
api/apps/sdk/session.py
CHANGED
@@ -33,7 +33,7 @@ from api.utils.api_utils import get_result, token_required
|
|
33 |
from api.db.services.llm_service import LLMBundle
|
34 |
|
35 |
|
36 |
-
@manager.route('/chats/<chat_id>/sessions', methods=['POST'])
|
37 |
@token_required
|
38 |
def create(tenant_id, chat_id):
|
39 |
req = request.json
|
@@ -60,7 +60,7 @@ def create(tenant_id, chat_id):
|
|
60 |
return get_result(data=conv)
|
61 |
|
62 |
|
63 |
-
@manager.route('/agents/<agent_id>/sessions', methods=['POST'])
|
64 |
@token_required
|
65 |
def create_agent_session(tenant_id, agent_id):
|
66 |
e, cvs = UserCanvasService.get_by_id(agent_id)
|
@@ -86,7 +86,7 @@ def create_agent_session(tenant_id, agent_id):
|
|
86 |
return get_result(data=conv)
|
87 |
|
88 |
|
89 |
-
@manager.route('/chats/<chat_id>/sessions/<session_id>', methods=['PUT'])
|
90 |
@token_required
|
91 |
def update(tenant_id, chat_id, session_id):
|
92 |
req = request.json
|
@@ -108,7 +108,7 @@ def update(tenant_id, chat_id, session_id):
|
|
108 |
return get_result()
|
109 |
|
110 |
|
111 |
-
@manager.route('/chats/<chat_id>/completions', methods=['POST'])
|
112 |
@token_required
|
113 |
def completion(tenant_id, chat_id):
|
114 |
dia = DialogService.query(id=chat_id, tenant_id=tenant_id, status=StatusEnum.VALID.value)
|
@@ -219,7 +219,7 @@ def completion(tenant_id, chat_id):
|
|
219 |
return get_result(data=answer)
|
220 |
|
221 |
|
222 |
-
@manager.route('/agents/<agent_id>/completions', methods=['POST'])
|
223 |
@token_required
|
224 |
def agent_completion(tenant_id, agent_id):
|
225 |
req = request.json
|
@@ -379,7 +379,7 @@ def agent_completion(tenant_id, agent_id):
|
|
379 |
return get_result(data=result)
|
380 |
|
381 |
|
382 |
-
@manager.route('/chats/<chat_id>/sessions', methods=['GET'])
|
383 |
@token_required
|
384 |
def list_session(tenant_id, chat_id):
|
385 |
if not DialogService.query(tenant_id=tenant_id, id=chat_id, status=StatusEnum.VALID.value):
|
@@ -433,7 +433,7 @@ def list_session(tenant_id, chat_id):
|
|
433 |
return get_result(data=convs)
|
434 |
|
435 |
|
436 |
-
@manager.route('/agents/<agent_id>/sessions', methods=['GET'])
|
437 |
@token_required
|
438 |
def list_agent_session(tenant_id, agent_id):
|
439 |
if not UserCanvasService.query(user_id=tenant_id, id=agent_id):
|
@@ -488,7 +488,7 @@ def list_agent_session(tenant_id, agent_id):
|
|
488 |
return get_result(data=convs)
|
489 |
|
490 |
|
491 |
-
@manager.route('/chats/<chat_id>/sessions', methods=["DELETE"])
|
492 |
@token_required
|
493 |
def delete(tenant_id, chat_id):
|
494 |
if not DialogService.query(id=chat_id, tenant_id=tenant_id, status=StatusEnum.VALID.value):
|
@@ -514,7 +514,7 @@ def delete(tenant_id, chat_id):
|
|
514 |
return get_result()
|
515 |
|
516 |
|
517 |
-
@manager.route('/sessions/ask', methods=['POST'])
|
518 |
@token_required
|
519 |
def ask_about(tenant_id):
|
520 |
req = request.json
|
@@ -553,7 +553,7 @@ def ask_about(tenant_id):
|
|
553 |
return resp
|
554 |
|
555 |
|
556 |
-
@manager.route('/sessions/related_questions', methods=['POST'])
|
557 |
@token_required
|
558 |
def related_questions(tenant_id):
|
559 |
req = request.json
|
|
|
33 |
from api.db.services.llm_service import LLMBundle
|
34 |
|
35 |
|
36 |
+
@manager.route('/chats/<chat_id>/sessions', methods=['POST']) # noqa: F821
|
37 |
@token_required
|
38 |
def create(tenant_id, chat_id):
|
39 |
req = request.json
|
|
|
60 |
return get_result(data=conv)
|
61 |
|
62 |
|
63 |
+
@manager.route('/agents/<agent_id>/sessions', methods=['POST']) # noqa: F821
|
64 |
@token_required
|
65 |
def create_agent_session(tenant_id, agent_id):
|
66 |
e, cvs = UserCanvasService.get_by_id(agent_id)
|
|
|
86 |
return get_result(data=conv)
|
87 |
|
88 |
|
89 |
+
@manager.route('/chats/<chat_id>/sessions/<session_id>', methods=['PUT']) # noqa: F821
|
90 |
@token_required
|
91 |
def update(tenant_id, chat_id, session_id):
|
92 |
req = request.json
|
|
|
108 |
return get_result()
|
109 |
|
110 |
|
111 |
+
@manager.route('/chats/<chat_id>/completions', methods=['POST']) # noqa: F821
|
112 |
@token_required
|
113 |
def completion(tenant_id, chat_id):
|
114 |
dia = DialogService.query(id=chat_id, tenant_id=tenant_id, status=StatusEnum.VALID.value)
|
|
|
219 |
return get_result(data=answer)
|
220 |
|
221 |
|
222 |
+
@manager.route('/agents/<agent_id>/completions', methods=['POST']) # noqa: F821
|
223 |
@token_required
|
224 |
def agent_completion(tenant_id, agent_id):
|
225 |
req = request.json
|
|
|
379 |
return get_result(data=result)
|
380 |
|
381 |
|
382 |
+
@manager.route('/chats/<chat_id>/sessions', methods=['GET']) # noqa: F821
|
383 |
@token_required
|
384 |
def list_session(tenant_id, chat_id):
|
385 |
if not DialogService.query(tenant_id=tenant_id, id=chat_id, status=StatusEnum.VALID.value):
|
|
|
433 |
return get_result(data=convs)
|
434 |
|
435 |
|
436 |
+
@manager.route('/agents/<agent_id>/sessions', methods=['GET']) # noqa: F821
|
437 |
@token_required
|
438 |
def list_agent_session(tenant_id, agent_id):
|
439 |
if not UserCanvasService.query(user_id=tenant_id, id=agent_id):
|
|
|
488 |
return get_result(data=convs)
|
489 |
|
490 |
|
491 |
+
@manager.route('/chats/<chat_id>/sessions', methods=["DELETE"]) # noqa: F821
|
492 |
@token_required
|
493 |
def delete(tenant_id, chat_id):
|
494 |
if not DialogService.query(id=chat_id, tenant_id=tenant_id, status=StatusEnum.VALID.value):
|
|
|
514 |
return get_result()
|
515 |
|
516 |
|
517 |
+
@manager.route('/sessions/ask', methods=['POST']) # noqa: F821
|
518 |
@token_required
|
519 |
def ask_about(tenant_id):
|
520 |
req = request.json
|
|
|
553 |
return resp
|
554 |
|
555 |
|
556 |
+
@manager.route('/sessions/related_questions', methods=['POST']) # noqa: F821
|
557 |
@token_required
|
558 |
def related_questions(tenant_id):
|
559 |
req = request.json
|
api/apps/system_app.py
CHANGED
@@ -38,7 +38,7 @@ from timeit import default_timer as timer
|
|
38 |
from rag.utils.redis_conn import REDIS_CONN
|
39 |
|
40 |
|
41 |
-
@manager.route("/version", methods=["GET"])
|
42 |
@login_required
|
43 |
def version():
|
44 |
"""
|
@@ -61,7 +61,7 @@ def version():
|
|
61 |
return get_json_result(data=get_ragflow_version())
|
62 |
|
63 |
|
64 |
-
@manager.route("/status", methods=["GET"])
|
65 |
@login_required
|
66 |
def status():
|
67 |
"""
|
@@ -170,7 +170,7 @@ def status():
|
|
170 |
return get_json_result(data=res)
|
171 |
|
172 |
|
173 |
-
@manager.route("/new_token", methods=["POST"])
|
174 |
@login_required
|
175 |
def new_token():
|
176 |
"""
|
@@ -219,7 +219,7 @@ def new_token():
|
|
219 |
return server_error_response(e)
|
220 |
|
221 |
|
222 |
-
@manager.route("/token_list", methods=["GET"])
|
223 |
@login_required
|
224 |
def token_list():
|
225 |
"""
|
@@ -261,7 +261,7 @@ def token_list():
|
|
261 |
return server_error_response(e)
|
262 |
|
263 |
|
264 |
-
@manager.route("/token/<token>", methods=["DELETE"])
|
265 |
@login_required
|
266 |
def rm(token):
|
267 |
"""
|
|
|
38 |
from rag.utils.redis_conn import REDIS_CONN
|
39 |
|
40 |
|
41 |
+
@manager.route("/version", methods=["GET"]) # noqa: F821
|
42 |
@login_required
|
43 |
def version():
|
44 |
"""
|
|
|
61 |
return get_json_result(data=get_ragflow_version())
|
62 |
|
63 |
|
64 |
+
@manager.route("/status", methods=["GET"]) # noqa: F821
|
65 |
@login_required
|
66 |
def status():
|
67 |
"""
|
|
|
170 |
return get_json_result(data=res)
|
171 |
|
172 |
|
173 |
+
@manager.route("/new_token", methods=["POST"]) # noqa: F821
|
174 |
@login_required
|
175 |
def new_token():
|
176 |
"""
|
|
|
219 |
return server_error_response(e)
|
220 |
|
221 |
|
222 |
+
@manager.route("/token_list", methods=["GET"]) # noqa: F821
|
223 |
@login_required
|
224 |
def token_list():
|
225 |
"""
|
|
|
261 |
return server_error_response(e)
|
262 |
|
263 |
|
264 |
+
@manager.route("/token/<token>", methods=["DELETE"]) # noqa: F821
|
265 |
@login_required
|
266 |
def rm(token):
|
267 |
"""
|
api/apps/tenant_app.py
CHANGED
@@ -26,7 +26,7 @@ from api.utils import get_uuid, delta_seconds
|
|
26 |
from api.utils.api_utils import get_json_result, validate_request, server_error_response, get_data_error_result
|
27 |
|
28 |
|
29 |
-
@manager.route("/<tenant_id>/user/list", methods=["GET"])
|
30 |
@login_required
|
31 |
def user_list(tenant_id):
|
32 |
if current_user.id != tenant_id:
|
@@ -44,7 +44,7 @@ def user_list(tenant_id):
|
|
44 |
return server_error_response(e)
|
45 |
|
46 |
|
47 |
-
@manager.route('/<tenant_id>/user', methods=['POST'])
|
48 |
@login_required
|
49 |
@validate_request("email")
|
50 |
def create(tenant_id):
|
@@ -80,7 +80,7 @@ def create(tenant_id):
|
|
80 |
return get_json_result(data=usr)
|
81 |
|
82 |
|
83 |
-
@manager.route('/<tenant_id>/user/<user_id>', methods=['DELETE'])
|
84 |
@login_required
|
85 |
def rm(tenant_id, user_id):
|
86 |
if current_user.id != tenant_id and current_user.id != user_id:
|
@@ -96,7 +96,7 @@ def rm(tenant_id, user_id):
|
|
96 |
return server_error_response(e)
|
97 |
|
98 |
|
99 |
-
@manager.route("/list", methods=["GET"])
|
100 |
@login_required
|
101 |
def tenant_list():
|
102 |
try:
|
@@ -108,7 +108,7 @@ def tenant_list():
|
|
108 |
return server_error_response(e)
|
109 |
|
110 |
|
111 |
-
@manager.route("/agree/<tenant_id>", methods=["PUT"])
|
112 |
@login_required
|
113 |
def agree(tenant_id):
|
114 |
try:
|
|
|
26 |
from api.utils.api_utils import get_json_result, validate_request, server_error_response, get_data_error_result
|
27 |
|
28 |
|
29 |
+
@manager.route("/<tenant_id>/user/list", methods=["GET"]) # noqa: F821
|
30 |
@login_required
|
31 |
def user_list(tenant_id):
|
32 |
if current_user.id != tenant_id:
|
|
|
44 |
return server_error_response(e)
|
45 |
|
46 |
|
47 |
+
@manager.route('/<tenant_id>/user', methods=['POST']) # noqa: F821
|
48 |
@login_required
|
49 |
@validate_request("email")
|
50 |
def create(tenant_id):
|
|
|
80 |
return get_json_result(data=usr)
|
81 |
|
82 |
|
83 |
+
@manager.route('/<tenant_id>/user/<user_id>', methods=['DELETE']) # noqa: F821
|
84 |
@login_required
|
85 |
def rm(tenant_id, user_id):
|
86 |
if current_user.id != tenant_id and current_user.id != user_id:
|
|
|
96 |
return server_error_response(e)
|
97 |
|
98 |
|
99 |
+
@manager.route("/list", methods=["GET"]) # noqa: F821
|
100 |
@login_required
|
101 |
def tenant_list():
|
102 |
try:
|
|
|
108 |
return server_error_response(e)
|
109 |
|
110 |
|
111 |
+
@manager.route("/agree/<tenant_id>", methods=["PUT"]) # noqa: F821
|
112 |
@login_required
|
113 |
def agree(tenant_id):
|
114 |
try:
|
api/apps/user_app.py
CHANGED
@@ -44,7 +44,7 @@ from api.db.services.file_service import FileService
|
|
44 |
from api.utils.api_utils import get_json_result, construct_response
|
45 |
|
46 |
|
47 |
-
@manager.route("/login", methods=["POST", "GET"])
|
48 |
def login():
|
49 |
"""
|
50 |
User login endpoint.
|
@@ -115,7 +115,7 @@ def login():
|
|
115 |
)
|
116 |
|
117 |
|
118 |
-
@manager.route("/github_callback", methods=["GET"])
|
119 |
def github_callback():
|
120 |
"""
|
121 |
GitHub OAuth callback endpoint.
|
@@ -200,7 +200,7 @@ def github_callback():
|
|
200 |
return redirect("/?auth=%s" % user.get_id())
|
201 |
|
202 |
|
203 |
-
@manager.route("/feishu_callback", methods=["GET"])
|
204 |
def feishu_callback():
|
205 |
"""
|
206 |
Feishu OAuth callback endpoint.
|
@@ -335,7 +335,7 @@ def user_info_from_github(access_token):
|
|
335 |
return user_info
|
336 |
|
337 |
|
338 |
-
@manager.route("/logout", methods=["GET"])
|
339 |
@login_required
|
340 |
def log_out():
|
341 |
"""
|
@@ -357,7 +357,7 @@ def log_out():
|
|
357 |
return get_json_result(data=True)
|
358 |
|
359 |
|
360 |
-
@manager.route("/setting", methods=["POST"])
|
361 |
@login_required
|
362 |
def setting_user():
|
363 |
"""
|
@@ -429,7 +429,7 @@ def setting_user():
|
|
429 |
)
|
430 |
|
431 |
|
432 |
-
@manager.route("/info", methods=["GET"])
|
433 |
@login_required
|
434 |
def user_profile():
|
435 |
"""
|
@@ -531,7 +531,7 @@ def user_register(user_id, user):
|
|
531 |
return UserService.query(email=user["email"])
|
532 |
|
533 |
|
534 |
-
@manager.route("/register", methods=["POST"])
|
535 |
@validate_request("nickname", "email", "password")
|
536 |
def user_add():
|
537 |
"""
|
@@ -617,7 +617,7 @@ def user_add():
|
|
617 |
)
|
618 |
|
619 |
|
620 |
-
@manager.route("/tenant_info", methods=["GET"])
|
621 |
@login_required
|
622 |
def tenant_info():
|
623 |
"""
|
@@ -655,7 +655,7 @@ def tenant_info():
|
|
655 |
return server_error_response(e)
|
656 |
|
657 |
|
658 |
-
@manager.route("/set_tenant_info", methods=["POST"])
|
659 |
@login_required
|
660 |
@validate_request("tenant_id", "asr_id", "embd_id", "img2txt_id", "llm_id")
|
661 |
def set_tenant_info():
|
|
|
44 |
from api.utils.api_utils import get_json_result, construct_response
|
45 |
|
46 |
|
47 |
+
@manager.route("/login", methods=["POST", "GET"]) # noqa: F821
|
48 |
def login():
|
49 |
"""
|
50 |
User login endpoint.
|
|
|
115 |
)
|
116 |
|
117 |
|
118 |
+
@manager.route("/github_callback", methods=["GET"]) # noqa: F821
|
119 |
def github_callback():
|
120 |
"""
|
121 |
GitHub OAuth callback endpoint.
|
|
|
200 |
return redirect("/?auth=%s" % user.get_id())
|
201 |
|
202 |
|
203 |
+
@manager.route("/feishu_callback", methods=["GET"]) # noqa: F821
|
204 |
def feishu_callback():
|
205 |
"""
|
206 |
Feishu OAuth callback endpoint.
|
|
|
335 |
return user_info
|
336 |
|
337 |
|
338 |
+
@manager.route("/logout", methods=["GET"]) # noqa: F821
|
339 |
@login_required
|
340 |
def log_out():
|
341 |
"""
|
|
|
357 |
return get_json_result(data=True)
|
358 |
|
359 |
|
360 |
+
@manager.route("/setting", methods=["POST"]) # noqa: F821
|
361 |
@login_required
|
362 |
def setting_user():
|
363 |
"""
|
|
|
429 |
)
|
430 |
|
431 |
|
432 |
+
@manager.route("/info", methods=["GET"]) # noqa: F821
|
433 |
@login_required
|
434 |
def user_profile():
|
435 |
"""
|
|
|
531 |
return UserService.query(email=user["email"])
|
532 |
|
533 |
|
534 |
+
@manager.route("/register", methods=["POST"]) # noqa: F821
|
535 |
@validate_request("nickname", "email", "password")
|
536 |
def user_add():
|
537 |
"""
|
|
|
617 |
)
|
618 |
|
619 |
|
620 |
+
@manager.route("/tenant_info", methods=["GET"]) # noqa: F821
|
621 |
@login_required
|
622 |
def tenant_info():
|
623 |
"""
|
|
|
655 |
return server_error_response(e)
|
656 |
|
657 |
|
658 |
+
@manager.route("/set_tenant_info", methods=["POST"]) # noqa: F821
|
659 |
@login_required
|
660 |
@validate_request("tenant_id", "asr_id", "embd_id", "img2txt_id", "llm_id")
|
661 |
def set_tenant_info():
|
deepdoc/parser/pdf_parser.py
CHANGED
@@ -21,7 +21,6 @@ import re
|
|
21 |
import pdfplumber
|
22 |
from PIL import Image
|
23 |
import numpy as np
|
24 |
-
from timeit import default_timer as timer
|
25 |
from pypdf import PdfReader as pdf2_read
|
26 |
|
27 |
from api import settings
|
@@ -949,7 +948,6 @@ class RAGFlowPdfParser:
|
|
949 |
self.page_cum_height = [0]
|
950 |
self.page_layout = []
|
951 |
self.page_from = page_from
|
952 |
-
st = timer()
|
953 |
try:
|
954 |
self.pdf = pdfplumber.open(fnm) if isinstance(
|
955 |
fnm, str) else pdfplumber.open(BytesIO(fnm))
|
|
|
21 |
import pdfplumber
|
22 |
from PIL import Image
|
23 |
import numpy as np
|
|
|
24 |
from pypdf import PdfReader as pdf2_read
|
25 |
|
26 |
from api import settings
|
|
|
948 |
self.page_cum_height = [0]
|
949 |
self.page_layout = []
|
950 |
self.page_from = page_from
|
|
|
951 |
try:
|
952 |
self.pdf = pdfplumber.open(fnm) if isinstance(
|
953 |
fnm, str) else pdfplumber.open(BytesIO(fnm))
|
deepdoc/vision/ocr.py
CHANGED
@@ -18,7 +18,7 @@ import os
|
|
18 |
from huggingface_hub import snapshot_download
|
19 |
|
20 |
from api.utils.file_utils import get_project_base_directory
|
21 |
-
from .operators import *
|
22 |
import math
|
23 |
import numpy as np
|
24 |
import cv2
|
|
|
18 |
from huggingface_hub import snapshot_download
|
19 |
|
20 |
from api.utils.file_utils import get_project_base_directory
|
21 |
+
from .operators import * # noqa: F403
|
22 |
import math
|
23 |
import numpy as np
|
24 |
import cv2
|
deepdoc/vision/recognizer.py
CHANGED
@@ -23,7 +23,8 @@ import onnxruntime as ort
|
|
23 |
from huggingface_hub import snapshot_download
|
24 |
|
25 |
from api.utils.file_utils import get_project_base_directory
|
26 |
-
from .operators import *
|
|
|
27 |
|
28 |
class Recognizer(object):
|
29 |
def __init__(self, label_list, task_name, model_dir=None):
|
|
|
23 |
from huggingface_hub import snapshot_download
|
24 |
|
25 |
from api.utils.file_utils import get_project_base_directory
|
26 |
+
from .operators import * # noqa: F403
|
27 |
+
from .operators import preprocess
|
28 |
|
29 |
class Recognizer(object):
|
30 |
def __init__(self, label_list, task_name, model_dir=None):
|
graphrag/entity_embedding.py
CHANGED
@@ -10,6 +10,7 @@ import numpy as np
|
|
10 |
import networkx as nx
|
11 |
from dataclasses import dataclass
|
12 |
from graphrag.leiden import stable_largest_connected_component
|
|
|
13 |
|
14 |
|
15 |
@dataclass
|
|
|
10 |
import networkx as nx
|
11 |
from dataclasses import dataclass
|
12 |
from graphrag.leiden import stable_largest_connected_component
|
13 |
+
import graspologic as gc
|
14 |
|
15 |
|
16 |
@dataclass
|