dify/api/services/rag_pipeline/rag_pipeline.py
2025-04-10 18:00:22 +08:00

21 lines
599 B
Python

import datetime
import hashlib
import os
import uuid
from typing import Any, List, Literal, Union
from flask_login import current_user
from models.dataset import PipelineBuiltInTemplate, PipelineCustomizedTemplate # type: ignore
class RagPipelineService:
@staticmethod
def get_pipeline_templates(
type: Literal["built-in", "customized"] = "built-in",
) -> list[PipelineBuiltInTemplate | PipelineCustomizedTemplate]:
if type == "built-in":
return PipelineBuiltInTemplate.query.all()
else:
return PipelineCustomizedTemplate.query.all()