ignore type check warning

infiniflow · Nov 25, 2024 · f4c292a · f4c292a
1 parent f4e8e08
commit f4c292a
Show file tree

Hide file tree

Showing 4 changed files with 31 additions and 26 deletions.
diff --git a/api/apps/conversation_app.py b/api/apps/conversation_app.py
@@ -31,7 +31,7 @@
 from graphrag.mind_map_extractor import MindMapExtractor
 
 
-@manager.route('/set', methods=['POST'])
+@manager.route('/set', methods=['POST']) # type: ignore
 @login_required
 def set_conversation():
     req = request.json
@@ -72,7 +72,7 @@ def set_conversation():
         return server_error_response(e)
 
 
-@manager.route('/get', methods=['GET'])
+@manager.route('/get', methods=['GET']) # type: ignore
 @login_required
 def get():
     conv_id = request.args["conversation_id"]
@@ -94,7 +94,7 @@ def get():
         return server_error_response(e)
 
 
-@manager.route('/rm', methods=['POST'])
+@manager.route('/rm', methods=['POST'])# type: ignore
 @login_required
 def rm():
     conv_ids = request.json["conversation_ids"]
@@ -117,7 +117,7 @@ def rm():
         return server_error_response(e)
 
 
-@manager.route('/list', methods=['GET'])
+@manager.route('/list', methods=['GET'])# type: ignore
 @login_required
 def list_convsersation():
     dialog_id = request.args["dialog_id"]
@@ -136,7 +136,7 @@ def list_convsersation():
         return server_error_response(e)
 
 
-@manager.route('/completion', methods=['POST'])
+@manager.route('/completion', methods=['POST'])# type: ignore
 @login_required
 @validate_request("conversation_id", "messages")
 def completion():
@@ -209,7 +209,7 @@ def stream():
         return server_error_response(e)
 
 
-@manager.route('/tts', methods=['POST'])
+@manager.route('/tts', methods=['POST'])# type: ignore
 @login_required
 def tts():
     req = request.json
@@ -243,7 +243,7 @@ def stream_audio():
     return resp
 
 
-@manager.route('/delete_msg', methods=['POST'])
+@manager.route('/delete_msg', methods=['POST'])# type: ignore
 @login_required
 @validate_request("conversation_id", "message_id")
 def delete_msg():
@@ -266,7 +266,7 @@ def delete_msg():
     return get_json_result(data=conv)
 
 
-@manager.route('/thumbup', methods=['POST'])
+@manager.route('/thumbup', methods=['POST'])# type: ignore
 @login_required
 @validate_request("conversation_id", "message_id")
 def thumbup():
@@ -291,7 +291,7 @@ def thumbup():
     return get_json_result(data=conv)
 
 
-@manager.route('/ask', methods=['POST'])
+@manager.route('/ask', methods=['POST'])# type: ignore
 @login_required
 @validate_request("question", "kb_ids")
 def ask_about():
@@ -317,7 +317,7 @@ def stream():
     return resp
 
 
-@manager.route('/mindmap', methods=['POST'])
+@manager.route('/mindmap', methods=['POST'])# type: ignore
 @login_required
 @validate_request("question", "kb_ids")
 def mindmap():
@@ -339,7 +339,7 @@ def mindmap():
     return get_json_result(data=mind_map)
 
 
-@manager.route('/related_questions', methods=['POST'])
+@manager.route('/related_questions', methods=['POST'])# type: ignore
 @login_required
 @validate_request("question")
 def related_questions():

diff --git a/api/apps/sdk/chat.py b/api/apps/sdk/chat.py
@@ -26,7 +26,7 @@
 
 
 
-@manager.route('/chats', methods=['POST'])
+@manager.route('/chats', methods=['POST']) # type: ignore
 @token_required
 def create(tenant_id):
     req=request.json
@@ -235,7 +235,7 @@ def update(tenant_id,chat_id):
     return get_result()
 
 
-@manager.route('/chats', methods=['DELETE'])
+@manager.route('/chats', methods=['DELETE']) # type: ignore
 @token_required
 def delete(tenant_id):
     req = request.json
@@ -257,7 +257,7 @@ def delete(tenant_id):
         DialogService.update_by_id(id, temp_dict)
     return get_result()
 
-@manager.route('/chats', methods=['GET'])
+@manager.route('/chats', methods=['GET']) # type: ignore
 @token_required
 def list_chat(tenant_id):
     id = request.args.get("id")

diff --git a/api/apps/sdk/dataset.py b/api/apps/sdk/dataset.py
@@ -34,7 +34,7 @@
 )
 
 
