parameter_entities.py 898 B

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. from enum import StrEnum
  2. class CommonParameterType(StrEnum):
  3. SECRET_INPUT = "secret-input"
  4. TEXT_INPUT = "text-input"
  5. SELECT = "select"
  6. STRING = "string"
  7. NUMBER = "number"
  8. FILE = "file"
  9. FILES = "files"
  10. SYSTEM_FILES = "system-files"
  11. BOOLEAN = "boolean"
  12. APP_SELECTOR = "app-selector"
  13. MODEL_SELECTOR = "model-selector"
  14. TOOLS_SELECTOR = "array[tools]"
  15. # TOOL_SELECTOR = "tool-selector"
  16. class AppSelectorScope(StrEnum):
  17. ALL = "all"
  18. CHAT = "chat"
  19. WORKFLOW = "workflow"
  20. COMPLETION = "completion"
  21. class ModelSelectorScope(StrEnum):
  22. LLM = "llm"
  23. TEXT_EMBEDDING = "text-embedding"
  24. RERANK = "rerank"
  25. TTS = "tts"
  26. SPEECH2TEXT = "speech2text"
  27. MODERATION = "moderation"
  28. VISION = "vision"
  29. class ToolSelectorScope(StrEnum):
  30. ALL = "all"
  31. CUSTOM = "custom"
  32. BUILTIN = "builtin"
  33. WORKFLOW = "workflow"