Explorar o código

遮蔽系统提示词

boweniac hai 4 meses
pai
achega
f03467b1a5
Modificáronse 1 ficheiros con 7 adicións e 1 borrados
  1. 7 1
      internal/logic/UsageDetail/get_usage_detail_list_logic.go

+ 7 - 1
internal/logic/UsageDetail/get_usage_detail_list_logic.go

@@ -48,6 +48,12 @@ func (l *GetUsageDetailListLogic) GetUsageDetailList(req *types.UsageDetailListR
 	resp.Data.Total = data.PageDetails.Total
 	resp.Data.Total = data.PageDetails.Total
 
 
 	for _, v := range data.List {
 	for _, v := range data.List {
+		request := v.Request
+		if v.App == 2 {
+			request = "问题优化"
+		} else if v.App == 4 {
+			request = "节点判断"
+		}
 		resp.Data.Data = append(resp.Data.Data,
 		resp.Data.Data = append(resp.Data.Data,
 			types.UsageDetailInfo{
 			types.UsageDetailInfo{
 				BaseIDInfo: types.BaseIDInfo{
 				BaseIDInfo: types.BaseIDInfo{
@@ -60,7 +66,7 @@ func (l *GetUsageDetailListLogic) GetUsageDetailList(req *types.UsageDetailListR
 				BotId:            &v.BotID,
 				BotId:            &v.BotID,
 				App:              &v.App,
 				App:              &v.App,
 				SessionId:        &v.SessionID,
 				SessionId:        &v.SessionID,
-				Request:          &v.Request,
+				Request:          &request,
 				Response:         &v.Response,
 				Response:         &v.Response,
 				TotalTokens:      &v.TotalTokens,
 				TotalTokens:      &v.TotalTokens,
 				PromptTokens:     &v.PromptTokens,
 				PromptTokens:     &v.PromptTokens,