extract • AI Agent SDK

extract: Extract text from a document or image.

extract

Method

</div>

This is a method of the OCRAgent class in the ocr_agent module.

Extract text from a document or image.

Signature

python
def extract(source: str, include_image_base64: Optional[bool], pages: Optional[List[int]], image_limit: Optional[int], model: Optional[str]) -> Any

Parameters

ParamField str

URL or path to document/image

ParamField Optional[bool]

Include base64 images in response

ParamField Optional

Specific pages to extract (for PDFs)

ParamField Optional[int]

Maximum images per page

ParamField Optional[str]

Override model for this call **kwargs: Additional parameters

Returns

ResponseField Any

OCRResponse with pages, markdown content, and metadata

Usage

python
agent = OCRAgent(llm="mistral/mistral-ocr-latest")
    result = agent.extract("https://arxiv.org/pdf/2201.04234")
    for page in result.pages:
        print(f"Page {page.index}: {page.markdown}")

Uses

  • ocr

Used By

Source

View on GitHub

praisonaiagents/agent/ocr_agent.py at line 168