pq-web-ai/app/services/vl_service.py

from ..schemas.base import DataResponse, UnderstandImageRequest, BusinessLicenseExtractRequest
from ..providers.understand_image_provider import UnderstandImageProvider

class VLService:
    def __init__(self, provider: UnderstandImageProvider) -> None:
        self._provider = provider

    def understand_image(self, req: UnderstandImageRequest) -> DataResponse:
        return self._provider.understand_image(
            req.image_url,
            model = req.model
        )

    def extract_business_license(self, req: BusinessLicenseExtractRequest) -> DataResponse:
        return self._provider.extract_business_license(
            req.image_url,
            model=req.model,
        )