From 073808eb3016c55b747463a890c10edb1439719b Mon Sep 17 00:00:00 2001
From: yoge <yogecoder@gmail.com>
Date: Sat, 20 Dec 2025 22:57:14 +0800
Subject: [PATCH] feat: add mml

---
 internal/service/task.go | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/internal/service/task.go b/internal/service/task.go
index b42bddc..bc5f979 100644
--- a/internal/service/task.go
+++ b/internal/service/task.go
@@ -3,7 +3,6 @@ package service
 import (
 	"context"
 	"errors"
-	"fmt"
 	"strings"
 
 	"gitea.com/bitwsd/document_ai/internal/model/task"
@@ -87,13 +86,12 @@ func (svc *TaskService) GetTaskList(ctx context.Context, req *task.TaskListReque
 	for _, item := range tasks {
 		var latex string
 		var markdown string
+		var mathML string
 		recognitionResult := recognitionResultMap[item.ID]
 		if recognitionResult != nil {
 			latex = recognitionResult.Latex
 			markdown = recognitionResult.Markdown
-			if markdown == "" {
-				markdown = fmt.Sprintf("$$%s$$", latex)
-			}
+			mathML = recognitionResult.MathML
 		}
 		originURL, err := oss.GetDownloadURL(ctx, item.FileURL)
 		if err != nil {
@@ -102,6 +100,7 @@ func (svc *TaskService) GetTaskList(ctx context.Context, req *task.TaskListReque
 		resp.TaskList = append(resp.TaskList, &task.TaskListDTO{
 			Latex:     latex,
 			Markdown:  markdown,
+			MathML:    mathML,
 			TaskID:    item.TaskUUID,
 			FileName:  item.FileName,
 			Status:    int(item.Status),