Remove unused /datasets/<id>/embedding endpoint (#16936)

api/apps/restful_apis/dataset_api.py

@@ -910,21 +910,6 @@ def delete_index(tenant_id, dataset_id, index_type=None):
         return get_error_data_result(message="Internal server error")
 
 
-@manager.route("/datasets/<dataset_id>/embedding", methods=["POST"])  # noqa: F821
-@login_required
-@add_tenant_id_to_kwargs
-async def run_embedding(tenant_id, dataset_id):
-    try:
-        success, result = dataset_api_service.run_embedding(dataset_id, tenant_id)
-        if success:
-            return get_result(data=result)
-        else:
-            return get_error_data_result(message=result)
-    except Exception as e:
-        logging.exception(e)
-        return get_error_data_result(message="Internal server error")
-
-
 @manager.route("/datasets/<dataset_id>/embedding/check", methods=["POST"])  # noqa: F821
 @login_required
 @add_tenant_id_to_kwargs

api/apps/services/dataset_api_service.py

@@ -873,46 +873,6 @@ def delete_index(dataset_id: str, tenant_id: str, index_type: str, wipe: bool =
     return True, {}
 
 
-def run_embedding(dataset_id: str, tenant_id: str):
-    """
-    Run embedding for all documents in a dataset.
-
-    :param dataset_id: dataset ID
-    :param tenant_id: tenant ID
-    :return: (success, result) or (success, error_message)
-    """
-    if not dataset_id:
-        return False, 'Lack of "Dataset ID"'
-
-    if not KnowledgebaseService.accessible(dataset_id, tenant_id):
-        return False, "No authorization."
-
-    ok, kb = KnowledgebaseService.get_by_id(dataset_id)
-    if not ok:
-        return False, "Invalid Dataset ID"
-
-    documents, _ = DocumentService.get_by_kb_id(
-        kb_id=dataset_id,
-        page_number=0,
-        items_per_page=0,
-        orderby="create_time",
-        desc=False,
-        keywords="",
-        run_status=[],
-        types=[],
-        suffix=[],
-    )
-    if not documents:
-        return False, f"No documents in Dataset {dataset_id}"
-
-    kb_table_num_map = {}
-    for doc in documents:
-        doc["tenant_id"] = tenant_id
-        DocumentService.run(tenant_id, doc, kb_table_num_map)
-
-    return True, {"scheduled_count": len(documents)}
-
-
 def rename_tag(dataset_id: str, tenant_id: str, from_tag: str, to_tag: str):
     """
     Rename a tag in a dataset.