From 8820636a3521f1dc09971290b2556d5a8eb530ae Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dominik=20Schr=C3=B6ter?= Date: Mon, 10 Feb 2025 15:29:01 +0100 Subject: [PATCH] feat(ocr): add conditional initialization for OCR provider based on LLM settings --- .github/workflows/docker-build-and-push.yml | 1 + main.go | 11 ++++++++--- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/.github/workflows/docker-build-and-push.yml b/.github/workflows/docker-build-and-push.yml index 0dcdec9..be51587 100644 --- a/.github/workflows/docker-build-and-push.yml +++ b/.github/workflows/docker-build-and-push.yml @@ -241,6 +241,7 @@ jobs: run: npm run test:e2e env: CI: true + DEBUG: testcontainers:containers OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }} PAPERLESS_GPT_IMAGE: ${{ env.PAPERLESS_GPT_IMAGE }} - name: Upload Playwright Report diff --git a/main.go b/main.go index 75f85cb..68aefea 100644 --- a/main.go +++ b/main.go @@ -168,9 +168,14 @@ func main() { VisionLLMModel: visionLlmModel, } - ocrProvider, err = ocr.NewProvider(ocrConfig) - if err != nil { - log.Fatalf("Failed to initialize OCR provider: %v", err) + // If provider is LLM, but no VISION_LLM_PROVIDER is set, don't initialize OCR provider + if providerType == "llm" && visionLlmProvider == "" { + log.Warn("OCR provider is set to LLM, but no VISION_LLM_PROVIDER is set. Disabling OCR.") + } else { + ocrProvider, err = ocr.NewProvider(ocrConfig) + if err != nil { + log.Fatalf("Failed to initialize OCR provider: %v", err) + } } // Initialize App with dependencies