AI를 사용하여 구조적 출력을 위한 가장 빠르고 정확한 송장 데이터 추출기를 만듭니다.-파이썬 튜토리얼-php.cn

AI를 사용하여 구조적 출력을 위한 가장 빠르고 정확한 송장 데이터 추출기를 만듭니다.

WBOY

풀어 주다： 2024-08-07 06:36:13

원래의

794명이 탐색했습니다.

Create the fastest and precise invoice data extractor for structural output using AI

매장 영수증 추출을 위해 Pydantic 모델과 함께 LlamaExtract 사용

이 기사에서는 상점 영수증에서 구조화된 데이터를 추출하기 위해 Pydantic 모델의 스키마와 통합된 LlamaExtract를 사용하는 방법을 살펴보겠습니다. 이러한 접근 방식을 통해 영수증 정보를 체계적으로 정리하여 분석 및 관리가 더욱 쉬워집니다.

설정

먼저 라마 추출 클라이언트 라이브러리가 설치되어 있는지 확인하세요. 다음 명령을 사용하십시오:

pip install llama-extract pydantic

로그인 후 복사

참고: pip 업데이트에 대한 알림이 표시되면 제공된 명령을 사용하여 업데이트할 수 있습니다.

먼저 로그인하고 Llama Index Cloud에서 무료로 API 키를 받으세요

LlamaExtract API 키에 대한 환경 변수를 설정합니다.

import os

os.environ["LLAMA_CLOUD_API_KEY"] = "YOUR LLAMA INDEX CLOUD API HERE"

로그인 후 복사

데이터 로드

이 예에서는 PDF 형식의 매장 영수증 데이터세트가 있다고 가정해 보겠습니다. 이러한 파일을 영수증이라는 디렉토리에 넣으세요.

DATA_DIR = "data/receipts"
fnames = os.listdir(DATA_DIR)
fnames = [fname for fname in fnames if fname.endswith(".pdf")]
fpaths = [os.path.join(DATA_DIR, fname) for fname in fnames]
fpaths

로그인 후 복사

출력에는 영수증의 파일 경로가 나열되어야 합니다.

['data/receipts/receipt.pdf']

로그인 후 복사

Pydantic 모델 정의

우리는 Pydantic을 사용하여 데이터 모델을 정의할 것입니다. 이는 우리가 예상하거나 PDF에서 추출하려는 필드/데이터를 API에 알려줍니다. 매장 영수증의 경우 매장명, 날짜, 총액, 구매한 품목 목록을 추출하는 데 관심이 있을 수 있습니다.

from pydantic import BaseModel
from typing import List

class Item(BaseModel):
    name: str
    quantity: int
    price: float

class Receipt(BaseModel):
    store_name: str
    date: str
    total_amount: float
    items: List[Item]

로그인 후 복사

스키마 생성

이제 Pydantic 모델을 사용하여 LlamaExtract에서 추출 스키마를 정의할 수 있습니다.

from llama_extract import LlamaExtract

extractor = LlamaExtract(verbose=True)
schema_response = await extractor.acreate_schema("Receipt Schema", data_schema=Receipt)
schema_response.data_schema

로그인 후 복사

출력 스키마는 다음과 유사해야 합니다.

{
    'type': 'object',
    '$defs': {
        'Item': {
            'type': 'object',
            'title': 'Item',
            'required': ['name', 'quantity', 'price'],
            'properties': {
                'name': {'type': 'string', 'title': 'Name'},
                'quantity': {'type': 'integer', 'title': 'Quantity'},
                'price': {'type': 'number', 'title': 'Price'}
            }
        }
    },
    'title': 'Receipt',
    'required': ['store_name', 'date', 'total_amount', 'items'],
    'properties': {
        'store_name': {'type': 'string', 'title': 'Store Name'},
        'date': {'type': 'string', 'title': 'Date'},
        'total_amount': {'type': 'number', 'title': 'Total Amount'},
        'items': {
            'type': 'array',
            'title': 'Items',
            'items': {'$ref': '#/$defs/Item'}
        }
    }
}

로그인 후 복사

추출 실행

스키마를 정의하면 이제 영수증 파일에서 구조화된 데이터를 추출할 수 있습니다. Receipt를 응답 모델로 지정하여 추출된 데이터가 검증되고 구조화되었는지 확인합니다.

responses = await extractor.aextract(
    schema_response.id, fpaths, response_model=Receipt
)

로그인 후 복사

필요한 경우 원시 JSON 출력에 액세스할 수 있습니다.

data = responses[0].data
print(data)

로그인 후 복사

JSON 출력 예:

{
    'store_name': 'ABC Electronics',
    'date': '2024-08-05',
    'total_amount': 123.45,
    'items': [
        {'name': 'Laptop', 'quantity': 1, 'price': 999.99},
        {'name': 'Mouse', 'quantity': 1, 'price': 25.00},
        {'name': 'Keyboard', 'quantity': 1, 'price': 50.00}
    ]
}

로그인 후 복사