-@manager.route("/datasets", methods=["POST"])
+@manager.route("/datasets", methods=["POST"]) # type: ignore
 @token_required
 def create(tenant_id):
     """
@@ -190,7 +190,7 @@ def create(tenant_id):
     return get_result(data=renamed_data)
 
 
-@manager.route("/datasets", methods=["DELETE"])
+@manager.route("/datasets", methods=["DELETE"]) # type: ignore
 @token_required
 def delete(tenant_id):
     """

diff --git a/api/apps/sdk/session.py b/api/apps/sdk/session.py
@@ -13,6 +13,7 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 #
+import logging
 import re
 import json
 from functools import partial
@@ -33,7 +34,7 @@
 from api.db.services.llm_service import LLMBundle
 
 
-@manager.route('/chats/<chat_id>/sessions', methods=['POST'])
+@manager.route('/chats/<chat_id>/sessions', methods=['POST'])# type: ignore
 @token_required
 def create(tenant_id,chat_id):
     req = request.json
@@ -60,7 +61,7 @@ def create(tenant_id,chat_id):
     return get_result(data=conv)
 
 
-@manager.route('/agents/<agent_id>/sessions', methods=['POST'])
+@manager.route('/agents/<agent_id>/sessions', methods=['POST']) # type: ignore
 @token_required
 def create_agent_session(tenant_id, agent_id):
     req = request.json
@@ -86,10 +87,14 @@ def create_agent_session(tenant_id, agent_id):
     return get_result(data=conv)
 
 
-@manager.route('/chats/<chat_id>/sessions/<session_id>', methods=['PUT'])
+@manager.route('/chats/<chat_id>/sessions/<session_id>', methods=['PUT']) # type: ignore
 @token_required
 def update(tenant_id,chat_id,session_id):
     req = request.json
+    logging.info(tenant_id)
+    logging.info(session_id)
+    logging.info(chat_id)
+    logging.info("json %s",req["name"])
     req["dialog_id"] = chat_id
     conv_id = session_id
     conv = ConversationService.query(id=conv_id,dialog_id=chat_id)
@@ -108,7 +113,7 @@ def update(tenant_id,chat_id,session_id):
     return get_result()
 
 
-@manager.route('/chats/<chat_id>/completions', methods=['POST'])
+@manager.route('/chats/<chat_id>/completions', methods=['POST']) # type: ignore
 @token_required
 def completion(tenant_id, chat_id):
     req = request.json
@@ -117,7 +122,7 @@ def completion(tenant_id, chat_id):
             "id": get_uuid(),
             "dialog_id": chat_id,
             "name": req.get("name", "New session"),
-            "message": [{"role": "assistant", "content": "Hi! I am your assistant，can I help you?"}]
+            "message": [{"role": "assistant", "content": "Hi! I am your assistant, can I help you?"}]
         }
         if not conv.get("name"):
             return get_error_data_result(message="`name` can not be empty.")
@@ -215,7 +220,7 @@ def stream():
         return get_result(data=answer)
 
 
-@manager.route('/agents/<agent_id>/completions', methods=['POST'])
+@manager.route('/agents/<agent_id>/completions', methods=['POST']) # type: ignore
 @token_required
 def agent_completion(tenant_id, agent_id):
     req = request.json
@@ -373,7 +378,7 @@ def sse():
         return get_result(data=result)
 
 
-@manager.route('/chats/<chat_id>/sessions', methods=['GET'])
+@manager.route('/chats/<chat_id>/sessions', methods=['GET']) # type: ignore
 @token_required
 def list_session(chat_id,tenant_id):
     if not DialogService.query(tenant_id=tenant_id, id=chat_id, status=StatusEnum.VALID.value):
@@ -427,7 +432,7 @@ def list_session(chat_id,tenant_id):
     return get_result(data=convs)
 
 
-@manager.route('/chats/<chat_id>/sessions', methods=["DELETE"])
+@manager.route('/chats/<chat_id>/sessions', methods=["DELETE"]) # type: ignore
 @token_required
 def delete(tenant_id,chat_id):
     if not DialogService.query(id=chat_id, tenant_id=tenant_id, status=StatusEnum.VALID.value):
@@ -452,7 +457,7 @@ def delete(tenant_id,chat_id):
         ConversationService.delete_by_id(id)
     return get_result()
 
-@manager.route('/sessions/ask', methods=['POST'])
+@manager.route('/sessions/ask', methods=['POST']) # type: ignore
 @token_required
 def ask_about(tenant_id):
     req = request.json
@@ -490,7 +495,7 @@ def stream():
     return resp
 
 
-@manager.route('/sessions/related_questions', methods=['POST'])
+@manager.route('/sessions/related_questions', methods=['POST']) # type: ignore
 @token_required
 def related_questions(tenant_id):
     req = request.json