mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-07-26 10:23:28 +08:00
Go: add context to lots of interface (#17253)
Signed-off-by: Jin Hai <haijin.chn@gmail.com>
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
package parser
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"ragflow/internal/common"
|
||||
"strings"
|
||||
@@ -8,7 +9,7 @@ import (
|
||||
models "ragflow/internal/entity/models"
|
||||
)
|
||||
|
||||
func parsePDFWithPaddleOCR(filename string, data []byte, parser *PDFParser) ParseResult {
|
||||
func parsePDFWithPaddleOCR(ctx context.Context, filename string, data []byte, parser *PDFParser) ParseResult {
|
||||
if len(data) == 0 {
|
||||
return emptyPDFResult(filename)
|
||||
}
|
||||
@@ -45,7 +46,7 @@ func parsePDFWithPaddleOCR(filename string, data []byte, parser *PDFParser) Pars
|
||||
apiConfig.ApiKey = &apiKey
|
||||
}
|
||||
|
||||
resp, err := driver.OCRFile(&algorithm, data, &filename, apiConfig, &models.OCRConfig{
|
||||
resp, err := driver.OCRFile(ctx, &algorithm, data, &filename, apiConfig, &models.OCRConfig{
|
||||
Algorithm: algorithm,
|
||||
}, nil)
|
||||
if err != nil {
|
||||
@@ -58,5 +59,5 @@ func parsePDFWithPaddleOCR(filename string, data []byte, parser *PDFParser) Pars
|
||||
if resp.Text != nil && strings.TrimSpace(*resp.Text) == "" {
|
||||
pageCount = 0
|
||||
}
|
||||
return parseMinerUMarkdownResult(filename, *resp.Text, parser.OutputFormat, pageCount)
|
||||
return parseMinerUMarkdownResult(ctx, filename, *resp.Text, parser.OutputFormat, pageCount)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user