feat: add mml

This commit is contained in:
2025-12-20 22:57:14 +08:00
parent 770c334083
commit 073808eb30

View File

@@ -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),