From 959de144ccdaab1ac2317d78b75249892440cb18 Mon Sep 17 00:00:00 2001 From: Hussein Ahmad Date: Wed, 22 Jan 2025 21:22:07 +0000 Subject: [PATCH] Update doc_utils.py Fix DocumentContentFormat DI --- code/doc_utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/code/doc_utils.py b/code/doc_utils.py index b02d2a1..559a535 100644 --- a/code/doc_utils.py +++ b/code/doc_utils.py @@ -47,7 +47,7 @@ from azure.core.credentials import AzureKeyCredential from azure.ai.documentintelligence import DocumentIntelligenceClient from azure.ai.documentintelligence.models import AnalyzeResult -from azure.ai.documentintelligence.models import ContentFormat +from azure.ai.documentintelligence.models import DocumentContentFormat load_dotenv() @@ -953,7 +953,7 @@ def extract_doc_using_doc_int(ingestion_pipeline_dict): with open(doc_path, "rb") as f: poller = document_intelligence_client.begin_analyze_document( - "prebuilt-layout", analyze_request=f, output_content_format=ContentFormat.MARKDOWN, content_type="application/octet-stream" + "prebuilt-layout", analyze_request=f, output_content_format=DocumentContentFormat.MARKDOWN, content_type="application/octet-stream" ) result: AnalyzeResult = poller.result() @@ -2957,4 +2957,4 @@ def search(query, learnings = None, top=7, approx_tag_limit=15, conversation_his print("Final Answer", final_answer) - return final_answer, references, output_excel, search_results, files \ No newline at end of file + return final_answer, references, output_excel, search_results, files