zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-08-14 20:54:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Go: fix env variables (#18032)

Browse Source 
Signed-off-by: Jin Hai <haijin.chn@gmail.com>
This commit is contained in:
Jin Hai
2026-08-10 14:37:05 +08:00
committed by GitHub
parent c0582b8e18
commit c0bc146fcb
24 changed files with 774 additions and 95 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
internal/parser/parser/pdf_parser_docling.go
View File

@@ -49,7 +49,7 @@ func parsePDFWithDocling(ctx context.Context, filename string, data []byte, pars
}
apiKey := strings.TrimSpace(parser.DoclingAPIKey)
if apiKey == "" {
apiKey = strings.TrimSpace(common.GetEnv(common.EnvDoclingApiKey))
apiKey = strings.TrimSpace(common.GetEnv(common.EnvDoclingAPIKey))
}
baseURL := strings.TrimRight(serverURL, "/")

4
internal/parser/parser/pdf_parser_mineru.go
View File

@@ -19,14 +19,14 @@ func parsePDFWithMinerU(ctx context.Context, filename string, data []byte, parse
}
apiServer := strings.TrimSpace(parser.MinerUAPIServer)
if apiServer == "" {
apiServer = strings.TrimSpace(common.GetEnv(common.EnvMineruApiServer))
apiServer = strings.TrimSpace(common.GetEnv(common.EnvMineruAPIServer))
}
if apiServer == "" {
return ParseResult{Err: fmt.Errorf("parser: MinerU requires mineru_apiserver or MINERU_APISERVER")}
}
apiKey := parser.MinerUAPIKey
if strings.TrimSpace(apiKey) == "" {
apiKey = strings.TrimSpace(common.GetEnv(common.EnvMineruApiKey))
apiKey = strings.TrimSpace(common.GetEnv(common.EnvMineruAPIKey))
}
backend := strings.TrimSpace(parser.MinerUBackend)
if backend == "" {

4
internal/parser/parser/pdf_parser_opendataloader.go
View File

@@ -19,14 +19,14 @@ func parsePDFWithOpenDataLoader(ctx context.Context, filename string, data []byt
}
baseURL := strings.TrimSpace(parser.OpenDataLoaderAPIServer)
if baseURL == "" {
baseURL = strings.TrimSpace(common.GetEnv(common.EnvOpenDataLoaderApiServer))
baseURL = strings.TrimSpace(common.GetEnv(common.EnvOpenDataLoaderAPIServer))
}
if baseURL == "" {
return ParseResult{Err: fmt.Errorf("parser: OpenDataLoader requires opendataloader_apiserver or OPENDATALOADER_APISERVER")}
}
apiKey := strings.TrimSpace(parser.OpenDataLoaderAPIKey)
if apiKey == "" {
apiKey = strings.TrimSpace(common.GetEnv(common.EnvOpenDataLoaderApiKey))
apiKey = strings.TrimSpace(common.GetEnv(common.EnvOpenDataLoaderAPIKey))
}
bodyReader, contentType, err := openDataLoaderMultipart(filename, data, parser)

2
internal/parser/parser/pdf_parser_paddleocr.go
View File

@@ -18,7 +18,7 @@ func parsePDFWithPaddleOCR(ctx context.Context, filename string, data []byte, pa
baseURL = strings.TrimSpace(common.GetEnv(common.EnvPaddleOCRBaseUrl))
}
if baseURL == "" {
baseURL = strings.TrimSpace(common.GetEnv(common.EnvPaddleOCRApiURL))
baseURL = strings.TrimSpace(common.GetEnv(common.EnvPaddleOCRAPIURL))
}
if baseURL == "" {
return ParseResult{Err: fmt.Errorf("parser: PaddleOCR requires paddleocr_base_url or PADDLEOCR_BASE_URL")}

2
internal/parser/parser/pdf_parser_somark.go
View File

@@ -28,7 +28,7 @@ func parsePDFWithSoMark(filename string, data []byte, parser *PDFParser) ParseRe
}
apiKey := parser.SoMarkAPIKey
if strings.TrimSpace(apiKey) == "" {
apiKey = strings.TrimSpace(common.GetEnv(common.EnvSOMarkApiKey))
apiKey = strings.TrimSpace(common.GetEnv(common.EnvSOMarkAPIKey))
}
taskID, err := soMarkSubmit(strings.TrimRight(baseURL, "/"), filename, data, parser, apiKey)
if err != nil {

4
internal/parser/parser/pdf_parser_tcadp.go
View File

@@ -21,14 +21,14 @@ func parsePDFWithTCADP(filename string, data []byte, parser *PDFParser) ParseRes
}
baseURL := strings.TrimSpace(parser.TCADPAPIServer)
if baseURL == "" {
baseURL = strings.TrimSpace(common.GetEnv(common.EnvTCADPApiServerURL))
baseURL = strings.TrimSpace(common.GetEnv(common.EnvTCADPAPIServerURL))
}
if baseURL == "" {
return ParseResult{Err: fmt.Errorf("parser: TCADP requires tcadp_apiserver or TCADP_APISERVER")}
}
apiKey := strings.TrimSpace(parser.TCADPAPIKey)
if apiKey == "" {
apiKey = strings.TrimSpace(common.GetEnv(common.EnvTCADPApiKey))
apiKey = strings.TrimSpace(common.GetEnv(common.EnvTCADPAPIKey))
}
requestBody := map[string]any{
"file_type": "PDF",