fix:edit NewResty

hook/dify/dify.go

@@ -37,9 +37,9 @@ type History struct {
 }
 
 // GetSessionList 获取会话列表
-func GetSessionList(user, lastId, limit string) (*SessionResponse, error) {
+func GetSessionList(baseUrl, token, user, lastId, limit string) (*SessionResponse, error) {
 	data := &SessionResponse{}
-	resp, err := NewResty().
+	resp, err := NewDifyResty(baseUrl, token).
 		R().
 		SetResult(&data).
 		SetQueryParam("user", user).
@@ -59,9 +59,9 @@ func GetSessionList(user, lastId, limit string) (*SessionResponse, error) {
 }
 
 // GetChatHistory 获取历史会话
-func GetChatHistory(user, conversationId, firstId, limit string) (*HistoryResponse, error) {
+func GetChatHistory(baseUrl, token, user, conversationId, firstId, limit string) (*HistoryResponse, error) {
 	data := &HistoryResponse{}
-	resp, err := NewResty().
+	resp, err := NewDifyResty(baseUrl, token).
 		R().
 		SetResult(&data).
 		SetQueryParam("user", user).
@@ -82,12 +82,12 @@ func GetChatHistory(user, conversationId, firstId, limit string) (*HistoryRespon
 }
 
 // NewResty 实例化dify实例
-func NewResty() *resty.Client {
+func NewDifyResty(baseUrl, token string) *resty.Client {
 	client := resty.New()
 	client.SetRetryCount(2).
 		SetHeader("Content-Type", "application/json").
-		SetBaseURL("https://dify.gkscrm.com/v1").
-		SetHeader("Authorization", "Bearer "+GetToken())
+		SetBaseURL(baseUrl).
+		SetHeader("Authorization", "Bearer "+token)
 
 	return client
 }

internal/logic/chatrecords/gpts_get_api_message_logic.go

@@ -7,6 +7,7 @@ import (
 	"github.com/suyuan32/simple-admin-core/rpc/types/core"
 	"github.com/zeromicro/go-zero/core/errorx"
 	"strconv"
+	"wechat-api/ent/employee"
 	"wechat-api/hook/dify"
 	"wechat-api/internal/utils"
 	jwtutils "wechat-api/internal/utils/jwt"
@@ -57,13 +58,18 @@ func (l *GptsGetApiMessageLogic) GptsGetApiMessage(req *types.GptMessageReq, tok
 		limit = strconv.Itoa(*req.Limit)
 	}
 
+	employeeItem, err := l.svcCtx.DB.Employee.Query().Where(employee.ID(*req.AgentId)).Only(l.ctx)
+	if err != nil {
+		return nil, errorx.NewInvalidArgumentError("智能体不存在")
+	}
+
 	user := fmt.Sprintf("%s-%d", userId, *req.AgentId)
-	fmt.Printf("user=%v ConversationId=%v firstId=%v, limit=%v\n", user, *req.ConversationId, firstId, limit)
-	response, err := dify.GetChatHistory(user, *req.ConversationId, firstId, limit)
+	//fmt.Printf("user=%v ConversationId=%v firstId=%v, limit=%v\n", user, *req.ConversationId, firstId, limit)
+	response, err := dify.GetChatHistory(employeeItem.APIBase, employeeItem.APIKey, user, *req.ConversationId, firstId, limit)
 	if err != nil {
 		return nil, err
 	}
-	fmt.Printf("response=%v\n", response)
+	//fmt.Printf("response=%v\n", response)
 
 	resp.Data.HasMore = response.HasMore
 	for _, v := range response.Data {

internal/logic/chatrecords/gpts_get_api_session_logic.go

@@ -7,6 +7,7 @@ import (
 	"github.com/suyuan32/simple-admin-core/rpc/types/core"
 	"github.com/zeromicro/go-zero/core/errorx"
 	"strconv"
+	"wechat-api/ent/employee"
 	"wechat-api/hook/dify"
 	"wechat-api/internal/utils"
 	jwtutils "wechat-api/internal/utils/jwt"
@@ -55,13 +56,18 @@ func (l *GptsGetApiSessionLogic) GptsGetApiSession(req *types.GptsSessionReq, to
 		lastId = *req.LastId
 	}
 
+	employeeItem, err := l.svcCtx.DB.Employee.Query().Where(employee.ID(*req.AgentId)).Only(l.ctx)
+	if err != nil {
+		return nil, errorx.NewInvalidArgumentError("智能体不存在")
+	}
+
 	user := fmt.Sprintf("%s-%d", userId, *req.AgentId)
-	fmt.Printf("user=%v lastId=%v, limit=%v\n", user, lastId, limit)
-	response, err := dify.GetSessionList(user, lastId, limit)
+	//fmt.Printf("user=%v lastId=%v, limit=%v\n", user, lastId, limit)
+	response, err := dify.GetSessionList(employeeItem.APIBase, employeeItem.APIKey, user, lastId, limit)
 	if err != nil {
 		return nil, err
 	}
-	fmt.Printf("response=%v\n", response)
+	//fmt.Printf("response=%v\n", response)
 
 	resp.Data.HasMore = response.HasMore
 	for _, v := range response.Data {