mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-07-07 21:31:48 +08:00
21 lines
599 B
Python
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()
|