|
@@ -26,24 +26,21 @@ func NewGetAgentDataListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *
|
|
|
|
|
|
func (l *GetAgentDataListLogic) GetAgentDataList(req *types.DataListReq) (*types.DataListResp, error) {
|
|
|
dataResp := types.DataListResp{}
|
|
|
- dataResp.PageNum = req.PageNum
|
|
|
- dataResp.PageSize = req.PageSize
|
|
|
- var total int
|
|
|
|
|
|
var params fastgpt.GetDataListReq
|
|
|
params.CollectionId = *req.CollectionId
|
|
|
params.PageSize = *req.PageSize
|
|
|
- params.PageNum = *req.PageNum
|
|
|
+ params.PageNum = *req.Page
|
|
|
resp, err := fastgpt.GetDataList(¶ms)
|
|
|
if err != nil {
|
|
|
return nil, errorx.NewInvalidArgumentError("fastgpt get data list failed " + err.Error())
|
|
|
}
|
|
|
|
|
|
- dataResp.Data = make([]types.DataSimpleInfo, 0, 1)
|
|
|
- dataResp.Total = &total
|
|
|
+ dataResp.Data = types.BatchSimpleInfo{}
|
|
|
+ dataResp.Data.Total = uint64(resp.Data.Total)
|
|
|
if resp.Code == 200 && resp.Data.Total > 0 {
|
|
|
for _, val := range resp.Data.Data {
|
|
|
- dataResp.Data = append(dataResp.Data, types.DataSimpleInfo{
|
|
|
+ dataResp.Data.Data = append(dataResp.Data.Data, types.DataSimpleInfo{
|
|
|
ID: &val.ID,
|
|
|
Q: &val.Q,
|
|
|
A: &val.A,
|
|
@@ -52,7 +49,6 @@ func (l *GetAgentDataListLogic) GetAgentDataList(req *types.DataListReq) (*types
|
|
|
CollectionId: &val.CollectionID,
|
|
|
})
|
|
|
}
|
|
|
- dataResp.Total = &resp.Data.Total
|
|
|
}
|
|
|
|
|
|
return &dataResp, nil
|