Back to Llama Index

Setup Box Client

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

0.14.21795 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"] = "your-key"

from llama_index.core.agent.workflow import FunctionAgent
from llama_index.llms.openai import OpenAI
python
from llama_index.tools.box import BoxAIPromptToolSpec


document_id = "your_document_id"
ai_prompt = "Summarize the document"


box_tool = BoxAIPromptToolSpec(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)