delete_agent_base_logic.go 1013 B

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. package agent_base
  2. import (
  3. "context"
  4. "github.com/suyuan32/simple-admin-common/msg/errormsg"
  5. "github.com/zeromicro/go-zero/core/errorx"
  6. "wechat-api/hook/fastgpt"
  7. "wechat-api/internal/svc"
  8. "wechat-api/internal/types"
  9. "github.com/zeromicro/go-zero/core/logx"
  10. )
  11. type DeleteAgentBaseLogic struct {
  12. logx.Logger
  13. ctx context.Context
  14. svcCtx *svc.ServiceContext
  15. }
  16. func NewDeleteAgentBaseLogic(ctx context.Context, svcCtx *svc.ServiceContext) *DeleteAgentBaseLogic {
  17. return &DeleteAgentBaseLogic{
  18. Logger: logx.WithContext(ctx),
  19. ctx: ctx,
  20. svcCtx: svcCtx}
  21. }
  22. func (l *DeleteAgentBaseLogic) DeleteAgentBase(req *types.DeleteDataReq) (*types.BaseDataInfo, error) {
  23. resp, err := fastgpt.DeleteData(*req.ID)
  24. if err != nil {
  25. return nil, errorx.NewInvalidArgumentError("fastgpt create data failed " + err.Error())
  26. }
  27. if resp.Code == 200 {
  28. return &types.BaseDataInfo{
  29. Code: 0,
  30. Msg: errormsg.Success,
  31. Data: "",
  32. }, nil
  33. }
  34. return nil, errorx.NewInvalidArgumentError(resp.StatusText)
  35. }