Back to Llama Index

Setup Box Client

llama-index-integrations/tools/llama-index-tools-box/examples/box_ai_extract.ipynb

0.14.21887 B
Original Source
python
# Setup Box Client
from box_sdk_gen import DeveloperTokenConfig, BoxDeveloperTokenAuth, BoxClient

BOX_DEV_TOKEN = "your_box_dev_token"

config = DeveloperTokenConfig(BOX_DEV_TOKEN)
auth = BoxDeveloperTokenAuth(config)
box_client = BoxClient(auth)
python
import os

os.environ["OPENAI_API_KEY"] = "sk-proj-1234567890"
python
from llama_index.core.agent.workflow import FunctionAgent
from llama_index.llms.openai import OpenAI
python
from llama_index.tools.box import BoxAIExtractToolSpec


document_id = "test_txt_invoice_id"
ai_prompt = (
    '{"doc_type","date","total","vendor","invoice_number","purchase_order_number"}'
)

box_tool = BoxAIExtractToolSpec(box_client=box_client)

agent = FunctionAgent(
    tools=box_tool.to_tool_list(),
    llm=OpenAI(model="gpt-4.1"),
)

answer = await agent.run(f"{ai_prompt} for {document_id}")
print(answer)