Emil_Shanaty/open-webui-ai4me
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #16826 from selenecodes/feat/azure-document-intelligence-azure-entra-auth

Browse Source 
feat: Authenticate Azure Document Intelligence using DefaultAzureCredential
This commit is contained in:
Tim Jaeryang Baek
2025-08-26 14:32:04 +04:00
committed by GitHub
parent e3207f35d7 5051bfe7ab
commit 4267e22d4a
2 changed files with 16 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/retrieval/loaders/main.py
+13 -6
View File
@@ -4,6 +4,7 @@ import ftfy
import sys import sys
import json import json
from azure.identity import DefaultAzureCredential
from langchain_community.document_loaders import ( from langchain_community.document_loaders import (
AzureAIDocumentIntelligenceLoader, AzureAIDocumentIntelligenceLoader,
BSHTMLLoader, BSHTMLLoader,
@@ -327,7 +328,6 @@ class Loader:
elif ( elif (
self.engine == "document_intelligence" self.engine == "document_intelligence"
and self.kwargs.get("DOCUMENT_INTELLIGENCE_ENDPOINT") != "" and self.kwargs.get("DOCUMENT_INTELLIGENCE_ENDPOINT") != ""
and self.kwargs.get("DOCUMENT_INTELLIGENCE_KEY") != ""
and ( and (
file_ext in ["pdf", "xls", "xlsx", "docx", "ppt", "pptx"] file_ext in ["pdf", "xls", "xlsx", "docx", "ppt", "pptx"]
or file_content_type or file_content_type
@@ -340,11 +340,18 @@ class Loader:
] ]
) )
): ):
loader = AzureAIDocumentIntelligenceLoader( if self.kwargs.get("DOCUMENT_INTELLIGENCE_KEY") != "":
file_path=file_path, loader = AzureAIDocumentIntelligenceLoader(
api_endpoint=self.kwargs.get("DOCUMENT_INTELLIGENCE_ENDPOINT"), file_path=file_path,
api_key=self.kwargs.get("DOCUMENT_INTELLIGENCE_KEY"), api_endpoint=self.kwargs.get("DOCUMENT_INTELLIGENCE_ENDPOINT"),
) api_key=self.kwargs.get("DOCUMENT_INTELLIGENCE_KEY"),
)
else:
loader = AzureAIDocumentIntelligenceLoader(
file_path=file_path,
api_endpoint=self.kwargs.get("DOCUMENT_INTELLIGENCE_ENDPOINT"),
azure_credential=DefaultAzureCredential(),
)
elif ( elif (
self.engine == "mistral_ocr" self.engine == "mistral_ocr"
and self.kwargs.get("MISTRAL_OCR_API_KEY") != "" and self.kwargs.get("MISTRAL_OCR_API_KEY") != ""
src/lib/components/admin/Settings/Documents.svelte
+3 -3
View File
@@ -185,10 +185,9 @@
if ( if (
RAGConfig.CONTENT_EXTRACTION_ENGINE === 'document_intelligence' && RAGConfig.CONTENT_EXTRACTION_ENGINE === 'document_intelligence' &&
(RAGConfig.DOCUMENT_INTELLIGENCE_ENDPOINT === '' || RAGConfig.DOCUMENT_INTELLIGENCE_ENDPOINT === ''
RAGConfig.DOCUMENT_INTELLIGENCE_KEY === '')
) { ) {
toast.error($i18n.t('Document Intelligence endpoint and key required.')); toast.error($i18n.t('Document Intelligence endpoint required.'));
return; return;
} }
if ( if (
@@ -644,6 +643,7 @@
<SensitiveInput <SensitiveInput
placeholder={$i18n.t('Enter Document Intelligence Key')} placeholder={$i18n.t('Enter Document Intelligence Key')}
bind:value={RAGConfig.DOCUMENT_INTELLIGENCE_KEY} bind:value={RAGConfig.DOCUMENT_INTELLIGENCE_KEY}
required={false}
/> />
</div> </div>
{:else if RAGConfig.CONTENT_EXTRACTION_ENGINE === 'mistral_ocr'} {:else if RAGConfig.CONTENT_EXTRACTION_ENGINE === 'mistral_ocr'}