invcation.go 1.6 KB

1234567891011121314151617181920212223242526272829303132333435
  1. package dify_invocation
  2. import (
  3. "github.com/langgenius/dify-plugin-daemon/internal/types/entities/model_entities"
  4. "github.com/langgenius/dify-plugin-daemon/internal/types/entities/tool_entities"
  5. "github.com/langgenius/dify-plugin-daemon/internal/utils/stream"
  6. )
  7. type BackwardsInvocation interface {
  8. // InvokeLLM
  9. InvokeLLM(payload *InvokeLLMRequest) (*stream.Stream[model_entities.LLMResultChunk], error)
  10. // InvokeTextEmbedding
  11. InvokeTextEmbedding(payload *InvokeTextEmbeddingRequest) (*model_entities.TextEmbeddingResult, error)
  12. // InvokeRerank
  13. InvokeRerank(payload *InvokeRerankRequest) (*model_entities.RerankResult, error)
  14. // InvokeTTS
  15. InvokeTTS(payload *InvokeTTSRequest) (*stream.Stream[model_entities.TTSResult], error)
  16. // InvokeSpeech2Text
  17. InvokeSpeech2Text(payload *InvokeSpeech2TextRequest) (*model_entities.Speech2TextResult, error)
  18. // InvokeModeration
  19. InvokeModeration(payload *InvokeModerationRequest) (*model_entities.ModerationResult, error)
  20. // InvokeTool
  21. InvokeTool(payload *InvokeToolRequest) (*stream.Stream[tool_entities.ToolResponseChunk], error)
  22. // InvokeApp
  23. InvokeApp(payload *InvokeAppRequest) (*stream.Stream[map[string]any], error)
  24. // InvokeParameterExtractor
  25. InvokeParameterExtractor(payload *InvokeParameterExtractorRequest) (*InvokeNodeResponse, error)
  26. // InvokeQuestionClassifier
  27. InvokeQuestionClassifier(payload *InvokeQuestionClassifierRequest) (*InvokeNodeResponse, error)
  28. // InvokeEncrypt
  29. InvokeEncrypt(payload *InvokeEncryptRequest) (map[string]any, error)
  30. // InvokeSummary
  31. InvokeSummary(payload *InvokeSummaryRequest) (*InvokeSummaryResponse, error)
  32. }