(StepCode string, Message string, Data map[string]interface{})
| 351 | } |
| 352 | |
| 353 | func (ps *ProcessSender) SendEndStep(StepCode string, Message string, Data map[string]interface{}) error { |
| 354 | // ⭐ 在发送前处理 data,截取 content |
| 355 | processedData := ps.truncateSourcesContent(Data) |
| 356 | |
| 357 | step := ProcessStep{ |
| 358 | StepCode: StepCode, |
| 359 | Status: STEP_STATUS_COMPLETED, |
| 360 | Message: Message, |
| 361 | Timestamp: time.Now().Unix(), |
| 362 | Data: processedData, // 使用处理后的数据 |
| 363 | } |
| 364 | if ps != nil && ps.chatRequest != nil && ps.chatRequest.EnableProcessSteps { |
| 365 | ps.recordProcessStep(step) |
| 366 | } |
| 367 | if !ps.CheckNeedSend() { |
| 368 | return nil |
| 369 | } |
| 370 | if err := sendProcessStep(ps.c, ps.requestId, step); err != nil { |
| 371 | logger.SysErrorf("发送处理步骤失败: %s", err.Error()) |
| 372 | return err |
| 373 | } |
| 374 | return nil |
| 375 | } |
| 376 | |
| 377 | // truncateSourcesContent 截取 data 中 sources 的 content |
| 378 | func (ps *ProcessSender) truncateSourcesContent(data map[string]interface{}) map[string]interface{} { |
no test coverage detected