manage_plugin.go 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  1. package service
  2. import (
  3. "time"
  4. "github.com/langgenius/dify-plugin-daemon/internal/db"
  5. "github.com/langgenius/dify-plugin-daemon/internal/types/entities"
  6. "github.com/langgenius/dify-plugin-daemon/internal/types/entities/plugin_entities"
  7. "github.com/langgenius/dify-plugin-daemon/internal/types/models"
  8. "github.com/langgenius/dify-plugin-daemon/internal/utils/cache"
  9. )
  10. func ListPlugins(tenant_id string, page int, page_size int) *entities.Response {
  11. type installation struct {
  12. ID string `json:"id"`
  13. Name string `json:"name"`
  14. PluginID string `json:"plugin_id"`
  15. InstallationID string `json:"installation_id"`
  16. Description *plugin_entities.PluginDeclaration `json:"description"`
  17. RuntimeType plugin_entities.PluginRuntimeType `json:"runtime_type"`
  18. Version string `json:"version"`
  19. CreatedAt time.Time `json:"created_at"`
  20. UpdatedAt time.Time `json:"updated_at"`
  21. }
  22. plugin_installations, err := db.GetAll[models.PluginInstallation](
  23. db.Equal("tenant_id", tenant_id),
  24. db.Page(page, page_size),
  25. )
  26. if err != nil {
  27. return entities.NewErrorResponse(-500, err.Error())
  28. }
  29. data := make([]installation, 0, len(plugin_installations))
  30. for _, plugin_installation := range plugin_installations {
  31. plugin, err := cache.Get[models.Plugin](plugin_installation.PluginUniqueIdentifier)
  32. if err != nil {
  33. return entities.NewErrorResponse(-500, err.Error())
  34. }
  35. declaration, err := plugin.GetDeclaration()
  36. if err != nil {
  37. return entities.NewErrorResponse(-500, err.Error())
  38. }
  39. data = append(data, installation{
  40. ID: plugin_installation.ID,
  41. Name: declaration.Name,
  42. PluginID: plugin.ID,
  43. InstallationID: plugin_installation.ID,
  44. Description: declaration,
  45. RuntimeType: plugin_entities.PluginRuntimeType(plugin_installation.RuntimeType),
  46. Version: declaration.Version,
  47. CreatedAt: plugin_installation.CreatedAt,
  48. UpdatedAt: plugin_installation.UpdatedAt,
  49. })
  50. }
  51. return entities.NewSuccessResponse(data)
  52. }
  53. func ListTools(tenant_id string, page int, page_size int) *entities.Response {
  54. return nil
  55. }
  56. func ListModels(tenant_id string, page int, page_size int) *entities.Response {
  57. return nil
  58. }