From c18f47c5c032c49bf3175aecd2236df37c0e9ae1 Mon Sep 17 00:00:00 2001 From: Michele Dolfi <97102151+dolfim-ibm@users.noreply.github.com> Date: Fri, 7 Feb 2025 17:51:31 +0100 Subject: [PATCH] fix: remove unused httpx (#919) * remove unused httpx Signed-off-by: Michele Dolfi * use requests instead of httpx Signed-off-by: Michele Dolfi * remove more usage of httpx Signed-off-by: Michele Dolfi --------- Signed-off-by: Michele Dolfi --- docling/models/easyocr_model.py | 2 -- docling/models/picture_description_api_model.py | 10 +++------- 2 files changed, 3 insertions(+), 9 deletions(-) diff --git a/docling/models/easyocr_model.py b/docling/models/easyocr_model.py index 9b1b2a0..0eccb98 100644 --- a/docling/models/easyocr_model.py +++ b/docling/models/easyocr_model.py @@ -4,9 +4,7 @@ import zipfile from pathlib import Path from typing import Iterable, List, Optional -import httpx import numpy -import torch from docling_core.types.doc import BoundingBox, CoordOrigin from docling.datamodel.base_models import Cell, OcrCell, Page diff --git a/docling/models/picture_description_api_model.py b/docling/models/picture_description_api_model.py index 6c7e02f..86b7694 100644 --- a/docling/models/picture_description_api_model.py +++ b/docling/models/picture_description_api_model.py @@ -3,11 +3,7 @@ import io import logging from typing import Iterable, List, Optional -import httpx -from docling_core.types.doc import PictureItem -from docling_core.types.doc.document import ( # TODO: move import to docling_core.types.doc - PictureDescriptionData, -) +import requests from PIL import Image from pydantic import BaseModel, ConfigDict @@ -90,13 +86,13 @@ class PictureDescriptionApiModel(PictureDescriptionBaseModel): **self.options.params, } - r = httpx.post( + r = requests.post( str(self.options.url), headers=self.options.headers, json=payload, timeout=self.options.timeout, ) - if not r.is_success: + if not r.ok: _log.error(f"Error calling the API. Reponse was {r.text}") r.raise_for_status()