[improvement][chat] Move python code out of chat-core module

chat/python/services_router/preset_query_service.py

@@ -0,0 +1,71 @@
+# -*- coding:utf-8 -*-
+import os
+import sys
+from typing import Any, List, Mapping, Optional, Union
+
+sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+
+from fastapi import APIRouter, Depends, HTTPException
+
+from services.query_retrieval.run import preset_query_retriever
+
+router = APIRouter()
+
+@router.post("/preset_query_retrival")
+def preset_query_retrival(query_text_list: List[str], n_results: int = 5):
+    parsed_retrieval_res_format = preset_query_retriever.retrieval_query_run(query_texts_list=query_text_list, filter_condition=None, n_results=n_results)
+
+    return parsed_retrieval_res_format
+
+
+@router.post("/preset_query_add")
+def preset_query_add(preset_info_list: List[Mapping[str, str]]):
+    preset_queries = []
+    preset_query_ids = []
+
+    for preset_info in preset_info_list:
+        preset_queries.append(preset_info['preset_query'])
+        preset_query_ids.append(preset_info['preset_query_id'])
+
+    preset_query_retriever.add_queries(query_text_list=preset_queries, query_id_list=preset_query_ids, metadatas=None)
+
+    return "success"
+
+@router.post("/preset_query_update")
+def preset_query_update(preset_info_list: List[Mapping[str, str]]):
+    preset_queries = []
+    preset_query_ids = []
+
+    for preset_info in preset_info_list:
+        preset_queries.append(preset_info['preset_query'])
+        preset_query_ids.append(preset_info['preset_query_id'])
+
+    preset_query_retriever.update_queries(query_text_list=preset_queries, query_id_list=preset_query_ids, metadatas=None)
+
+    return "success"
+
+
+@router.get("/preset_query_empty")
+def preset_query_empty():
+    preset_query_retriever.empty_query_collection()
+
+    return "success"
+
+@router.post("/preset_delete_by_ids")
+def preset_delete_by_ids(preset_query_ids: List[str]):
+    preset_query_retriever.delete_queries_by_ids(preset_query_ids)
+
+    return "success"
+
+@router.post("/preset_get_by_ids")
+def preset_get_by_ids(preset_query_ids: List[str]):
+    preset_queries = preset_query_retriever.get_query_by_ids(preset_query_ids)
+
+    return preset_queries
+
+@router.get("/preset_query_size")
+def preset_query_size():
+    size = preset_query_retriever.get_query_size()
+
+    return size