jimmyyem před 2 dny
rodič
revize
84302c2107

+ 2 - 0
desc/wechat/agent.api

@@ -252,6 +252,8 @@ type (
 		PageSize  *int    `json:"pageSize" validate:"required,number,lt=100000"`
 
 		CollectionId  *string `json:"collectionId" validate:"required"`
+
+		AgentId  *uint64 `json:"agentId,optional"`
 	}
 
 	// Data list response | Data List信息返回体

+ 8 - 6
hook/fastgpt/data.go

@@ -2,6 +2,7 @@ package fastgpt
 
 import (
 	"errors"
+	"time"
 )
 
 type DataResp struct {
@@ -61,12 +62,13 @@ type DataInfo struct {
 		Text         string `json:"text"`
 		ID           string `json:"_id"`
 	} `json:"indexes"`
-	DatasetID    string `json:"datasetId"`
-	CollectionID string `json:"collectionId"`
-	SourceName   string `json:"sourceName"`
-	SourceID     string `json:"sourceId"`
-	IsOwner      bool   `json:"isOwner"`
-	CanWrite     bool   `json:"canWrite"`
+	DatasetID    string    `json:"datasetId"`
+	CollectionID string    `json:"collectionId"`
+	SourceName   string    `json:"sourceName"`
+	SourceID     string    `json:"sourceId"`
+	IsOwner      bool      `json:"isOwner"`
+	CanWrite     bool      `json:"canWrite"`
+	UpdateTime   time.Time `json:"updateTime"`
 }
 
 type DataListResp struct {

+ 6 - 2
internal/logic/agent_base/get_agent_base_list_logic.go

@@ -37,9 +37,13 @@ func (l *GetAgentBaseListLogic) GetAgentBaseList(req *types.DataListReq) (*types
 		return nil, errorx.NewInvalidArgumentError("fastgpt get data list failed " + err.Error())
 	}
 
-	resp2, err := fastgpt.GetCollectionDetail(*req.CollectionId)
+	agent, err := l.svcCtx.DB.Agent.Get(l.ctx, *req.AgentId)
+	if err != nil {
+		return nil, errorx.NewInvalidArgumentError("get agent failed " + err.Error())
+	}
+
+	resp2, err := fastgpt.GetDataDetail(agent.DatasetID)
 	fmt.Printf("resp2=%v  resp2.Data=%v resp2.Data.UpdateTime=%v\n", resp2, resp2.Data, resp2.Data.UpdateTime)
-	fmt.Println(resp2.Data.UpdateTime.UnixMilli(), resp2.Code)
 	if err != nil {
 		return nil, errorx.NewInvalidArgumentError("fastgpt get collection detail failed " + err.Error())
 	}

+ 1 - 0
internal/types/types.go

@@ -706,6 +706,7 @@ type DataListReq struct {
 	PageSize *int `json:"pageSize" validate:"required,number,lt=100000"`
 	// required : true
 	CollectionId *string `json:"collectionId" validate:"required"`
+	AgentId      *uint64 `json:"agentId,optional"`
 }
 
 // Data list response | Data List信息返回体