Bump version to v0.6.3 (#636)

* v0.6.0

* nit release

* nit

.github/ISSUE_TEMPLATE/bug_report.md

@@ -0,0 +1,31 @@
+---
+name: Bug report
+about: Create a report to help us improve
+title: ''
+labels: bug
+assignees: ''
+
+---
+
+**Describe the bug**
+Write a concise description of what the bug is.
+
+**Files**
+If possible, please provide the PDF file causing the issue.
+
+**Job ID**
+If you have it, please provide the ID of the job you ran.
+You can find it here: https://cloud.llamaindex.ai/parse in the "History" tab.
+
+**Client:**
+Please remove untested options:
+ - Python Library
+ - API
+ - Frontend (cloud.llamaindex.ai)
+ - Typescript Library
+ - Notebook
+
+**Additional context**
+Add any additional context about the problem here.
+What options did you use? Premium mode, multimodal, fast mode, parsing instructions, etc.
+Screenshots, code snippets, etc.

.github/ISSUE_TEMPLATE/custom-issue.md

@@ -0,0 +1,10 @@
+---
+name: Custom issue
+about: Not a bug nor a feature request
+title: ''
+labels: ''
+assignees: ''
+
+---
+
+

.github/ISSUE_TEMPLATE/feature_request.md

@@ -0,0 +1,10 @@
+---
+name: Feature request
+about: Suggest an idea for this project
+title: ''
+labels: enhancement
+assignees: ''
+
+---
+
+

.github/workflows/build_package.yml

@@ -45,4 +45,4 @@ jobs:
       - name: Test import
         shell: bash
         working-directory: ${{ vars.RUNNER_TEMP }}
-        run: python -c "import llama_parse"
+        run: python -c "import llama_cloud_services"

.github/workflows/publish_release.yml

@@ -14,7 +14,7 @@ env:
 jobs:
   build-n-publish:
     name: Build and publish to PyPI
-    if: github.repository == 'run-llama/llama_parse'
+    if: github.repository == 'run-llama/llama_cloud_services'
     runs-on: ubuntu-latest
 
     steps:
@@ -23,18 +23,28 @@ jobs:
         uses: actions/setup-python@v4
         with:
           python-version: ${{ env.PYTHON_VERSION }}
+
       - name: Install Poetry
         uses: snok/install-poetry@v1
         with:
           version: ${{ env.POETRY_VERSION }}
+
       - name: Install deps
         shell: bash
         run: pip install -e .
-      - name: Build and publish to pypi
-        uses: JRubics/poetry-publish@v1.17
+
+      - name: Build and publish llama-cloud-services
+        uses: JRubics/poetry-publish@v2.1
         with:
           pypi_token: ${{ secrets.LLAMA_PARSE_PYPI_TOKEN }}
-          ignore_dev_requirements: "yes"
+          poetry_install_options: "--without dev"
+
+      - name: Build and publish llama-parse
+        uses: JRubics/poetry-publish@v2.1
+        with:
+          working_directory: "llama_parse"
+          pypi_token: ${{ secrets.LLAMA_PARSE_PYPI_TOKEN }}
+          poetry_install_options: "--without dev"
 
       - name: Create GitHub Release
         id: create_release
@@ -52,6 +62,7 @@ jobs:
           export PKG=$(ls dist/ | grep tar)
           set -- $PKG
           echo "name=$1" >> $GITHUB_ENV
+
       - name: Upload Release Asset (sdist) to GitHub
         id: upload-release-asset
         uses: actions/upload-release-asset@v1

.github/workflows/unit_test.yml

@@ -17,7 +17,7 @@ jobs:
       # You can use PyPy versions in python-version.
       # For example, pypy-2.7 and pypy-3.8
       matrix:
-        python-version: ["3.8", "3.10", "3.11"]
+        python-version: ["3.9", "3.10", "3.11", "3.12"]
     steps:
       - uses: actions/checkout@v3
         with:

.gitignore

@@ -2,3 +2,6 @@
 __pycache__/
 *.pyc
 .DS_Store
+.idea
+.env*
+.ipynb_checkpoints*

.pre-commit-config.yaml

@@ -33,6 +33,7 @@ repos:
     rev: v1.0.1
     hooks:
       - id: mypy
+        exclude: ^tests/
         additional_dependencies:
           [
             "types-requests",
@@ -46,7 +47,7 @@ repos:
           [
             --disallow-untyped-defs,
             --ignore-missing-imports,
-            --python-version=3.8,
+            --python-version=3.10,
           ]
   - repo: https://github.com/adamchainz/blacken-docs
     rev: 1.16.0
@@ -84,3 +85,5 @@ repos:
     hooks:
       - id: toml-sort-fix
         exclude: ".*poetry.lock"
+
+exclude: .github/ISSUE_TEMPLATE

README.md

@@ -1,126 +1,51 @@
-# LlamaParse
+[![PyPI - Downloads](https://img.shields.io/pypi/dm/llama-cloud-services)](https://pypi.org/project/llama-cloud-services/)
+[![GitHub contributors](https://img.shields.io/github/contributors/run-llama/llama_cloud_services)](https://github.com/run-llama/llama_cloud_services/graphs/contributors)
+[![Discord](https://img.shields.io/discord/1059199217496772688)](https://discord.gg/dGcwcsnxhU)
 
-LlamaParse is an API created by LlamaIndex to efficiently parse and represent files for efficient retrieval and context augmentation using LlamaIndex frameworks.
+# Llama Cloud Services
 
-LlamaParse directly integrates with [LlamaIndex](https://github.com/run-llama/llama_index).
+This repository contains the code for hand-written SDKs and clients for interacting with LlamaCloud.
 
-Free plan is up to 1000 pages a day. Paid plan is free 7k pages per week + 0.3c per additional page.
+This includes:
 
-Read below for some quickstart information, or see the [full documentation](https://docs.cloud.llamaindex.ai/).
+- [LlamaParse](./parse.md) - A GenAI-native document parser that can parse complex document data for any downstream LLM use case (Agents, RAG, data processing, etc.).
+- [LlamaReport (beta/invite-only)](./report.md) - A prebuilt agentic report builder that can be used to build reports from a variety of data sources.
+- [LlamaExtract (beta/invite-only)](./extract.md) - A prebuilt agentic data extractor that can be used to transform data into a structured JSON representation.
 
 ## Getting Started
 
-First, login and get an api-key from [**https://cloud.llamaindex.ai ↗**](https://cloud.llamaindex.ai).
+Install the package:
 
-Then, make sure you have the latest LlamaIndex version installed.
-
-**NOTE:** If you are upgrading from v0.9.X, we recommend following our [migration guide](https://pretty-sodium-5e0.notion.site/v0-10-0-Migration-Guide-6ede431dcb8841b09ea171e7f133bd77), as well as uninstalling your previous version first.
-
-```
-pip uninstall llama-index  # run this if upgrading from v0.9.x or older
-pip install -U llama-index --upgrade --no-cache-dir --force-reinstall
+```bash
+pip install llama-cloud-services
 ```
 
-Lastly, install the package:
+Then, get your API key from [LlamaCloud](https://cloud.llamaindex.ai/).
 
-`pip install llama-parse`
-
-Now you can run the following to parse your first PDF file:
+Then, you can use the services in your code:
 
 ```python
-import nest_asyncio
+from llama_cloud_services import LlamaParse, LlamaReport, LlamaExtract
 
-nest_asyncio.apply()
-
-from llama_parse import LlamaParse
-
-parser = LlamaParse(
-    api_key="llx-...",  # can also be set in your env as LLAMA_CLOUD_API_KEY
-    result_type="markdown",  # "markdown" and "text" are available
-    num_workers=4,  # if multiple files passed, split in `num_workers` API calls
-    verbose=True,
-    language="en",  # Optionally you can define a language, default=en
-)
-
-# sync
-documents = parser.load_data("./my_file.pdf")
-
-# sync batch
-documents = parser.load_data(["./my_file1.pdf", "./my_file2.pdf"])
-
-# async
-documents = await parser.aload_data("./my_file.pdf")
-
-# async batch
-documents = await parser.aload_data(["./my_file1.pdf", "./my_file2.pdf"])
+parser = LlamaParse(api_key="YOUR_API_KEY")
+report = LlamaReport(api_key="YOUR_API_KEY")
+extract = LlamaExtract(api_key="YOUR_API_KEY")
 ```
 
-## Using with file object
+See the quickstart guides for each service for more information:
 
-You can parse a file object directly:
-
-```python
-import nest_asyncio
-
-nest_asyncio.apply()
-
-from llama_parse import LlamaParse
-
-parser = LlamaParse(
-    api_key="llx-...",  # can also be set in your env as LLAMA_CLOUD_API_KEY
-    result_type="markdown",  # "markdown" and "text" are available
-    num_workers=4,  # if multiple files passed, split in `num_workers` API calls
-    verbose=True,
-    language="en",  # Optionally you can define a language, default=en
-)
-
-with open("./my_file1.pdf", "rb") as f:
-    documents = parser.load_data(f)
-
-# you can also pass file bytes directly
-with open("./my_file1.pdf", "rb") as f:
-    file_bytes = f.read()
-    documents = parser.load_data(file_bytes)
-```
-
-## Using with `SimpleDirectoryReader`
-
-You can also integrate the parser as the default PDF loader in `SimpleDirectoryReader`:
-
-```python
-import nest_asyncio
-
-nest_asyncio.apply()
-
-from llama_parse import LlamaParse
-from llama_index.core import SimpleDirectoryReader
-
-parser = LlamaParse(
-    api_key="llx-...",  # can also be set in your env as LLAMA_CLOUD_API_KEY
-    result_type="markdown",  # "markdown" and "text" are available
-    verbose=True,
-)
-
-file_extractor = {".pdf": parser}
-documents = SimpleDirectoryReader(
-    "./data", file_extractor=file_extractor
-).load_data()
-```
-
-Full documentation for `SimpleDirectoryReader` can be found on the [LlamaIndex Documentation](https://docs.llamaindex.ai/en/stable/module_guides/loading/simpledirectoryreader.html).
-
-## Examples
-
-Several end-to-end indexing examples can be found in the examples folder
-
-- [Getting Started](examples/demo_basic.ipynb)
-- [Advanced RAG Example](examples/demo_advanced.ipynb)
-- [Raw API Usage](examples/demo_api.ipynb)
+- [LlamaParse](./parse.md)
+- [LlamaReport (beta/invite-only)](./report.md)
+- [LlamaExtract (beta/invite-only)](./extract.md)
 
 ## Documentation
 
-[https://docs.cloud.llamaindex.ai/](https://docs.cloud.llamaindex.ai/)
+You can see complete SDK and API documentation for each service on [our official docs](https://docs.cloud.llamaindex.ai/).
 
 ## Terms of Service
 
 See the [Terms of Service Here](./TOS.pdf).
+
+## Get in Touch (LlamaCloud)
+
+You can get in touch with us by following our [contact link](https://www.llamaindex.ai/contact).

examples/extract/resume_screening.ipynb

@@ -0,0 +1,834 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "# Extracting data from Resumes\n",
+    "\n",
+    "Let us assume that we are running a hiring process for a company and we have received a list of resumes from candidates. We want to extract structured data from the resumes so that we can run a screening process and shortlist candidates. \n",
+    "\n",
+    "Take a look at one of the resumes in the `data/resumes` directory. "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/html": [
+       "\n",
+       "        <iframe\n",
+       "            width=\"600\"\n",
+       "            height=\"400\"\n",
+       "            src=\"./data/resumes/ai_researcher.pdf\"\n",
+       "            frameborder=\"0\"\n",
+       "            allowfullscreen\n",
+       "            \n",
+       "        ></iframe>\n",
+       "        "
+      ],
+      "text/plain": [
+       "<IPython.lib.display.IFrame at 0x109a7dcd0>"
+      ]
+     },
+     "execution_count": null,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "from IPython.display import IFrame\n",
+    "\n",
+    "IFrame(src=\"./data/resumes/ai_researcher.pdf\", width=600, height=400)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "You will notice that all the resumes have different layouts but contain common information like name, email, experience, education, etc. \n",
+    "\n",
+    "With LlamaExtract, we will show you how to:\n",
+    "- *Define* a data schema to extract the information of interest. \n",
+    "- *Iterate* over the data schema to generalize the schema for multiple resumes.\n",
+    "- *Finalize* the schema and schedule extractions for multiple resumes.\n",
+    "\n",
+    "We will start by defining a `LlamaExtract` client which provides a Python interface to the LlamaExtract API. "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from dotenv import load_dotenv\n",
+    "from llama_cloud_services import LlamaExtract\n",
+    "\n",
+    "\n",
+    "# Load environment variables (put LLAMA_CLOUD_API_KEY in your .env file)\n",
+    "load_dotenv(override=True)\n",
+    "\n",
+    "# Optionally, add your project id/organization id\n",
+    "llama_extract = LlamaExtract()"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Defining the data schema\n",
+    "\n",
+    "Next, let us try to extract two fields from the resume: `name` and `email`. We can either use a Python dictionary structure to define the `data_schema` as a JSON or use a Pydantic model instead, for brevity and convenience. In either case, our output is guaranteed to validate against this schema."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from pydantic import BaseModel, Field\n",
+    "\n",
+    "\n",
+    "class Resume(BaseModel):\n",
+    "    name: str = Field(description=\"The name of the candidate\")\n",
+    "    email: str = Field(description=\"The email address of the candidate\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Uploading files: 100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 1/1 [00:02<00:00,  2.20s/it]\n",
+      "Creating extraction jobs: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 1/1 [00:02<00:00,  2.93s/it]\n",
+      "Extracting files: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 1/1 [00:02<00:00,  2.94s/it]\n",
+      "Uploading files: 100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 1/1 [00:00<00:00,  1.13it/s]\n",
+      "Creating extraction jobs: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 1/1 [00:00<00:00,  1.80it/s]\n",
+      "Extracting files: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 1/1 [00:15<00:00, 15.18s/it]\n",
+      "Uploading files: 100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 1/1 [00:00<00:00,  1.16it/s]\n",
+      "Creating extraction jobs: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 1/1 [00:00<00:00,  2.33it/s]\n",
+      "Extracting files: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 1/1 [00:32<00:00, 32.86s/it]\n"
+     ]
+    }
+   ],
+   "source": [
+    "from llama_cloud.core.api_error import ApiError\n",
+    "\n",
+    "try:\n",
+    "    existing_agent = llama_extract.get_agent(name=\"resume-screening\")\n",
+    "    if existing_agent:\n",
+    "        llama_extract.delete_agent(existing_agent.id)\n",
+    "except ApiError as e:\n",
+    "    if e.status_code == 404:\n",
+    "        pass\n",
+    "    else:\n",
+    "        raise\n",
+    "\n",
+    "agent = llama_extract.create_agent(name=\"resume-screening\", data_schema=Resume)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "[ExtractionAgent(id=1fef43b5-8230-43b4-9e80-c1cddf53889c, name=resume-screening),\n",
+       " ExtractionAgent(id=93f8508b-3570-46f0-ae62-6315b40043bd, name=receipt/noisebridge_receipt.pdf_56db3d92),\n",
+       " ExtractionAgent(id=08315f0e-7146-430b-99b8-9701cb3ace6a, name=receipt/noisebridge_receipt.pdf_5c4730a7),\n",
+       " ExtractionAgent(id=cfcd7756-015d-4dbd-b142-a3eefcb16cd3, name=resume/software_architect_resume.html_4a11cf15),\n",
+       " ExtractionAgent(id=17cb83d9-601e-4f5c-a7aa-286e3045bcb4, name=resume/software_architect_resume.html_0b7d84a8),\n",
+       " ExtractionAgent(id=adc8e88c-44d3-4613-a5aa-d666ef007494, name=slide/saas_slide.pdf_bcc627a5),\n",
+       " ExtractionAgent(id=189f14cd-6370-4476-a6ad-36eafbc62618, name=slide/saas_slide.pdf_065aa22b),\n",
+       " ExtractionAgent(id=b9938ca5-6225-43cb-89ea-b0065237792f, name=test2),\n",
+       " ExtractionAgent(id=574d37b8-59dc-41e9-bde0-5c506a8eb670, name=test)]"
+      ]
+     },
+     "execution_count": null,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "llama_extract.list_agents()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "{'name': 'Dr. Rachel Zhang', 'email': 'rachel.zhang@email.com'}"
+      ]
+     },
+     "execution_count": null,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "resume = agent.extract(\"./data/resumes/ai_researcher.pdf\")\n",
+    "resume.data"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Iterating over the data schema\n",
+    "\n",
+    "Now that we have created a data schema, let us add more fields to the schema. We will add `experience` and `education` fields to the schema. \n",
+    "- We can create a new Pydantic model for each of these fields and represent `experience` and `education` as lists of these models. Doing this will allow us to extract multiple entities from the resume without having to pre-define how many experiences or education the candidate has. \n",
+    "- We have added a `description` parameter to provide more context for extraction. We can use `description` to provide example inputs/outputs for the extraction. \n",
+    "- Note that we have annotated the `start_date` and `end_date` fields with `Optional[str]` to indicate that these fields are optional. This is *important* because the schema will be used to extract data from multiple resumes and not all resumes will have the same format. A field must only be required if it is guaranteed to be present in all the resumes. \n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from typing import List, Optional\n",
+    "\n",
+    "\n",
+    "class Education(BaseModel):\n",
+    "    institution: str = Field(description=\"The institution of the candidate\")\n",
+    "    degree: str = Field(description=\"The degree of the candidate\")\n",
+    "    start_date: Optional[str] = Field(\n",
+    "        default=None, description=\"The start date of the candidate's education\"\n",
+    "    )\n",
+    "    end_date: Optional[str] = Field(\n",
+    "        default=None, description=\"The end date of the candidate's education\"\n",
+    "    )\n",
+    "\n",
+    "\n",
+    "class Experience(BaseModel):\n",
+    "    company: str = Field(description=\"The name of the company\")\n",
+    "    title: str = Field(description=\"The title of the candidate\")\n",
+    "    description: Optional[str] = Field(\n",
+    "        default=None, description=\"The description of the candidate's experience\"\n",
+    "    )\n",
+    "    start_date: Optional[str] = Field(\n",
+    "        default=None, description=\"The start date of the candidate's experience\"\n",
+    "    )\n",
+    "    end_date: Optional[str] = Field(\n",
+    "        default=None, description=\"The end date of the candidate's experience\"\n",
+    "    )\n",
+    "\n",
+    "\n",
+    "class Resume(BaseModel):\n",
+    "    name: str = Field(description=\"The name of the candidate\")\n",
+    "    email: str = Field(description=\"The email address of the candidate\")\n",
+    "    links: List[str] = Field(\n",
+    "        description=\"The links to the candidate's social media profiles\"\n",
+    "    )\n",
+    "    experience: List[Experience] = Field(description=\"The candidate's experience\")\n",
+    "    education: List[Education] = Field(description=\"The candidate's education\")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Next, we will update the `data_schema` for the `resume-screening` agent to use the new `Resume` model. "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "{'name': 'Dr. Rachel Zhang',\n",
+       " 'email': 'rachel.zhang@email.com',\n",
+       " 'links': ['linkedin.com/in/rachelzhang',\n",
+       "  'github.com/rzhang-ai',\n",
+       "  'scholar.google.com/rachelzhang'],\n",
+       " 'experience': [{'company': 'DeepMind',\n",
+       "   'title': 'Senior Research Scientist',\n",
+       "   'description': '- Lead researcher on large-scale multi-task learning systems, developing novel architectures that improve cross-task generalization by 40%\\n- Pioneered new approach to zero-shot learning using contrastive training, published in NeurIPS 2023\\n- Built and led team of 6 researchers working on foundational ML models\\n- Developed novel regularization techniques for large language models, reducing catastrophic forgetting by 35%',\n",
+       "   'start_date': '2019',\n",
+       "   'end_date': 'Present'},\n",
+       "  {'company': 'Google Research',\n",
+       "   'title': 'Research Scientist',\n",
+       "   'description': '- Developed probabilistic frameworks for robust ML, published in ICML 2018\\n- Created novel attention mechanisms for computer vision models, improving accuracy by 25%\\n- Led collaboration with Google Brain team on efficient training methods for transformer models\\n- Mentored 4 PhD interns and collaborated with academic institutions',\n",
+       "   'start_date': '2015',\n",
+       "   'end_date': '2019'},\n",
+       "  {'company': 'Columbia University',\n",
+       "   'title': 'Research Assistant Professor',\n",
+       "   'description': '- Published seminal work on Bayesian optimization methods (cited 1000+ times)\\n- Taught graduate-level courses in Machine Learning and Statistical Learning Theory\\n- Supervised 5 PhD students and 3 MSc students\\n- Secured $500K in research grants for probabilistic ML research',\n",
+       "   'start_date': '2011',\n",
+       "   'end_date': '2015'}],\n",
+       " 'education': [{'institution': 'Columbia University',\n",
+       "   'degree': 'Ph.D. in Computer Science',\n",
+       "   'start_date': '2007',\n",
+       "   'end_date': '2011'},\n",
+       "  {'institution': 'Stanford University',\n",
+       "   'degree': 'M.S. in Computer Science',\n",
+       "   'start_date': '2005',\n",
+       "   'end_date': '2007'}]}"
+      ]
+     },
+     "execution_count": null,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "agent.data_schema = Resume\n",
+    "resume = agent.extract(\"./data/resumes/ai_researcher.pdf\")\n",
+    "resume.data"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "This is a good start. Let us add a few more fields to the schema and re-run the extraction. "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "class TechnicalSkills(BaseModel):\n",
+    "    programming_languages: List[str] = Field(\n",
+    "        description=\"The programming languages the candidate is proficient in.\"\n",
+    "    )\n",
+    "    frameworks: List[str] = Field(\n",
+    "        description=\"The tools/frameworks the candidate is proficient in, e.g. React, Django, PyTorch, etc.\"\n",
+    "    )\n",
+    "    skills: List[str] = Field(\n",
+    "        description=\"Other general skills the candidate is proficient in, e.g. Data Engineering, Machine Learning, etc.\"\n",
+    "    )\n",
+    "\n",
+    "\n",
+    "class Resume(BaseModel):\n",
+    "    name: str = Field(description=\"The name of the candidate\")\n",
+    "    email: str = Field(description=\"The email address of the candidate\")\n",
+    "    links: List[str] = Field(\n",
+    "        description=\"The links to the candidate's social media profiles\"\n",
+    "    )\n",
+    "    experience: List[Experience] = Field(description=\"The candidate's experience\")\n",
+    "    education: List[Education] = Field(description=\"The candidate's education\")\n",
+    "    technical_skills: TechnicalSkills = Field(\n",
+    "        description=\"The candidate's technical skills\"\n",
+    "    )\n",
+    "    key_accomplishments: str = Field(\n",
+    "        description=\"Summarize the candidates highest achievements.\"\n",
+    "    )"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "{'name': 'Dr. Rachel Zhang, Ph.D.',\n",
+       " 'email': 'rachel.zhang@email.com',\n",
+       " 'links': ['linkedin.com/in/rachelzhang',\n",
+       "  'github.com/rzhang-ai',\n",
+       "  'scholar.google.com/rachelzhang'],\n",
+       " 'experience': [{'company': 'DeepMind',\n",
+       "   'title': 'Senior Research Scientist',\n",
+       "   'description': 'Lead researcher on large-scale multi-task learning systems, developing novel architectures that improve cross-task generalization by 40%\\nPioneered new approach to zero-shot learning using contrastive training, published in NeurIPS 2023\\nBuilt and led team of 6 researchers working on foundational ML models\\nDeveloped novel regularization techniques for large language models, reducing catastrophic forgetting by 35%',\n",
+       "   'start_date': '2019',\n",
+       "   'end_date': 'Present'},\n",
+       "  {'company': 'Google Research',\n",
+       "   'title': 'Research Scientist',\n",
+       "   'description': 'Developed probabilistic frameworks for robust ML, published in ICML 2018\\nCreated novel attention mechanisms for computer vision models, improving accuracy by 25%\\nLed collaboration with Google Brain team on efficient training methods for transformer models\\nMentored 4 PhD interns and collaborated with academic institutions',\n",
+       "   'start_date': '2015',\n",
+       "   'end_date': '2019'},\n",
+       "  {'company': 'Columbia University',\n",
+       "   'title': 'Research Assistant Professor',\n",
+       "   'description': 'Published seminal work on Bayesian optimization methods (cited 1000+ times)\\nTaught graduate-level courses in Machine Learning and Statistical Learning Theory\\nSupervised 5 PhD students and 3 MSc students\\nSecured $500K in research grants for probabilistic ML research',\n",
+       "   'start_date': '2011',\n",
+       "   'end_date': '2015'}],\n",
+       " 'education': [{'institution': 'Columbia University',\n",
+       "   'degree': 'Ph.D. in Computer Science',\n",
+       "   'start_date': '2007',\n",
+       "   'end_date': '2011'},\n",
+       "  {'institution': 'Stanford University',\n",
+       "   'degree': 'M.S. in Computer Science',\n",
+       "   'start_date': '2005',\n",
+       "   'end_date': '2007'}],\n",
+       " 'technical_skills': {'programming_languages': ['Python',\n",
+       "   'C++',\n",
+       "   'Julia',\n",
+       "   'CUDA'],\n",
+       "  'frameworks': ['PyTorch', 'TensorFlow', 'JAX', 'Ray'],\n",
+       "  'skills': ['Deep Learning',\n",
+       "   'Reinforcement Learning',\n",
+       "   'Probabilistic Models',\n",
+       "   'Multi-Task Learning',\n",
+       "   'Zero-Shot Learning',\n",
+       "   'Neural Architecture Search']},\n",
+       " 'key_accomplishments': 'AI researcher with 12+ years of experience spanning classical machine learning, deep learning, and probabilistic modeling. Led groundbreaking research in reinforcement learning, generative models, and multi-task learning. Published 25+ papers in top-tier conferences (NeurIPS, ICML, ICLR). Strong track record of transitioning theoretical advances into practical applications in both academic and industrial settings.'}"
+      ]
+     },
+     "execution_count": null,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "agent.data_schema = Resume\n",
+    "resume = agent.extract(\"./data/resumes/ai_researcher.pdf\")\n",
+    "resume.data"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Finalizing the schema\n",
+    "\n",
+    "This is great! We have extracted a lot of key information from the resume that is well-typed and can be used downstream for further processing. Until now, this data is ephemeral and will be lost if we close the session. Let us save the state of our extraction and use it to extract data from multiple resumes. "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "agent.save()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "{'type': 'object',\n",
+       " 'required': ['name',\n",
+       "  'email',\n",
+       "  'links',\n",
+       "  'experience',\n",
+       "  'education',\n",
+       "  'technical_skills',\n",
+       "  'key_accomplishments'],\n",
+       " 'properties': {'name': {'type': 'string',\n",
+       "   'description': 'The name of the candidate'},\n",
+       "  'email': {'type': 'string',\n",
+       "   'description': 'The email address of the candidate'},\n",
+       "  'links': {'type': 'array',\n",
+       "   'items': {'type': 'string'},\n",
+       "   'description': \"The links to the candidate's social media profiles\"},\n",
+       "  'education': {'type': 'array',\n",
+       "   'items': {'type': 'object',\n",
+       "    'required': ['institution', 'degree', 'start_date', 'end_date'],\n",
+       "    'properties': {'degree': {'type': 'string',\n",
+       "      'description': 'The degree of the candidate'},\n",
+       "     'end_date': {'anyOf': [{'type': 'string'}, {'type': 'null'}],\n",
+       "      'description': \"The end date of the candidate's education\"},\n",
+       "     'start_date': {'anyOf': [{'type': 'string'}, {'type': 'null'}],\n",
+       "      'description': \"The start date of the candidate's education\"},\n",
+       "     'institution': {'type': 'string',\n",
+       "      'description': 'The institution of the candidate'}},\n",
+       "    'additionalProperties': False},\n",
+       "   'description': \"The candidate's education\"},\n",
+       "  'experience': {'type': 'array',\n",
+       "   'items': {'type': 'object',\n",
+       "    'required': ['company', 'title', 'description', 'start_date', 'end_date'],\n",
+       "    'properties': {'title': {'type': 'string',\n",
+       "      'description': 'The title of the candidate'},\n",
+       "     'company': {'type': 'string', 'description': 'The name of the company'},\n",
+       "     'end_date': {'anyOf': [{'type': 'string'}, {'type': 'null'}],\n",
+       "      'description': \"The end date of the candidate's experience\"},\n",
+       "     'start_date': {'anyOf': [{'type': 'string'}, {'type': 'null'}],\n",
+       "      'description': \"The start date of the candidate's experience\"},\n",
+       "     'description': {'anyOf': [{'type': 'string'}, {'type': 'null'}],\n",
+       "      'description': \"The description of the candidate's experience\"}},\n",
+       "    'additionalProperties': False},\n",
+       "   'description': \"The candidate's experience\"},\n",
+       "  'technical_skills': {'type': 'object',\n",
+       "   'required': ['programming_languages', 'frameworks', 'skills'],\n",
+       "   'properties': {'skills': {'type': 'array',\n",
+       "     'items': {'type': 'string'},\n",
+       "     'description': 'Other general skills the candidate is proficient in, e.g. Data Engineering, Machine Learning, etc.'},\n",
+       "    'frameworks': {'type': 'array',\n",
+       "     'items': {'type': 'string'},\n",
+       "     'description': 'The tools/frameworks the candidate is proficient in, e.g. React, Django, PyTorch, etc.'},\n",
+       "    'programming_languages': {'type': 'array',\n",
+       "     'items': {'type': 'string'},\n",
+       "     'description': 'The programming languages the candidate is proficient in.'}},\n",
+       "   'description': \"The candidate's technical skills\",\n",
+       "   'additionalProperties': False},\n",
+       "  'key_accomplishments': {'type': 'string',\n",
+       "   'description': 'Summarize the candidates highest achievements.'}},\n",
+       " 'additionalProperties': False}"
+      ]
+     },
+     "execution_count": null,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "agent = llama_extract.get_agent(\"resume-screening\")\n",
+    "agent.data_schema  # Latest schema should be returned"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "#### Queueing extractions"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "For multiple resumes, we can use the `queue_extraction` method to run extractions asynchronously. This is ideal for processing batch extraction jobs."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Uploading files: 100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 3/3 [00:01<00:00,  2.13it/s]\n",
+      "Creating extraction jobs: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 3/3 [00:00<00:00,  5.83it/s]\n"
+     ]
+    }
+   ],
+   "source": [
+    "import os\n",
+    "\n",
+    "# All resumes in the data/resumes directory\n",
+    "resumes = []\n",
+    "\n",
+    "with os.scandir(\"./data/resumes\") as entries:\n",
+    "    for entry in entries:\n",
+    "        if entry.is_file():\n",
+    "            resumes.append(entry.path)\n",
+    "\n",
+    "jobs = await agent.queue_extraction(resumes)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "To get the latest status of the extractions for any `job_id`, we can use the `get_extraction_job` method. \n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "[<StatusEnum.PENDING: 'PENDING'>,\n",
+       " <StatusEnum.PENDING: 'PENDING'>,\n",
+       " <StatusEnum.PENDING: 'PENDING'>]"
+      ]
+     },
+     "execution_count": null,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "[agent.get_extraction_job(job_id=job.id).status for job in jobs]"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "We notice that all extraction runs are in a PENDING state. We can check back again to see if the extractions have completed. "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "[<StatusEnum.SUCCESS: 'SUCCESS'>,\n",
+       " <StatusEnum.SUCCESS: 'SUCCESS'>,\n",
+       " <StatusEnum.SUCCESS: 'SUCCESS'>]"
+      ]
+     },
+     "execution_count": null,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "[agent.get_extraction_job(job_id=job.id).status for job in jobs]"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "#### Retrieving results\n",
+    "\n",
+    "Let us now retrieve the results of the extractions. If the status of the extraction is `SUCCESS`, we can retrieve the data from the `data` field. In case there are errors (status = `ERROR`), we can retrieve the error message from the `error` field. \n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "results = []\n",
+    "for job in jobs:\n",
+    "    extract_run = agent.get_extraction_run_for_job(job.id)\n",
+    "    if extract_run.status == \"SUCCESS\":\n",
+    "        results.append(extract_run.data)\n",
+    "    else:\n",
+    "        print(f\"Extraction status for job {job.id}: {extract_run.status}\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "{'name': 'Dr. Rachel Zhang, Ph.D.',\n",
+       " 'email': 'rachel.zhang@email.com',\n",
+       " 'links': ['linkedin.com/in/rachelzhang',\n",
+       "  'github.com/rzhang-ai',\n",
+       "  'scholar.google.com/rachelzhang'],\n",
+       " 'education': [{'degree': 'Ph.D. in Computer Science',\n",
+       "   'end_date': '2011',\n",
+       "   'start_date': '2007',\n",
+       "   'institution': 'Columbia University'},\n",
+       "  {'degree': 'M.S. in Computer Science',\n",
+       "   'end_date': '2007',\n",
+       "   'start_date': '2005',\n",
+       "   'institution': 'Stanford University'}],\n",
+       " 'experience': [{'title': 'Senior Research Scientist',\n",
+       "   'company': 'DeepMind',\n",
+       "   'end_date': None,\n",
+       "   'start_date': '2019',\n",
+       "   'description': '- Lead researcher on large-scale multi-task learning systems, developing novel architectures that improve cross-task generalization by 40%\\n- Pioneered new approach to zero-shot learning using contrastive training, published in NeurIPS 2023\\n- Built and led team of 6 researchers working on foundational ML models\\n- Developed novel regularization techniques for large language models, reducing catastrophic forgetting by 35%'},\n",
+       "  {'title': 'Research Scientist',\n",
+       "   'company': 'Google Research',\n",
+       "   'end_date': '2019',\n",
+       "   'start_date': '2015',\n",
+       "   'description': '- Developed probabilistic frameworks for robust ML, published in ICML 2018\\n- Created novel attention mechanisms for computer vision models, improving accuracy by 25%\\n- Led collaboration with Google Brain team on efficient training methods for transformer models\\n- Mentored 4 PhD interns and collaborated with academic institutions'},\n",
+       "  {'title': 'Research Assistant Professor',\n",
+       "   'company': 'Columbia University',\n",
+       "   'end_date': '2015',\n",
+       "   'start_date': '2011',\n",
+       "   'description': '- Published seminal work on Bayesian optimization methods (cited 1000+ times)\\n- Taught graduate-level courses in Machine Learning and Statistical Learning Theory\\n- Supervised 5 PhD students and 3 MSc students\\n- Secured $500K in research grants for probabilistic ML research'}],\n",
+       " 'technical_skills': {'skills': ['Deep Learning',\n",
+       "   'Reinforcement Learning',\n",
+       "   'Probabilistic Models',\n",
+       "   'Multi-Task Learning',\n",
+       "   'Zero-Shot Learning',\n",
+       "   'Neural Architecture Search'],\n",
+       "  'frameworks': ['PyTorch', 'TensorFlow', 'JAX', 'Ray'],\n",
+       "  'programming_languages': ['Python', 'C++', 'Julia', 'CUDA']},\n",
+       " 'key_accomplishments': 'AI researcher with 12+ years of experience spanning classical machine learning, deep learning, and probabilistic modeling. Led groundbreaking research in reinforcement learning, generative models, and multi-task learning. Published 25+ papers in top-tier conferences (NeurIPS, ICML, ICLR). Strong track record of transitioning theoretical advances into practical applications in both academic and industrial settings.'}"
+      ]
+     },
+     "execution_count": null,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "results[0]"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "{'name': 'Alex Park',\n",
+       " 'email': 'alex park@email.com',\n",
+       " 'links': ['linkedin.com/in/alexpark'],\n",
+       " 'education': [{'degree': 'M.S. Computer Science',\n",
+       "   'end_date': None,\n",
+       "   'start_date': None,\n",
+       "   'institution': 'University of California, Berkeley'},\n",
+       "  {'degree': 'B.S. Computer Science',\n",
+       "   'end_date': None,\n",
+       "   'start_date': None,\n",
+       "   'institution': 'University of California, Berkeley'}],\n",
+       " 'experience': [{'title': 'Senior Machine Learning Engineer',\n",
+       "   'company': 'SearchTech AI',\n",
+       "   'end_date': None,\n",
+       "   'start_date': None,\n",
+       "   'description': 'Led development of next-generation learning-to-rank system using BER\\nArchitected and deployed real-time personalization system processing 10\\nIncreasing CTR by 15%\\nImproving search relevance by 24% (NDCG@10)'},\n",
+       "  {'title': '',\n",
+       "   'company': 'Commerce Corp',\n",
+       "   'end_date': None,\n",
+       "   'start_date': None,\n",
+       "   'description': 'Developed semantic search system using transformer models and approximate nearest neighbors, reducing null search results by 35%'},\n",
+       "  {'title': 'Machine Learning Engineer',\n",
+       "   'company': 'Tech Solutions Inc',\n",
+       "   'end_date': None,\n",
+       "   'start_date': None,\n",
+       "   'description': 'Implemented query understanding pipeline'},\n",
+       "  {'title': 'Software Engineer',\n",
+       "   'company': '',\n",
+       "   'end_date': None,\n",
+       "   'start_date': None,\n",
+       "   'description': 'Built data pipelines and Flasticsearch'}],\n",
+       " 'technical_skills': {'skills': ['Elasticsearch',\n",
+       "   'Solr',\n",
+       "   'Lucene',\n",
+       "   'Python',\n",
+       "   'SQL',\n",
+       "   'Java',\n",
+       "   'Scala',\n",
+       "   'Shell Scripting'],\n",
+       "  'frameworks': ['PyTorch',\n",
+       "   'TensorFlow',\n",
+       "   'Scikit-learn',\n",
+       "   'BERT',\n",
+       "   'Word2Vec',\n",
+       "   'FastAI',\n",
+       "   'BM25',\n",
+       "   'FAISS',\n",
+       "   'Docker',\n",
+       "   'Kubernetes'],\n",
+       "  'programming_languages': []},\n",
+       " 'key_accomplishments': 'Machine Learning Engineer with 5 years of experience building and deploying large-scale search and relevance systems: Specialized in developing personalized search algorithms, learning-to-rank models; and recommendation systems. Strong track record of improving search relevance metrics and user engagement through ML-driven solutions:'}"
+      ]
+     },
+     "execution_count": null,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "results[1]"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "{'name': 'Sarah Chen',\n",
+       " 'email': 'sarah.chen@email.com',\n",
+       " 'links': [],\n",
+       " 'education': [{'degree': 'Master of Science in Computer Science',\n",
+       "   'end_date': '2013',\n",
+       "   'start_date': None,\n",
+       "   'institution': 'Stanford University'},\n",
+       "  {'degree': 'Bachelor of Science in Computer Engineering',\n",
+       "   'end_date': '2011',\n",
+       "   'start_date': None,\n",
+       "   'institution': 'University of California, Berkeley'}],\n",
+       " 'experience': [{'title': 'Senior Software Architect',\n",
+       "   'company': 'TechCorp Solutions',\n",
+       "   'end_date': None,\n",
+       "   'start_date': '2020',\n",
+       "   'description': '- Led architectural design and implementation of a cloud-native platform serving 2M+ users\\n- Established architectural guidelines and best practices adopted across 12 development teams\\n- Reduced system latency by 40% through implementation of event-driven architecture\\n- Mentored 15+ senior developers in cloud-native development practices'},\n",
+       "  {'title': 'Lead Software Engineer',\n",
+       "   'company': 'DataFlow Systems',\n",
+       "   'end_date': '2020',\n",
+       "   'start_date': '2016',\n",
+       "   'description': '- Architected and led development of distributed data processing platform handling 5TB daily\\n- Designed microservices architecture reducing deployment time by 65%\\n- Led migration of legacy monolith to cloud-native architecture\\n- Managed team of 8 engineers across 3 international locations'},\n",
+       "  {'title': 'Senior Software Engineer',\n",
+       "   'company': 'InnovateTech',\n",
+       "   'end_date': '2016',\n",
+       "   'start_date': '2013',\n",
+       "   'description': '- Developed high-performance trading platform processing 100K transactions per second\\n- Implemented real-time analytics engine reducing processing latency by 75%\\n- Led adoption of container orchestration reducing deployment costs by 35%'}],\n",
+       " 'technical_skills': {'skills': ['Architecture & Design',\n",
+       "   'Microservices',\n",
+       "   'Event-Driven Architecture',\n",
+       "   'Domain-Driven Design',\n",
+       "   'REST APIs',\n",
+       "   'Cloud Platforms'],\n",
+       "  'frameworks': ['AWS (Advanced)', 'Azure', 'Google Cloud Platform'],\n",
+       "  'programming_languages': ['Java', 'Python', 'Go', 'JavaScript/TypeScript']},\n",
+       " 'key_accomplishments': '- Co-inventor on three patents for distributed systems architecture\\n- Published paper on \"Scalable Microservices Architecture\" at IEEE Cloud Computing Conference 2022\\n- Keynote Speaker, CloudCon 2023: \"Future of Cloud-Native Architecture\"\\n- Regular presenter at local tech meetups and conferences'}"
+      ]
+     },
+     "execution_count": null,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "results[2]"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Congratulations! You now have an agent that can extract structured data from resumes. \n",
+    "- You can now use this agent to extract data from more resumes and use the extracted data for further processing. \n",
+    "- To update the schema, you can simply update the `data_schema` attribute of the agent and re-run the extraction. \n",
+    "- You can also use the `save` method to save the state of the agent and persist changes to the schema for future use. \n",
+    "\n"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 4
+}

examples/parse/agents/demo_simple_openai_agent.ipynb

@@ -22,7 +22,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!pip install llama-parse llama-index llama-index-postprocessor-sbert-rerank"
+    "!pip install llama-cloud-services llama-index llama-index-postprocessor-sbert-rerank"
    ]
   },
   {
@@ -82,7 +82,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from llama_parse import LlamaParse\n",
+    "from llama_cloud_services import LlamaParse\n",
     "\n",
     "parser = LlamaParse(\n",
     "    result_type=\"markdown\",\n",

examples/parse/caltrain/caltrain_text_mode.ipynb

@@ -7,7 +7,7 @@
    "source": [
     "# RAG over the Caltrain Weekend Schedule \n",
     "\n",
-    "<a href=\"https://colab.research.google.com/github/run-llama/llama_parse/blob/main/examples/caltrain/caltrain_text_mode.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
+    "<a href=\"https://colab.research.google.com/github/run-llama/llama_cloud_services/blob/main/examples/parse/caltrain/caltrain_text_mode.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
     "\n",
     "This example shows off LlamaParse parsing capabilities to build a functioning query pipeline over the Caltrain weekend schedule, a big timetable containing all trains northbound and southbound and their stops in various cities.\n",
     "\n",
@@ -81,7 +81,7 @@
     }
    ],
    "source": [
-    "from llama_parse import LlamaParse\n",
+    "from llama_cloud_services import LlamaParse\n",
     "\n",
     "docs = LlamaParse(result_type=\"text\").load_data(\"./caltrain_schedule_weekend.pdf\")"
    ]

examples/parse/demo_advanced.ipynb

@@ -26,7 +26,7 @@
     "!pip install llama-index-embeddings-openai\n",
     "!pip install llama-index-postprocessor-flag-embedding-reranker\n",
     "!pip install git+https://github.com/FlagOpen/FlagEmbedding.git\n",
-    "!pip install llama-parse"
+    "!pip install llama-cloud-services"
    ]
   },
   {
@@ -108,7 +108,7 @@
     }
    ],
    "source": [
-    "from llama_parse import LlamaParse\n",
+    "from llama_cloud_services import LlamaParse\n",
     "\n",
     "documents = LlamaParse(result_type=\"markdown\").load_data(\"./apple_2021_10k.pdf\")"
    ]

examples/parse/demo_advanced_astradb.ipynb

@@ -22,7 +22,7 @@
     "%pip install llama-index-embeddings-openai\n",
     "%pip install llama-index-postprocessor-flag-embedding-reranker\n",
     "%pip install git+https://github.com/FlagOpen/FlagEmbedding.git\n",
-    "%pip install llama-parse\n",
+    "%pip install llama-cloud-services\n",
     "%pip install llama-index-vector-stores-astra-db"
    ]
   },
@@ -107,7 +107,7 @@
     }
    ],
    "source": [
-    "from llama_parse import LlamaParse\n",
+    "from llama_cloud_services import LlamaParse\n",
     "\n",
     "documents = LlamaParse(result_type=\"markdown\").load_data(\"./uber_10q_march_2022.pdf\")"
    ]

examples/parse/demo_advanced_weaviate.ipynb

@@ -6,7 +6,7 @@
    "source": [
     "# Advanced RAG with LlamaParse + Weaviate\n",
     "\n",
-    "<a href=\"https://colab.research.google.com/github/run-llama/llama_parse/blob/main/examples/demo_advanced_weaviate.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\\\" alt=\"Open In Colab\"/></a>\n",
+    "<a href=\"https://colab.research.google.com/github/run-llama/llama_parse/blob/main/examples/parse/demo_advanced_weaviate.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\\\" alt=\"Open In Colab\"/></a>\n",
     "\n",
     "This notebook shows you how to use `LlamaParse` for advancd RAG applications with `LlamaIndex` and [Weaviate](https://weaviate.io/).\n",
     "\n",
@@ -176,7 +176,7 @@
     }
    ],
    "source": [
-    "from llama_parse import LlamaParse\n",
+    "from llama_cloud_services import LlamaParse\n",
     "\n",
     "documents = LlamaParse(result_type=\"markdown\").load_data(\"./uber_10q_march_2022.pdf\")"
    ]

examples/parse/demo_astradb.ipynb

@@ -130,7 +130,7 @@
     }
    ],
    "source": [
-    "from llama_parse import LlamaParse\n",
+    "from llama_cloud_services import LlamaParse\n",
     "\n",
     "documents = LlamaParse(result_type=\"text\").load_data(file_path)"
    ]

examples/parse/demo_basic.ipynb

@@ -73,7 +73,7 @@
     }
    ],
    "source": [
-    "from llama_parse import LlamaParse\n",
+    "from llama_cloud_services import LlamaParse\n",
     "\n",
     "documents = LlamaParse(result_type=\"text\").load_data(\"./attention.pdf\")"
    ]
@@ -120,7 +120,7 @@
     }
    ],
    "source": [
-    "from llama_parse import LlamaParse\n",
+    "from llama_cloud_services import LlamaParse\n",
     "\n",
     "documents = LlamaParse(result_type=\"markdown\").load_data(\"./attention.pdf\")"
    ]

examples/parse/demo_excel.ipynb

@@ -6,7 +6,7 @@
    "source": [
     "# RAG with Excel Spreadsheet using LlamaPrase\n",
     "\n",
-    "<a href=\"https://colab.research.google.com/github/run-llama/llama_parse/blob/main/examples/demo_excel.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
+    "<a href=\"https://colab.research.google.com/github/run-llama/llama_cloud_services/blob/main/examples/demo_excel.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
     "\n",
     "This notebook shows you using LlamaParse with Excel Spreadsheet.\n",
     "\n",
@@ -21,7 +21,7 @@
    "outputs": [],
    "source": [
     "%pip install llama-index\n",
-    "%pip install llama-parse"
+    "%pip install llama-cloud-services"
    ]
   },
   {
@@ -41,7 +41,7 @@
     "\n",
     "nest_asyncio.apply()\n",
     "\n",
-    "from llama_parse import LlamaParse\n",
+    "from llama_cloud_services import LlamaParse\n",
     "\n",
     "api_key = \"llx-\"  # get from cloud.llamaindex.ai"
    ]

examples/parse/demo_insurance.ipynb

@@ -6,7 +6,7 @@
    "source": [
     "# LlamaParse - Fast checking Insurance Contract for Coverage\n",
     "\n",
-    "<a href=\"https://colab.research.google.com/github/run-llama/llama_parse/blob/main/examples/demo_insurance.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
+    "<a href=\"https://colab.research.google.com/github/run-llama/llama_cloud_services/blob/main/examples/demo_insurance.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
     "\n",
     "In this notebook we will look at how LlamaParse can be used to extract structured coverage information from an insurance policy."
    ]
@@ -116,7 +116,7 @@
     }
    ],
    "source": [
-    "from llama_parse import LlamaParse\n",
+    "from llama_cloud_services import LlamaParse\n",
     "\n",
     "documents = LlamaParse(result_type=\"markdown\").load_data(\"./policy.pdf\")"
    ]

examples/parse/demo_json.ipynb

@@ -7,7 +7,7 @@
    "source": [
     "# LlamaParse JSON Mode + Multimodal RAG\n",
     "\n",
-    "<a href=\"https://colab.research.google.com/github/run-llama/llama_parse/blob/main/examples/demo_json.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
+    "<a href=\"https://colab.research.google.com/github/run-llama/llama_cloud_services/blob/main/examples/parse/demo_json.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
     "\n",
     "This notebook shows you how to use LlamaParse JSON mode with LlamaIndex to build a simple multimodal RAG pipeline.\n",
     "\n",
@@ -35,7 +35,7 @@
     "!pip install llama-index-core\n",
     "!pip install llama-index-llms-anthropic llama-index-multi-modal-llms-anthropic\n",
     "!pip install llama-index-embeddings-huggingface\n",
-    "!pip install llama-parse"
+    "!pip install llama-cloud-services"
    ]
   },
   {
@@ -129,7 +129,7 @@
     }
    ],
    "source": [
-    "from llama_parse import LlamaParse\n",
+    "from llama_cloud_services import LlamaParse\n",
     "\n",
     "parser = LlamaParse(verbose=True)\n",
     "json_objs = parser.get_json_result(\"./uber_10q_march_2022.pdf\")\n",
@@ -342,7 +342,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "llama-parse-aNC435Vv-py3.10",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
   },

examples/parse/demo_json_parsing.ipynb

@@ -7,7 +7,7 @@
    "source": [
     "# LlamaParse JSON Mode + Advanced RAG with `LlamaParseJsonNodeParser`\n",
     "\n",
-    "<a href=\"https://colab.research.google.com/github/run-llama/llama_parse/blob/main/examples/demo_json_parsing.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
+    "<a href=\"https://colab.research.google.com/github/run-llama/llama_cloud_services/blob/main/examples/demo_json_parsing.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
     "\n",
     "This notebook shows you how to use LlamaParse JSON mode with LlamaIndex to build a simple recursive retrieval RAG pipeline using `LlamaParseJsonNodeParser`.\n",
     "\n",
@@ -37,7 +37,7 @@
     "%pip install llama-index-core\n",
     "%pip install llama-index-llms-anthropic llama-index-multi-modal-llms-anthropic\n",
     "%pip install llama-index-embeddings-huggingface\n",
-    "%pip install llama-parse"
+    "%pip install llama-cloud-services"
    ]
   },
   {
@@ -110,7 +110,7 @@
     }
    ],
    "source": [
-    "from llama_parse import LlamaParse\n",
+    "from llama_cloud_services import LlamaParse\n",
     "\n",
     "parser = LlamaParse(verbose=True)\n",
     "json_objs = parser.get_json_result(\"./uber_10q_march_2022.pdf\")\n",

examples/parse/demo_languages.ipynb

@@ -9,7 +9,7 @@
     "\n",
     "LlamaParse supports users to specify a `language` parameter before uploading documents, giving users better OCR capabilities over non-English PDFs, parsing images into more accurate representations.\n",
     "\n",
-    "You can specify 80+ different languages: see this file for a full list of supported languages: https://github.com/run-llama/llama_parse/blob/main/llama_parse/base.py.\n",
+    "You can specify 80+ different languages: see this file for a full list of supported languages: https://github.com/run-llama/llama_cloud_services/blob/main/llama_parse/base.py.\n",
     "\n",
     "This notebook shows a demo of this in action. "
    ]
@@ -77,7 +77,7 @@
     }
    ],
    "source": [
-    "from llama_parse import LlamaParse\n",
+    "from llama_cloud_services import LlamaParse\n",
     "\n",
     "parser = LlamaParse(result_type=\"text\", language=\"fr\")\n",
     "documents = parser.load_data(\"./treasury_report.pdf\")"
@@ -250,7 +250,7 @@
     }
    ],
    "source": [
-    "from llama_parse import LlamaParse\n",
+    "from llama_cloud_services import LlamaParse\n",
     "\n",
     "parser = LlamaParse(result_type=\"text\", language=\"ch_sim\")\n",
     "documents = parser.load_data(\"./chinese_pdf.pdf\")"
@@ -404,7 +404,7 @@
     }
    ],
    "source": [
-    "from llama_parse import LlamaParse\n",
+    "from llama_cloud_services import LlamaParse\n",
     "\n",
     "base_parser = LlamaParse(result_type=\"text\", language=\"en\")\n",
     "base_documents = parser.load_data(\"./chinese_pdf2.pdf\")"

examples/parse/demo_mongodb.ipynb

@@ -7,7 +7,7 @@
    "source": [
     "# LlamaParse With MongoDB\n",
     "\n",
-    "<a href=\"https://colab.research.google.com/github/run-llama/llama_parse/blob/main/examples/demo_mongodb.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
+    "<a href=\"https://colab.research.google.com/github/run-llama/llama_cloud_services/blob/main/examples/parse/demo_mongodb.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
     "\n",
     "In this notebook, we provide a straightforward example of using LlamaParse with MongoDB Atlas VectorSearch.\n",
     "\n",
@@ -69,7 +69,7 @@
     "import pymongo\n",
     "\n",
     "from llama_index.vector_stores.mongodb import MongoDBAtlasVectorSearch\n",
-    "from llama_parse import LlamaParse\n",
+    "from llama_cloud_services import LlamaParse\n",
     "from llama_index.embeddings.openai import OpenAIEmbedding\n",
     "from llama_index.core import VectorStoreIndex, StorageContext\n",
     "from llama_index.core.node_parser import SimpleNodeParser"

examples/parse/demo_parsing_instructions.ipynb

@@ -114,7 +114,7 @@
     }
    ],
    "source": [
-    "%pip install llama-parse"
+    "%pip install llama-cloud-services"
    ]
   },
   {
@@ -169,7 +169,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from llama_parse import LlamaParse"
+    "from llama_cloud_services import LlamaParse"
    ]
   },
   {

examples/parse/demo_starter_multimodal.ipynb

@@ -0,0 +1,357 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "id": "97c79c38-38a3-40f3-ba2e-250649347d63",
+   "metadata": {},
+   "source": [
+    "<a href=\"https://colab.research.google.com/github/run-llama/llama_cloud_services/blob/main/examples/parse/demo_starter_multimodal.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "4e081457",
+   "metadata": {},
+   "source": [
+    "# Multimodal Parsing using LlamaParse\n",
+    "\n",
+    "This cookbook shows you how to use LlamaParse to parse any document with the multimodal capabilities of Multi-Modal LLMs from Anthropic/ OpenAI.\n",
+    "\n",
+    "LlamaParse allows you to plug in external, multimodal model vendors for parsing - we handle the error correction, validation, and scalability/reliability for you.\n"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "qOdqBxCS51Ow",
+   "metadata": {},
+   "source": [
+    "### Installation"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "H_Vqcylb50vm",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install llama-cloud-services"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "15e60ecf-519c-41fc-911b-765adaf8bad4",
+   "metadata": {},
+   "source": [
+    "### Setup\n",
+    "\n",
+    "Here we setup `LLAMA_CLOUD_API_KEY` for using `LlamaParse`."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "91a9e532-1454-40e0-bbf0-fd442c350121",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import nest_asyncio\n",
+    "\n",
+    "nest_asyncio.apply()\n",
+    "\n",
+    "import os\n",
+    "\n",
+    "# API access to llama-cloud\n",
+    "os.environ[\"LLAMA_CLOUD_API_KEY\"] = \"<YOUR LLAMACLOUD API KEY>\""
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "LGwBNPNotZRQ",
+   "metadata": {},
+   "source": [
+    "## Download Data\n",
+    "\n",
+    "For this demonstration, we will use OpenAI's recent paper `Evaluation of OpenAI o1: Opportunities and Challenges of AGI`."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "IjtKDQRLrylI",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "--2024-12-05 18:54:24--  https://arxiv.org/pdf/2409.18486\n",
+      "Resolving arxiv.org (arxiv.org)... 151.101.67.42, 151.101.131.42, 151.101.3.42, ...\n",
+      "Connecting to arxiv.org (arxiv.org)|151.101.67.42|:443... connected.\n",
+      "HTTP request sent, awaiting response... 200 OK\n",
+      "Length: 13986265 (13M) [application/pdf]\n",
+      "Saving to: ‘o1.pdf’\n",
+      "\n",
+      "o1.pdf              100%[===================>]  13.34M  11.8MB/s    in 1.1s    \n",
+      "\n",
+      "2024-12-05 18:54:26 (11.8 MB/s) - ‘o1.pdf’ saved [13986265/13986265]\n",
+      "\n"
+     ]
+    }
+   ],
+   "source": [
+    "!wget \"https://arxiv.org/pdf/2409.18486\" -O \"o1.pdf\""
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "4e29a9d7-5bd9-4fb8-8ec1-4c128a748662",
+   "metadata": {},
+   "source": [
+    "## Initialize LlamaParse\n",
+    "\n",
+    "Initialize LlamaParse in multimodal mode, and specify the vendor.\n",
+    "\n",
+    "**NOTE**: optionally you can specify the Anthropic/ OpenAI API key. If you choose to do so LlamaParse will only charge you 1 credit (0.3c) per page. \n",
+    "\n",
+    "\n",
+    "Using your own API key may incur additional costs from your model provider and could result in failed pages or documents if you do not have sufficient usage limits."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "dc921729-3446-42ca-8e1b-a6fd26195ed9",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from llama_index.core.schema import TextNode\n",
+    "from typing import List\n",
+    "\n",
+    "\n",
+    "def get_text_nodes(json_list: List[dict]):\n",
+    "    text_nodes = []\n",
+    "    for idx, page in enumerate(json_list):\n",
+    "        text_node = TextNode(text=page[\"md\"], metadata={\"page\": page[\"page\"]})\n",
+    "        text_nodes.append(text_node)\n",
+    "    return text_nodes"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "1b5d6da6",
+   "metadata": {},
+   "source": [
+    "### With anthropic-sonnet-3.5"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "f2e9d9cf-8189-4fcb-b34f-cde6cc0b59c8",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Started parsing the file under job_id dd9d5e0f-160e-486a-89a2-6005e5a1c2ac\n"
+     ]
+    }
+   ],
+   "source": [
+    "from llama_cloud_services import LlamaParse\n",
+    "\n",
+    "parser = LlamaParse(\n",
+    "    result_type=\"markdown\",\n",
+    "    use_vendor_multimodal_model=True,\n",
+    "    vendor_multimodal_model_name=\"anthropic-sonnet-3.5\",\n",
+    "    target_pages=\"24\"\n",
+    "    # invalidate_cache=True\n",
+    ")\n",
+    "json_objs = parser.get_json_result(\"o1.pdf\")\n",
+    "json_list = json_objs[0][\"pages\"]\n",
+    "docs = get_text_nodes(json_list)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "4f3c51b0-7878-48d7-9bc3-02b516500128",
+   "metadata": {},
+   "source": [
+    "### With GPT-4o\n",
+    "\n",
+    "For comparison, we will also parse the document using GPT-4o."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "6fc3f258-50ae-4988-b904-c105463a498f",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Started parsing the file under job_id 6a4dea44-4f90-406b-b290-9e98620b1232\n"
+     ]
+    }
+   ],
+   "source": [
+    "from llama_cloud_services import LlamaParse\n",
+    "\n",
+    "parser_gpt4o = LlamaParse(\n",
+    "    result_type=\"markdown\",\n",
+    "    use_vendor_multimodal_model=True,\n",
+    "    vendor_multimodal_model=\"openai-gpt4o\",\n",
+    "    target_pages=\"24\",\n",
+    "    # invalidate_cache=True\n",
+    ")\n",
+    "json_objs_gpt4o = parser_gpt4o.get_json_result(\"o1.pdf\")\n",
+    "json_list_gpt4o = json_objs_gpt4o[0][\"pages\"]\n",
+    "docs_gpt4o = get_text_nodes(json_list_gpt4o)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "44c20f7a-2901-4dd0-b635-a4b33c5664c1",
+   "metadata": {},
+   "source": [
+    "### View Results\n",
+    "\n",
+    "Let's visualize the results along with the original document page."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "778698aa-da7e-4081-b3b5-0372f228536f",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "page: 25\n",
+      "\n",
+      "| Participant_ID | clinical Description Reference |\n",
+      "|-----------------|----------------------------------|\n",
+      "| Attribute | Value | Basic Personal Information: Subject 098_S_0896 is a 72.0-year-old Female who has completed 15 years of education. The ethnicity is Not Hisp/Latino and race is White. Marital status is Married. Initially diagnosed as AD, as of the date 2007-10-24, the final diagnosis was Dementia. |\n",
+      "| Age | 72.0 |\n",
+      "| Sex | Female |\n",
+      "| Education | 15 |\n",
+      "| Race | White | Biomarker Measurements: The subject's genetic profile includes an ApoE4 status of 0.0... |\n",
+      "| DX_bl | AD |\n",
+      "| DX | Dementia |\n",
+      "| ... | ... | Cognitive and Neurofunctional Assessments: The Mini-Mental State Examination score stands at 29.0. The Clinical Dementia Rating, sum of boxes, is 1.0. ADAS 11 and 13 scores are 4.67 and 4.67 respectively, with a score of 1.0 in delayed word recall... |\n",
+      "| APOE4 | 1.0 |\n",
+      "| TAU | 212.5 |\n",
+      "| ... | ... |\n",
+      "| MMSE | 29.0 | Volumetric Data: Under MRI conditions at a field strength of 1.5 Tesla MRI Tesla, using Cross Sectional FreeSurfer (FreeSurfer Version 4.3), the imaging data recorded includes ventricles volume at 54422.0, hippocampus volume at 6677.0, whole brain volume at 1147980.0, entorhinal cortex volume at 2782.0, fusiform gyrus volume at 19432.0, and middle temporal area volume at 24951.0. The intracranial volume measured is 1799580.0.... |\n",
+      "| CDRSB | 0.0 |\n",
+      "| ... | ... |\n",
+      "| FLDSTRENG | 1.5 Tesla MRI |\n",
+      "| Ventricles | 84599 |\n",
+      "| Hippocampus | 5319 |\n",
+      "| ... | ... |\n",
+      "\n",
+      "Figure 2: An example of a patient table and its corresponding clinical description.\n",
+      "\n",
+      "skills. Mathematics, as a highly structured and logic-driven discipline, provides an ideal testing ground for evaluating this reasoning ability. To investigate o1-preview's performance, we designed a series of tests covering various difficulty levels. We begin with high school-level math competition problems in this section, followed by college-level mathematics problems in the next section, allowing us to observe the model's logical reasoning across varying levels of complexity.\n",
+      "\n",
+      "In this section, we selected two primary areas of mathematics: algebra and counting and probability in this section. We chose these two topics because of their heavy reliance on problem-solving skills and their frequent use in assessing logical and abstract thinking [46]. The dataset used in testing is from the MATH dataset [46]. The problems in the dataset cover a wide range of subjects, including Prealgebra, Intermediate Algebra, Algebra, Geometry, Counting and Probability, Number Theory, and Precalculus. Each problem is categorized based on difficulty, ranked from level 1 to 5, according to the Art of Problem Solving (AoPS). The dataset mainly comprises problems from various high school math competitions, including the American Mathematics Competitions (AMC) 10 and 12, as well as the American Invitational Mathematics Examination (AIME), and other similar contests. Each problem comes with detailed reference solutions, allowing for a comprehensive comparison of o1-preview's solutions.\n",
+      "\n",
+      "In addition to evaluating the final answers produced by o1-preview, our analysis delves into the step-by-step reasoning process of the o1-preview's solutions. By comparing o1-preview's solutions with the dataset's solutions, we assess its ability to engage in logical reasoning, handle abstract problem-solving tasks, and apply structured approaches to reach correct answers. This deeper analysis offers insights into o1-preview's overall reasoning capabilities, using mathematics as a reliable indicator for logical and structured thought processes.\n"
+     ]
+    }
+   ],
+   "source": [
+    "# using Sonnet-3.5\n",
+    "print(docs[0].get_content(metadata_mode=\"all\"))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "1511a30f-3efc-4142-9668-7dc056a24d0c",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "page: 25\n",
+      "\n",
+      "\n",
+      "| Participant_ID | clinical Description Reference |\n",
+      "|----------------|--------------------------------|\n",
+      "| **Attribute**  | **Value**                      |\n",
+      "| Age            | 72.0                           |\n",
+      "| Sex            | Female                         |\n",
+      "| Education      | 15                             |\n",
+      "| Race           | White                          |\n",
+      "| DX_bl          | AD                             |\n",
+      "| DX             | Dementia                       |\n",
+      "| ...            | ...                            |\n",
+      "| APOE4          | 1.0                            |\n",
+      "| TAU            | 212.5                          |\n",
+      "| ...            | ...                            |\n",
+      "| MMSE           | 29.0                           |\n",
+      "| CDRSB          | 0.0                            |\n",
+      "| ...            | ...                            |\n",
+      "| FLDSTRENG      | 1.5 Tesla MRI                  |\n",
+      "| Ventricles     | 84599                          |\n",
+      "| Hippocampus    | 5319                           |\n",
+      "| ...            | ...                            |\n",
+      "\n",
+      "**Basic Personal Information:** Subject 098_S_0896 is a 72.0-year-old Female who has completed 15 years of education. The ethnicity is Not Hisp/Latino and race is White. Marital status is Married. Initially diagnosed as AD, as of the date 2007-10-24, the final diagnosis was Dementia.\n",
+      "\n",
+      "**Biomarker Measurements:** The subject's genetic profile includes an ApoE4 status of 0.0...\n",
+      "\n",
+      "**Cognitive and Neurofunctional Assessments:** The Mini-Mental State Examination score stands at 29.0. The Clinical Dementia Rating, sum of boxes, is 1.0. ADAS 11 and 13 scores are 4.67 and 4.67 respectively, with a score of 1.0 in delayed word recall...\n",
+      "\n",
+      "**Volumetric Data:** Under MRI conditions at a field strength of 1.5 Tesla MRI Tesla, using Cross-Sectional FreeSurfer (FreeSurfer Version 4.3), the imaging data recorded includes ventricles volume at 84422.0, hippocampus volume at 6677.0, whole brain volume at 1147980.0, entorhinal cortex volume at 27820.0, fusiform gyrus volume at 19432.0, and middle temporal area volume at 24951.0. The intracranial volume measured is 1799580.0...\n",
+      "\n",
+      "Figure 2: An example of a patient table and its corresponding clinical description.\n",
+      "\n",
+      "----\n",
+      "\n",
+      "Skills. Mathematics, as a highly structured and logic-driven discipline, provides an ideal testing ground for evaluating this reasoning ability. To investigate o1-preview’s performance, we designed a series of tests covering various difficulty levels. We begin with high school-level math competition problems in this section, followed by college-level mathematics problems in the next section, allowing us to observe the model’s logical reasoning across varying levels of complexity.\n",
+      "\n",
+      "In this section, we selected two primary areas of mathematics: algebra and counting and probability in this section. We chose these two topics because of their heavy reliance on problem-solving skills and their frequent use in assessing logical and abstract thinking [46]. The dataset used in testing is from the MATH dataset [46]. The problems in the dataset cover a wide range of subjects, including Prealgebra, Intermediate Algebra, Algebra, Geometry, Counting and Probability, Number Theory, and Precalculus. Each problem is categorized based on difficulty, ranked from level 1 to 5, according to the Art of Problem Solving (AoPS). The dataset mainly comprises problems from various high school math competitions, including the American Mathematics Competitions (AMC) 10 and 12, as well as the American Invitational Mathematics Examination (AIME), and other similar contests. Each problem comes with detailed reference solutions, allowing for a comprehensive comparison of o1-preview’s solutions.\n",
+      "\n",
+      "In addition to evaluating the final answers produced by o1-preview, our analysis delves into the step-by-step reasoning process of the o1-preview’s solutions. By comparing o1-preview’s solutions with the dataset’s solutions, we assess its ability to engage in logical reasoning, handle abstract problem-solving tasks, and apply structured approaches to reach correct answers. This deeper analysis offers insights into o1-preview’s overall reasoning capabilities, using mathematics as a reliable indicator for logical and structured thought processes.\n"
+     ]
+    }
+   ],
+   "source": [
+    "# using GPT-4o\n",
+    "print(docs_gpt4o[0].get_content(metadata_mode=\"all\"))"
+   ]
+  }
+ ],
+ "metadata": {
+  "colab": {
+   "provenance": []
+  },
+  "kernelspec": {
+   "display_name": "llamacloud",
+   "language": "python",
+   "name": "llamacloud"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

examples/parse/demo_starter_parse_selected_pages.ipynb

@@ -0,0 +1,170 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<a href=\"https://colab.research.google.com/github/run-llama/llama_cloud_services/blob/main/examples/parse/demo_starter_parse_selected_pages.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "# Parse Selected Pages \n",
+    "\n",
+    "In this notebook we will demonstrate how to parse selected pages in a document using LlamaParse."
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Installation\n",
+    "\n",
+    "Here we install `llama-parse` used for parsing the document"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install llama-cloud-services"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Set API Key"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# llama-parse is async-first, running the async code in a notebook requires the use of nest_asyncio\n",
+    "import nest_asyncio\n",
+    "\n",
+    "nest_asyncio.apply()\n",
+    "\n",
+    "import os\n",
+    "\n",
+    "# API access to llama-cloud\n",
+    "os.environ[\"LLAMA_CLOUD_API_KEY\"] = \"<YOUR LLAMACLOUD API KEY>\""
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Download Data\n",
+    "\n",
+    "Here we download Uber 2021 10K SEC filings data for the demonstration."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "--2024-12-05 11:40:59--  https://raw.githubusercontent.com/run-llama/llama_index/main/docs/docs/examples/data/10k/uber_2021.pdf\n",
+      "Resolving raw.githubusercontent.com (raw.githubusercontent.com)... 2606:50c0:8000::154, 2606:50c0:8002::154, 2606:50c0:8003::154, ...\n",
+      "Connecting to raw.githubusercontent.com (raw.githubusercontent.com)|2606:50c0:8000::154|:443... connected.\n",
+      "HTTP request sent, awaiting response... 200 OK\n",
+      "Length: 1880483 (1.8M) [application/octet-stream]\n",
+      "Saving to: ‘./uber_2021.pdf’\n",
+      "\n",
+      "./uber_2021.pdf     100%[===================>]   1.79M  --.-KB/s    in 0.1s    \n",
+      "\n",
+      "2024-12-05 11:40:59 (14.2 MB/s) - ‘./uber_2021.pdf’ saved [1880483/1880483]\n",
+      "\n"
+     ]
+    }
+   ],
+   "source": [
+    "!wget 'https://raw.githubusercontent.com/run-llama/llama_index/main/docs/docs/examples/data/10k/uber_2021.pdf' -O './uber_2021.pdf'"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Parse the PDF file in selected pages\n",
+    "\n",
+    "Here we will parse the PDF file in selected pages and get the text in `markdown` format."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Started parsing the file under job_id ad1087c1-b085-4dc7-9aa8-d13cdd440f2b\n"
+     ]
+    }
+   ],
+   "source": [
+    "from llama_cloud_services import LlamaParse\n",
+    "\n",
+    "parser = LlamaParse(target_pages=\"0,1,2\", result_type=\"markdown\")\n",
+    "\n",
+    "documents = parser.load_data(\"./uber_2021.pdf\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "[Document(id_='d0b34f4a-27ef-48e2-a92a-386e5e265f4c', embedding=None, metadata={}, excluded_embed_metadata_keys=[], excluded_llm_metadata_keys=[], relationships={}, metadata_template='{key}: {value}', metadata_separator='\\n', text='# UNITED STATES SECURITIES AND EXCHANGE COMMISSION\\n\\n# Washington, D.C. 20549\\n\\n# FORM 10-K\\n\\n(Mark One)\\n\\n☒ ANNUAL REPORT PURSUANT TO SECTION 13 OR 15(d) OF THE SECURITIES EXCHANGE ACT OF 1934\\n\\nFor the fiscal year ended December 31, 2021\\n\\nOR\\n\\n☐ TRANSITION REPORT PURSUANT TO SECTION 13 OR 15(d) OF THE SECURITIES EXCHANGE ACT OF 1934\\n\\nFor the transition period from _____ to _____\\n\\nCommission File Number: 001-38902\\n\\n# UBER TECHNOLOGIES, INC.\\n\\n(Exact name of registrant as specified in its charter)\\n\\nDelaware\\n\\n45-2647441\\n\\n(State or other jurisdiction of incorporation or organization) (I.R.S. Employer Identification No.)\\n\\n1515 3rd Street\\n\\nSan Francisco, California 94158\\n\\n(Address of principal executive offices, including zip code)\\n\\n(415) 612-8582\\n\\n(Registrant’s telephone number, including area code)\\n\\n# Securities registered pursuant to Section 12(b) of the Act:\\n\\n|Title of each class|Trading Symbol(s)|Name of each exchange on which registered|\\n|---|---|---|\\n|Common Stock, par value $0.00001 per share|UBER|New York Stock Exchange|\\n\\nSecurities registered pursuant to Section 12(g) of the Act: None\\n\\nIndicate by check mark whether the registrant is a well-known seasoned issuer, as defined in Rule 405 of the Securities Act. Yes ☒ No ☐\\n\\nIndicate by check mark whether the registrant is not required to file reports pursuant to Section 13 or Section 15(d) of the Act. Yes ☐ No ☒\\n\\nIndicate by check mark whether the registrant (1) has filed all reports required to be filed by Section 13 or 15(d) of the Securities Exchange Act of 1934 during the preceding 12 months (or for such shorter period that the registrant was required to file such reports), and (2) has been subject to such filing requirements for the past 90 days. Yes ☒ No ☐\\n\\nIndicate by check mark whether the registrant has submitted electronically every Interactive Data File required to be submitted pursuant to Rule 405 of Regulation S-T (§232.405 of this chapter) during the preceding 12 months (or for such shorter period that the registrant was required to submit such files). Yes ☒ No ☐\\n\\nIndicate by check mark whether the registrant is a large accelerated filer, an accelerated filer, a non-accelerated filer, a smaller reporting company, or an emerging growth company. See the definitions of “large accelerated filer,” “accelerated filer,” “smaller reporting company,” and “emerging growth company” in Rule 12b-2 of the Exchange Act.', mimetype='text/plain', start_char_idx=None, end_char_idx=None, metadata_seperator='\\n', text_template='{metadata_str}\\n\\n{content}'),\n",
+       " Document(id_='253b1141-a260-466e-b164-b39df67ef799', embedding=None, metadata={}, excluded_embed_metadata_keys=[], excluded_llm_metadata_keys=[], relationships={}, metadata_template='{key}: {value}', metadata_separator='\\n', text=\"# Large accelerated filer\\n\\n☒\\n\\n# Accelerated filer\\n\\n☐\\n\\n# Non-accelerated filer\\n\\n☐\\n\\n# Smaller reporting company\\n\\n☐\\n\\n# Emerging growth company\\n\\n☐\\n\\nIf an emerging growth company, indicate by check mark if the registrant has elected not to use the extended transition period for complying with any new or revised financial accounting standards provided pursuant to Section 13(a) of the Exchange Act.\\n\\n☐\\n\\nIndicate by check mark whether the registrant has filed a report on and attestation to its management’s assessment of the effectiveness of its internal control over financial reporting under Section 404(b) of the Sarbanes-Oxley Act (15 U.S.C. 7262(b)) by the registered public accounting firm that prepared or issued\\n\\n☒\\n\\nIndicate by check mark whether the registrant is a shell company (as defined in Rule 12b-2 of the Exchange Act). Yes\\n\\n☐\\n\\nNo\\n\\n☒\\n\\nThe aggregate market value of the voting and non-voting common equity held by non-affiliates of the registrant as of June 30, 2021, the last business day of the registrant's most recently completed second fiscal quarter, was approximately $90.5 billion based upon the closing price reported for such date on the New York Stock Exchange.\\n\\nThe number of shares of the registrant's common stock outstanding as of February 22, 2022 was 1,954,464,088.\\n\\n# DOCUMENTS INCORPORATED BY REFERENCE\\n\\nPortions of the registrant’s Definitive Proxy Statement relating to the Annual Meeting of Stockholders are incorporated by reference into Part III of this Annual Report on Form 10-K where indicated. Such Definitive Proxy Statement will be filed with the Securities and Exchange Commission within 120 days after the end of the registrant’s fiscal year ended December 31, 2021.\", mimetype='text/plain', start_char_idx=None, end_char_idx=None, metadata_seperator='\\n', text_template='{metadata_str}\\n\\n{content}'),\n",
+       " Document(id_='ad988239-3ab5-498d-85ba-a29241db24d4', embedding=None, metadata={}, excluded_embed_metadata_keys=[], excluded_llm_metadata_keys=[], relationships={}, metadata_template='{key}: {value}', metadata_separator='\\n', text='# UBER TECHNOLOGIES, INC.\\n\\n# TABLE OF CONTENTS\\n\\n|Special Note Regarding Forward-Looking Statements|2|\\n|---|---|\\n|PART I|PART I|\\n|Item 1. Business|4|\\n|Item 1A. Risk Factors|11|\\n|Item 1B. Unresolved Staff Comments|46|\\n|Item 2. Properties|46|\\n|Item 3. Legal Proceedings|46|\\n|Item 4. Mine Safety Disclosures|47|\\n|PART II|PART II|\\n|Item 5. Market for Registrant’s Common Equity, Related Stockholder Matters and Issuer Purchases of Equity Securities|47|\\n|Item 6. [Reserved]|48|\\n|Item 7. Management’s Discussion and Analysis of Financial Condition and Results of Operations|48|\\n|Item 7A. Quantitative and Qualitative Disclosures About Market Risk|69|\\n|Item 8. Financial Statements and Supplementary Data|70|\\n|Item 9. Changes in and Disagreements with Accountants on Accounting and Financial Disclosure|146|\\n|Item 9A. Controls and Procedures|147|\\n|Item 9B. Other Information|147|\\n|Item 9C. Disclosure Regarding Foreign Jurisdictions that Prevent Inspections|147|\\n|PART III|PART III|\\n|Item 10. Directors, Executive Officers and Corporate Governance|147|\\n|Item 11. Executive Compensation|147|\\n|Item 12. Security Ownership of Certain Beneficial Owners and Management and Related Stockholder Matters|148|\\n|Item 13. Certain Relationships and Related Transactions, and Director Independence|148|\\n|Item 14. Principal Accounting Fees and Services|148|\\n|PART IV|PART IV|\\n|Item 15. Exhibits, Financial Statement Schedules|148|\\n|Item 16. Form 10-K Summary|148|\\n|Exhibit Index|149|\\n|Signatures|152|', mimetype='text/plain', start_char_idx=None, end_char_idx=None, metadata_seperator='\\n', text_template='{metadata_str}\\n\\n{content}')]"
+      ]
+     },
+     "execution_count": null,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "documents"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "llamacloud",
+   "language": "python",
+   "name": "llamacloud"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}

examples/parse/demo_table_comparisons.ipynb

@@ -6,7 +6,7 @@
    "source": [
     "# RAG for Table Comparisons with LlamaParse + LlamaIndex\n",
     "\n",
-    "<a href=\"https://colab.research.google.com/github/run-llama/llama_parse/blob/main/examples/demo_table_comparisons.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
+    "<a href=\"https://colab.research.google.com/github/run-llama/llama_cloud_services/blob/main/examples/parse/demo_table_comparisons.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
     "\n",
     "This notebook shows you how to do comparisons across both tabular and text data across multiple PDF documents.\n",
     "\n",
@@ -34,7 +34,7 @@
     "%pip install llama-index-question-gen-openai\n",
     "%pip install llama-index-postprocessor-flag-embedding-reranker\n",
     "%pip install git+https://github.com/FlagOpen/FlagEmbedding.git\n",
-    "%pip install llama-parse"
+    "%pip install llama-cloud-services"
    ]
   },
   {
@@ -109,7 +109,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from llama_parse import LlamaParse\n",
+    "from llama_cloud_services import LlamaParse\n",
     "\n",
     "docs_2021 = LlamaParse(result_type=\"markdown\").load_data(\"./apple_2021_10k.pdf\")\n",
     "docs_2020 = LlamaParse(result_type=\"markdown\").load_data(\"./apple_2020_10k.pdf\")"

examples/parse/excel/dcf_rag.ipynb

@@ -7,7 +7,7 @@
    "source": [
     "# RAG with Excel Spreadsheet using LlamaPrase\n",
     "\n",
-    "<a href=\"https://colab.research.google.com/github/run-llama/llama_parse/blob/main/examples/demo_excel.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
+    "<a href=\"https://colab.research.google.com/github/run-llama/llama_cloud_services/blob/main/examples/parse/excel/dcf_rag.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
     "\n",
     "This notebook constructs a RAG pipeline over a simple DCF template [here](https://eqvista.com/app/uploads/2020/09/Eqvista_DCF-Excel-Template.xlsx).\n",
     "\n"
@@ -31,7 +31,7 @@
    "outputs": [],
    "source": [
     "%pip install llama-index\n",
-    "%pip install llama-parse"
+    "%pip install llama-cloud-services"
    ]
   },
   {
@@ -53,7 +53,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from llama_parse import LlamaParse\n",
+    "from llama_cloud_services import LlamaParse\n",
     "\n",
     "# api_key = \"llx-\"  # get from cloud.llamaindex.ai"
    ]

examples/parse/knowledge_graphs/kg_agent.ipynb

@@ -7,7 +7,7 @@
    "source": [
     "# Knowledge Graph Agent with LlamaParse\n",
     "\n",
-    "<a href=\"https://colab.research.google.com/github/run-llama/llama_parse/blob/main/examples/knowledge_graphs/kg_agent.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
+    "<a href=\"https://colab.research.google.com/github/run-llama/llama_cloud_services/blob/main/examples/parse/knowledge_graphs/kg_agent.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
     "\n",
     "Here we build a knowledge graph agent over the SF 2023 Budget Proposal. We use LlamaIndex abstractions to construct a knowledge graph, and we store the property graph in neo4j. We then build an agent that can interact with the knowledge graph as a tool."
    ]
@@ -33,7 +33,7 @@
     "!pip install llama-index-postprocessor-flag-embedding-reranker\n",
     "!pip install git+https://github.com/FlagOpen/FlagEmbedding.git\n",
     "!pip install llama-index-graph-stores-neo4j\n",
-    "!pip install llama-parse"
+    "!pip install llama-cloud-services"
    ]
   },
   {
@@ -125,7 +125,7 @@
     }
    ],
    "source": [
-    "from llama_parse import LlamaParse\n",
+    "from llama_cloud_services import LlamaParse\n",
     "\n",
     "docs = LlamaParse(result_type=\"text\").load_data(\"./data/budget_2023.pdf\")"
    ]

examples/parse/multimodal/claude_parse.ipynb

@@ -0,0 +1,635 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "id": "97c79c38-38a3-40f3-ba2e-250649347d63",
+   "metadata": {},
+   "source": [
+    "# Multimodal Parsing using Anthropic Claude (Sonnet 3.5)\n",
+    "\n",
+    "<a href=\"https://colab.research.google.com/github/run-llama/llama_cloud_services/blob/main/examples/parse/multimodal/claude_parse.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
+    "\n",
+    "This cookbook shows you how to use LlamaParse to parse any document with the multimodal capabilities of Sonnet 3.5. \n",
+    "\n",
+    "LlamaParse allows you to plug in external, multimodal model vendors for parsing - we handle the error correction, validation, and scalability/reliability for you.\n"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "15e60ecf-519c-41fc-911b-765adaf8bad4",
+   "metadata": {},
+   "source": [
+    "## Setup\n",
+    "\n",
+    "Download the data. Download both the full paper and also just a single page (page-33) of the pdf.\n",
+    "\n",
+    "Swap in `data/llama2-p33.pdf` for `data/llama2.pdf` in the code blocks below if you want to save on parsing tokens. \n",
+    "\n",
+    "An image of this page is shown below."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "91a9e532-1454-40e0-bbf0-fd442c350121",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import nest_asyncio\n",
+    "\n",
+    "nest_asyncio.apply()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "0d9fb0aa-74cd-476f-8161-efd9e04248bf",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "--2024-07-11 23:44:38--  https://arxiv.org/pdf/2307.09288\n",
+      "Resolving arxiv.org (arxiv.org)... 151.101.195.42, 151.101.131.42, 151.101.3.42, ...\n",
+      "Connecting to arxiv.org (arxiv.org)|151.101.195.42|:443... connected.\n",
+      "HTTP request sent, awaiting response... 200 OK\n",
+      "Length: 13661300 (13M) [application/pdf]\n",
+      "Saving to: ‘data/llama2.pdf’\n",
+      "\n",
+      "data/llama2.pdf     100%[===================>]  13.03M  69.3MB/s    in 0.2s    \n",
+      "\n",
+      "2024-07-11 23:44:38 (69.3 MB/s) - ‘data/llama2.pdf’ saved [13661300/13661300]\n",
+      "\n"
+     ]
+    }
+   ],
+   "source": [
+    "!wget \"https://arxiv.org/pdf/2307.09288\" -O data/llama2.pdf\n",
+    "!wget \"https://www.dropbox.com/scl/fi/wpql661uu98vf6e2of2i0/llama2-p33.pdf?rlkey=64weubzkwpmf73y58vbmc8pyi&st=khgx5161&dl=1\" -O data/llama2-p33.pdf"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "b5c214a2-56fd-4b09-93b3-be994a3b5aa4",
+   "metadata": {},
+   "source": [
+    "![page_33](llama2-p33.png)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "4e29a9d7-5bd9-4fb8-8ec1-4c128a748662",
+   "metadata": {},
+   "source": [
+    "## Initialize LlamaParse\n",
+    "\n",
+    "Initialize LlamaParse in multimodal mode, and specify the vendor.\n",
+    "\n",
+    "**NOTE**: optionally you can specify the Anthropic API key. If you do so you will be charged our base LlamaParse price of 0.3c per page. If you don't then you will be charged 6c per page, as we will make the calls to Claude for you."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "dc921729-3446-42ca-8e1b-a6fd26195ed9",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from llama_index.core.schema import TextNode\n",
+    "from typing import List\n",
+    "import json\n",
+    "\n",
+    "\n",
+    "def get_text_nodes(json_list: List[dict]):\n",
+    "    text_nodes = []\n",
+    "    for idx, page in enumerate(json_list):\n",
+    "        text_node = TextNode(text=page[\"md\"], metadata={\"page\": page[\"page\"]})\n",
+    "        text_nodes.append(text_node)\n",
+    "    return text_nodes\n",
+    "\n",
+    "\n",
+    "def save_jsonl(data_list, filename):\n",
+    "    \"\"\"Save a list of dictionaries as JSON Lines.\"\"\"\n",
+    "    with open(filename, \"w\") as file:\n",
+    "        for item in data_list:\n",
+    "            json.dump(item, file)\n",
+    "            file.write(\"\\n\")\n",
+    "\n",
+    "\n",
+    "def load_jsonl(filename):\n",
+    "    \"\"\"Load a list of dictionaries from JSON Lines.\"\"\"\n",
+    "    data_list = []\n",
+    "    with open(filename, \"r\") as file:\n",
+    "        for line in file:\n",
+    "            data_list.append(json.loads(line))\n",
+    "    return data_list"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "f2e9d9cf-8189-4fcb-b34f-cde6cc0b59c8",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Started parsing the file under job_id 811a29d8-8bcd-4100-bee3-6a83fbde1697\n"
+     ]
+    }
+   ],
+   "source": [
+    "from llama_cloud_services import LlamaParse\n",
+    "\n",
+    "parser = LlamaParse(\n",
+    "    result_type=\"markdown\",\n",
+    "    use_vendor_multimodal_model=True,\n",
+    "    vendor_multimodal_model_name=\"anthropic-sonnet-3.5\",\n",
+    "    # invalidate_cache=True\n",
+    ")\n",
+    "json_objs = parser.get_json_result(\"./data/llama2.pdf\")\n",
+    "# json_objs = parser.get_json_result(\"./data/llama2-p33.pdf\")\n",
+    "json_list = json_objs[0][\"pages\"]\n",
+    "docs = get_text_nodes(json_list)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "96a81df0-1026-4e30-a930-f677dc31e344",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Optional: Save\n",
+    "save_jsonl([d.dict() for d in docs], \"docs.jsonl\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "ee2e6920-8893-4b39-ae12-94d13c651406",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Optional: Load\n",
+    "from llama_index.core import Document\n",
+    "\n",
+    "docs_dicts = load_jsonl(\"docs.jsonl\")\n",
+    "docs = [Document.parse_obj(d) for d in docs_dicts]"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "4f3c51b0-7878-48d7-9bc3-02b516500128",
+   "metadata": {},
+   "source": [
+    "### Setup GPT-4o baseline\n",
+    "\n",
+    "For comparison, we will also parse the document using GPT-4o (3c per page)."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "6fc3f258-50ae-4988-b904-c105463a498f",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Started parsing the file under job_id 04c69ecc-e45d-4ad9-ba72-3045af38268b\n"
+     ]
+    }
+   ],
+   "source": [
+    "from llama_cloud_services import LlamaParse\n",
+    "\n",
+    "parser_gpt4o = LlamaParse(\n",
+    "    result_type=\"markdown\",\n",
+    "    use_vendor_multimodal_model=True,\n",
+    "    vendor_multimodal_model=\"openai-gpt4o\",\n",
+    "    # invalidate_cache=True\n",
+    ")\n",
+    "json_objs_gpt4o = parser_gpt4o.get_json_result(\"./data/llama2.pdf\")\n",
+    "# json_objs_gpt4o = parser.get_json_result(\"./data/llama2-p33.pdf\")\n",
+    "json_list_gpt4o = json_objs_gpt4o[0][\"pages\"]\n",
+    "docs_gpt4o = get_text_nodes(json_list_gpt4o)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "6a47f04e-12e1-4c80-a71d-ef7721f96401",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Optional: Save\n",
+    "save_jsonl([d.dict() for d in docs_gpt4o], \"docs_gpt4o.jsonl\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "c38b5ca3-fa87-434b-b477-bf6a4962eb3d",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Optional: Load\n",
+    "from llama_index.core import Document\n",
+    "\n",
+    "docs_gpt4o_dicts = load_jsonl(\"docs_gpt4o.jsonl\")\n",
+    "docs_gpt4o = [Document.parse_obj(d) for d in docs_gpt4o_dicts]"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "44c20f7a-2901-4dd0-b635-a4b33c5664c1",
+   "metadata": {},
+   "source": [
+    "## View Results\n",
+    "\n",
+    "Let's visualize the results along with the original document page.\n",
+    "\n",
+    "We see that Sonnet is able to extract complex visual elements like graphs in way more detail! \n",
+    "\n",
+    "**NOTE**: If you're using llama2-p33, just use `docs[0]`"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "778698aa-da7e-4081-b3b5-0372f228536f",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "page: 33\n",
+      "\n",
+      "| Temperature | RLHF v3 | RLHF v2 | RLHF v1 | SFT |\n",
+      "|-------------|---------|---------|---------|-----|\n",
+      "| 0.4         | 98      | 98      | 97      | 95  |\n",
+      "| 0.6         | 97      | 97      | 95      | 94  |\n",
+      "| 0.8         | 97      | 96      | 94      | 92  |\n",
+      "| 1.0         | 96      | 94      | 92      | 89  |\n",
+      "| 1.2         | 95      | 92      | 88      | 83  |\n",
+      "| 1.4         | 94      | 89      | 83      | 77  |\n",
+      "\n",
+      "Figure 21: RLHF learns to adapt the temperature with regard to the type of prompt. Lower Self-BLEU corresponds to more diversity: RLHF eliminates diversity in responses to factual prompts but retains more diversity when generating responses to creative prompts. We prompt each model with a diverse set of 10 creative and 10 factual instructions and sample 25 responses. This is repeated for the temperatures T ∈ {k/10 | k ∈ N : 1 ≤ k ≤ 15}. For each of the 25 responses we compute the Self-BLEU metric and report the mean and standard deviation against the temperature.\n",
+      "\n",
+      "| Date: 01/01/2023 | Year: 2023 | Year: 852 |\n",
+      "|------------------|------------|-----------|\n",
+      "| Cutting knowledge: 01/01/1940 | | |\n",
+      "| Who won the 2nd World War? | Is the Earth flat or round? | Is the Earth flat or round? |\n",
+      "| I'm not sure what you're referring to, as my knowledge stopped in 1940. | The Earth is round. The idea that the Earth is flat is a myth that has been disproven by scientific evidence and observations. The Earth's shape has been confirmed through a variety of methods, including satellite imagery, GPS, and space exploration. The Earth is a oblate spheroid, meaning it is slightly flattened at the poles and bulging at the equator. | In 852, the Earth was believed to be flat by many people. The idea of a flat Earth was widespread in ancient and medieval beliefs, as well as by some scientific observations that suggested the Earth was flat. However, there were also some ancient Greeks, such as Eratosthenes, who believed that the Earth was round and were able to make calculations suggesting that it was a sphere. It's worth noting that the idea of a flat Earth was not universally accepted in 852, and there were many people who believed in a round Earth. Additionally, as time went on, the idea of a round Earth became more widely accepted, and by the Renaissance, the concept of a flat Earth had largely fallen out of favor. |\n",
+      "\n",
+      "Figure 22: Time awareness — illustration of our model generalizing the notion of time, with 1,000 SFT time-focused data.\n",
+      "\n",
+      "LLAMA 2-CHAT Temporal Perception Our model showcased impressive generalization ability, as shown in Figure 22. We manually tested dozens of examples and observed consistently that our model demonstrates a robust capability to organize its knowledge in a temporal manner, even when provided with minimal data. To instill a concept of time in LLAMA 2-CHAT, we collected a set of 1,000 SFT examples that were related to specific dates. These examples included questions like \"How long ago did Barack Obama become president?\" Each was associated with two critical pieces of metadata: the date when the query was posed — which influenced the response — and the event date, a point in time prior to which the question would be nonsensical.\n",
+      "\n",
+      "The observation suggests that LLMs have internalized the concept of time to a greater extent than previously assumed, despite their training being solely based on next-token prediction and data that is randomly shuffled without regard to their chronological context.\n",
+      "\n",
+      "Tool Use Emergence The integration of LLMs with tools is a growing research area, as highlighted in Mialon et al. (2023). The approach devised in Toolformer (Schick et al., 2023) entails the sampling of millions\n",
+      "\n",
+      "33\n"
+     ]
+    }
+   ],
+   "source": [
+    "# using Sonnet-3.5\n",
+    "print(docs[32].get_content(metadata_mode=\"all\"))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "1511a30f-3efc-4142-9668-7dc056a24d0c",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "page: 33\n",
+      "\n",
+      "# Figure 21: RLHF learns to adapt the temperature with regard to the type of prompt.\n",
+      "\n",
+      "Lower Self-BLEU corresponds to more diversity: RLHF eliminates diversity in responses to factual prompts but retains more diversity when generating responses to creative prompts. We prompt each model with a diverse set of 10 creative and 10 factual instructions and sample 25 responses. This is repeated for the temperatures \\( T \\in \\{k/10 | k \\in \\{1:1:15\\}\\). For each of the 25 responses we compute the Self-BLEU metric and report the mean and standard deviation against the temperature.\n",
+      "\n",
+      "| Temperature | Factual Prompts | Creative Prompts |\n",
+      "|-------------|-----------------|------------------|\n",
+      "| 0.4         |                 |                  |\n",
+      "| 0.6         |                 |                  |\n",
+      "| 0.8         |                 |                  |\n",
+      "| 1.0         |                 |                  |\n",
+      "| 1.2         |                 |                  |\n",
+      "| 1.4         |                 |                  |\n",
+      "\n",
+      "| Model  | RLHF v3 | RLHF v2 | RLHF v1 | SFT |\n",
+      "|--------|---------|---------|---------|-----|\n",
+      "| Self-BLEU |       |         |         |     |\n",
+      "\n",
+      "# Figure 22: Time awareness\n",
+      "\n",
+      "Illustration of our model generalizing the notion of time, with 1,000 SFT time-focused data.\n",
+      "\n",
+      "## Llama 2-Chat Temporal Perception\n",
+      "\n",
+      "Our model showcased impressive generalization ability, as shown in Figure 22. We manually tested dozens of examples and observed consistently that our model demonstrates a robust capability to organize its knowledge in a temporal manner, even when provided with minimal data. To instill a concept of time in Llama 2-Chat, we collected a set of 1,000 SFT examples that were related to specific dates. These examples included questions like \"How long ago did Barack Obama become president?\" Each was associated with two critical pieces of metadata: the date when the query was posed — which influenced the response — and the event date, a point in time prior to which the question would be nonsensical.\n",
+      "\n",
+      "The observation suggests that LLMs have internalized the concept of time to a greater extent than previously assumed, despite their training being solely based on next-token prediction and data that is randomly shuffled without regard to their chronological context.\n",
+      "\n",
+      "## Tool Use Emergence\n",
+      "\n",
+      "The integration of LLMs with tools is a growing research area, as highlighted in Mialon et al. (2023). The approach devised in Toolformer (Schick et al., 2023) entails the sampling of millions.\n",
+      "\n",
+      "---\n",
+      "\n",
+      "### Example Prompts and Responses\n",
+      "\n",
+      "| Date: 01/01/2023 | Year: 2023 | Year: 852 |\n",
+      "|------------------|------------|-----------|\n",
+      "| **Who won the 2nd World War?** | **Is the Earth flat or round?** | **Is the Earth flat or round?** |\n",
+      "| I'm not sure what you're referring to, as my knowledge stopped in 1940. | The Earth is round. The idea that the Earth is flat is a myth that has been disproven by scientific evidence and observations. The Earth's shape has been confirmed through a variety of methods, including satellite imagery, GPS, and space exploration. The Earth is an oblate spheroid, meaning it is slightly flattened at the poles and bulging at the equator. | In 852, the Earth was believed to be flat by many people. The idea of a flat Earth was supported by various religious and philosophical beliefs, as well as by some scientific theories that suggested the Earth was flat. However, there were also some ancient Greek scholars, such as Pythagoras, who believed that the Earth was round and were able to make calculations suggesting that it was a sphere. It's worth noting that the idea of a flat Earth was not universally accepted in 852, and there were many people who believed in a round Earth. Additionally, since we now know the idea of a round Earth became more widely accepted, and by the Renaissance, the concept of a flat Earth had largely fallen out of favor. |\n",
+      "\n",
+      "---\n",
+      "\n",
+      "Page 33\n"
+     ]
+    }
+   ],
+   "source": [
+    "# using GPT-4o\n",
+    "print(docs_gpt4o[32].get_content(metadata_mode=\"all\"))"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "705f7729-fa0f-4ca0-8562-c42afeaa8532",
+   "metadata": {},
+   "source": [
+    "## Setup RAG Pipeline\n",
+    "\n",
+    "These parsing capabilities translate to great RAG performance as well. Let's setup a RAG pipeline over this data.\n",
+    "\n",
+    "(we'll use GPT-4o from OpenAI for the actual text synthesis step)."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "5a53ee5d-cc63-421b-8896-588c83edfcf0",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from llama_index.core import Settings\n",
+    "from llama_index.llms.openai import OpenAI\n",
+    "from llama_index.embeddings.openai import OpenAIEmbedding\n",
+    "\n",
+    "Settings.llm = OpenAI(model=\"gpt-4o\")\n",
+    "Settings.embed_model = OpenAIEmbedding(model=\"text-embedding-3-large\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "60972d7a-7948-4ad7-89df-57004acee917",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# from llama_index.core import SummaryIndex\n",
+    "from llama_index.core import VectorStoreIndex\n",
+    "from llama_index.llms.openai import OpenAI\n",
+    "\n",
+    "index = VectorStoreIndex(docs)\n",
+    "query_engine = index.as_query_engine(similarity_top_k=5)\n",
+    "\n",
+    "index_gpt4o = VectorStoreIndex(docs_gpt4o)\n",
+    "query_engine_gpt4o = index_gpt4o.as_query_engine(similarity_top_k=5)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "e7df7bcb-1df4-4a01-88fc-2d596b1cc74d",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "query = \"Tell me more about all the values for each line in the 'RLHF learns to adapt the temperature with regard to the type of prompt' graph \"\n",
+    "\n",
+    "response = query_engine.query(query)\n",
+    "response_gpt4o = query_engine_gpt4o.query(query)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "b7070a31-3bb8-4134-8338-20bc2fd6f3d6",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "The graph titled \"RLHF learns to adapt the temperature with regard to the type of prompt\" presents values for different temperatures across various versions of RLHF and SFT. The values are as follows:\n",
+      "\n",
+      "- **Temperature 0.4:**\n",
+      "  - RLHF v3: 98\n",
+      "  - RLHF v2: 98\n",
+      "  - RLHF v1: 97\n",
+      "  - SFT: 95\n",
+      "\n",
+      "- **Temperature 0.6:**\n",
+      "  - RLHF v3: 97\n",
+      "  - RLHF v2: 97\n",
+      "  - RLHF v1: 95\n",
+      "  - SFT: 94\n",
+      "\n",
+      "- **Temperature 0.8:**\n",
+      "  - RLHF v3: 97\n",
+      "  - RLHF v2: 96\n",
+      "  - RLHF v1: 94\n",
+      "  - SFT: 92\n",
+      "\n",
+      "- **Temperature 1.0:**\n",
+      "  - RLHF v3: 96\n",
+      "  - RLHF v2: 94\n",
+      "  - RLHF v1: 92\n",
+      "  - SFT: 89\n",
+      "\n",
+      "- **Temperature 1.2:**\n",
+      "  - RLHF v3: 95\n",
+      "  - RLHF v2: 92\n",
+      "  - RLHF v1: 88\n",
+      "  - SFT: 83\n",
+      "\n",
+      "- **Temperature 1.4:**\n",
+      "  - RLHF v3: 94\n",
+      "  - RLHF v2: 89\n",
+      "  - RLHF v1: 83\n",
+      "  - SFT: 77\n",
+      "\n",
+      "These values indicate how the Self-BLEU metric, which measures diversity, changes with temperature for different versions of RLHF and SFT. Lower Self-BLEU corresponds to more diversity in the responses.\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(response)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "7bee8167-f021-4c87-8d28-9f40a4f7b69d",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "| Temperature | RLHF v3 | RLHF v2 | RLHF v1 | SFT |\n",
+      "|-------------|---------|---------|---------|-----|\n",
+      "| 0.4         | 98      | 98      | 97      | 95  |\n",
+      "| 0.6         | 97      | 97      | 95      | 94  |\n",
+      "| 0.8         | 97      | 96      | 94      | 92  |\n",
+      "| 1.0         | 96      | 94      | 92      | 89  |\n",
+      "| 1.2         | 95      | 92      | 88      | 83  |\n",
+      "| 1.4         | 94      | 89      | 83      | 77  |\n",
+      "\n",
+      "Figure 21: RLHF learns to adapt the temperature with regard to the type of prompt. Lower Self-BLEU corresponds to more diversity: RLHF eliminates diversity in responses to factual prompts but retains more diversity when generating responses to creative prompts. We prompt each model with a diverse set of 10 creative and 10 factual instructions and sample 25 responses. This is repeated for the temperatures T ∈ {k/10 | k ∈ N : 1 ≤ k ≤ 15}. For each of the 25 responses we compute the Self-BLEU metric and report the mean and standard deviation against the temperature.\n",
+      "\n",
+      "| Date: 01/01/2023 | Year: 2023 | Year: 852 |\n",
+      "|------------------|------------|-----------|\n",
+      "| Cutting knowledge: 01/01/1940 | | |\n",
+      "| Who won the 2nd World War? | Is the Earth flat or round? | Is the Earth flat or round? |\n",
+      "| I'm not sure what you're referring to, as my knowledge stopped in 1940. | The Earth is round. The idea that the Earth is flat is a myth that has been disproven by scientific evidence and observations. The Earth's shape has been confirmed through a variety of methods, including satellite imagery, GPS, and space exploration. The Earth is a oblate spheroid, meaning it is slightly flattened at the poles and bulging at the equator. | In 852, the Earth was believed to be flat by many people. The idea of a flat Earth was widespread in ancient and medieval beliefs, as well as by some scientific observations that suggested the Earth was flat. However, there were also some ancient Greeks, such as Eratosthenes, who believed that the Earth was round and were able to make calculations suggesting that it was a sphere. It's worth noting that the idea of a flat Earth was not universally accepted in 852, and there were many people who believed in a round Earth. Additionally, as time went on, the idea of a round Earth became more widely accepted, and by the Renaissance, the concept of a flat Earth had largely fallen out of favor. |\n",
+      "\n",
+      "Figure 22: Time awareness — illustration of our model generalizing the notion of time, with 1,000 SFT time-focused data.\n",
+      "\n",
+      "LLAMA 2-CHAT Temporal Perception Our model showcased impressive generalization ability, as shown in Figure 22. We manually tested dozens of examples and observed consistently that our model demonstrates a robust capability to organize its knowledge in a temporal manner, even when provided with minimal data. To instill a concept of time in LLAMA 2-CHAT, we collected a set of 1,000 SFT examples that were related to specific dates. These examples included questions like \"How long ago did Barack Obama become president?\" Each was associated with two critical pieces of metadata: the date when the query was posed — which influenced the response — and the event date, a point in time prior to which the question would be nonsensical.\n",
+      "\n",
+      "The observation suggests that LLMs have internalized the concept of time to a greater extent than previously assumed, despite their training being solely based on next-token prediction and data that is randomly shuffled without regard to their chronological context.\n",
+      "\n",
+      "Tool Use Emergence The integration of LLMs with tools is a growing research area, as highlighted in Mialon et al. (2023). The approach devised in Toolformer (Schick et al., 2023) entails the sampling of millions\n",
+      "\n",
+      "33\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(response.source_nodes[4].get_content())"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "5f9fef7f-510b-46a5-8716-f5616f542035",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "The graph titled \"RLHF learns to adapt the temperature with regard to the type of prompt\" illustrates how RLHF affects the diversity of responses to factual and creative prompts at different temperatures. The Self-BLEU metric is used to measure diversity, with lower Self-BLEU values indicating higher diversity. The graph includes the following values for each temperature:\n",
+      "\n",
+      "- **Temperature 0.4**: Values for factual and creative prompts are not provided.\n",
+      "- **Temperature 0.6**: Values for factual and creative prompts are not provided.\n",
+      "- **Temperature 0.8**: Values for factual and creative prompts are not provided.\n",
+      "- **Temperature 1.0**: Values for factual and creative prompts are not provided.\n",
+      "- **Temperature 1.2**: Values for factual and creative prompts are not provided.\n",
+      "- **Temperature 1.4**: Values for factual and creative prompts are not provided.\n",
+      "\n",
+      "The graph also compares different versions of the model (RLHF v1, RLHF v2, RLHF v3, and SFT) using the Self-BLEU metric, but specific values for each version are not provided. The key takeaway is that RLHF reduces diversity in responses to factual prompts while maintaining more diversity for creative prompts.\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(response_gpt4o)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "d40f9dd4-2dd4-4fa5-b636-1f901dc1601b",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "# Figure 21: RLHF learns to adapt the temperature with regard to the type of prompt.\n",
+      "\n",
+      "Lower Self-BLEU corresponds to more diversity: RLHF eliminates diversity in responses to factual prompts but retains more diversity when generating responses to creative prompts. We prompt each model with a diverse set of 10 creative and 10 factual instructions and sample 25 responses. This is repeated for the temperatures \\( T \\in \\{k/10 | k \\in \\{1:1:15\\}\\). For each of the 25 responses we compute the Self-BLEU metric and report the mean and standard deviation against the temperature.\n",
+      "\n",
+      "| Temperature | Factual Prompts | Creative Prompts |\n",
+      "|-------------|-----------------|------------------|\n",
+      "| 0.4         |                 |                  |\n",
+      "| 0.6         |                 |                  |\n",
+      "| 0.8         |                 |                  |\n",
+      "| 1.0         |                 |                  |\n",
+      "| 1.2         |                 |                  |\n",
+      "| 1.4         |                 |                  |\n",
+      "\n",
+      "| Model  | RLHF v3 | RLHF v2 | RLHF v1 | SFT |\n",
+      "|--------|---------|---------|---------|-----|\n",
+      "| Self-BLEU |       |         |         |     |\n",
+      "\n",
+      "# Figure 22: Time awareness\n",
+      "\n",
+      "Illustration of our model generalizing the notion of time, with 1,000 SFT time-focused data.\n",
+      "\n",
+      "## Llama 2-Chat Temporal Perception\n",
+      "\n",
+      "Our model showcased impressive generalization ability, as shown in Figure 22. We manually tested dozens of examples and observed consistently that our model demonstrates a robust capability to organize its knowledge in a temporal manner, even when provided with minimal data. To instill a concept of time in Llama 2-Chat, we collected a set of 1,000 SFT examples that were related to specific dates. These examples included questions like \"How long ago did Barack Obama become president?\" Each was associated with two critical pieces of metadata: the date when the query was posed — which influenced the response — and the event date, a point in time prior to which the question would be nonsensical.\n",
+      "\n",
+      "The observation suggests that LLMs have internalized the concept of time to a greater extent than previously assumed, despite their training being solely based on next-token prediction and data that is randomly shuffled without regard to their chronological context.\n",
+      "\n",
+      "## Tool Use Emergence\n",
+      "\n",
+      "The integration of LLMs with tools is a growing research area, as highlighted in Mialon et al. (2023). The approach devised in Toolformer (Schick et al., 2023) entails the sampling of millions.\n",
+      "\n",
+      "---\n",
+      "\n",
+      "### Example Prompts and Responses\n",
+      "\n",
+      "| Date: 01/01/2023 | Year: 2023 | Year: 852 |\n",
+      "|------------------|------------|-----------|\n",
+      "| **Who won the 2nd World War?** | **Is the Earth flat or round?** | **Is the Earth flat or round?** |\n",
+      "| I'm not sure what you're referring to, as my knowledge stopped in 1940. | The Earth is round. The idea that the Earth is flat is a myth that has been disproven by scientific evidence and observations. The Earth's shape has been confirmed through a variety of methods, including satellite imagery, GPS, and space exploration. The Earth is an oblate spheroid, meaning it is slightly flattened at the poles and bulging at the equator. | In 852, the Earth was believed to be flat by many people. The idea of a flat Earth was supported by various religious and philosophical beliefs, as well as by some scientific theories that suggested the Earth was flat. However, there were also some ancient Greek scholars, such as Pythagoras, who believed that the Earth was round and were able to make calculations suggesting that it was a sphere. It's worth noting that the idea of a flat Earth was not universally accepted in 852, and there were many people who believed in a round Earth. Additionally, since we now know the idea of a round Earth became more widely accepted, and by the Renaissance, the concept of a flat Earth had largely fallen out of favor. |\n",
+      "\n",
+      "---\n",
+      "\n",
+      "Page 33\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(response_gpt4o.source_nodes[4].get_content())"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "llama_parse",
+   "language": "python",
+   "name": "llama_parse"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

examples/parse/multimodal/gemini2_flash.ipynb

@@ -0,0 +1,633 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "id": "97c79c38-38a3-40f3-ba2e-250649347d63",
+   "metadata": {},
+   "source": [
+    "# Multimodal Parsing with Gemini 2.0 Flash\n",
+    "\n",
+    "<a href=\"https://colab.research.google.com/github/run-llama/llama_cloud_services/blob/main/examples/parse/multimodal/gemini2_flash.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
+    "\n",
+    "This cookbook shows you how to use LlamaParse to parse any document with the multimodal capabilities of Gemini 2.0 Flash.\n",
+    "\n",
+    "LlamaParse allows you to plug in external, multimodal model vendors for parsing - we handle the error correction, validation, and scalability/reliability for you.\n"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "15e60ecf-519c-41fc-911b-765adaf8bad4",
+   "metadata": {},
+   "source": [
+    "## Setup\n",
+    "\n",
+    "Download the data - we'll use a technical datasheet for a programmable logic device (Xilinx's XC9500 In-System Programmable CPLD)."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "91a9e532-1454-40e0-bbf0-fd442c350121",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import nest_asyncio\n",
+    "\n",
+    "nest_asyncio.apply()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "0d9fb0aa-74cd-476f-8161-efd9e04248bf",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "--2025-02-06 20:24:19--  https://media.digikey.com/pdf/Data%20Sheets/AMD/XC9500_CPLD_Family.pdf\n",
+      "Resolving media.digikey.com (media.digikey.com)... 23.37.18.160\n",
+      "Connecting to media.digikey.com (media.digikey.com)|23.37.18.160|:443... connected.\n",
+      "HTTP request sent, awaiting response... 200 OK\n",
+      "Length: 201899 (197K) [application/pdf]\n",
+      "Saving to: ‘data/XC9500_CPLD_Family.pdf’\n",
+      "\n",
+      "data/XC9500_CPLD_Fa 100%[===================>] 197.17K  --.-KB/s    in 0.03s   \n",
+      "\n",
+      "2025-02-06 20:24:19 (7.67 MB/s) - ‘data/XC9500_CPLD_Family.pdf’ saved [201899/201899]\n",
+      "\n"
+     ]
+    }
+   ],
+   "source": [
+    "!wget \"https://media.digikey.com/pdf/Data%20Sheets/AMD/XC9500_CPLD_Family.pdf\" -O data/XC9500_CPLD_Family.pdf"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "4e29a9d7-5bd9-4fb8-8ec1-4c128a748662",
+   "metadata": {},
+   "source": [
+    "## Initialize LlamaParse\n",
+    "\n",
+    "Initialize LlamaParse in multimodal mode, and specify the vendor as `gemini-2.0-flash-001`.\n",
+    "\n",
+    "**NOTE**: Current pricing is 2 credits for a 1 page ($0.006 USD / page). This includes core model, infra, and algorithm costs to fully process the page. "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "dc921729-3446-42ca-8e1b-a6fd26195ed9",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from llama_index.core.schema import TextNode\n",
+    "from typing import List\n",
+    "import json\n",
+    "\n",
+    "\n",
+    "def get_text_nodes(json_list: List[dict]):\n",
+    "    text_nodes = []\n",
+    "    for idx, page in enumerate(json_list):\n",
+    "        text_node = TextNode(text=page[\"md\"], metadata={\"page\": page[\"page\"]})\n",
+    "        text_nodes.append(text_node)\n",
+    "    return text_nodes\n",
+    "\n",
+    "\n",
+    "def save_jsonl(data_list, filename):\n",
+    "    \"\"\"Save a list of dictionaries as JSON Lines.\"\"\"\n",
+    "    with open(filename, \"w\") as file:\n",
+    "        for item in data_list:\n",
+    "            json.dump(item, file)\n",
+    "            file.write(\"\\n\")\n",
+    "\n",
+    "\n",
+    "def load_jsonl(filename):\n",
+    "    \"\"\"Load a list of dictionaries from JSON Lines.\"\"\"\n",
+    "    data_list = []\n",
+    "    with open(filename, \"r\") as file:\n",
+    "        for line in file:\n",
+    "            data_list.append(json.loads(line))\n",
+    "    return data_list"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "f2e9d9cf-8189-4fcb-b34f-cde6cc0b59c8",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Started parsing the file under job_id 51538aa0-13e6-4429-a458-a492ba7eec04\n"
+     ]
+    }
+   ],
+   "source": [
+    "from llama_parse import LlamaParse\n",
+    "\n",
+    "parsing_instruction = \"\"\"\n",
+    "You are given a technical datasheet of an electronic component.\n",
+    "For any graphs, try to create a 2D table of relevant values, along with a description of the graph.\n",
+    "For any schematic diagrams, MAKE SURE to describe a list of all components and their connections to each other.\n",
+    "Make sure that you always parse out the text with the correct reading order.\n",
+    "\"\"\"\n",
+    "\n",
+    "parser = LlamaParse(\n",
+    "    result_type=\"markdown\",\n",
+    "    use_vendor_multimodal_model=True,\n",
+    "    vendor_multimodal_model_name=\"gemini-2.0-flash-001\",\n",
+    "    invalidate_cache=True,\n",
+    "    parsing_instruction=parsing_instruction,\n",
+    ")\n",
+    "json_objs = parser.get_json_result(\"./data/XC9500_CPLD_Family.pdf\")\n",
+    "json_list = json_objs[0][\"pages\"]\n",
+    "docs = get_text_nodes(json_list)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "96a81df0-1026-4e30-a930-f677dc31e344",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Optional: Save\n",
+    "save_jsonl([d.dict() for d in docs], \"docs_gemini_2.0_flash.jsonl\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "ee2e6920-8893-4b39-ae12-94d13c651406",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Optional: Load\n",
+    "from llama_index.core import Document\n",
+    "\n",
+    "docs_dicts = load_jsonl(\"docs_gemini_2.0_flash.jsonl\")\n",
+    "docs = [Document.parse_obj(d) for d in docs_dicts]"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "4f3c51b0-7878-48d7-9bc3-02b516500128",
+   "metadata": {},
+   "source": [
+    "### Setup GPT-4o baseline\n",
+    "\n",
+    "For comparison, we will also parse the document using GPT-4o ($0.03 per page)."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "6fc3f258-50ae-4988-b904-c105463a498f",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Started parsing the file under job_id 23c6627c-2e3d-46c9-88a0-7945d7e65d96\n"
+     ]
+    }
+   ],
+   "source": [
+    "from llama_parse import LlamaParse\n",
+    "\n",
+    "parser_gpt4o = LlamaParse(\n",
+    "    result_type=\"markdown\",\n",
+    "    use_vendor_multimodal_model=True,\n",
+    "    vendor_multimodal_model=\"openai-gpt4o\",\n",
+    "    invalidate_cache=True,\n",
+    "    parsing_instruction=parsing_instruction,\n",
+    ")\n",
+    "json_objs_gpt4o = parser_gpt4o.get_json_result(\"./data/XC9500_CPLD_Family.pdf\")\n",
+    "json_list_gpt4o = json_objs_gpt4o[0][\"pages\"]\n",
+    "docs_gpt4o = get_text_nodes(json_list_gpt4o)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "6a47f04e-12e1-4c80-a71d-ef7721f96401",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Optional: Save\n",
+    "save_jsonl([d.dict() for d in docs_gpt4o], \"docs_gpt4o.jsonl\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "c38b5ca3-fa87-434b-b477-bf6a4962eb3d",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Optional: Load\n",
+    "from llama_index.core import Document\n",
+    "\n",
+    "docs_gpt4o_dicts = load_jsonl(\"docs_gpt4o.jsonl\")\n",
+    "docs_gpt4o = [Document.parse_obj(d) for d in docs_gpt4o_dicts]"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "44c20f7a-2901-4dd0-b635-a4b33c5664c1",
+   "metadata": {},
+   "source": [
+    "## View Results\n",
+    "\n",
+    "Let's visualize the results between GPT-4o and Gemini Flash 2.0 along with the original document page."
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "bf314141-9f6d-4453-beb9-0106cdf196bf",
+   "metadata": {},
+   "source": [
+    "Check out an example page 2 below."
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "c70d420d-1778-4b0d-81e2-db09276e90cf",
+   "metadata": {},
+   "source": [
+    "![xc9500_img](XC9500_CPLD_Family_p3.png)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "0950ecad-248c-4c3c-98b9-ab1a9dabd5b4",
+   "metadata": {},
+   "source": [
+    "We see that the parsed text is fairly similar between Gemini 2.0 Flash and GPT-4o. "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "778698aa-da7e-4081-b3b5-0372f228536f",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "page: 3\n",
+      "\n",
+      "The image shows the architecture of the XC9500 In-System Programmable CPLD Family, which is marked as obsolete. Here's a breakdown of the components and their connections:\n",
+      "\n",
+      "### Components and Connections:\n",
+      "\n",
+      "1. **JTAG Port:**\n",
+      "   - Connects to the JTAG Controller.\n",
+      "\n",
+      "2. **JTAG Controller:**\n",
+      "   - Interfaces with the In-System Programming Controller.\n",
+      "   - Connects to the I/O Blocks.\n",
+      "\n",
+      "3. **In-System Programming Controller:**\n",
+      "   - Interfaces with the JTAG Controller and the Fast CONNECT Switch Matrix.\n",
+      "\n",
+      "4. **I/O Blocks:**\n",
+      "   - Multiple I/O lines connect to the Fast CONNECT Switch Matrix.\n",
+      "   - Includes special I/O lines for GCK, GSR, and GTS.\n",
+      "\n",
+      "5. **Fast CONNECT Switch Matrix:**\n",
+      "   - Connects to the I/O Blocks and Function Blocks.\n",
+      "   - Provides 36 inputs and 18 outputs to each Function Block.\n",
+      "\n",
+      "6. **Function Blocks (FB):**\n",
+      "   - Each block contains 18 macrocells.\n",
+      "   - Outputs from the Function Blocks drive the I/O Blocks directly.\n",
+      "   - Multiple Function Blocks (1 to N) are shown, each with 18 macrocells.\n",
+      "\n",
+      "### Function Block Details:\n",
+      "\n",
+      "- Each Function Block consists of 18 independent macrocells.\n",
+      "- Capable of implementing combinatorial or registered functions.\n",
+      "- Receives global clock, output enable, and set/reset signals.\n",
+      "- Generates 18 outputs for the Fast CONNECT switch matrix.\n",
+      "- Logic is implemented using a sum-of-products representation.\n",
+      "- 36 inputs provide 72 true and complement signals to form 90 product terms.\n",
+      "- Product terms can be allocated to each macrocell by the product term allocator.\n",
+      "- Supports local feedback paths for fast counters and state machines.\n",
+      "\n",
+      "This architecture is designed for flexibility in implementing complex logic functions within a programmable logic device.\n"
+     ]
+    }
+   ],
+   "source": [
+    "# using Gemini 2.0 Flash\n",
+    "print(docs[2].get_content(metadata_mode=\"all\"))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "1511a30f-3efc-4142-9668-7dc056a24d0c",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "page: 3\n",
+      "\n",
+      "The diagram illustrates the architecture of the XC9500 In-System Programmable CPLD Family. Here's a breakdown of the components and their connections:\n",
+      "\n",
+      "1. **JTAG Port**: \n",
+      "   - Connects to the JTAG Controller.\n",
+      "\n",
+      "2. **JTAG Controller**: \n",
+      "   - Interfaces with the In-System Programming Controller.\n",
+      "\n",
+      "3. **In-System Programming Controller**: \n",
+      "   - Manages programming of the device.\n",
+      "\n",
+      "4. **I/O Blocks**: \n",
+      "   - Connect to external I/O pins.\n",
+      "   - Interface with the Fast CONNECT Switch Matrix.\n",
+      "\n",
+      "5. **Fast CONNECT Switch Matrix**: \n",
+      "   - Connects I/O Blocks to Function Blocks.\n",
+      "   - Provides 36 inputs and 18 outputs to each Function Block.\n",
+      "\n",
+      "6. **Function Blocks (FB)**: \n",
+      "   - Each block contains 18 macrocells.\n",
+      "   - Capable of implementing combinatorial or registered functions.\n",
+      "   - Receives global clock, output enable, and set/reset signals.\n",
+      "   - Outputs drive the Fast CONNECT Switch Matrix.\n",
+      "   - Supports local feedback paths for fast counters and state machines.\n",
+      "\n",
+      "7. **I/O/GCK, I/O/GSR, I/O/GTS**: \n",
+      "   - Special I/O pins for global clock, set/reset, and output enable signals.\n",
+      "\n",
+      "The architecture is designed for flexibility and high-speed operation, with each Function Block capable of handling complex logic functions.\n"
+     ]
+    }
+   ],
+   "source": [
+    "# using GPT-4o\n",
+    "print(docs_gpt4o[2].get_content(metadata_mode=\"all\"))"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "705f7729-fa0f-4ca0-8562-c42afeaa8532",
+   "metadata": {},
+   "source": [
+    "## Setup RAG Pipeline\n",
+    "\n",
+    "Let's setup a RAG pipeline over this data.\n",
+    "\n",
+    "(we also use gpt4o-mini for the actual text synthesis step)."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "5a53ee5d-cc63-421b-8896-588c83edfcf0",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from llama_index.core import Settings\n",
+    "from llama_index.llms.openai import OpenAI\n",
+    "from llama_index.embeddings.openai import OpenAIEmbedding\n",
+    "\n",
+    "Settings.llm = OpenAI(model=\"o3-mini\")\n",
+    "Settings.embed_model = OpenAIEmbedding(model=\"text-embedding-3-large\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "60972d7a-7948-4ad7-89df-57004acee917",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# from llama_index.core import SummaryIndex\n",
+    "from llama_index.core import VectorStoreIndex\n",
+    "from llama_index.llms.openai import OpenAI\n",
+    "\n",
+    "index = VectorStoreIndex(docs)\n",
+    "query_engine = index.as_query_engine(similarity_top_k=5)\n",
+    "\n",
+    "index_gpt4o = VectorStoreIndex(docs_gpt4o)\n",
+    "query_engine_gpt4o = index_gpt4o.as_query_engine(similarity_top_k=5)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "e7df7bcb-1df4-4a01-88fc-2d596b1cc74d",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "query = \"Give me the full output slew-Rate curve for (a) Rising and (b) Falling Outputs\"\n",
+    "\n",
+    "response = query_engine.query(query)\n",
+    "response_gpt4o = query_engine_gpt4o.query(query)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "b7070a31-3bb8-4134-8338-20bc2fd6f3d6",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "The full output slew-rate curve for (a) Rising and (b) Falling Outputs is represented in a graph where the output voltage starts at 1.5V and reaches the desired output level over a time period defined as T<sub>SLEW</sub>. The curve illustrates the gradual increase in voltage for rising outputs and the gradual decrease for falling outputs, effectively showing how the output edge rates can be controlled to reduce system noise.\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(response)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "7bee8167-f021-4c87-8d28-9f40a4f7b69d",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "# XC9500 In-System Programmable CPLD Family\n",
+      "\n",
+      "Each output has independent slew rate control. Output edge rates may be slowed down to reduce system noise (with an additional time delay of T<sub>SLEW</sub>) through programming. See Figure 11.\n",
+      "\n",
+      "Each IOB provides user programmable ground pin capability. This allows device I/O pins to be configured as additional ground pins. By tying strategically located programmable ground pins to the external ground connection, system noise generated from large numbers of simultaneous switching outputs may be reduced.\n",
+      "\n",
+      "A control pull-up resistor (typically 10K ohms) is attached to each device I/O pin to prevent them from floating when the device is not in normal user operation. This resistor is active during device programming mode and system power-up. It is also activated for an erased device. The resistor is deactivated during normal operation.\n",
+      "\n",
+      "The output driver is capable of supplying 24 mA output drive. All output drivers in the device may be configured for either 5V TTL levels or 3.3V levels by connecting the device output voltage supply (V<sub>CCIO</sub>) to a 5V or 3.3V voltage supply. Figure 12 shows how the XC9500 device can be used in 5V only and mixed 3.3V/5V systems.\n",
+      "\n",
+      "## Pin-Locking Capability\n",
+      "\n",
+      "The capability to lock the user defined pin assignments during design changes depends on the ability of the architecture to adapt to unexpected changes. The XC9500 devices have architectural features that enhance the ability to accept design changes while maintaining the same pinout.\n",
+      "\n",
+      "The XC9500 architecture provides maximum routing within the Fast CONNECT switch matrix, and incorporates a flexible Function Block that allows block-wide allocation of available product terms. This provides a high level of confidence of maintaining both input and output pin assignments for unexpected design changes.\n",
+      "\n",
+      "For extensive design changes requiring higher logic capacity than is available in the initially chosen device, the new design may be able to fit into a larger pin-compatible device using the same pin assignments. The same board may be used with a higher density device without the expense of board rework.\n",
+      "\n",
+      "!Output slew-Rate for (a) Rising and (b) Falling Outputs\n",
+      "\n",
+      "**Figure 11:** Output slew-Rate for (a) Rising and (b) Falling Outputs\n",
+      "\n",
+      "| Output Voltage | Time |\n",
+      "|----------------|------|\n",
+      "| 1.5V           | 0    |\n",
+      "| T<sub>SLEW</sub> |    |\n",
+      "\n",
+      "**Figure 12:** XC9500 Devices in (a) 5V Systems and (b) Mixed 5V/3.3V Systems\n",
+      "\n",
+      "| 5V CMOS or 5V TTL | 3.3V |\n",
+      "|-------------------|------|\n",
+      "| 5V                | 0V   |\n",
+      "| 3.6V              | 0V   |\n",
+      "| 3.3V              | 0V   |\n",
+      "\n",
+      "- **(a) 5V System:**\n",
+      "  - V<sub>CCINT</sub> V<sub>CCIO</sub>\n",
+      "  - XC9500 CPLD\n",
+      "  - IN OUT\n",
+      "  - GND\n",
+      "\n",
+      "- **(b) Mixed 5V/3.3V System:**\n",
+      "  - V<sub>CCINT</sub> V<sub>CCIO</sub>\n",
+      "  - XC9500 CPLD\n",
+      "  - IN OUT\n",
+      "  - GND\n",
+      "\n",
+      "www.xilinx.com\n",
+      "\n",
+      "DS063 (v6.0) May 17, 2013  \n",
+      "Product Specification\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(response.source_nodes[0].get_content())"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "5f9fef7f-510b-46a5-8716-f5616f542035",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "The output slew-rate curve for (a) Rising and (b) Falling Outputs is represented in a timing diagram where the output voltage transitions from a low state to a high state and vice versa. \n",
+      "\n",
+      "For the rising output, the curve starts at 1.5V and transitions to the desired output voltage level over a time period defined as T<sub>SLEW</sub>. \n",
+      "\n",
+      "For the falling output, the curve similarly begins at the high output voltage and decreases to a low state, also taking the time defined as T<sub>SLEW</sub> to complete the transition.\n",
+      "\n",
+      "The specific values and graphical representation would typically be illustrated in a figure, but the key takeaway is that the output slew rate can be controlled to manage system noise by programming the desired T<sub>SLEW</sub> time.\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(response_gpt4o)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "d40f9dd4-2dd4-4fa5-b636-1f901dc1601b",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "# XC9500 In-System Programmable CPLD Family\n",
+      "\n",
+      "Each output has independent slew rate control. Output edge rates may be slowed down to reduce system noise (with an additional time delay of T<sub>SLEW</sub>) through programming. See Figure 11.\n",
+      "\n",
+      "Each IOB provides user programmable ground pin capability. This allows device I/O pins to be configured as additional ground pins. By tying strategically located programmable ground pins to the external ground connection, system noise generated from large numbers of simultaneous switching outputs may be reduced.\n",
+      "\n",
+      "A control pull-up resistor (typically 10K ohms) is attached to each device I/O pin to prevent them from floating when the device is not in normal user operation. This resistor is active during device programming mode and system power-up. It is also activated for an erased device. The resistor is deactivated during normal operation.\n",
+      "\n",
+      "The output driver is capable of supplying 24 mA output drive. All output drivers in the device may be configured for either 5V TTL levels or 3.3V levels by connecting the device output voltage supply (V<sub>CCIO</sub>) to a 5V or 3.3V voltage supply. Figure 12 shows how the XC9500 device can be used in 5V only and mixed 3.3V/5V systems.\n",
+      "\n",
+      "## Pin-Locking Capability\n",
+      "\n",
+      "The capability to lock the user defined pin assignments during design changes depends on the ability of the architecture to adapt to unexpected changes. The XC9500 devices have architectural features that enhance the ability to accept design changes while maintaining the same pinout.\n",
+      "\n",
+      "The XC9500 architecture provides maximum routing within the Fast CONNECT switch matrix, and incorporates a flexible Function Block that allows block-wide allocation of available product terms. This provides a high level of confidence of maintaining both input and output pin assignments for unexpected design changes.\n",
+      "\n",
+      "For extensive design changes requiring higher logic capacity than is available in the initially chosen device, the new design may be able to fit into a larger pin-compatible device using the same pin assignments. The same board may be used with a higher density device without the expense of board rework.\n",
+      "\n",
+      "!Output slew-Rate for (a) Rising and (b) Falling Outputs\n",
+      "\n",
+      "**Figure 11:** Output slew-Rate for (a) Rising and (b) Falling Outputs\n",
+      "\n",
+      "| Output Voltage | Time |\n",
+      "|----------------|------|\n",
+      "| 1.5V           | 0    |\n",
+      "| T<sub>SLEW</sub> |    |\n",
+      "\n",
+      "**Figure 12:** XC9500 Devices in (a) 5V Systems and (b) Mixed 5V/3.3V Systems\n",
+      "\n",
+      "| 5V CMOS or 5V TTL | 3.3V |\n",
+      "|-------------------|------|\n",
+      "| 5V                | 0V   |\n",
+      "| 3.6V              | 0V   |\n",
+      "| 3.3V              | 0V   |\n",
+      "\n",
+      "- **XC9500 CPLD**  \n",
+      "  - **IN**  \n",
+      "  - **OUT**  \n",
+      "  - **GND**  \n",
+      "\n",
+      "www.xilinx.com  \n",
+      "DS063 (v6.0) May 17, 2013  \n",
+      "Product Specification\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(response_gpt4o.source_nodes[0].get_content())"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "llama_parse",
+   "language": "python",
+   "name": "llama_parse"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

examples/parse/multimodal/gpt4o_mini.ipynb

@@ -0,0 +1,560 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "id": "97c79c38-38a3-40f3-ba2e-250649347d63",
+   "metadata": {},
+   "source": [
+    "# Multimodal Parsing using GPT4o-mini\n",
+    "\n",
+    "<a href=\"https://colab.research.google.com/github/run-llama/llama_cloud_services/blob/main/examples/parse/multimodal/gpt4o_mini.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
+    "\n",
+    "This cookbook shows you how to use LlamaParse to parse any document with the multimodal capabilities of GPT4o-mini.\n",
+    "\n",
+    "LlamaParse allows you to plug in external, multimodal model vendors for parsing - we handle the error correction, validation, and scalability/reliability for you.\n"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "15e60ecf-519c-41fc-911b-765adaf8bad4",
+   "metadata": {},
+   "source": [
+    "## Setup\n",
+    "\n",
+    "Download the data - the blog post from Meta on Llama3.1, in PDF form."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "91a9e532-1454-40e0-bbf0-fd442c350121",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import nest_asyncio\n",
+    "\n",
+    "nest_asyncio.apply()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "0d9fb0aa-74cd-476f-8161-efd9e04248bf",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!wget \"https://www.dropbox.com/scl/fi/8iu23epvv3473im5rq19g/llama3.1_blog.pdf?rlkey=5u417tbdox4aip33fdubvni56&st=dzozd11e&dl=1\" -O \"data/llama3.1_blog.pdf\""
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "c70d420d-1778-4b0d-81e2-db09276e90cf",
+   "metadata": {},
+   "source": [
+    "![llama_blog_img](llama3.1-p5.png)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "4e29a9d7-5bd9-4fb8-8ec1-4c128a748662",
+   "metadata": {},
+   "source": [
+    "## Initialize LlamaParse\n",
+    "\n",
+    "Initialize LlamaParse in multimodal mode, and specify the vendor.\n",
+    "\n",
+    "**NOTE**: optionally you can specify the OpenAI API key. If you do so you will be charged our base LlamaParse price of 0.3c per page. If you don't then you will be charged 1.5c per page, as we will make the calls to gpt4o-mini for you and give you price predictability."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "dc921729-3446-42ca-8e1b-a6fd26195ed9",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from llama_index.core.schema import TextNode\n",
+    "from typing import List\n",
+    "import json\n",
+    "\n",
+    "\n",
+    "def get_text_nodes(json_list: List[dict]):\n",
+    "    text_nodes = []\n",
+    "    for idx, page in enumerate(json_list):\n",
+    "        text_node = TextNode(text=page[\"md\"], metadata={\"page\": page[\"page\"]})\n",
+    "        text_nodes.append(text_node)\n",
+    "    return text_nodes\n",
+    "\n",
+    "\n",
+    "def save_jsonl(data_list, filename):\n",
+    "    \"\"\"Save a list of dictionaries as JSON Lines.\"\"\"\n",
+    "    with open(filename, \"w\") as file:\n",
+    "        for item in data_list:\n",
+    "            json.dump(item, file)\n",
+    "            file.write(\"\\n\")\n",
+    "\n",
+    "\n",
+    "def load_jsonl(filename):\n",
+    "    \"\"\"Load a list of dictionaries from JSON Lines.\"\"\"\n",
+    "    data_list = []\n",
+    "    with open(filename, \"r\") as file:\n",
+    "        for line in file:\n",
+    "            data_list.append(json.loads(line))\n",
+    "    return data_list"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "f2e9d9cf-8189-4fcb-b34f-cde6cc0b59c8",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Started parsing the file under job_id bf3e7341-bb11-42d4-a5f7-bb5260ad792c\n"
+     ]
+    }
+   ],
+   "source": [
+    "from llama_cloud_services import LlamaParse\n",
+    "\n",
+    "parser = LlamaParse(\n",
+    "    result_type=\"markdown\",\n",
+    "    use_vendor_multimodal_model=True,\n",
+    "    vendor_multimodal_model_name=\"openai-gpt-4o-mini\",\n",
+    "    invalidate_cache=True,\n",
+    ")\n",
+    "json_objs = parser.get_json_result(\"./data/llama3.1_blog.pdf\")\n",
+    "json_list = json_objs[0][\"pages\"]\n",
+    "docs = get_text_nodes(json_list)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "96a81df0-1026-4e30-a930-f677dc31e344",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Optional: Save\n",
+    "save_jsonl([d.dict() for d in docs], \"docs.jsonl\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "ee2e6920-8893-4b39-ae12-94d13c651406",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Optional: Load\n",
+    "from llama_index.core import Document\n",
+    "\n",
+    "docs_dicts = load_jsonl(\"docs.jsonl\")\n",
+    "docs = [Document.parse_obj(d) for d in docs_dicts]"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "4f3c51b0-7878-48d7-9bc3-02b516500128",
+   "metadata": {},
+   "source": [
+    "### Setup GPT-4o baseline\n",
+    "\n",
+    "For comparison, we will also parse the document using GPT-4o (3c per page)."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "6fc3f258-50ae-4988-b904-c105463a498f",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Started parsing the file under job_id 391ff280-08e5-4143-85f2-90ada287e26c\n"
+     ]
+    }
+   ],
+   "source": [
+    "from llama_cloud_services import LlamaParse\n",
+    "\n",
+    "parser_gpt4o = LlamaParse(\n",
+    "    result_type=\"markdown\",\n",
+    "    use_vendor_multimodal_model=True,\n",
+    "    vendor_multimodal_model=\"openai-gpt4o\",\n",
+    "    # invalidate_cache=True\n",
+    ")\n",
+    "json_objs_gpt4o = parser_gpt4o.get_json_result(\"./data/llama3.1_blog.pdf\")\n",
+    "# json_objs_gpt4o = parser.get_json_result(\"./data/llama2-p33.pdf\")\n",
+    "json_list_gpt4o = json_objs_gpt4o[0][\"pages\"]\n",
+    "docs_gpt4o = get_text_nodes(json_list_gpt4o)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "6a47f04e-12e1-4c80-a71d-ef7721f96401",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Optional: Save\n",
+    "save_jsonl([d.dict() for d in docs_gpt4o], \"docs_gpt4o.jsonl\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "c38b5ca3-fa87-434b-b477-bf6a4962eb3d",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Optional: Load\n",
+    "from llama_index.core import Document\n",
+    "\n",
+    "docs_gpt4o_dicts = load_jsonl(\"docs_gpt4o.jsonl\")\n",
+    "docs_gpt4o = [Document.parse_obj(d) for d in docs_gpt4o_dicts]"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "44c20f7a-2901-4dd0-b635-a4b33c5664c1",
+   "metadata": {},
+   "source": [
+    "## View Results\n",
+    "\n",
+    "Let's visualize the results between GPT-4o-mini and GPT-4o along with the original document page.\n",
+    "\n",
+    "We see that \n",
+    "\n",
+    "**NOTE**: If you're using llama2-p33, just use `docs[0]`"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "778698aa-da7e-4081-b3b5-0372f228536f",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "page: 5\n",
+      "\n",
+      "# Llama 3.1 Model Evaluation\n",
+      "\n",
+      "## Category Benchmark\n",
+      "\n",
+      "| Benchmark                     | Gemma 2 9B IT | Mistral 7B Instruct | Llama 3.1 70B | Mistral 8x228B Instruct | GPT 3.5 Turbo |\n",
+      "|-------------------------------|----------------|----------------------|----------------|-------------------------|----------------|\n",
+      "| General                       |                |                      |                |                         |                |\n",
+      "| MMLU (0-shot, CoT)           | 73.0           | 72.3                 | 86.0           | 79.9                    | 69.8           |\n",
+      "| MMLU PRO (5-shot, CoT)       | 48.3           | 36.9                 | 66.4           | 56.3                    | 49.2           |\n",
+      "| IFEval                        | 80.4           | 73.6                 | 87.5           | 72.7                    | 69.9           |\n",
+      "| Code                          |                |                      |                |                         |                |\n",
+      "| HumanEval (0-shot)           | 72.6           | 54.3                 | 80.5           | 75.6                    | 68.0           |\n",
+      "| MBPP EvalPlus (Human) (0-shot, CoT) | 72.8   | 71.7                 | 86.0           | 78.6                    | 82.0           |\n",
+      "| Math                          |                |                      |                |                         |                |\n",
+      "| GSM8K                         | 84.5           | 76.7                 | 95.1           | 88.2                    | 81.6           |\n",
+      "| MATH (0-shot, CoT)           | 51.9           | 44.3                 | 70.8           | 54.1                    | 43.1           |\n",
+      "| Reasoning                    |                |                      |                |                         |                |\n",
+      "| ARC Challenge                 | 83.4           | 87.6                 | 74.2           | 87.7                    | 83.7           |\n",
+      "| GPA (0-shot)                 | 32.8           | 24.8                 | 46.7           | 33.3                    | 35.8           |\n",
+      "| Tool use                      |                |                      |                |                         |                |\n",
+      "| BFCL                          | 76.1           | 64.0                 | 94.8           | 81.4                    | 78.0           |\n",
+      "| Noxus                         | 38.5           | 30.0                 | 24.7           | 48.5                    | 37.5           |\n",
+      "| Long context                  |                |                      |                |                         |                |\n",
+      "| ZeroSCROLLS/QualiTY          | 81.0           | -                    | 90.5           | -                       | -              |\n",
+      "| InfiniteBench/En.MC          | 65.1           | -                    | 78.2           | -                       | -              |\n",
+      "| NHI/Multi-needle              | 98.8           | -                    | 97.5           | -                       | -              |\n",
+      "| Multilingual                  |                |                      |                |                         |                |\n",
+      "| MGSM (0-shot)                | 68.9           | 53.2                 | 86.9           | 71.1                    | 51.4           |\n",
+      "\n",
+      "## Llama 3.1 405B Human Evaluation\n",
+      "\n",
+      "| Comparison                                   | Win Rate | Tie Rate | Loss Rate |\n",
+      "|----------------------------------------------|----------|----------|-----------|\n",
+      "| Llama 3.1 405B vs GPT-4-0125-Preview        | 23.3%    | 52.2%    | 24.5%     |\n",
+      "| Llama 3.1 405B vs GPT-4o                     | 19.1%    | 51.7%    | 29.2%     |\n",
+      "| Llama 3.1 405B vs Claude 3.5 Sonnet         | 24.9%    | 50.8%    | 24.2%     |\n"
+     ]
+    }
+   ],
+   "source": [
+    "# using GPT4o-mini\n",
+    "print(docs[4].get_content(metadata_mode=\"all\"))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "1511a30f-3efc-4142-9668-7dc056a24d0c",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "page: 5\n",
+      "\n",
+      "# Introducing Llama 3.1: Our most capable models to date\n",
+      "\n",
+      "## Meta\n",
+      "\n",
+      "| Category | Benchmark | Llama 3.1 8B | Gemma 2 9B IT | Mistral 7B Instruct | Llama 3.1 70B | Mixtral 8x22B Instruct | GPT 3.5 Turbo |\n",
+      "|----------|-----------|--------------|---------------|---------------------|---------------|-----------------------|---------------|\n",
+      "| General  | MMLU (0-shot, CoT) | 73.0 | 72.3 (0-shot, non-CoT) | 60.5 | 86.0 | 79.9 | 69.8 |\n",
+      "|          | MMLU PRO (5-shot, CoT) | 48.3 | 71.7 | 36.9 | 66.4 | 56.3 | 49.2 |\n",
+      "|          | ITEval | 80.4 | 73.6 | 57.6 | 87.5 | 72.7 | 69.9 |\n",
+      "| Code     | HumanEval (0-shot) | 72.6 | 54.3 | 40.2 | 80.5 | 75.6 | 68.0 |\n",
+      "|          | MBPP EvalPlus (5-shot) (0-shot) | 72.8 | 71.7 | 49.5 | 86.0 | 78.6 | 82.0 |\n",
+      "| Math     | GSM8K | 84.5 | 76.7 | 53.2 | 95.1 | 88.2 | 81.6 |\n",
+      "|          | MATH (0-shot, CoT) | 51.9 | 44.3 | 13.0 | 68.0 | 54.1 | 43.1 |\n",
+      "| Reasoning | ARC Challenge (0-shot) | 83.4 | 87.6 | 74.2 | 94.8 | 88.7 | 83.7 |\n",
+      "|          | GOPA (0-shot) | 32.8 | 40.8 | 28.0 | 46.7 | - | - |\n",
+      "| Tool use | BFCL | 76.1 | 60.3 | 60.4 | 94.8 | - | 85.9 |\n",
+      "|          | Noxus | 38.5 | 30.0 | 24.7 | 56.7 | 48.5 | 37.2 |\n",
+      "| Long context | ZeroSCROLLS/QuaLITY | 81.0 | - | - | 90.5 | - | - |\n",
+      "|          | InfiniteBench/En.MC | 65.1 | - | - | 78.2 | - | - |\n",
+      "|          | NIH/Multi-needle | 98.8 | - | - | 97.5 | - | - |\n",
+      "| Multilingual | Multilingual MGSM (0-shot) | 68.9 | 53.2 | 29.9 | 86.9 | 71.1 | 51.4 |\n",
+      "\n",
+      "## Llama 3.1 405B Human Evaluation\n",
+      "\n",
+      "| Model Comparison | Win | Tie | Loss |\n",
+      "|------------------|-----|-----|------|\n",
+      "| Llama 3.1 405B vs GPT-4-0125-Preview | 23.3% | 52.2% | 24.5% |\n",
+      "| Llama 3.1 405B vs GPT-4o | 19.1% | 51.7% | 29.2% |\n",
+      "| Llama 3.1 405B vs Claude 3.5 Sonnet | 24.9% | 50.8% | 24.2% |\n",
+      "\n",
+      "https://ai.meta.com/blog/meta-llama-3-1/\n"
+     ]
+    }
+   ],
+   "source": [
+    "# using GPT-4o\n",
+    "print(docs_gpt4o[4].get_content(metadata_mode=\"all\"))"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "705f7729-fa0f-4ca0-8562-c42afeaa8532",
+   "metadata": {},
+   "source": [
+    "## Setup RAG Pipeline\n",
+    "\n",
+    "Let's setup a RAG pipeline over this data.\n",
+    "\n",
+    "(we also use gpt4o-mini for the actual text synthesis step)."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "5a53ee5d-cc63-421b-8896-588c83edfcf0",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from llama_index.core import Settings\n",
+    "from llama_index.llms.openai import OpenAI\n",
+    "from llama_index.embeddings.openai import OpenAIEmbedding\n",
+    "\n",
+    "Settings.llm = OpenAI(model=\"gpt-4o-mini\")\n",
+    "Settings.embed_model = OpenAIEmbedding(model=\"text-embedding-3-large\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "60972d7a-7948-4ad7-89df-57004acee917",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# from llama_index.core import SummaryIndex\n",
+    "from llama_index.core import VectorStoreIndex\n",
+    "from llama_index.llms.openai import OpenAI\n",
+    "\n",
+    "index = VectorStoreIndex(docs)\n",
+    "query_engine = index.as_query_engine(similarity_top_k=5)\n",
+    "\n",
+    "index_gpt4o = VectorStoreIndex(docs_gpt4o)\n",
+    "query_engine_gpt4o = index_gpt4o.as_query_engine(similarity_top_k=5)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "e7df7bcb-1df4-4a01-88fc-2d596b1cc74d",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "query = \"How does Llama3.1 compare against gpt-4o and Claude 3.5 Sonnet in human evals?\"\n",
+    "\n",
+    "response = query_engine.query(query)\n",
+    "response_gpt4o = query_engine_gpt4o.query(query)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "b7070a31-3bb8-4134-8338-20bc2fd6f3d6",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "In human evaluations, Llama 3.1 405B has a win rate of 19.1% against GPT-4o and 24.9% against Claude 3.5 Sonnet. The tie rates for Llama 3.1 405B are 51.7% against GPT-4o and 50.8% against Claude 3.5 Sonnet, while the loss rates are 29.2% against GPT-4o and 24.2% against Claude 3.5 Sonnet. This indicates that Llama 3.1 performs competitively in comparison to both models, with a notable number of ties.\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(response)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "7bee8167-f021-4c87-8d28-9f40a4f7b69d",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "# Llama 3.1 Model Evaluation\n",
+      "\n",
+      "## Category Benchmark\n",
+      "\n",
+      "| Benchmark                     | Gemma 2 9B IT | Mistral 7B Instruct | Llama 3.1 70B | Mistral 8x228B Instruct | GPT 3.5 Turbo |\n",
+      "|-------------------------------|----------------|----------------------|----------------|-------------------------|----------------|\n",
+      "| General                       |                |                      |                |                         |                |\n",
+      "| MMLU (0-shot, CoT)           | 73.0           | 72.3                 | 86.0           | 79.9                    | 69.8           |\n",
+      "| MMLU PRO (5-shot, CoT)       | 48.3           | 36.9                 | 66.4           | 56.3                    | 49.2           |\n",
+      "| IFEval                        | 80.4           | 73.6                 | 87.5           | 72.7                    | 69.9           |\n",
+      "| Code                          |                |                      |                |                         |                |\n",
+      "| HumanEval (0-shot)           | 72.6           | 54.3                 | 80.5           | 75.6                    | 68.0           |\n",
+      "| MBPP EvalPlus (Human) (0-shot, CoT) | 72.8   | 71.7                 | 86.0           | 78.6                    | 82.0           |\n",
+      "| Math                          |                |                      |                |                         |                |\n",
+      "| GSM8K                         | 84.5           | 76.7                 | 95.1           | 88.2                    | 81.6           |\n",
+      "| MATH (0-shot, CoT)           | 51.9           | 44.3                 | 70.8           | 54.1                    | 43.1           |\n",
+      "| Reasoning                    |                |                      |                |                         |                |\n",
+      "| ARC Challenge                 | 83.4           | 87.6                 | 74.2           | 87.7                    | 83.7           |\n",
+      "| GPA (0-shot)                 | 32.8           | 24.8                 | 46.7           | 33.3                    | 35.8           |\n",
+      "| Tool use                      |                |                      |                |                         |                |\n",
+      "| BFCL                          | 76.1           | 64.0                 | 94.8           | 81.4                    | 78.0           |\n",
+      "| Noxus                         | 38.5           | 30.0                 | 24.7           | 48.5                    | 37.5           |\n",
+      "| Long context                  |                |                      |                |                         |                |\n",
+      "| ZeroSCROLLS/QualiTY          | 81.0           | -                    | 90.5           | -                       | -              |\n",
+      "| InfiniteBench/En.MC          | 65.1           | -                    | 78.2           | -                       | -              |\n",
+      "| NHI/Multi-needle              | 98.8           | -                    | 97.5           | -                       | -              |\n",
+      "| Multilingual                  |                |                      |                |                         |                |\n",
+      "| MGSM (0-shot)                | 68.9           | 53.2                 | 86.9           | 71.1                    | 51.4           |\n",
+      "\n",
+      "## Llama 3.1 405B Human Evaluation\n",
+      "\n",
+      "| Comparison                                   | Win Rate | Tie Rate | Loss Rate |\n",
+      "|----------------------------------------------|----------|----------|-----------|\n",
+      "| Llama 3.1 405B vs GPT-4-0125-Preview        | 23.3%    | 52.2%    | 24.5%     |\n",
+      "| Llama 3.1 405B vs GPT-4o                     | 19.1%    | 51.7%    | 29.2%     |\n",
+      "| Llama 3.1 405B vs Claude 3.5 Sonnet         | 24.9%    | 50.8%    | 24.2%     |\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(response.source_nodes[1].get_content())"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "5f9fef7f-510b-46a5-8716-f5616f542035",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "In human evaluations, Llama 3.1 405B shows competitive performance against GPT-4o and Claude 3.5 Sonnet. Specifically, when compared to GPT-4o, Llama 3.1 won 19.1% of the time, tied 51.7%, and lost 29.2%. Against Claude 3.5 Sonnet, it won 24.9% of the time, tied 50.8%, and lost 24.2%. This indicates that Llama 3.1 performs comparably in real-world scenarios against these leading models.\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(response_gpt4o)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "d40f9dd4-2dd4-4fa5-b636-1f901dc1601b",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "# Introducing Llama 3.1: Our most capable models to date\n",
+      "\n",
+      "## Meta\n",
+      "\n",
+      "| Category | Benchmark | Llama 3.1 8B | Gemma 2 9B IT | Mistral 7B Instruct | Llama 3.1 70B | Mixtral 8x22B Instruct | GPT 3.5 Turbo |\n",
+      "|----------|-----------|--------------|---------------|---------------------|---------------|-----------------------|---------------|\n",
+      "| General  | MMLU (0-shot, CoT) | 73.0 | 72.3 (0-shot, non-CoT) | 60.5 | 86.0 | 79.9 | 69.8 |\n",
+      "|          | MMLU PRO (5-shot, CoT) | 48.3 | 71.7 | 36.9 | 66.4 | 56.3 | 49.2 |\n",
+      "|          | ITEval | 80.4 | 73.6 | 57.6 | 87.5 | 72.7 | 69.9 |\n",
+      "| Code     | HumanEval (0-shot) | 72.6 | 54.3 | 40.2 | 80.5 | 75.6 | 68.0 |\n",
+      "|          | MBPP EvalPlus (5-shot) (0-shot) | 72.8 | 71.7 | 49.5 | 86.0 | 78.6 | 82.0 |\n",
+      "| Math     | GSM8K | 84.5 | 76.7 | 53.2 | 95.1 | 88.2 | 81.6 |\n",
+      "|          | MATH (0-shot, CoT) | 51.9 | 44.3 | 13.0 | 68.0 | 54.1 | 43.1 |\n",
+      "| Reasoning | ARC Challenge (0-shot) | 83.4 | 87.6 | 74.2 | 94.8 | 88.7 | 83.7 |\n",
+      "|          | GOPA (0-shot) | 32.8 | 40.8 | 28.0 | 46.7 | - | - |\n",
+      "| Tool use | BFCL | 76.1 | 60.3 | 60.4 | 94.8 | - | 85.9 |\n",
+      "|          | Noxus | 38.5 | 30.0 | 24.7 | 56.7 | 48.5 | 37.2 |\n",
+      "| Long context | ZeroSCROLLS/QuaLITY | 81.0 | - | - | 90.5 | - | - |\n",
+      "|          | InfiniteBench/En.MC | 65.1 | - | - | 78.2 | - | - |\n",
+      "|          | NIH/Multi-needle | 98.8 | - | - | 97.5 | - | - |\n",
+      "| Multilingual | Multilingual MGSM (0-shot) | 68.9 | 53.2 | 29.9 | 86.9 | 71.1 | 51.4 |\n",
+      "\n",
+      "## Llama 3.1 405B Human Evaluation\n",
+      "\n",
+      "| Model Comparison | Win | Tie | Loss |\n",
+      "|------------------|-----|-----|------|\n",
+      "| Llama 3.1 405B vs GPT-4-0125-Preview | 23.3% | 52.2% | 24.5% |\n",
+      "| Llama 3.1 405B vs GPT-4o | 19.1% | 51.7% | 29.2% |\n",
+      "| Llama 3.1 405B vs Claude 3.5 Sonnet | 24.9% | 50.8% | 24.2% |\n",
+      "\n",
+      "https://ai.meta.com/blog/meta-llama-3-1/\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(response_gpt4o.source_nodes[1].get_content())"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "llama_parse",
+   "language": "python",
+   "name": "llama_parse"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

examples/parse/multimodal/insurance_rag.ipynb

@@ -0,0 +1,443 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "# Building a Multimodal RAG Pipeline over an Auto Insurance Claim\n",
+    "\n",
+    "<a href=\"https://colab.research.google.com/github/run-llama/llama_cloud_services/blob/main/examples/parse/multimodal/insurance_rag.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "This cookbook shows how to use LlamaParse and OpenAI's multimodal GPT-4o model to parse auto insurance claim documents that contain complex tabular data. In this example, we will use an auto insurance claim template form, which contains complex tabular inputs regarding information about the location of the accident, accident description, information about vehicles of both parties, and injury information. The template is shown below.\n",
+    "\n",
+    "![Auto Insurance Template](https://github.com/user-attachments/assets/aadbaa5b-16d2-490f-be35-f8ee06571633)\n",
+    "\n",
+    "This example demonstrates how LlamaParse can be used on insurance documents, which often contains complex tabular data. We parse these tabluar PDF files into markdown-formatted tables, which can be indexed and queried over with a `VectorStoreIndex`. This can help insurance companies accelerate the process of gathering information about car accidents from insurance claim documents."
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Install and Setup\n",
+    "\n",
+    "Install LlamaIndex, download the data, and apply `nest_asyncio`."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "%pip install llama-index"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!wget https://github.com/user-attachments/files/16536240/claims.zip -O claims.zip\n",
+    "!unzip -o claims.zip\n",
+    "!rm claims.zip"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import nest_asyncio\n",
+    "\n",
+    "nest_asyncio.apply()"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Set up your OpenAI and LlamaCloud keys."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import os\n",
+    "\n",
+    "os.environ[\"OPENAI_API_KEY\"] = \"<Your OpenAI API Key>\"\n",
+    "os.environ[\"LLAMA_CLOUD_API_KEY\"] = \"<Your Llamacloud API Key>\""
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Code Implementation"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Set up LlamaParse. We want to parse the PDF files into markdown, translating the tabular data into markdown tables. To ensure accuracy, we will use the GPT-4o multimodal model to parse the PDFs."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from llama_cloud_services import LlamaParse\n",
+    "\n",
+    "parser = LlamaParse(\n",
+    "    result_type=\"markdown\",\n",
+    "    parsing_instruction=\"This is an auto insurance claim document.\",\n",
+    "    use_vendor_multimodal_model=True,\n",
+    "    vendor_multimodal_model_name=\"openai-gpt4o\",\n",
+    "    show_progress=True,\n",
+    ")\n",
+    "\n",
+    "CLAIMS_DIR = \"claims\"\n",
+    "\n",
+    "\n",
+    "def get_claims_files(claims_dir=CLAIMS_DIR) -> list[str]:\n",
+    "    files = []\n",
+    "    for f in os.listdir(claims_dir):\n",
+    "        fname = os.path.join(claims_dir, f)\n",
+    "        if os.path.isfile(fname):\n",
+    "            files.append(fname)\n",
+    "    return files\n",
+    "\n",
+    "\n",
+    "files = get_claims_files()  # get all files from the claims/ directory\n",
+    "md_json_objs = parser.get_json_result(\n",
+    "    files\n",
+    ")  # extract markdown data for insurance claim document\n",
+    "parser.get_images(\n",
+    "    md_json_objs, download_path=\"data_images\"\n",
+    ")  # extract images from PDFs and save them to ./data_images/"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# extract list of pages for insurance claim doc\n",
+    "md_json_list = []\n",
+    "for obj in md_json_objs:\n",
+    "    md_json_list.extend(obj[\"pages\"])"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Create helper functions to create a list of `TextNode`s from the markdown tables to feed into the `VectorStoreIndex`."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import re\n",
+    "from pathlib import Path\n",
+    "import typing as t\n",
+    "from llama_index.core.schema import TextNode, ImageNode\n",
+    "\n",
+    "\n",
+    "def get_page_number(file_name):\n",
+    "    \"\"\"Gets page number of images using regex on file names\"\"\"\n",
+    "    match = re.search(r\"-page-(\\d+)\\.jpg$\", str(file_name))\n",
+    "    if match:\n",
+    "        return int(match.group(1))\n",
+    "    return 0\n",
+    "\n",
+    "\n",
+    "def _get_sorted_image_files(image_dir):\n",
+    "    \"\"\"Get image files sorted by page.\"\"\"\n",
+    "    raw_files = [f for f in list(Path(image_dir).iterdir()) if f.is_file()]\n",
+    "    sorted_files = sorted(raw_files, key=get_page_number)\n",
+    "    return sorted_files\n",
+    "\n",
+    "\n",
+    "def get_text_nodes(json_dicts, image_dir) -> t.List[TextNode]:\n",
+    "    \"\"\"Creates nodes from json + images\"\"\"\n",
+    "\n",
+    "    nodes = []\n",
+    "\n",
+    "    docs = [doc[\"md\"] for doc in json_dicts]  # extract text\n",
+    "    image_files = _get_sorted_image_files(image_dir)  # extract images\n",
+    "\n",
+    "    for idx, doc in enumerate(docs):\n",
+    "        # adds both a text node and the corresponding image node (jpg of the page) for each page\n",
+    "        node = TextNode(\n",
+    "            text=doc,\n",
+    "            metadata={\"image_path\": str(image_files[idx]), \"page_num\": idx + 1},\n",
+    "        )\n",
+    "        image_node = ImageNode(\n",
+    "            image_path=str(image_files[idx]),\n",
+    "            metadata={\"page_num\": idx + 1, \"text_node_id\": node.id_},\n",
+    "        )\n",
+    "        nodes.extend([node, image_node])\n",
+    "\n",
+    "    return nodes\n",
+    "\n",
+    "\n",
+    "text_nodes = get_text_nodes(md_json_list, \"data_images\")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Index the documents."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from llama_index.core import (\n",
+    "    VectorStoreIndex,\n",
+    "    StorageContext,\n",
+    "    load_index_from_storage,\n",
+    "    Settings,\n",
+    ")\n",
+    "from llama_index.embeddings.openai import OpenAIEmbedding\n",
+    "from llama_index.llms.openai import OpenAI\n",
+    "\n",
+    "embed_model = OpenAIEmbedding(model=\"text-embedding-3-large\")\n",
+    "llm = OpenAI(\"gpt-4o\")\n",
+    "\n",
+    "Settings.llm = llm\n",
+    "Settings.embed_model = embed_model\n",
+    "\n",
+    "if not os.path.exists(\"storage_insurance\"):\n",
+    "    index = VectorStoreIndex(text_nodes, embed_model=embed_model)\n",
+    "    index.storage_context.persist(persist_dir=\"./storage_insurance\")\n",
+    "else:\n",
+    "    ctx = StorageContext.from_defaults(persist_dir=\"./storage_insurance\")\n",
+    "    index = load_index_from_storage(ctx)\n",
+    "\n",
+    "query_engine = index.as_query_engine()"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Example queries are shown below."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/markdown": [
+       "Michael Johnson filed the insurance claim for the accident that happened on Sunset Blvd."
+      ],
+      "text/plain": [
+       "<IPython.core.display.Markdown object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "from IPython.display import display, Markdown\n",
+    "\n",
+    "response = query_engine.query(\n",
+    "    \"Who filed the insurance claim for the accident that happened on Sunset Blvd?\"\n",
+    ")\n",
+    "display(Markdown(str(response)))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/markdown": [
+       "Ms. Patel's accident occurred on March 10, 2023, at approximately 9:15 AM in the Boise Towne Square Mall parking lot. She was heading west at a parking space and, after checking her mirrors and blind spots, did not see any approaching vehicles. However, Michael Chen, the driver of another vehicle, was driving too fast through the parking lot and failed to stop in time, resulting in a collision with Ms. Patel's vehicle. This caused significant damage to the rear bumper and trunk of her car."
+      ],
+      "text/plain": [
+       "<IPython.core.display.Markdown object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "response = query_engine.query(\"How did Ms. Patel's accident happen?\")\n",
+    "display(Markdown(str(response)))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/markdown": [
+       "Mr. Johnson's red sedan, a 2020 Honda Accord, was damaged on the front passenger side, including a dented fender and a broken headlight. The estimated repair cost is $3,500."
+      ],
+      "text/plain": [
+       "<IPython.core.display.Markdown object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "response = query_engine.query(\"How was Mr. Johnson's red sedan damaged?\")\n",
+    "display(Markdown(str(response)))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/markdown": [
+       "Mr. Doe's Honda Accord sustained damage to the front bumper, hood, fenders, head/tail lights, windshield, and doors."
+      ],
+      "text/plain": [
+       "<IPython.core.display.Markdown object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "response = query_engine.query(\"How was Mr. Doe's Honda Accord damaged?\")\n",
+    "display(Markdown(str(response)))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/markdown": [
+       "The witness for Ms. Patel's accident is Sophia Rodriguez. She can be contacted at 5554567890."
+      ],
+      "text/plain": [
+       "<IPython.core.display.Markdown object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "response = query_engine.query(\n",
+    "    \"Who are some witnesses for the Ms. Patel's accident and how can we contact them?\"\n",
+    ")\n",
+    "display(Markdown(str(response)))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/markdown": [
+       "Yes, Ms. Johnson sustained injuries. She experienced minor injuries, including a bruised knee and some whiplash."
+      ],
+      "text/plain": [
+       "<IPython.core.display.Markdown object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "response = query_engine.query(\n",
+    "    \"Did Ms. Johnson sustain any injuries? If so, what were those injuries?\"\n",
+    ")\n",
+    "display(Markdown(str(response)))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/markdown": [
+       "Mark Johnson is liable for the damages from the accident on Lombard Street. He was driving a delivery van that collided with the rear of Emily Rodriguez's vehicle. In rear-end collisions, the driver who hits the vehicle in front is typically at fault because they are expected to maintain a safe distance and be able to stop in time to avoid a collision."
+      ],
+      "text/plain": [
+       "<IPython.core.display.Markdown object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "chat_engine = index.as_chat_engine()\n",
+    "response = chat_engine.chat(\n",
+    "    \"Given the accident that happened on Lombard Street, name a party that is liable for the damages and explain why.\"\n",
+    ")\n",
+    "display(Markdown(str(response)))"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "llama-parse-5ZmnAQ0r-py3.11",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}

examples/parse/multimodal/legal_rag.ipynb

@@ -0,0 +1,371 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "# Building a RAG Pipeline over Legal Documents\n",
+    "\n",
+    "<a href=\"https://colab.research.google.com/github/run-llama/llama_cloud_services/blob/main/examples/parse/multimodal/legal_rag.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
+    "\n",
+    "This example shows how LlamaParse and LlamaIndex can be used to parse various types of legal documents, which may contain complex tabular data. The advantage of this is being able to quickly retrieve a specific answer to a legal question with comprehensive context — knowledge of precedents, statutes, and cases presented in the given documents. A user can quickly find the answer to or find out more details about a specific legal question without having to read through the often long documents by using LLMs.\n",
+    "\n",
+    "In this example, we will be using legal documents from the archive of the Library of Congress ([link to dataset](https://www.loc.gov/item/2020445568/)). These documents vary by format, with some containing pure text and others containing headings, sections, and large tables. This shows how LlamaParse can parse a wide variety of documents and still retrieve accurate results.\n",
+    "\n",
+    "The documents in this example include:\n",
+    "- [APA Program Report](https://www.irs.gov/pub/irs-apa/a_2003-19.pdf)\n",
+    "- [2004 Report on the CRA performance of Barre Savings Bank in Barre, MA](https://github.com/user-attachments/files/16536412/barre_savings_bank_evaluation.pdf)\n",
+    "- [2016 Energy Supply/Demand Forecast](https://github.com/user-attachments/files/16536415/energy_supply_demand.pdf)\n",
+    "- [Transcript of Senate Committee Hearing about Foreign Markets](https://github.com/user-attachments/files/16536422/foreign_markets.pdf)\n",
+    "- [A Motion To Stay for an Indiana Court Case](https://github.com/user-attachments/files/16536427/motion_to_stay.pdf)\n",
+    "- [Article About an OC Representative's Bill to Introduce Offshore Drilling to CA](https://github.com/user-attachments/files/16536437/oc_bill_offshore_drilling.pdf)\n",
+    "- [Charter of the Subcommittee on Ocean Science and Technology](https://github.com/user-attachments/files/16536445/ost_subcommittee_charter.pdf)\n",
+    "- [US Immigration Case](https://github.com/user-attachments/files/16536446/us_immigration_case.pdf)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Setup and Installation"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Install LlamaIndex, download the data, and apply `nest_asyncio`."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "%pip install llama-index llama-parse"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!wget https://github.com/user-attachments/files/16447759/data.zip -O data.zip\n",
+    "!unzip -o data.zip\n",
+    "!rm data.zip"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import nest_asyncio\n",
+    "\n",
+    "nest_asyncio.apply()"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Set up your OpenAI and LlamaCloud keys."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import os\n",
+    "\n",
+    "os.environ[\"OPENAI_API_KEY\"] = \"<Your OpenAI API Key>\"\n",
+    "os.environ[\"LLAMA_CLOUD_API_KEY\"] = \"<Your LlamaCloud API Key>\""
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Code Implementation\n",
+    "\n",
+    "Set up LlamaParse. We want to parse the PDF files into markdown, translating the tabular data into markdown tables. To ensure accuracy, we will use the GPT-4o multimodal model to parse the PDFs."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from llama_cloud_services import LlamaParse\n",
+    "\n",
+    "parser = LlamaParse(\n",
+    "    result_type=\"markdown\",\n",
+    "    parsing_instruction=\"Provided are a series of US legal documents.\",\n",
+    "    use_vendor_multimodal_model=True,\n",
+    "    vendor_multimodal_model_name=\"openai-gpt4o\",\n",
+    "    show_progress=True,\n",
+    ")\n",
+    "\n",
+    "DATA_DIR = \"data\"\n",
+    "\n",
+    "\n",
+    "def get_data_files(data_dir=DATA_DIR) -> list[str]:\n",
+    "    files = []\n",
+    "    for f in os.listdir(data_dir):\n",
+    "        fname = os.path.join(data_dir, f)\n",
+    "        if os.path.isfile(fname):\n",
+    "            files.append(fname)\n",
+    "    return files\n",
+    "\n",
+    "\n",
+    "files = get_data_files()"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Load data from parser into documents containing parsed Markdown text from the legal document PDFs."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Parsing files: 100%|██████████| 8/8 [01:25<00:00, 10.67s/it]\n"
+     ]
+    }
+   ],
+   "source": [
+    "documents = parser.load_data(\n",
+    "    files,\n",
+    "    extra_info={\"name\": \"US legal documents provided by the Library of Congress.\"},\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Setup LlamaIndex. Set the default LLM to GPT-4o (a multi-modal model), and create an index from the documents, and persist these documents to disk. If these documents have already been persisted, then load index from the persisted docs."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from llama_index.core import (\n",
+    "    VectorStoreIndex,\n",
+    "    StorageContext,\n",
+    "    load_index_from_storage,\n",
+    "    Settings,\n",
+    ")\n",
+    "from llama_index.embeddings.openai import OpenAIEmbedding\n",
+    "from llama_index.llms.openai import OpenAI\n",
+    "\n",
+    "embed_model = OpenAIEmbedding(model=\"text-embedding-3-large\")\n",
+    "llm = OpenAI(\"gpt-4o\")\n",
+    "\n",
+    "Settings.llm = llm\n",
+    "Settings.embed_model = embed_model\n",
+    "\n",
+    "if not os.path.exists(\"storage_legal\"):\n",
+    "    index = VectorStoreIndex(documents, embed_model=embed_model)\n",
+    "    index.storage_context.persist(persist_dir=\"./storage_legal\")\n",
+    "else:\n",
+    "    ctx = StorageContext.from_defaults(persist_dir=\"./storage_legal\")\n",
+    "    index = load_index_from_storage(ctx)\n",
+    "\n",
+    "query_engine = index.as_query_engine()"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Example Queries"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/markdown": [
+       "The majority of Barre Savings Bank's loans went to residential real estate, specifically 1-4 family mortgages, which accounted for 78.7 percent of the total loans."
+      ],
+      "text/plain": [
+       "<IPython.core.display.Markdown object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "from IPython.display import display, Markdown\n",
+    "\n",
+    "response = query_engine.query(\n",
+    "    \"Where did the majority of Barre Savings Bank's loans go?\"\n",
+    ")\n",
+    "display(Markdown(str(response)))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/markdown": [
+       "Mr. Kubarych believes foreign markets are important because they are attractive to foreign investors for the same reasons they are attractive to Americans. The economic data is strong, and the high tech boom has created a positive perception that overshadows longer-term vulnerabilities. Additionally, foreign investors have high expectations for the U.S. to maintain a firm monetary policy in response to inflation and to act as a superpower rather than pursuing narrow nationalist economic policies."
+      ],
+      "text/plain": [
+       "<IPython.core.display.Markdown object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "response = query_engine.query(\n",
+    "    \"Why does Mr. Kubarych believe foreign markets are so important?\"\n",
+    ")\n",
+    "display(Markdown(str(response)))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/markdown": [
+       "House Speaker Nancy Pelosi and the Democratic majority are against the proposal of offshore drilling in California. Pelosi stated that offshore drilling is \"off the table,\" and Democrats have been consistently unwilling to bend environmental rules. They argue that oil companies are not using the 68 million acres of federal lands already leased to them, either because it takes a long time or they lack the necessary equipment."
+      ],
+      "text/plain": [
+       "<IPython.core.display.Markdown object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "response = query_engine.query(\n",
+    "    \"Who is against the proposal of offshore drilling in CA and why?\"\n",
+    ")\n",
+    "display(Markdown(str(response)))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/markdown": [
+       "The purpose of the Ocean Science and Technology Subcommittee (SOST) is to advise and assist the Committee on Environment, Natural Resources, and Sustainability on national issues of ocean science and technology. The SOST aims to contribute to the goals for Federal ocean science and technology by developing coordinated interagency strategies. It also retains the functions of the previously-chartered Joint Subcommittee on Ocean Science and Technology and serves as the Ocean Science and Technology Interagency Policy Committee for the National Ocean Council."
+      ],
+      "text/plain": [
+       "<IPython.core.display.Markdown object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "response = query_engine.query(\n",
+    "    \"What is the purpose of the Ocean Science and Technology Subcommittee?\"\n",
+    ")\n",
+    "display(Markdown(str(response)))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/markdown": [
+       "The immigration appeal is dismissed because the petitioner is not a U.S. citizen, and therefore, is not eligible to file a Petition for Alien Fiancé(e) (Form I-129F) on behalf of the beneficiary. The relevant law provides nonimmigrant classification only to aliens who are the fiancé(e)s of U.S. citizens."
+      ],
+      "text/plain": [
+       "<IPython.core.display.Markdown object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "response = query_engine.query(\"Why is the immigration appeal dismissed?\")\n",
+    "display(Markdown(str(response)))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/markdown": [
+       "An advance pricing agreement (APA) is a binding contract between a taxpayer and the IRS that establishes an approved transfer pricing method (TPM) for specific transactions. This agreement aims to prevent disputes over transfer pricing by ensuring that the taxpayer's tax returns for the covered years are consistent with the agreed TPM. APAs can be unilateral, involving only the taxpayer and the IRS, or bilateral/multilateral, involving agreements with one or more foreign tax authorities to avoid double taxation."
+      ],
+      "text/plain": [
+       "<IPython.core.display.Markdown object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "response = query_engine.query(\"What is an advance pricing agreement?\")\n",
+    "display(Markdown(str(response)))"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "llama-parse-5ZmnAQ0r-py3.11",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}

examples/parse/multimodal/multimodal_rag_slide_deck.ipynb

@@ -0,0 +1,999 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "id": "93ae9bad-b8cc-43de-ba7d-387e0155674c",
+   "metadata": {},
+   "source": [
+    "# Building a Natively Multimodal RAG Pipeline (over a Slide Deck)\n",
+    "\n",
+    "<a href=\"https://colab.research.google.com/github/run-llama/llama_cloud_services/blob/main/examples/parse/multimodal/multimodal_rag_slide_deck.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
+    "\n",
+    "In this cookbook we show you how to build a multimodal RAG pipeline over a slide deck, with text, tables, images, diagrams, and complex layouts.\n",
+    "\n",
+    "A gap of text-based RAG is that they struggle with purely text-based representations of complex documents. For instance, if a page contains a lot of images and diagrams, a text parser would need to rely on raw OCR to extract out text. You can also use a multimodal model (e.g. gpt-4o and up) to do text extraction, but this is inherently a lossy conversion.\n",
+    "\n",
+    "Instead a **native multimodal pipeline** stores both a text and image representation of a document chunk. They are indexed via embeddings (text or image), and during synthesis both text and image are directly fed to the multimodal model for synthesis.\n",
+    "\n",
+    "This can have the following advantages:\n",
+    "- **Robustness**: This solution is more robust than a pure text or even a pure image-based approach. In a pure text RAG approach, the parsing piece can be lossy. In a pure image-based approach, multimodal OCR is not perfect and may lose out against text parsing for text-heavy documents.\n",
+    "- **Cost Optimization**: You may choose to dynamically include text-only, or text + image depending on the content of the page.\n",
+    "\n",
+    "![mm_rag_diagram](./multimodal_rag_slide_deck_img.png)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "54e8d9a7-5036-4d32-818f-00b2e888521f",
+   "metadata": {},
+   "source": [
+    "## Setup"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "70ccdd53-e68a-4199-aacb-cfe71ad1ff0b",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import nest_asyncio\n",
+    "\n",
+    "nest_asyncio.apply()"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "225c5556-a789-4386-a1ee-cce01dbeb6cf",
+   "metadata": {},
+   "source": [
+    "### Setup Observability\n",
+    "\n",
+    "We setup an integration with LlamaTrace (integration with Arize).\n",
+    "\n",
+    "If you haven't already done so, make sure to create an account here: https://llamatrace.com/login. Then create an API key and put it in the `PHOENIX_API_KEY` variable below."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "0eabee1f-290a-4c85-b362-54f45c8559ae",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install -U llama-index-callbacks-arize-phoenix"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "aaeb245c-730b-4c34-ad68-708fdde0e6cb",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# setup Arize Phoenix for logging/observability\n",
+    "import llama_index.core\n",
+    "import os\n",
+    "\n",
+    "PHOENIX_API_KEY = \"<PHOENIX_API_KEY>\"\n",
+    "os.environ[\"OTEL_EXPORTER_OTLP_HEADERS\"] = f\"api_key={PHOENIX_API_KEY}\"\n",
+    "llama_index.core.set_global_handler(\n",
+    "    \"arize_phoenix\", endpoint=\"https://llamatrace.com/v1/traces\"\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "fbb362db-b1b1-4eea-be1a-b1f78b0779d7",
+   "metadata": {},
+   "source": [
+    "### Load Data\n",
+    "\n",
+    "Here we load the [Conoco Phillips 2023 investor meeting slide deck](https://static.conocophillips.com/files/2023-conocophillips-aim-presentation.pdf)."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "8bce3407-a7d2-47e8-9eaf-ab297a94750c",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!mkdir data\n",
+    "!mkdir data_images\n",
+    "!wget \"https://static.conocophillips.com/files/2023-conocophillips-aim-presentation.pdf\" -O data/conocophillips.pdf"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "246ba6b0-51af-42f9-b1b2-8d3e721ef782",
+   "metadata": {},
+   "source": [
+    "### Model Setup\n",
+    "\n",
+    "Setup models that will be used for downstream orchestration."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "16e2071d-bbc2-4707-8ae7-cb4e1fecafd3",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from llama_index.core import Settings\n",
+    "from llama_index.llms.openai import OpenAI\n",
+    "from llama_index.embeddings.openai import OpenAIEmbedding\n",
+    "\n",
+    "embed_model = OpenAIEmbedding(model=\"text-embedding-3-large\")\n",
+    "llm = OpenAI(model=\"gpt-4o\")\n",
+    "\n",
+    "Settings.embed_model = embed_model\n",
+    "Settings.llm = llm"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "e3f6416f-f580-4722-aaa9-7f3500408547",
+   "metadata": {},
+   "source": [
+    "## Use LlamaParse to Parse Text and Images\n",
+    "\n",
+    "In this example, use LlamaParse to parse both the text and images from the document.\n",
+    "\n",
+    "We parse out the text in two ways: \n",
+    "- in regular `text` mode using our default text layout algorithm\n",
+    "- in `markdown` mode using GPT-4o (`gpt4o_mode=True`). This also allows us to capture page screenshots"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "570089e5-238a-4dcc-af65-96e7393c2b4d",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from llama_cloud_services import LlamaParse\n",
+    "\n",
+    "\n",
+    "parser_text = LlamaParse(result_type=\"text\")\n",
+    "parser_gpt4o = LlamaParse(result_type=\"markdown\", gpt4o_mode=True)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "ef82a985-4088-4bb7-9a21-0318e1b9207d",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Parsing text...\n",
+      "Started parsing the file under job_id 62f157a9-9ef9-4e5b-95ac-67093fa25800\n",
+      "..........Parsing PDF file...\n",
+      "Started parsing the file under job_id 1ddd5654-062b-4e19-b488-d66efc9c509d\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(f\"Parsing text...\")\n",
+    "docs_text = parser_text.load_data(\"data/conocophillips.pdf\")\n",
+    "print(f\"Parsing PDF file...\")\n",
+    "md_json_objs = parser_gpt4o.get_json_result(\"data/conocophillips.pdf\")\n",
+    "md_json_list = md_json_objs[0][\"pages\"]"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "5318fb7b-fe6a-4a8a-b82e-4ed7b4512c37",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "# Commitment to Disciplined Reinvestment Rate\n",
+      "\n",
+      "| Period       | Description                          | Reinvestment Rate | WTI Average |\n",
+      "|--------------|--------------------------------------|-------------------|-------------|\n",
+      "| 2012-2016    | Industry Growth Focus                | >100%             | ~$75/BBL    |\n",
+      "| 2017-2022    | ConocoPhillips Strategy Reset        | <60%              | ~$63/BBL    |\n",
+      "| 2023E        |                                      |                   | at $80/BBL  |\n",
+      "| 2024-2028    | Disciplined Reinvestment Rate        | ~50%              | at $60/BBL  |\n",
+      "| 2029-2032    |                                      | ~6% CFO CAGR      | at $60/BBL  |\n",
+      "\n",
+      "- **Historic Reinvestment Rate**: Gray bars\n",
+      "- **Reinvestment Rate at $60/BBL WTI**: Blue bars\n",
+      "- **Reinvestment Rate at $80/BBL WTI**: Dashed blue lines\n",
+      "\n",
+      "Reinvestment rate and cash from operations (CFO) are non-GAAP measures. Definitions and reconciliations are included in the Appendix.\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(md_json_list[10][\"md\"])"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "eeadb16c-97eb-4622-9551-b34d7f90d72f",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "image_dicts = parser_gpt4o.get_images(md_json_objs, download_path=\"data_images\")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "fd3e098b-0606-4429-b48d-d4fe0140fc0e",
+   "metadata": {},
+   "source": [
+    "## Build Multimodal Index\n",
+    "\n",
+    "In this section we build the multimodal index over the parsed deck. \n",
+    "\n",
+    "We do this by creating **text** nodes from the document that contain metadata referencing the original image path.\n",
+    "\n",
+    "In this example we're indexing the text node for retrieval. The text node has a reference to both the parsed text as well as the image screenshot."
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "3aae2dee-9d85-4604-8a51-705d4db527f7",
+   "metadata": {},
+   "source": [
+    "#### Get Text Nodes"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "18c24174-05ce-417f-8dd2-79c3f375db03",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from llama_index.core.schema import TextNode\n",
+    "from typing import Optional"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "8e331dfe-a627-4e23-8c57-70ab1d9342e4",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# get pages loaded through llamaparse\n",
+    "import re\n",
+    "\n",
+    "\n",
+    "def get_page_number(file_name):\n",
+    "    match = re.search(r\"-page-(\\d+)\\.jpg$\", str(file_name))\n",
+    "    if match:\n",
+    "        return int(match.group(1))\n",
+    "    return 0\n",
+    "\n",
+    "\n",
+    "def _get_sorted_image_files(image_dir):\n",
+    "    \"\"\"Get image files sorted by page.\"\"\"\n",
+    "    raw_files = [f for f in list(Path(image_dir).iterdir()) if f.is_file()]\n",
+    "    sorted_files = sorted(raw_files, key=get_page_number)\n",
+    "    return sorted_files"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "346fe5ef-171e-4a54-9084-7a7805103a13",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from copy import deepcopy\n",
+    "from pathlib import Path\n",
+    "\n",
+    "\n",
+    "# attach image metadata to the text nodes\n",
+    "def get_text_nodes(docs, image_dir=None, json_dicts=None):\n",
+    "    \"\"\"Split docs into nodes, by separator.\"\"\"\n",
+    "    nodes = []\n",
+    "\n",
+    "    image_files = _get_sorted_image_files(image_dir) if image_dir is not None else None\n",
+    "    md_texts = [d[\"md\"] for d in json_dicts] if json_dicts is not None else None\n",
+    "\n",
+    "    doc_chunks = [c for d in docs for c in d.text.split(\"---\")]\n",
+    "    for idx, doc_chunk in enumerate(doc_chunks):\n",
+    "        chunk_metadata = {\"page_num\": idx + 1}\n",
+    "        if image_files is not None:\n",
+    "            image_file = image_files[idx]\n",
+    "            chunk_metadata[\"image_path\"] = str(image_file)\n",
+    "        if md_texts is not None:\n",
+    "            chunk_metadata[\"parsed_text_markdown\"] = md_texts[idx]\n",
+    "        chunk_metadata[\"parsed_text\"] = doc_chunk\n",
+    "        node = TextNode(\n",
+    "            text=\"\",\n",
+    "            metadata=chunk_metadata,\n",
+    "        )\n",
+    "        nodes.append(node)\n",
+    "\n",
+    "    return nodes"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "f591669c-5a8e-491d-9cef-0b754abbf26f",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# this will split into pages\n",
+    "text_nodes = get_text_nodes(docs_text, image_dir=\"data_images\", json_dicts=md_json_list)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "32c13950-c1db-435f-b5b4-89d62b8b7744",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "page_num: 11\n",
+      "image_path: data_images/1ddd5654-062b-4e19-b488-d66efc9c509d-page_39.jpg\n",
+      "parsed_text_markdown: # Commitment to Disciplined Reinvestment Rate\n",
+      "\n",
+      "| Period       | Description                          | Reinvestment Rate | WTI Average |\n",
+      "|--------------|--------------------------------------|-------------------|-------------|\n",
+      "| 2012-2016    | Industry Growth Focus                | >100%             | ~$75/BBL    |\n",
+      "| 2017-2022    | ConocoPhillips Strategy Reset        | <60%              | ~$63/BBL    |\n",
+      "| 2023E        |                                      |                   | at $80/BBL  |\n",
+      "| 2024-2028    | Disciplined Reinvestment Rate        | ~50%              | at $60/BBL  |\n",
+      "| 2029-2032    |                                      | ~6% CFO CAGR      | at $60/BBL  |\n",
+      "\n",
+      "- **Historic Reinvestment Rate**: Gray bars\n",
+      "- **Reinvestment Rate at $60/BBL WTI**: Blue bars\n",
+      "- **Reinvestment Rate at $80/BBL WTI**: Dashed blue lines\n",
+      "\n",
+      "Reinvestment rate and cash from operations (CFO) are non-GAAP measures. Definitions and reconciliations are included in the Appendix.\n",
+      "parsed_text: Commitment to Disciplined Reinvestment Rate\n",
+      "                         Industry                    ConocoPhillips\n",
+      "                                                     Strategy Reset                   Disciplined Reinvestment Rate is the Foundation for Superior\n",
+      "                      Growth Focus                                                    Returns on and of Capital, while Driving Durable CFO Growth\n",
+      "                             100%                           <60%                                        50%                 6%         at $60/BBL WTI\n",
+      "                       Reinvestment Rate               Reinvestment Rate                          Reinvestment Rate10-YearCFO CAGR          Planning PriceMid-Cycle\n",
+      "                                                                                                                         2024-2032\n",
+      "    2   100%\n",
+      "    1    75%\n",
+      "    1    50%\n",
+      "    1                                                                                                                                         WTIat $80/BBL                at S80/BBL\n",
+      "         25%                'S75/BBL                        $63/BBL                                                                                                        WTI\n",
+      "                              WTI                             WTI                         at S80/BBL                       at S60/BBL                      at S60/BBL\n",
+      "                           Average                         Average                            WTI                              WTI                             WTI\n",
+      "          0%\n",
+      "                          2012-2016                        2017-2022                         2023E                          2024-2028                       2029-2032\n",
+      "                                    Historic Reinvestment Rate            Reinvestment Rate at $60/BBL WTI                 Reinvestment Rate at $80/BBL WTI\n",
+      " Reinvestment rate and cash from operations (CFO) are non-GAAP measures: Definitions and reconciliations are included in the Appendix                                  ConocoPhillips\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(text_nodes[10].get_content(metadata_mode=\"all\"))"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "4f404f56-db1e-4ed7-9ba1-ead763546348",
+   "metadata": {},
+   "source": [
+    "#### Build Index\n",
+    "\n",
+    "Once the text nodes are ready, we feed into our vector store index abstraction, which will index these nodes into a simple in-memory vector store (of course, you should definitely check out our 40+ vector store integrations!)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "6ea53c31-0e38-421c-8d9b-0e3adaa1677e",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "/Users/jerryliu/Programming/gpt_index/.venv/lib/python3.10/site-packages/tiktoken/core.py:50: RuntimeWarning: coroutine 'LlamaParse.aload_data' was never awaited\n",
+      "  self._core_bpe = _tiktoken.CoreBPE(mergeable_ranks, special_tokens, pat_str)\n",
+      "RuntimeWarning: Enable tracemalloc to get the object allocation traceback\n"
+     ]
+    }
+   ],
+   "source": [
+    "import os\n",
+    "from llama_index.core import (\n",
+    "    StorageContext,\n",
+    "    VectorStoreIndex,\n",
+    "    load_index_from_storage,\n",
+    ")\n",
+    "\n",
+    "if not os.path.exists(\"storage_nodes\"):\n",
+    "    index = VectorStoreIndex(text_nodes, embed_model=embed_model)\n",
+    "    # save index to disk\n",
+    "    index.set_index_id(\"vector_index\")\n",
+    "    index.storage_context.persist(\"./storage_nodes\")\n",
+    "else:\n",
+    "    # rebuild storage context\n",
+    "    storage_context = StorageContext.from_defaults(persist_dir=\"storage_nodes\")\n",
+    "    # load index\n",
+    "    index = load_index_from_storage(storage_context, index_id=\"vector_index\")\n",
+    "\n",
+    "retriever = index.as_retriever()"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "5f0e33a4-9422-498d-87ee-d917bdf74d80",
+   "metadata": {},
+   "source": [
+    "## Build Multimodal Query Engine\n",
+    "\n",
+    "We now use LlamaIndex abstractions to build a **custom query engine**. In contrast to a standard RAG query engine that will retrieve the text node and only put that into the prompt (response synthesis module), this custom query engine will also load the image document, and put both the text and image document into the response synthesis module."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "35a94be2-e289-41a6-92e4-d3cb428fb0c8",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from llama_index.core.query_engine import CustomQueryEngine, SimpleMultiModalQueryEngine\n",
+    "from llama_index.core.retrievers import BaseRetriever\n",
+    "from llama_index.multi_modal_llms.openai import OpenAIMultiModal\n",
+    "from llama_index.core.schema import ImageNode, NodeWithScore, MetadataMode\n",
+    "from llama_index.core.prompts import PromptTemplate\n",
+    "from llama_index.core.base.response.schema import Response\n",
+    "from typing import Optional\n",
+    "\n",
+    "\n",
+    "gpt_4o = OpenAIMultiModal(model=\"gpt-4o\", max_new_tokens=4096)\n",
+    "\n",
+    "QA_PROMPT_TMPL = \"\"\"\\\n",
+    "Below we give parsed text from slides in two different formats, as well as the image.\n",
+    "\n",
+    "We parse the text in both 'markdown' mode as well as 'raw text' mode. Markdown mode attempts \\\n",
+    "to convert relevant diagrams into tables, whereas raw text tries to maintain the rough spatial \\\n",
+    "layout of the text.\n",
+    "\n",
+    "Use the image information first and foremost. ONLY use the text/markdown information \n",
+    "if you can't understand the image.\n",
+    "\n",
+    "---------------------\n",
+    "{context_str}\n",
+    "---------------------\n",
+    "Given the context information and not prior knowledge, answer the query. Explain whether you got the answer\n",
+    "from the parsed markdown or raw text or image, and if there's discrepancies, and your reasoning for the final answer.\n",
+    "\n",
+    "Query: {query_str}\n",
+    "Answer: \"\"\"\n",
+    "\n",
+    "QA_PROMPT = PromptTemplate(QA_PROMPT_TMPL)\n",
+    "\n",
+    "\n",
+    "class MultimodalQueryEngine(CustomQueryEngine):\n",
+    "    \"\"\"Custom multimodal Query Engine.\n",
+    "\n",
+    "    Takes in a retriever to retrieve a set of document nodes.\n",
+    "    Also takes in a prompt template and multimodal model.\n",
+    "\n",
+    "    \"\"\"\n",
+    "\n",
+    "    qa_prompt: PromptTemplate\n",
+    "    retriever: BaseRetriever\n",
+    "    multi_modal_llm: OpenAIMultiModal\n",
+    "\n",
+    "    def __init__(self, qa_prompt: Optional[PromptTemplate] = None, **kwargs) -> None:\n",
+    "        \"\"\"Initialize.\"\"\"\n",
+    "        super().__init__(qa_prompt=qa_prompt or QA_PROMPT, **kwargs)\n",
+    "\n",
+    "    def custom_query(self, query_str: str):\n",
+    "        # retrieve text nodes\n",
+    "        nodes = self.retriever.retrieve(query_str)\n",
+    "        # create ImageNode items from text nodes\n",
+    "        image_nodes = [\n",
+    "            NodeWithScore(node=ImageNode(image_path=n.metadata[\"image_path\"]))\n",
+    "            for n in nodes\n",
+    "        ]\n",
+    "\n",
+    "        # create context string from text nodes, dump into the prompt\n",
+    "        context_str = \"\\n\\n\".join(\n",
+    "            [r.get_content(metadata_mode=MetadataMode.LLM) for r in nodes]\n",
+    "        )\n",
+    "        fmt_prompt = self.qa_prompt.format(context_str=context_str, query_str=query_str)\n",
+    "\n",
+    "        # synthesize an answer from formatted text and images\n",
+    "        llm_response = self.multi_modal_llm.complete(\n",
+    "            prompt=fmt_prompt,\n",
+    "            image_documents=[image_node.node for image_node in image_nodes],\n",
+    "        )\n",
+    "        return Response(\n",
+    "            response=str(llm_response),\n",
+    "            source_nodes=nodes,\n",
+    "            metadata={\"text_nodes\": text_nodes, \"image_nodes\": image_nodes},\n",
+    "        )\n",
+    "\n",
+    "        return response"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "0890be59-fb12-4bb5-959b-b2d9600f7774",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "query_engine = MultimodalQueryEngine(\n",
+    "    retriever=index.as_retriever(similarity_top_k=9), multi_modal_llm=gpt_4o\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "a92aa4f1-7501-4711-b054-f02338e54e74",
+   "metadata": {},
+   "source": [
+    "### Define Baseline\n",
+    "\n",
+    "In addition, we define a \"baseline\" where we rely only on text-based indexing. Here we define an index using only the nodes that are parsed in text-mode from LlamaParse. \n",
+    "\n",
+    "**NOTE**: We don't currently include the markdown-parsed text because that was parsed with GPT-4o, so already uses a multimodal model during the text extraction phase.\n",
+    "\n",
+    "It is of course a valid experiment to compare RAG where multimodal extraction only happens during indexing, vs. the current multimodal RAG implementation where images are fed during synthesis to the LLM. "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "c0b15a48-d177-4666-aec2-98ee90664642",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "def get_nodes(docs):\n",
+    "    \"\"\"Split docs into nodes, by separator.\"\"\"\n",
+    "    nodes = []\n",
+    "    for doc in docs:\n",
+    "        doc_chunks = doc.text.split(\"\\n---\\n\")\n",
+    "        for doc_chunk in doc_chunks:\n",
+    "            node = TextNode(\n",
+    "                text=doc_chunk,\n",
+    "                metadata=deepcopy(doc.metadata),\n",
+    "            )\n",
+    "            nodes.append(node)\n",
+    "\n",
+    "    return nodes"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "2065d2c6-d6ba-4ee3-8e9e-dbc83cbcec1b",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "base_nodes = get_nodes(docs_text)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "bcaea1a8-26c9-4385-8f62-32855aa898b6",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Our Differentiated Portfolio: Deep; Durable and Diverse\n",
+      "                              20 BBOE of Resource                                           Diverse Production Base\n",
+      "                            Under $40/BBL Cost of Supply                              10-Year Plan Cumulative Production (BBOE)\n",
+      "      S50                   S32/BBL                                                Lower 48                           Alaska\n",
+      "                    Average Cost of Supply\n",
+      "  3 $40                                                                                                                       GKA        GWA\n",
+      "                                                                                                                      GPA     WNS\n",
+      "      $30                                                                                                             EMENA\n",
+      "  3                                                                                                                              Norway\n",
+      "  8   $20\n",
+      "  E                                                                                                                   Qatar      Libya\n",
+      "                                                                                                                      Asia Pacific Canada\n",
+      "      $10                                                                          Permian\n",
+      "                                                                                                                      APLNG        Montney\n",
+      "       S0\n",
+      "                                               10              15              20                         Bakken\n",
+      "                                         Resource (BBOE)                           Eagle Ford             Other       Malaysia ChinaSurmont\n",
+      "                 Lower 48      Canada       Alaska      EMENA    Asia Pacific\n",
+      "Costs assumemid-cycle price environment of S60/BBL WTI:\n",
+      "                                                                                                                                           ConocoPhillips\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(base_nodes[13].get_content(metadata_mode=\"all\"))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "f6bcfbc6-4e9b-41ad-ad81-1c4245b95cd5",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "base_index = VectorStoreIndex(base_nodes, embed_model=embed_model)\n",
+    "base_query_engine = base_index.as_query_engine(llm=llm, similarity_top_k=9)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "1f94ef26-0df5-4468-a156-903d686f02ce",
+   "metadata": {},
+   "source": [
+    "## Build a Multimodal Agent\n",
+    "\n",
+    "Build an agent around the multimodal query engine. This gives you agent capabilities like query planning/decomposition and memory around a central QA interface."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "5b7a8c5f-39fc-4d04-8c56-3642f5718437",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from llama_index.core.tools import QueryEngineTool\n",
+    "from llama_index.core.agent import FunctionCallingAgentWorker\n",
+    "\n",
+    "\n",
+    "vector_tool = QueryEngineTool.from_defaults(\n",
+    "    query_engine=query_engine,\n",
+    "    name=\"vector_tool\",\n",
+    "    description=(\n",
+    "        \"Useful for retrieving specific context from the data. Do NOT select if question asks for a summary of the data.\"\n",
+    "    ),\n",
+    ")\n",
+    "agent = FunctionCallingAgentWorker.from_tools(\n",
+    "    [vector_tool], llm=llm, verbose=True\n",
+    ").as_agent()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "2b4f7eb1-d247-45fa-bb41-c02fc353a22a",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# define a similar agent for the baseline\n",
+    "base_vector_tool = QueryEngineTool.from_defaults(\n",
+    "    query_engine=base_query_engine,\n",
+    "    name=\"vector_tool\",\n",
+    "    description=(\n",
+    "        \"Useful for retrieving specific context from the data. Do NOT select if question asks for a summary of the data.\"\n",
+    "    ),\n",
+    ")\n",
+    "base_agent = FunctionCallingAgentWorker.from_tools(\n",
+    "    [base_vector_tool], llm=llm, verbose=True\n",
+    ").as_agent()"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "2336f98b-c0a1-413a-849d-8a89bacb90b5",
+   "metadata": {},
+   "source": [
+    "## Try out Queries\n",
+    "\n",
+    "Let's try out queries against these documents and compare against each other."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "d78e53cf-35cb-4ef8-b03e-1b47ba15ae64",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Added user message to memory: Tell me about the diverse geographies where Conoco Phillips has a production base\n",
+      "=== Calling Function ===\n",
+      "Calling function: vector_tool with args: {\"input\": \"Conoco Phillips production base geographies\"}\n",
+      "=== Function Output ===\n",
+      "ConocoPhillips' production base geographies include:\n",
+      "\n",
+      "1. **Lower 48** (Permian, Eagle Ford, Bakken, Other)\n",
+      "2. **Alaska** (GKA, GWA, GPA, WNS)\n",
+      "3. **EMENA** (Norway, Libya, Qatar)\n",
+      "4. **Asia Pacific** (APLNG, Malaysia, China)\n",
+      "5. **Canada** (Montney, Surmont)\n",
+      "\n",
+      "This information was derived from the image on page 14, which provides a detailed breakdown of the diverse production base and the regions involved. The parsed markdown and raw text also support this information, but the image provides the clearest and most comprehensive view. There are no discrepancies between the image and the parsed text in this case.\n",
+      "=== LLM Response ===\n",
+      "ConocoPhillips has a diverse production base spread across various geographies, including:\n",
+      "\n",
+      "1. **Lower 48**:\n",
+      "   - Permian Basin\n",
+      "   - Eagle Ford\n",
+      "   - Bakken\n",
+      "   - Other regions within the continental United States\n",
+      "\n",
+      "2. **Alaska**:\n",
+      "   - Greater Kuparuk Area (GKA)\n",
+      "   - Greater Prudhoe Area (GPA)\n",
+      "   - Greater Willow Area (GWA)\n",
+      "   - Western North Slope (WNS)\n",
+      "\n",
+      "3. **EMENA (Europe, Middle East, and North Africa)**:\n",
+      "   - Norway\n",
+      "   - Libya\n",
+      "   - Qatar\n",
+      "\n",
+      "4. **Asia Pacific**:\n",
+      "   - Australia Pacific LNG (APLNG)\n",
+      "   - Malaysia\n",
+      "   - China\n",
+      "\n",
+      "5. **Canada**:\n",
+      "   - Montney\n",
+      "   - Surmont\n",
+      "\n",
+      "These regions highlight the global reach and diverse geographical footprint of ConocoPhillips' production operations.\n",
+      "Added user message to memory: Tell me about the diverse geographies where Conoco Phillips has a production base\n",
+      "=== Calling Function ===\n",
+      "Calling function: vector_tool with args: {\"input\": \"diverse geographies where Conoco Phillips has a production base\"}\n",
+      "=== Function Output ===\n",
+      "ConocoPhillips has a diverse production base that includes the Lower 48 (Permian, Bakken, Eagle Ford), Alaska, Canada (Montney, Surmont), EMENA (Norway, Libya), Asia Pacific (Malaysia, China, APLNG), and Qatar.\n",
+      "=== LLM Response ===\n",
+      "ConocoPhillips has a diverse production base spanning several key geographies:\n",
+      "\n",
+      "1. **Lower 48 (United States)**: This includes major production areas such as the Permian Basin, Bakken Formation, and Eagle Ford Shale.\n",
+      "2. **Alaska**: Significant operations in the North Slope region.\n",
+      "3. **Canada**: Operations in the Montney Formation and the Surmont oil sands project.\n",
+      "4. **EMENA (Europe, Middle East, and North Africa)**: Notable operations in Norway and Libya.\n",
+      "5. **Asia Pacific**: Includes operations in Malaysia, China, and the Australia Pacific LNG (APLNG) project.\n",
+      "6. **Qatar**: Involvement in the country's energy sector.\n",
+      "\n",
+      "These regions highlight the company's extensive and varied geographical footprint in the energy production industry.\n"
+     ]
+    }
+   ],
+   "source": [
+    "query = (\n",
+    "    \"Tell me about the diverse geographies where Conoco Phillips has a production base\"\n",
+    ")\n",
+    "response = agent.query(query)\n",
+    "base_response = base_agent.query(query)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "355d2aa4-c26f-480e-b512-4446acbd9227",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "ConocoPhillips has a diverse production base spread across various geographies, including:\n",
+      "\n",
+      "1. **Lower 48**:\n",
+      "   - Permian Basin\n",
+      "   - Eagle Ford\n",
+      "   - Bakken\n",
+      "   - Other regions within the continental United States\n",
+      "\n",
+      "2. **Alaska**:\n",
+      "   - Greater Kuparuk Area (GKA)\n",
+      "   - Greater Prudhoe Area (GPA)\n",
+      "   - Greater Willow Area (GWA)\n",
+      "   - Western North Slope (WNS)\n",
+      "\n",
+      "3. **EMENA (Europe, Middle East, and North Africa)**:\n",
+      "   - Norway\n",
+      "   - Libya\n",
+      "   - Qatar\n",
+      "\n",
+      "4. **Asia Pacific**:\n",
+      "   - Australia Pacific LNG (APLNG)\n",
+      "   - Malaysia\n",
+      "   - China\n",
+      "\n",
+      "5. **Canada**:\n",
+      "   - Montney\n",
+      "   - Surmont\n",
+      "\n",
+      "These regions highlight the global reach and diverse geographical footprint of ConocoPhillips' production operations.\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(str(response))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "d584c560-8f49-4c10-a4db-2e0d3b7085d2",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "page_num: 14\n",
+      "image_path: data_images/1ddd5654-062b-4e19-b488-d66efc9c509d-page_12.jpg\n",
+      "parsed_text_markdown: # Our Differentiated Portfolio: Deep, Durable and Diverse\n",
+      "\n",
+      "## ~20 BBOE of Resource\n",
+      "Under $40/BBL Cost of Supply\n",
+      "\n",
+      "### ~ $32/BBL\n",
+      "Average Cost of Supply\n",
+      "\n",
+      "### WTI Cost of Supply ($/BBL)\n",
+      "\n",
+      "| Cost ($/BBL) | Resource (BBOE) |\n",
+      "|--------------|-----------------|\n",
+      "| $0           | 0               |\n",
+      "| $10          |                 |\n",
+      "| $20          |                 |\n",
+      "| $30          |                 |\n",
+      "| $40          |                 |\n",
+      "| $50          |                 |\n",
+      "\n",
+      "- **Legend:**\n",
+      "  - Lower 48\n",
+      "  - Canada\n",
+      "  - Alaska\n",
+      "  - EMENA\n",
+      "  - Asia Pacific\n",
+      "\n",
+      "*Costs assume a mid-cycle price environment of $60/BBL WTI.*\n",
+      "\n",
+      "## Diverse Production Base\n",
+      "10-Year Plan Cumulative Production (BBOE)\n",
+      "\n",
+      "| Region       | Sub-region      |\n",
+      "|--------------|-----------------|\n",
+      "| Lower 48     | Permian         |\n",
+      "|              | Eagle Ford      |\n",
+      "|              | Bakken          |\n",
+      "|              | Other           |\n",
+      "| Alaska       | GKA             |\n",
+      "|              | GWA             |\n",
+      "|              | GPA             |\n",
+      "|              | WNS             |\n",
+      "| EMENA        | Norway          |\n",
+      "|              | Libya           |\n",
+      "|              | Qatar           |\n",
+      "| Asia Pacific | APLNG           |\n",
+      "|              | Malaysia        |\n",
+      "|              | China           |\n",
+      "| Canada       | Montney         |\n",
+      "|              | Surmont         |\n",
+      "parsed_text: Our Differentiated Portfolio: Deep; Durable and Diverse\n",
+      "                              20 BBOE of Resource                                           Diverse Production Base\n",
+      "                            Under $40/BBL Cost of Supply                              10-Year Plan Cumulative Production (BBOE)\n",
+      "      S50                   S32/BBL                                                Lower 48                           Alaska\n",
+      "                    Average Cost of Supply\n",
+      "  3 $40                                                                                                                       GKA        GWA\n",
+      "                                                                                                                      GPA     WNS\n",
+      "      $30                                                                                                             EMENA\n",
+      "  3                                                                                                                              Norway\n",
+      "  8   $20\n",
+      "  E                                                                                                                   Qatar      Libya\n",
+      "                                                                                                                      Asia Pacific Canada\n",
+      "      $10                                                                          Permian\n",
+      "                                                                                                                      APLNG        Montney\n",
+      "       S0\n",
+      "                                               10              15              20                         Bakken\n",
+      "                                         Resource (BBOE)                           Eagle Ford             Other       Malaysia ChinaSurmont\n",
+      "                 Lower 48      Canada       Alaska      EMENA    Asia Pacific\n",
+      "Costs assumemid-cycle price environment of S60/BBL WTI:\n",
+      "                                                                                                                                           ConocoPhillips\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(response.source_nodes[7].get_content(metadata_mode=\"all\"))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "d21d694b-6618-4d04-a6f6-8b0c2625f539",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "ConocoPhillips has a diverse production base spanning several key geographies:\n",
+      "\n",
+      "1. **Lower 48 (United States)**: This includes major production areas such as the Permian Basin, Bakken Formation, and Eagle Ford Shale.\n",
+      "2. **Alaska**: Significant operations in the North Slope region.\n",
+      "3. **Canada**: Operations in the Montney Formation and the Surmont oil sands project.\n",
+      "4. **EMENA (Europe, Middle East, and North Africa)**: Notable operations in Norway and Libya.\n",
+      "5. **Asia Pacific**: Includes operations in Malaysia, China, and the Australia Pacific LNG (APLNG) project.\n",
+      "6. **Qatar**: Involvement in the country's energy sector.\n",
+      "\n",
+      "These regions highlight the company's extensive and varied geographical footprint in the energy production industry.\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(str(base_response))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "d3afccae-ad8d-4c5d-9d93-810dba413a5d",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Our Differentiated Portfolio: Deep; Durable and Diverse\n",
+      "                              20 BBOE of Resource                                           Diverse Production Base\n",
+      "                            Under $40/BBL Cost of Supply                              10-Year Plan Cumulative Production (BBOE)\n",
+      "      S50                   S32/BBL                                                Lower 48                           Alaska\n",
+      "                    Average Cost of Supply\n",
+      "  3 $40                                                                                                                       GKA        GWA\n",
+      "                                                                                                                      GPA     WNS\n",
+      "      $30                                                                                                             EMENA\n",
+      "  3                                                                                                                              Norway\n",
+      "  8   $20\n",
+      "  E                                                                                                                   Qatar      Libya\n",
+      "                                                                                                                      Asia Pacific Canada\n",
+      "      $10                                                                          Permian\n",
+      "                                                                                                                      APLNG        Montney\n",
+      "       S0\n",
+      "                                               10              15              20                         Bakken\n",
+      "                                         Resource (BBOE)                           Eagle Ford             Other       Malaysia ChinaSurmont\n",
+      "                 Lower 48      Canada       Alaska      EMENA    Asia Pacific\n",
+      "Costs assumemid-cycle price environment of S60/BBL WTI:\n",
+      "                                                                                                                                           ConocoPhillips\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(base_response.source_nodes[1].get_content(metadata_mode=\"all\"))"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "llama_index_v3",
+   "language": "python",
+   "name": "llama_index_v3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

examples/parse/multimodal/product_manual_rag.ipynb

@@ -0,0 +1,834 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "# Building a RAG Pipeline over IKEA Product Instruction Manuals\n",
+    "\n",
+    "<a href=\"https://colab.research.google.com/github/run-llama/llama_cloud_services/blob/main/examples/parse/multimodal/product_manual_rag.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "This cookbook shows how to use LlamaParse and OpenAI's multimodal models to query over IKEA instruction manual PDFs, which mainly contain images and diagrams to show how one can assemble the product.\n",
+    "\n",
+    "LlamaParse and multimodal LLMs can interpret these diagrams and translate them into textual instructions. With textual assistance, confusing visual instructions within the IKEA product manuals can be made easier to understand and interpret. Additionally, textual instructions can be helpful for those who are visually impaired."
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Install and Setup\n",
+    "\n",
+    "Install LlamaIndex, download the data, and apply `nest_asyncio`."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "%pip install llama-index llama-parse llama-index-multi-modal-llms-openai git+https://github.com/openai/CLIP.git"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!wget https://github.com/user-attachments/files/16461058/data.zip -O data.zip\n",
+    "!unzip -o data.zip\n",
+    "!rm data.zip"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import nest_asyncio\n",
+    "\n",
+    "nest_asyncio.apply()"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Set up your OpenAI and LlamaCloud keys."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import os\n",
+    "\n",
+    "os.environ[\"OPENAI_API_KEY\"] = \"<Your OpenAI API Key>\"\n",
+    "os.environ[\"LLAMA_CLOUD_API_KEY\"] = \"<Your LlamaCloud API Key>\""
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Code Implementation"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Set up LlamaParse. We will parse the PDF files into markdown and use the GPT-4o multimodal model to parse the PDFs."
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Load data from the parser."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from llama_cloud_services import LlamaParse\n",
+    "\n",
+    "parser = LlamaParse(\n",
+    "    result_type=\"markdown\",\n",
+    "    parsing_instruction=\"You are given IKEA assembly instruction manuals\",\n",
+    "    use_vendor_multimodal_model=True,\n",
+    "    vendor_multimodal_model_name=\"openai-gpt4o\",\n",
+    "    show_progress=True,\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "DATA_DIR = \"data\"\n",
+    "\n",
+    "\n",
+    "def get_data_files(data_dir=DATA_DIR) -> list[str]:\n",
+    "    files = []\n",
+    "    for f in os.listdir(data_dir):\n",
+    "        fname = os.path.join(data_dir, f)\n",
+    "        if os.path.isfile(fname):\n",
+    "            files.append(fname)\n",
+    "    return files\n",
+    "\n",
+    "\n",
+    "files = get_data_files()"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Load data into docs, and save images from PDFs into `data_images` directory."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "md_json_objs = parser.get_json_result(files)\n",
+    "md_json_list = md_json_objs[0][\"pages\"]\n",
+    "image_dicts = parser.get_images(md_json_objs, download_path=\"data_images\")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Create helper functions to create a list of `TextNode`s from the markdown tables to feed into the `VectorStoreIndex`."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import re\n",
+    "from pathlib import Path\n",
+    "import typing as t\n",
+    "from llama_index.core.schema import TextNode\n",
+    "\n",
+    "\n",
+    "def get_page_number(file_name):\n",
+    "    \"\"\"Gets page number of images using regex on file names\"\"\"\n",
+    "    match = re.search(r\"-page-(\\d+)\\.jpg$\", str(file_name))\n",
+    "    if match:\n",
+    "        return int(match.group(1))\n",
+    "    return 0\n",
+    "\n",
+    "\n",
+    "def _get_sorted_image_files(image_dir):\n",
+    "    \"\"\"Get image files sorted by page.\"\"\"\n",
+    "    raw_files = [f for f in list(Path(image_dir).iterdir()) if f.is_file()]\n",
+    "    sorted_files = sorted(raw_files, key=get_page_number)\n",
+    "    return sorted_files\n",
+    "\n",
+    "\n",
+    "def get_text_nodes(json_dicts, image_dir) -> t.List[TextNode]:\n",
+    "    \"\"\"Creates nodes from json + images\"\"\"\n",
+    "\n",
+    "    nodes = []\n",
+    "\n",
+    "    docs = [doc[\"md\"] for doc in json_dicts]  # extract text\n",
+    "    image_files = _get_sorted_image_files(image_dir)  # extract images\n",
+    "\n",
+    "    for idx, doc in enumerate(docs):\n",
+    "        # adds both a text node and the corresponding image node (jpg of the page) for each page\n",
+    "        node = TextNode(\n",
+    "            text=doc,\n",
+    "            metadata={\"image_path\": str(image_files[idx]), \"page_num\": idx + 1},\n",
+    "        )\n",
+    "        nodes.append(node)\n",
+    "\n",
+    "    return nodes\n",
+    "\n",
+    "\n",
+    "text_nodes = get_text_nodes(md_json_list, \"data_images\")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Index the documents."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from llama_index.core import (\n",
+    "    VectorStoreIndex,\n",
+    "    StorageContext,\n",
+    "    load_index_from_storage,\n",
+    "    Settings,\n",
+    ")\n",
+    "from llama_index.embeddings.openai import OpenAIEmbedding\n",
+    "from llama_index.llms.openai import OpenAI\n",
+    "\n",
+    "embed_model = OpenAIEmbedding(model=\"text-embedding-3-large\")\n",
+    "llm = OpenAI(\"gpt-4o\")\n",
+    "\n",
+    "Settings.llm = llm\n",
+    "Settings.embed_model = embed_model\n",
+    "\n",
+    "if not os.path.exists(\"storage_ikea\"):\n",
+    "    index = VectorStoreIndex(text_nodes, embed_model=embed_model)\n",
+    "    index.storage_context.persist(persist_dir=\"./storage_ikea\")\n",
+    "else:\n",
+    "    ctx = StorageContext.from_defaults(persist_dir=\"./storage_ikea\")\n",
+    "    index = load_index_from_storage(ctx)\n",
+    "\n",
+    "retriever = index.as_retriever()"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Create a custom query engine that uses GPT-4o's multimodal model."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from llama_index.core.query_engine import CustomQueryEngine\n",
+    "from llama_index.core.retrievers import BaseRetriever\n",
+    "from llama_index.multi_modal_llms.openai import OpenAIMultiModal\n",
+    "from llama_index.core.schema import NodeWithScore, MetadataMode\n",
+    "from llama_index.core.base.response.schema import Response\n",
+    "from llama_index.core.prompts import PromptTemplate\n",
+    "from llama_index.core.schema import ImageNode\n",
+    "\n",
+    "QA_PROMPT_TMPL = \"\"\"\\\n",
+    "Below we give parsed text from slides in two different formats, as well as the image.\n",
+    "\n",
+    "We parse the text in both 'markdown' mode as well as 'raw text' mode. Markdown mode attempts \\\n",
+    "to convert relevant diagrams into tables, whereas raw text tries to maintain the rough spatial \\\n",
+    "layout of the text.\n",
+    "\n",
+    "Use the image information first and foremost. ONLY use the text/markdown information \n",
+    "if you can't understand the image.\n",
+    "\n",
+    "---------------------\n",
+    "{context_str}\n",
+    "---------------------\n",
+    "Given the context information and not prior knowledge, answer the query. Explain whether you got the answer\n",
+    "from the parsed markdown or raw text or image, and if there's discrepancies, and your reasoning for the final answer.\n",
+    "\n",
+    "Query: {query_str}\n",
+    "Answer: \"\"\"\n",
+    "\n",
+    "QA_PROMPT = PromptTemplate(QA_PROMPT_TMPL)\n",
+    "\n",
+    "gpt_4o_mm = OpenAIMultiModal(model=\"gpt-4o\", max_new_tokens=4096)\n",
+    "\n",
+    "\n",
+    "class MultimodalQueryEngine(CustomQueryEngine):\n",
+    "    qa_prompt: PromptTemplate\n",
+    "    retriever: BaseRetriever\n",
+    "    multi_modal_llm: OpenAIMultiModal\n",
+    "\n",
+    "    def __init__(\n",
+    "        self,\n",
+    "        qa_prompt: PromptTemplate,\n",
+    "        retriever: BaseRetriever,\n",
+    "        multi_modal_llm: OpenAIMultiModal,\n",
+    "    ):\n",
+    "        super().__init__(\n",
+    "            qa_prompt=qa_prompt, retriever=retriever, multi_modal_llm=multi_modal_llm\n",
+    "        )\n",
+    "\n",
+    "    def custom_query(self, query_str: str):\n",
+    "        # retrieve most relevant nodes\n",
+    "        nodes = self.retriever.retrieve(query_str)\n",
+    "\n",
+    "        # create image nodes from the image associated with those nodes\n",
+    "        image_nodes = [\n",
+    "            NodeWithScore(node=ImageNode(image_path=n.node.metadata[\"image_path\"]))\n",
+    "            for n in nodes\n",
+    "        ]\n",
+    "\n",
+    "        # create context string from parsed markdown text\n",
+    "        ctx_str = \"\\n\\n\".join(\n",
+    "            [r.node.get_content(metadata_mode=MetadataMode.LLM) for r in nodes]\n",
+    "        )\n",
+    "        # prompt for the LLM\n",
+    "        fmt_prompt = self.qa_prompt.format(context_str=ctx_str, query_str=query_str)\n",
+    "\n",
+    "        # use the multimodal LLM to interpret images and generate a response to the prompt\n",
+    "        llm_repsonse = self.multi_modal_llm.complete(\n",
+    "            prompt=fmt_prompt,\n",
+    "            image_documents=[image_node.node for image_node in image_nodes],\n",
+    "        )\n",
+    "        return Response(\n",
+    "            response=str(llm_repsonse),\n",
+    "            source_nodes=nodes,\n",
+    "            metadata={\"text_nodes\": text_nodes, \"image_nodes\": image_nodes},\n",
+    "        )"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Create a query engine instance."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "query_engine = MultimodalQueryEngine(\n",
+    "    qa_prompt=QA_PROMPT,\n",
+    "    retriever=index.as_retriever(similarity_top_k=9),\n",
+    "    multi_modal_llm=gpt_4o_mm,\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "\n",
+    "## Example Queries"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/markdown": [
+       "The query asks about the parts included in the Uppspel, but the provided images and parsed text do not contain any information about the Uppspel. Instead, they contain information about other IKEA products such as SMÅGÖRA, FREDDE, and TUFFING.\n",
+       "\n",
+       "Therefore, based on the provided images and parsed text, I cannot determine the parts included in the Uppspel. The answer cannot be derived from the given information."
+      ],
+      "text/plain": [
+       "<IPython.core.display.Markdown object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "from IPython.display import display, Markdown\n",
+    "\n",
+    "response = query_engine.query(\"What parts are included in the Uppspel?\")\n",
+    "display(Markdown(str(response)))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/markdown": [
+       "The Tuffing is a bunk bed frame with a minimalist design, featuring a metal frame and safety rails on the top bunk. The image provided shows the Tuffing bunk bed with a ladder for access to the top bunk and a simple, sturdy construction.\n",
+       "\n",
+       "I got the answer from the image provided. The image clearly shows the design and structure of the Tuffing bunk bed. There were no discrepancies between the parsed markdown or raw text and the image. The image was the primary source for understanding what the Tuffing looks like."
+      ],
+      "text/plain": [
+       "<IPython.core.display.Markdown object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "response = query_engine.query(\"What does the Tuffing look like?\")\n",
+    "display(Markdown(str(response)))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/markdown": [
+       "The query asks for step 4 of assembling the Nordli. Based on the provided information, step 4 is described in the parsed text as follows:\n",
+       "\n",
+       "**Step 4:**\n",
+       "- Insert the provided tool into the hole as shown.\n",
+       "- Ensure the structure is properly aligned and secure.\n",
+       "- Push down firmly to lock the structure in place.\n",
+       "\n",
+       "This information was derived from the parsed text, as the image provided does not contain step-by-step instructions for the Nordli assembly. There are no discrepancies between the parsed markdown and raw text for this step."
+      ],
+      "text/plain": [
+       "<IPython.core.display.Markdown object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "response = query_engine.query(\"What is step 4 of assembling the Nordli?\")\n",
+    "display(Markdown(str(response)))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/markdown": [
+       "If you're confused with reading the manual, you should contact IKEA customer service for assistance. This information is derived from the image on page 2, which shows a person with a question mark next to an IKEA box and another person making a phone call to IKEA. This visual cue indicates that contacting IKEA customer service is the recommended action if you need help."
+      ],
+      "text/plain": [
+       "<IPython.core.display.Markdown object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "response = query_engine.query(\n",
+    "    \"What should I do if I'm confused with reading the manual?\"\n",
+    ")\n",
+    "display(Markdown(str(response)))"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "You can also create an agent around the query engine and chat with the agent."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from llama_index.core.agent import FunctionCallingAgentWorker\n",
+    "from llama_index.core.tools import QueryEngineTool\n",
+    "\n",
+    "query_engine_tool = QueryEngineTool.from_defaults(\n",
+    "    query_engine=query_engine,\n",
+    "    name=\"query_engine_tool\",\n",
+    "    description=\"Useful for retrieving specific context from the data. Do NOT select if question asks for a summary of the data.\",\n",
+    ")\n",
+    "agent = FunctionCallingAgentWorker.from_tools(\n",
+    "    [query_engine_tool], llm=llm, verbose=True\n",
+    ").as_agent()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Added user message to memory: Give a step-by-step instruction guide on how to assemble the Smagora\n",
+      "=== Calling Function ===\n",
+      "Calling function: query_engine_tool with args: {\"input\": \"step-by-step instruction guide on how to assemble the Smagora\"}\n",
+      "=== Function Output ===\n",
+      "The step-by-step instruction guide on how to assemble the Smågåra crib is provided in the images. The images show detailed visual instructions for each step of the assembly process, including the tools required, the parts involved, and the specific actions to be taken.\n",
+      "\n",
+      "Here is a summary of the steps based on the images:\n",
+      "\n",
+      "1. **Tools Required**:\n",
+      "   - Flathead screwdriver\n",
+      "   - Phillips screwdriver\n",
+      "   - Hammer\n",
+      "\n",
+      "2. **Preparation**:\n",
+      "   - Do not assemble alone; assemble with a partner.\n",
+      "   - Do not assemble on a hard surface; use a soft surface to avoid damage.\n",
+      "   - If you have questions or need assistance, contact IKEA customer service.\n",
+      "\n",
+      "3. **Step 1**:\n",
+      "   - Insert 12 screws into the designated holes on the frame.\n",
+      "\n",
+      "4. **Step 2**:\n",
+      "   - Align the side panels with the headboard and footboard.\n",
+      "   - Use 4 connectors and secure them with bolts and washers.\n",
+      "   - Tighten using the provided tool.\n",
+      "   - Carefully flip the structure as shown.\n",
+      "\n",
+      "5. **Step 3**:\n",
+      "   - Use the provided Allen key to tighten the screws into the designated holes.\n",
+      "   - Ensure the screws are properly aligned and tightened.\n",
+      "   - Repeat this process for all four screws.\n",
+      "   - Make sure the screws are flush with the surface.\n",
+      "\n",
+      "6. **Step 4**:\n",
+      "   - Insert the provided tool into the hole as shown.\n",
+      "   - Ensure the structure is properly aligned and secure.\n",
+      "   - Push down firmly to lock the structure in place.\n",
+      "\n",
+      "7. **Step 5**:\n",
+      "   - Insert 4 dowels into the designated holes on the board.\n",
+      "\n",
+      "8. **Step 6**:\n",
+      "   - Align the board with the dowels and insert it into the corresponding slots on the frame.\n",
+      "\n",
+      "9. **Step 7**:\n",
+      "   - Insert the top panel into the side panels.\n",
+      "   - Use 4 screws to secure the top panel.\n",
+      "   - Ensure the screws are properly aligned and tightened using the provided tool.\n",
+      "\n",
+      "10. **Step 8**:\n",
+      "    - Carefully flip the assembled structure upright.\n",
+      "    - Use 2 screws to secure the bottom panel.\n",
+      "    - Tighten the screws with the provided tool.\n",
+      "\n",
+      "These steps are derived from the images provided, which offer a clear and detailed visual guide for assembling the Smågåra crib.\n",
+      "=== LLM Response ===\n",
+      "Here is a step-by-step instruction guide on how to assemble the Smågåra crib:\n",
+      "\n",
+      "### Tools Required:\n",
+      "- Flathead screwdriver\n",
+      "- Phillips screwdriver\n",
+      "- Hammer\n",
+      "- Allen key (provided in the package)\n",
+      "\n",
+      "### Preparation:\n",
+      "- **Safety First**: Assemble with a partner to ensure safety and ease.\n",
+      "- **Surface**: Assemble on a soft surface to avoid damaging the parts.\n",
+      "- **Assistance**: If you have questions or need help, contact IKEA customer service.\n",
+      "\n",
+      "### Step-by-Step Assembly:\n",
+      "\n",
+      "#### Step 1: Insert Screws into the Frame\n",
+      "1. Insert 12 screws into the designated holes on the frame.\n",
+      "2. Ensure the screws are properly aligned.\n",
+      "\n",
+      "#### Step 2: Align and Secure Side Panels\n",
+      "1. Align the side panels with the headboard and footboard.\n",
+      "2. Use 4 connectors and secure them with bolts and washers.\n",
+      "3. Tighten the bolts using the provided tool.\n",
+      "4. Carefully flip the structure as shown in the instructions.\n",
+      "\n",
+      "#### Step 3: Tighten Screws\n",
+      "1. Use the provided Allen key to tighten the screws into the designated holes.\n",
+      "2. Ensure the screws are properly aligned and tightened.\n",
+      "3. Repeat this process for all four screws.\n",
+      "4. Make sure the screws are flush with the surface.\n",
+      "\n",
+      "#### Step 4: Lock the Structure\n",
+      "1. Insert the provided tool into the hole as shown.\n",
+      "2. Ensure the structure is properly aligned and secure.\n",
+      "3. Push down firmly to lock the structure in place.\n",
+      "\n",
+      "#### Step 5: Insert Dowels\n",
+      "1. Insert 4 dowels into the designated holes on the board.\n",
+      "\n",
+      "#### Step 6: Align and Insert the Board\n",
+      "1. Align the board with the dowels.\n",
+      "2. Insert the board into the corresponding slots on the frame.\n",
+      "\n",
+      "#### Step 7: Secure the Top Panel\n",
+      "1. Insert the top panel into the side panels.\n",
+      "2. Use 4 screws to secure the top panel.\n",
+      "3. Ensure the screws are properly aligned and tightened using the provided tool.\n",
+      "\n",
+      "#### Step 8: Secure the Bottom Panel\n",
+      "1. Carefully flip the assembled structure upright.\n",
+      "2. Use 2 screws to secure the bottom panel.\n",
+      "3. Tighten the screws with the provided tool.\n",
+      "\n",
+      "By following these steps, you should be able to assemble the Smågåra crib successfully. If you encounter any issues, refer to the visual instructions provided in the package or contact IKEA customer service for assistance.\n"
+     ]
+    },
+    {
+     "data": {
+      "text/markdown": [
+       "Here is a step-by-step instruction guide on how to assemble the Smågåra crib:\n",
+       "\n",
+       "### Tools Required:\n",
+       "- Flathead screwdriver\n",
+       "- Phillips screwdriver\n",
+       "- Hammer\n",
+       "- Allen key (provided in the package)\n",
+       "\n",
+       "### Preparation:\n",
+       "- **Safety First**: Assemble with a partner to ensure safety and ease.\n",
+       "- **Surface**: Assemble on a soft surface to avoid damaging the parts.\n",
+       "- **Assistance**: If you have questions or need help, contact IKEA customer service.\n",
+       "\n",
+       "### Step-by-Step Assembly:\n",
+       "\n",
+       "#### Step 1: Insert Screws into the Frame\n",
+       "1. Insert 12 screws into the designated holes on the frame.\n",
+       "2. Ensure the screws are properly aligned.\n",
+       "\n",
+       "#### Step 2: Align and Secure Side Panels\n",
+       "1. Align the side panels with the headboard and footboard.\n",
+       "2. Use 4 connectors and secure them with bolts and washers.\n",
+       "3. Tighten the bolts using the provided tool.\n",
+       "4. Carefully flip the structure as shown in the instructions.\n",
+       "\n",
+       "#### Step 3: Tighten Screws\n",
+       "1. Use the provided Allen key to tighten the screws into the designated holes.\n",
+       "2. Ensure the screws are properly aligned and tightened.\n",
+       "3. Repeat this process for all four screws.\n",
+       "4. Make sure the screws are flush with the surface.\n",
+       "\n",
+       "#### Step 4: Lock the Structure\n",
+       "1. Insert the provided tool into the hole as shown.\n",
+       "2. Ensure the structure is properly aligned and secure.\n",
+       "3. Push down firmly to lock the structure in place.\n",
+       "\n",
+       "#### Step 5: Insert Dowels\n",
+       "1. Insert 4 dowels into the designated holes on the board.\n",
+       "\n",
+       "#### Step 6: Align and Insert the Board\n",
+       "1. Align the board with the dowels.\n",
+       "2. Insert the board into the corresponding slots on the frame.\n",
+       "\n",
+       "#### Step 7: Secure the Top Panel\n",
+       "1. Insert the top panel into the side panels.\n",
+       "2. Use 4 screws to secure the top panel.\n",
+       "3. Ensure the screws are properly aligned and tightened using the provided tool.\n",
+       "\n",
+       "#### Step 8: Secure the Bottom Panel\n",
+       "1. Carefully flip the assembled structure upright.\n",
+       "2. Use 2 screws to secure the bottom panel.\n",
+       "3. Tighten the screws with the provided tool.\n",
+       "\n",
+       "By following these steps, you should be able to assemble the Smågåra crib successfully. If you encounter any issues, refer to the visual instructions provided in the package or contact IKEA customer service for assistance."
+      ],
+      "text/plain": [
+       "<IPython.core.display.Markdown object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "response = agent.chat(\n",
+    "    \"Give a step-by-step instruction guide on how to assemble the Smagora\"\n",
+    ")\n",
+    "display(Markdown(str(response)))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Added user message to memory: How do I assemble the Fredde?\n",
+      "=== Calling Function ===\n",
+      "Calling function: query_engine_tool with args: {\"input\": \"step-by-step instruction guide on how to assemble the Fredde\"}\n",
+      "=== Function Output ===\n",
+      "The query asks for a step-by-step instruction guide on how to assemble the Fredde. However, based on the provided images and parsed text, there is no specific mention or visual representation of the Fredde assembly instructions. The images and text provided are related to other IKEA products such as Tuffing and Smågöra, but not Fredde.\n",
+      "\n",
+      "Therefore, I cannot provide the step-by-step instructions for assembling the Fredde from the given information. If you have the specific instructions for Fredde, please provide them, and I can assist you further.\n",
+      "=== LLM Response ===\n",
+      "It appears that the specific step-by-step instructions for assembling the Fredde desk are not available in the provided data. However, I can offer a general guide based on typical assembly procedures for IKEA furniture. For the most accurate and detailed instructions, please refer to the assembly manual that comes with the product.\n",
+      "\n",
+      "### General Assembly Guide for Fredde Desk:\n",
+      "\n",
+      "#### Tools Required:\n",
+      "- Phillips screwdriver\n",
+      "- Flathead screwdriver\n",
+      "- Allen key (usually provided in the package)\n",
+      "- Hammer (if needed for dowels)\n",
+      "\n",
+      "### Step-by-Step Assembly:\n",
+      "\n",
+      "#### Step 1: Unpack and Organize\n",
+      "1. **Unpack** all the parts and hardware.\n",
+      "2. **Organize** the parts by type and size to make the assembly process easier.\n",
+      "\n",
+      "#### Step 2: Assemble the Main Frame\n",
+      "1. **Connect the Side Panels**: Attach the side panels to the back panel using screws and dowels as indicated in the manual.\n",
+      "2. **Secure the Bottom Panel**: Attach the bottom panel to the side panels.\n",
+      "\n",
+      "#### Step 3: Attach the Shelves\n",
+      "1. **Install the Lower Shelves**: Insert the lower shelves into the designated slots and secure them with screws.\n",
+      "2. **Install the Upper Shelves**: Repeat the process for the upper shelves.\n",
+      "\n",
+      "#### Step 4: Attach the Desktop\n",
+      "1. **Align the Desktop**: Place the desktop on top of the frame, ensuring it is properly aligned.\n",
+      "2. **Secure the Desktop**: Use screws to secure the desktop to the frame.\n",
+      "\n",
+      "#### Step 5: Install Additional Features\n",
+      "1. **Attach Monitor Shelf**: If the Fredde desk includes a monitor shelf, attach it to the back panel using screws.\n",
+      "2. **Install Side Extensions**: Attach any side extensions or additional shelves as per the instructions.\n",
+      "\n",
+      "#### Step 6: Final Adjustments\n",
+      "1. **Check Stability**: Ensure all screws are tightened and the desk is stable.\n",
+      "2. **Adjust Height**: If the desk has adjustable height features, set it to the desired height.\n",
+      "\n",
+      "#### Step 7: Clean Up\n",
+      "1. **Remove Packaging**: Dispose of any packaging materials.\n",
+      "2. **Organize Tools**: Put away your tools and clean the workspace.\n",
+      "\n",
+      "For the most accurate and detailed instructions, please refer to the assembly manual that comes with the Fredde desk. If you encounter any issues, IKEA customer service can provide additional support.\n"
+     ]
+    },
+    {
+     "data": {
+      "text/markdown": [
+       "It appears that the specific step-by-step instructions for assembling the Fredde desk are not available in the provided data. However, I can offer a general guide based on typical assembly procedures for IKEA furniture. For the most accurate and detailed instructions, please refer to the assembly manual that comes with the product.\n",
+       "\n",
+       "### General Assembly Guide for Fredde Desk:\n",
+       "\n",
+       "#### Tools Required:\n",
+       "- Phillips screwdriver\n",
+       "- Flathead screwdriver\n",
+       "- Allen key (usually provided in the package)\n",
+       "- Hammer (if needed for dowels)\n",
+       "\n",
+       "### Step-by-Step Assembly:\n",
+       "\n",
+       "#### Step 1: Unpack and Organize\n",
+       "1. **Unpack** all the parts and hardware.\n",
+       "2. **Organize** the parts by type and size to make the assembly process easier.\n",
+       "\n",
+       "#### Step 2: Assemble the Main Frame\n",
+       "1. **Connect the Side Panels**: Attach the side panels to the back panel using screws and dowels as indicated in the manual.\n",
+       "2. **Secure the Bottom Panel**: Attach the bottom panel to the side panels.\n",
+       "\n",
+       "#### Step 3: Attach the Shelves\n",
+       "1. **Install the Lower Shelves**: Insert the lower shelves into the designated slots and secure them with screws.\n",
+       "2. **Install the Upper Shelves**: Repeat the process for the upper shelves.\n",
+       "\n",
+       "#### Step 4: Attach the Desktop\n",
+       "1. **Align the Desktop**: Place the desktop on top of the frame, ensuring it is properly aligned.\n",
+       "2. **Secure the Desktop**: Use screws to secure the desktop to the frame.\n",
+       "\n",
+       "#### Step 5: Install Additional Features\n",
+       "1. **Attach Monitor Shelf**: If the Fredde desk includes a monitor shelf, attach it to the back panel using screws.\n",
+       "2. **Install Side Extensions**: Attach any side extensions or additional shelves as per the instructions.\n",
+       "\n",
+       "#### Step 6: Final Adjustments\n",
+       "1. **Check Stability**: Ensure all screws are tightened and the desk is stable.\n",
+       "2. **Adjust Height**: If the desk has adjustable height features, set it to the desired height.\n",
+       "\n",
+       "#### Step 7: Clean Up\n",
+       "1. **Remove Packaging**: Dispose of any packaging materials.\n",
+       "2. **Organize Tools**: Put away your tools and clean the workspace.\n",
+       "\n",
+       "For the most accurate and detailed instructions, please refer to the assembly manual that comes with the Fredde desk. If you encounter any issues, IKEA customer service can provide additional support."
+      ],
+      "text/plain": [
+       "<IPython.core.display.Markdown object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "response = agent.chat(\"How do I assemble the Fredde?\")\n",
+    "display(Markdown(str(response)))"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "llama-parse-5ZmnAQ0r-py3.11",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}

examples/parse/other_files/demo_ppt_basic.ipynb

@@ -25,7 +25,7 @@
     "\n",
     "nest_asyncio.apply()\n",
     "\n",
-    "from llama_parse import LlamaParse"
+    "from llama_cloud_services import LlamaParse"
    ]
   },
   {

examples/parse/other_files/demo_ppt_financial.ipynb

@@ -27,7 +27,7 @@
    "outputs": [],
    "source": [
     "%pip install llama-index\n",
-    "%pip install llama-parse\n",
+    "%pip install llama-cloud-services\n",
     "%pip install torch transformers python-pptx Pillow"
    ]
   },
@@ -85,7 +85,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from llama_parse import LlamaParse"
+    "from llama_cloud_services import LlamaParse"
    ]
   },
   {

examples/parse/parsing_instructions/parsing_instructions.ipynb

@@ -0,0 +1,602 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<a href=\"https://colab.research.google.com/github/run-llama/llama_cloud_services/blob/main/examples/parse/parsing_instructions.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
+    "\n",
+    "# Parsing documents with Instructions\n",
+    "\n",
+    "Parsing instructions allow you to guide our parsing model in the same way you would instruct an LLM.\n",
+    "\n",
+    "These instructions can be useful for improving the parser's performance on complex document layouts, extracting data in a specific format, or transforming the document in other ways.\n",
+    "\n",
+    "### Why This Matters:\n",
+    "Traditional document parsing can be rigid and error-prone, often missing crucial context and nuances in complex layouts. Our instruction-based parsing allows you to:\n",
+    "\n",
+    "1. Extract specific information with pinpoint accuracy\n",
+    "2. Handle complex document layouts with ease\n",
+    "3. Transform unstructured data into structured formats effortlessly\n",
+    "4. Save hours of manual data entry and verification\n",
+    "5. Reduce errors in document processing workflows\n",
+    "\n",
+    "In this demonstration, we showcase how parsing instructions can be used to extract specific information from unstructured documents. Below are the documents we use for testing:\n",
+    "\n",
+    "1. McDonald's Receipt - Extracting the price of each order and the final amount to be paid.\n",
+    "\n",
+    "2. Expense Report Document - Extracting employee name, employee ID, position, department, date ranges, individual expense items with dates, categories, and amounts.\n",
+    "\n",
+    "3. Purchase Order Document - Identifying the PO number, vendor details, shipping terms, and an itemized list of products with quantities and unit prices.\n",
+    "\n",
+    "Let's jump into these real-world examples and see how parsing instructions can help us extract specific information."
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Installation"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install llama-cloud-services"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Setup API Key"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import nest_asyncio\n",
+    "\n",
+    "nest_asyncio.apply()\n",
+    "\n",
+    "import os\n",
+    "\n",
+    "os.environ[\"LLAMA_CLOUD_API_KEY\"] = \"llx-...\""
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### McDonald's Receipt\n",
+    "\n",
+    "Here we extract the price of each order and the final amount to be paid."
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<img src=\"mcdonalds_receipt.png\" alt=\"Alt Text\" width=\"500\">"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Started parsing the file under job_id 66643b81-e2f4-408b-890b-8e116472210b\n"
+     ]
+    }
+   ],
+   "source": [
+    "from llama_cloud_services import LlamaParse\n",
+    "\n",
+    "vanilaParsing = LlamaParse(result_type=\"markdown\").load_data(\"./mcdonalds_receipt.png\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "# Rate us HIGHLY SATISFIED\n",
+      "\n",
+      "Purchase any sandwich and receive a FREE ITEM\n",
+      "\n",
+      "Go to WWW.mcdvoice.com within 7 days of purchase of equal or lesser value and tell us about your visit.\n",
+      "\n",
+      "Validation Code: 31278-01121-21018-20481-00081-0\n",
+      "\n",
+      "Valid at participating US McDonald's\n",
+      "\n",
+      "Expires 30 days after receipt date\n",
+      "\n",
+      "# McDonald's Restaurant #312782378\n",
+      "\n",
+      "PINE RD NW\n",
+      "\n",
+      "RICE MN 56367-9740\n",
+      "\n",
+      "TEL# 320 393 4600\n",
+      "\n",
+      "KS# 12/08/2022 08:48 PM\n",
+      "\n",
+      "# Order\n",
+      "\n",
+      "|Happy Meal 6 Pc|$4.89|\n",
+      "|---|---|\n",
+      "|Creamy Ranch Cup| |\n",
+      "|Extra Kids Fry| |\n",
+      "|Wreck It Ralph 2 Snack| |\n",
+      "|Oreo McFlurry|$2.69|\n",
+      "\n",
+      "# Summary\n",
+      "\n",
+      "|Subtotal|$7.58|\n",
+      "|---|---|\n",
+      "|Tax|$0.52|\n",
+      "|Take-Out Total|$8.10|\n",
+      "|Cash Tendered|$10.00|\n",
+      "|Change|$1.90|\n",
+      "\n",
+      "### Not ACCEPTING APPLICATIONS *++ McDonald's Restaurant Rice\n",
+      "\n",
+      "Text to #36453 apply 31278\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(vanilaParsing[0].text)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Started parsing the file under job_id 1a04fdbb-5415-4a36-a1bd-26bfb5d618fa\n"
+     ]
+    }
+   ],
+   "source": [
+    "parsingInstruction = \"\"\"The provided document is a McDonald's receipt.\n",
+    " Provide the price of each order and final amount to be paid.\"\"\"\n",
+    "withInstructionParsing = LlamaParse(\n",
+    "    result_type=\"markdown\", parsing_instruction=parsingInstruction\n",
+    ").load_data(\"./mcdonalds_receipt.png\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Here are the prices for each order from the McDonald's receipt:\n",
+      "\n",
+      "1. Happy Meal 6 Pc: $4.89\n",
+      "2. Snack Oreo McFlurry: $2.69\n",
+      "\n",
+      "**Subtotal:** $7.58\n",
+      "**Tax:** $0.52\n",
+      "**Total Amount to be Paid:** $8.10\n",
+      "\n",
+      "The cash tendered was $10.00, and the change given was $1.90.\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(withInstructionParsing[0].text)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Expense Report Document\n",
+    "\n",
+    "Here we extract employee name, employee ID, position, department, date ranges, individual expense items with dates, categories, and amounts."
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<img src=\"expense_report_document.png\" alt=\"Alt Text\" width=\"500\">"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Started parsing the file under job_id b6bcc6e1-7d30-4522-9abd-ace196781a70\n"
+     ]
+    }
+   ],
+   "source": [
+    "vanilaParsing = LlamaParse(result_type=\"markdown\").load_data(\n",
+    "    \"./expense_report_document.pdf\"\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "# QUANTUM DYNAMICS CORPORATION\n",
+      "\n",
+      "# EMPLOYEE EXPENSE REPORT\n",
+      "\n",
+      "# FISCAL YEAR 2024\n",
+      "\n",
+      "# EMPLOYEE INFORMATION:\n",
+      "\n",
+      "Name: Dr. Alexandra Chen-Martinez, PhD\n",
+      "\n",
+      "Employee ID: QD-2022-1457\n",
+      "\n",
+      "Department: Advanced Research & Development\n",
+      "\n",
+      "Cost Center: CC-ARD-NA-003\n",
+      "\n",
+      "Project Codes: QD-QUANTUM-2024-01, QD-AI-2024-03\n",
+      "\n",
+      "Position: Principal Research Scientist\n",
+      "\n",
+      "Reporting Manager: Dr. James Thompson\n",
+      "\n",
+      "# TRIP/EXPENSE PERIOD:\n",
+      "\n",
+      "Start Date: November 15, 2024\n",
+      "\n",
+      "End Date: December 10, 2024\n",
+      "\n",
+      "Purpose: International Conference Attendance & Client Meetings\n",
+      "\n",
+      "Locations: Tokyo, Japan → Singapore → Sydney, Australia\n",
+      "\n",
+      "# CURRENCY CONVERSION RATES APPLIED:\n",
+      "\n",
+      "JPY (¥) → USD: 0.0068 (as of 11/15/2024)\n",
+      "\n",
+      "SGD (S$) → USD: 0.74 (as of 11/28/2024)\n",
+      "\n",
+      "AUD (A$) → USD: 0.65 (as of 12/03/2024)\n",
+      "\n",
+      "# ITEMIZED EXPENSES:\n",
+      "\n",
+      "|Date|Category|Description|Original|Currency|USD|\n",
+      "|---|---|---|---|---|---|\n",
+      "|11/15/2024|Transportation|JFK → NRT Business Class|4,250.00|USD|4,250.00|\n",
+      "|Booking Ref: QF78956 - Corporate Rate Applied|Booking Ref: QF78956 - Corporate Rate Applied|Booking Ref: QF78956 - Corporate Rate Applied|Booking Ref: QF78956 - Corporate Rate Applied|Booking Ref: QF78956 - Corporate Rate Applied|Booking Ref: QF78956 - Corporate Rate Applied|\n",
+      "|Project Code: QD-QUANTUM-2024-01|Project Code: QD-QUANTUM-2024-01|Project Code: QD-QUANTUM-2024-01|Project Code: QD-QUANTUM-2024-01|Project Code: QD-QUANTUM-2024-01|Project Code: QD-QUANTUM-2024-01|\n",
+      "|11/16/2024|Accommodation|Hilton Tokyo - 5 nights|225,000|JPY|1,530.00|\n",
+      "|Confirmation: HTK-2024-78956|Confirmation: HTK-2024-78956|Confirmation: HTK-2024-78956|Confirmation: HTK-2024-78956|Confirmation: HTK-2024-78956|Confirmation: HTK-2024-78956|\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(vanilaParsing[0].text)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Started parsing the file under job_id 7b0d05bb-947b-4475-8d0f-f10386f7446e\n"
+     ]
+    }
+   ],
+   "source": [
+    "parsingInstruction = \"\"\"You are provided with an expense report. \n",
+    "Extract employee name, employee id, position, department, date ranges, individual expense items with dates, categories, and amounts.\"\"\"\n",
+    "\n",
+    "withInstructionParsing = LlamaParse(\n",
+    "    result_type=\"markdown\", parsing_instruction=parsingInstruction\n",
+    ").load_data(\"./expense_report_document.pdf\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "**Employee Information:**\n",
+      "- **Name:** Dr. Alexandra Chen-Martinez, PhD\n",
+      "- **Employee ID:** QD-2022-1457\n",
+      "- **Position:** Principal Research Scientist\n",
+      "- **Department:** Advanced Research & Development\n",
+      "\n",
+      "**Trip/Expense Period:**\n",
+      "- **Start Date:** November 15, 2024\n",
+      "- **End Date:** December 10, 2024\n",
+      "\n",
+      "**Expense Items:**\n",
+      "1. **Date:** 11/15/2024\n",
+      "- **Category:** Transportation\n",
+      "- **Description:** JFK → NRT Business Class\n",
+      "- **Original Amount:** $4,250.00\n",
+      "- **Currency:** USD\n",
+      "- **USD Amount:** $4,250.00\n",
+      "- **Booking Reference:** QF78956 - Corporate Rate Applied\n",
+      "- **Project Code:** QD-QUANTUM-2024-01\n",
+      "\n",
+      "2. **Date:** 11/16/2024\n",
+      "- **Category:** Accommodation\n",
+      "- **Description:** Hilton Tokyo - 5 nights\n",
+      "- **Original Amount:** ¥225,000\n",
+      "- **Currency:** JPY\n",
+      "- **USD Amount:** $1,530.00\n",
+      "- **Confirmation:** HTK-2024-78956\n",
+      "\n",
+      "**Locations:**\n",
+      "- Tokyo, Japan\n",
+      "- Singapore\n",
+      "- Sydney, Australia\n",
+      "\n",
+      "**Currency Conversion Rates Applied:**\n",
+      "- JPY (¥) → USD: 0.0068 (as of 11/15/2024)\n",
+      "- SGD (S$) → USD: 0.74 (as of 11/28/2024)\n",
+      "- AUD (A$) → USD: 0.65 (as of 12/03/2024)\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(withInstructionParsing[0].text)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Purchase Order Document \n",
+    "\n",
+    "Here we identify the PO number, vendor details, shipping terms, and an itemized list of products with quantities and unit prices."
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<img src=\"purchase_order_document.png\" alt=\"Alt Text\" width=\"500\">"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Started parsing the file under job_id b8cb11c3-7dce-4e6a-94bb-1a4e50e45e55\n"
+     ]
+    }
+   ],
+   "source": [
+    "vanilaParsing = LlamaParse(result_type=\"markdown\").load_data(\n",
+    "    \"./purchase_order_document.pdf\"\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "# GLOBAL TECH SOLUTIONS, INC.\n",
+      "\n",
+      "# PURCHASE ORDER\n",
+      "\n",
+      "Document Reference: PO-2024-GT-9876/REV.2\n",
+      "\n",
+      "[Original: PO-2024-GT-9876]\n",
+      "\n",
+      "Amendment Date: 12/10/2024\n",
+      "\n",
+      "# VENDOR INFORMATION:\n",
+      "\n",
+      "Quantum Electronics Manufacturing\n",
+      "\n",
+      "DUNS: 78-456-7890\n",
+      "\n",
+      "Tax ID: EU8976543210\n",
+      "\n",
+      "Hoofdorp, Netherlands\n",
+      "\n",
+      "Vendor #: QEM-EU-2024-001\n",
+      "\n",
+      "# SHIP TO:\n",
+      "\n",
+      "Global Tech Solutions, Inc.\n",
+      "\n",
+      "Building 7A, Innovation Park\n",
+      "\n",
+      "2100 Technology Drive\n",
+      "\n",
+      "Austin, TX 78701\n",
+      "\n",
+      "USA\n",
+      "\n",
+      "Attn: Sarah Martinez, Receiving Manager\n",
+      "\n",
+      "Tel: +1 (512) 555-0123\n",
+      "\n",
+      "# PAYMENT TERMS:\n",
+      "\n",
+      "Net 45\n",
+      "\n",
+      "2% discount if paid within 15 days\n",
+      "\n",
+      "# SHIPPING TERMS:\n",
+      "\n",
+      "DDP (Delivered Duty Paid) - Incoterms 2020\n",
+      "\n",
+      "Insurance Required: Yes\n",
+      "\n",
+      "Preferred Carrier: DHL/FedEx\n",
+      "\n",
+      "Required Delivery Date: 01/15/2025\n",
+      "\n",
+      "# SPECIAL INSTRUCTIONS:\n",
+      "\n",
+      "1. All shipments must include Certificate of Conformance\n",
+      "2. ESD-sensitive items must be properly packaged\n",
+      "3. Temperature logging required for items marked with *\n",
+      "4. Partial shipments accepted with prior approval\n",
+      "5. Quote PO number on all correspondence\n",
+      "\n",
+      "# ITEM DETAILS:\n",
+      "\n",
+      "|Line|Part Number|Description|Qty|UOM|Unit Price|Total|\n",
+      "|---|---|---|---|---|---|---|\n",
+      "|1|QE-MCU-5590|Microcontroller Unit|500|EA|$12.50|$6,250.00|\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(vanilaParsing[0].text)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Started parsing the file under job_id d2731305-984d-4633-8a52-0493748cf10b\n"
+     ]
+    }
+   ],
+   "source": [
+    "parsingInstruction = \"\"\"You are provided with a purchase order. \n",
+    "Identify the PO number, vendor details, shipping terms, and itemized list of products with quantities and unit prices.\"\"\"\n",
+    "\n",
+    "withInstructionParsing = LlamaParse(\n",
+    "    result_type=\"markdown\", parsing_instruction=parsingInstruction\n",
+    ").load_data(\"./purchase_order_document.pdf\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Here are the details extracted from the purchase order:\n",
+      "\n",
+      "**PO Number:** PO-2024-GT-9876/REV.2\n",
+      "\n",
+      "**Vendor Details:**\n",
+      "- **Vendor Name:** Quantum Electronics Manufacturing\n",
+      "- **DUNS:** 78-456-7890\n",
+      "- **Tax ID:** EU8976543210\n",
+      "- **Address:** Hoofdorp, Netherlands\n",
+      "- **Vendor Number:** QEM-EU-2024-001\n",
+      "- **Contact Person:** Sarah Martinez, Receiving Manager\n",
+      "- **Phone:** +1 (512) 555-0123\n",
+      "\n",
+      "**Shipping Terms:**\n",
+      "- **Terms:** DDP (Delivered Duty Paid) - Incoterms 2020\n",
+      "- **Insurance Required:** Yes\n",
+      "- **Preferred Carrier:** DHL/FedEx\n",
+      "- **Required Delivery Date:** 01/15/2025\n",
+      "\n",
+      "**Itemized List of Products:**\n",
+      "1. **Part Number:** QE-MCU-5590\n",
+      "- **Description:** Microcontroller Unit\n",
+      "- **Quantity:** 500 EA\n",
+      "- **Unit Price:** $12.50\n",
+      "- **Total:** $6,250.00\n",
+      "\n",
+      "**Payment Terms:**\n",
+      "- Net 45\n",
+      "- 2% discount if paid within 15 days\n",
+      "\n",
+      "**Special Instructions:**\n",
+      "1. All shipments must include Certificate of Conformance\n",
+      "2. ESD-sensitive items must be properly packaged\n",
+      "3. Temperature logging required for items marked with *\n",
+      "4. Partial shipments accepted with prior approval\n",
+      "5. Quote PO number on all correspondence\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(withInstructionParsing[0].text)"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "llamacloud",
+   "language": "python",
+   "name": "llamacloud"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}

examples/parse/test_tesla_impact_report/test_gpt4o.ipynb

@@ -8,7 +8,7 @@
     "# LlamaParse with GPT-4o\n",
     "\n",
     "\n",
-    "<a href=\"https://colab.research.google.com/github/run-llama/llama_parse/blob/main/examples/test_tesla_impact_report/test_gpt4o.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
+    "<a href=\"https://colab.research.google.com/github/run-llama/llama_cloud_services/blob/main/examples/parse/test_tesla_impact_report/test_gpt4o.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
     "\n",
     "GPT-4o is a [fully multimodal model by OpenAI](https://openai.com/index/hello-gpt-4o/) released in May 2024. It matches GPT-4 Turbo performance in text and code, and has significantly improved vision and audio capabilities.\n",
     "\n",
@@ -46,7 +46,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "os.environ[\"LLAMA_CLOUD_API_KEY\"] = \"<LLAMA_CLOUD_API_KEY>"
+    "os.environ[\"LLAMA_CLOUD_API_KEY\"] = \"<LLAMA_CLOUD_API_KEY>\""
    ]
   },
   {
@@ -107,7 +107,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from llama_parse import LlamaParse\n",
+    "from llama_cloud_services import LlamaParse\n",
     "\n",
     "parser_gpt4o = LlamaParse(\n",
     "    result_type=\"markdown\",\n",

examples/report/basic_report.ipynb

@@ -0,0 +1,762 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "# Report Generation with LlamaReport\n",
+    "\n",
+    "In this notebook, we'll walk through the basic process of generating a report with LlamaReport, and highlight some of the key features of the library.\n",
+    "\n",
+    "TLDR:\n",
+    "1. Download source data to use as knowledge base for the report\n",
+    "2. Kick off report generation with a template\n",
+    "3. Get the plan and review/accept/reject suggestions\n",
+    "4. Get the final report\n",
+    "5. Review/accept/reject suggestions to edit the final report\n",
+    "6. Print the final report"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "%pip install llama-cloud-services"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## 1. Download Source Data\n",
+    "\n",
+    "Here, we download the `Attention is All You Need` paper as a PDF.\n",
+    "\n",
+    "LlamaReport currently supports up to 5 files as input, and essentially any file type that can be parsed by LlamaParse.\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!wget \"https://arxiv.org/pdf/1706.03762.pdf\" -O \"./attention.pdf\""
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## 2. Kick off Report Generation\n",
+    "\n",
+    "Here, we kick off report generation with a template.\n",
+    "\n",
+    "The template can either be a string or a file path, but here we'll use a string.\n",
+    "\n",
+    "In our experiments, anything works as a template, but some general guidelines:\n",
+    "\n",
+    "- Use markdown formatting + instructions in each section to guide the report generation\n",
+    "- If using an existing file as a template, provide extra instructions to guide the report generation\n",
+    "\n",
+    "**NOTE:** Since we are in a notebook, we will use async functions and `await` throughout. Synchronous methods that work without `await` are available by just removing the `a` from the method name and removing the `await` keyword."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from llama_cloud_services import LlamaReport\n",
+    "\n",
+    "llama_report = LlamaReport(\n",
+    "    api_key=\"llx-...\",\n",
+    ")\n",
+    "\n",
+    "report_client = await llama_report.acreate_report(\n",
+    "    name=\"my_cool_report_on_attention\",\n",
+    "    # can pass in file paths or bytes\n",
+    "    input_files=[\"./attention.pdf\"],\n",
+    "    template_text=\"\"\"\\\n",
+    "# [Some title]\\n\\n\n",
+    "## TLDR\\n\n",
+    "A quick summary of the paper.\\n\\n\n",
+    "## Details\\n\n",
+    "More details about the paper, possibly more than one section here.\\n\n",
+    "\"\"\",\n",
+    "    # optional additional instructions for the report generation\n",
+    "    # template_instructions=None,\n",
+    "    # optional file path to an existing template instead of template_text\n",
+    "    # template_file=None,\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "The returned `ReportClient` object is used to interact with the report generation process for this specific report."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Report(id=0a394b33-1a3e-463c-b5cb-7ff8ab827d0a, name=my_cool_report_on_attention)\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(report_client)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## 3. Get the plan\n",
+    "\n",
+    "The first phases of report generation involve ingesting the source data and generating a plan.\n",
+    "\n",
+    "The plan is a list of instructions for the report generation, and can be reviewed/accepted/rejected by the user.\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "plan = await report_client.await_for_plan(\n",
+    "    timeout=10000,\n",
+    "    poll_interval=10,\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "# {title}\n",
+      "[ReportQuery(field='title', prompt='Generate a clear and concise title for this paper about the Transformer model and attention mechanisms', context='The paper discusses the Transformer architecture for sequence transduction using attention mechanisms, focusing on machine translation applications')]\n",
+      "==================\n",
+      "## TLDR\n",
+      "\n",
+      "{tldr_content}\n",
+      "[ReportQuery(field='tldr_content', prompt='Write a brief, clear summary of the key points about the Transformer model', context='Focus on the main innovations: attention mechanisms, efficiency improvements, and state-of-the-art results in machine translation')]\n",
+      "==================\n",
+      "## Details\n",
+      "\n",
+      "{details_content}\n",
+      "[ReportQuery(field='details_content', prompt='Provide detailed information about the Transformer model architecture and its applications', context='Include information about:\\n- The attention mechanism implementation\\n- Advantages over recurrent and convolutional models\\n- Performance in machine translation tasks\\n- Training efficiency improvements')]\n",
+      "==================\n"
+     ]
+    }
+   ],
+   "source": [
+    "for plan_block in plan.blocks:\n",
+    "    print(plan_block.block.template)\n",
+    "    print(plan_block.queries)\n",
+    "    print(\"==================\")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "With the plan, we can either use it to kick off generation of the final report, or we can edit the plan and adjust it as needed.\n",
+    "\n",
+    "While we could manually edit the objects here and use `await report_client.aupdate_plan(action=\"edit\", updated_plan=plan)`, we can also use `LlamaReport` to agentically edit the plan."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "suggestions = await report_client.asuggest_edits(\n",
+    "    \"Can you split the details section into two sections?\"\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Justification for change: \n",
+      "I'll help you break down the details section into two distinct parts - one focusing on the architecture and another on the practical applications and performance. This will make the content more organized and easier to follow. The original block at index 2 will be replaced with these two new sections.\n",
+      "\n",
+      "Proposed changes:\n",
+      "\n",
+      "## Architecture Details\n",
+      "\n",
+      "{architecture_content}\n",
+      "\n",
+      "[ReportQuery(field='architecture_content', prompt='Describe the technical details of the Transformer model architecture', context='Focus on:\\n- Core components of the Transformer architecture\\n- Self-attention mechanism implementation\\n- Multi-head attention details\\n- Position encoding approach\\n- Feed-forward network structure')]\n",
+      "==================\n",
+      "\n",
+      "## Performance and Applications\n",
+      "\n",
+      "{applications_content}\n",
+      "\n",
+      "[ReportQuery(field='applications_content', prompt='Explain the practical applications and performance advantages of the Transformer model', context='Cover:\\n- Comparison with RNN and CNN models\\n- Machine translation results and benchmarks\\n- Training efficiency improvements\\n- Real-world applications and use cases\\n- Scalability benefits')]\n",
+      "==================\n"
+     ]
+    }
+   ],
+   "source": [
+    "for suggestion in suggestions:\n",
+    "    print(\"Justification for change:\", suggestion.justification)\n",
+    "    print(\"Proposed changes:\")\n",
+    "    for plan_block in suggestion.blocks:\n",
+    "        print(plan_block.block.template)\n",
+    "        print(plan_block.queries)\n",
+    "        print(\"==================\")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "This looks pretty good! We can also use the client to automatically accept and apply, or reject, these suggestions.\n",
+    "\n",
+    "This will (locally) keep track of the history of changes, so that future suggestions can be based on the previous changes."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "for suggestion in suggestions:\n",
+    "    await report_client.aaccept_edit(suggestion)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "What effect did that have on the tracked local history? Let's see!"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "[EditAction(block_idx=2, old_content='## Details\\n\\n{details_content}\\n\\nField: details_content, Prompt: Provide detailed information about the Transformer model architecture and its applications, Context: Include information about:\\n- The attention mechanism implementation\\n- Advantages over recurrent and convolutional models\\n- Performance in machine translation tasks\\n- Training efficiency improvements\\nDepends on: none', new_content='\\n## Architecture Details\\n\\n{architecture_content}\\n\\n\\nField: architecture_content, Prompt: Describe the technical details of the Transformer model architecture, Context: Focus on:\\n- Core components of the Transformer architecture\\n- Self-attention mechanism implementation\\n- Multi-head attention details\\n- Position encoding approach\\n- Feed-forward network structure\\nDepends on: none', action='approved', timestamp=datetime.datetime(2025, 2, 4, 20, 59, 55, 773558)),\n",
+       " EditAction(block_idx=3, old_content='[No old content]', new_content='\\n## Performance and Applications\\n\\n{applications_content}\\n\\n\\nField: applications_content, Prompt: Explain the practical applications and performance advantages of the Transformer model, Context: Cover:\\n- Comparison with RNN and CNN models\\n- Machine translation results and benchmarks\\n- Training efficiency improvements\\n- Real-world applications and use cases\\n- Scalability benefits\\nDepends on: previous', action='approved', timestamp=datetime.datetime(2025, 2, 4, 20, 59, 55, 773687))]"
+      ]
+     },
+     "execution_count": null,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "report_client.edit_history"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "[Message(role=<MessageRole.USER: 'user'>, content='Can you split the details section into two sections?', timestamp=datetime.datetime(2025, 2, 4, 20, 59, 47, 754848)),\n",
+       " Message(role=<MessageRole.ASSISTANT: 'assistant'>, content=\"\\nI'll help you break down the details section into two distinct parts - one focusing on the architecture and another on the practical applications and performance. This will make the content more organized and easier to follow. The original block at index 2 will be replaced with these two new sections.\\n\", timestamp=datetime.datetime(2025, 2, 4, 20, 59, 55, 482070))]"
+      ]
+     },
+     "execution_count": null,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "report_client.chat_history"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "These two items are used to provide context for future suggestions! You can always clear this, or provide your own history."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# report_client.suggest_edits(\"....\", chat_history=[{\"role\": \"user\", \"content\": \"...\"}, ...])"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## 4. Get the final report\n",
+    "\n",
+    "Now that we have a plan, we can kick off generation of the final report."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# kicks off report generation\n",
+    "await report_client.aupdate_plan(action=\"approve\")\n",
+    "\n",
+    "# waits for report generation to complete\n",
+    "report = await report_client.await_completion(\n",
+    "    timeout=10000,\n",
+    "    poll_interval=10,\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "# Attention Is All You Need: A Pure Attention-Based Architecture for Neural Machine Translation\n",
+      "\n",
+      "## TLDR\n",
+      "\n",
+      "The Transformer introduced a revolutionary architecture that relies entirely on attention mechanisms, eliminating the need for recurrence or convolution in sequence processing. Its key innovations include multi-head self-attention for parallel processing of input sequences, scaled dot-product attention for efficient computation, and positional encodings for sequence order awareness. The model achieved breakthrough results in machine translation (28.4 BLEU on English-to-German, 41.8 BLEU on English-to-French) while requiring significantly less training time than previous approaches, training in 3.5 days on 8 GPUs. This architecture demonstrated that attention mechanisms alone are sufficient for state-of-the-art sequence modeling, setting a new direction for natural language processing.\n",
+      "\n",
+      "\n",
+      "## Architecture Details\n",
+      "\n",
+      "The Transformer architecture represents a groundbreaking approach to sequence processing, built entirely on attention mechanisms without recurrence or convolution. Here are its key technical details:\n",
+      "\n",
+      "Core Components:\n",
+      "- Encoder-decoder architecture with stacked self-attention and point-wise feed-forward layers\n",
+      "- Each layer contains two main sub-layers: multi-head self-attention mechanism and position-wise feed-forward network\n",
+      "- Layer normalization and residual connections between sub-layers\n",
+      "- No recurrent or convolutional elements, enabling parallel processing\n",
+      "\n",
+      "Self-Attention Mechanism:\n",
+      "- Processes relationships between all positions in a sequence simultaneously\n",
+      "- Computes attention weights using queries, keys, and values derived from input representations\n",
+      "- Implements scaled dot-product attention to prevent gradient issues with large input dimensions\n",
+      "- Allows direct modeling of dependencies regardless of positional distance\n",
+      "- Uses masking in decoder to prevent leftward information flow and maintain auto-regressive property\n",
+      "\n",
+      "Multi-Head Attention:\n",
+      "- Employs multiple attention heads operating in parallel\n",
+      "- Each head processes information in different representation subspaces\n",
+      "- Three types of attention applications:\n",
+      "  1. Encoder self-attention (all positions attend to each other)\n",
+      "  2. Decoder self-attention (each position attends to previous positions)\n",
+      "  3. Encoder-decoder attention (decoder queries attend to encoder outputs)\n",
+      "- Counteracts reduced resolution from attention averaging through parallel processing\n",
+      "\n",
+      "Position-wise Feed-Forward Network:\n",
+      "- Applied identically to each position separately\n",
+      "- Consists of two linear transformations with ReLU activation\n",
+      "- Structure: FFN(x) = max(0, xW1 + b1)W2 + b2\n",
+      "- Input and output dimensionality: dmodel = 512\n",
+      "- Inner-layer dimensionality: dff = 2048\n",
+      "- Parameters vary between layers but remain constant across positions\n",
+      "\n",
+      "Position Encoding:\n",
+      "- Adds positional information to input embeddings\n",
+      "- Enables the model to consider sequential order without recurrence\n",
+      "- Implements sinusoidal position encodings to allow model to attend to relative positions\n",
+      "- Maintains constant number of operations between any two positions, unlike convolutional approaches\n",
+      "- Allows effective modeling of both local and long-range dependencies\n",
+      "\n",
+      "\n",
+      "\n",
+      "## Performance and Applications\n",
+      "\n",
+      "The Transformer model demonstrates significant performance advantages and practical applications across multiple domains:\n",
+      "\n",
+      "Performance Advantages over RNN/CNN Models:\n",
+      "- Eliminates sequential computation constraints present in RNNs, enabling superior parallelization\n",
+      "- Reduces operations needed for relating distant positions to a constant number, compared to linear/logarithmic scaling in CNNs\n",
+      "- Processes all input and output positions simultaneously through self-attention mechanisms\n",
+      "- Achieves state-of-the-art results while requiring significantly less computational resources\n",
+      "\n",
+      "Machine Translation Benchmarks:\n",
+      "- WMT 2014 English-to-German: 28.4 BLEU score, exceeding previous best results by over 2 BLEU points\n",
+      "- WMT 2014 English-to-French: 41.8 BLEU score (single-model state-of-the-art)\n",
+      "- Surpasses performance of existing model ensembles in translation tasks\n",
+      "\n",
+      "Training Efficiency:\n",
+      "- Requires only 3.5 days of training on eight GPUs for state-of-the-art performance\n",
+      "- Achieves superior results at \"a small fraction of the training costs\" compared to previous models\n",
+      "- Enables significantly faster training through parallel processing of input/output sequences\n",
+      "- Can reach production-quality performance in as little as twelve hours on modern GPU hardware\n",
+      "\n",
+      "Real-world Applications:\n",
+      "- Machine translation systems\n",
+      "- Natural language understanding tasks\n",
+      "- Reading comprehension\n",
+      "- Abstractive summarization\n",
+      "- Text entailment analysis\n",
+      "- Constituency parsing (achieving 92.7 F1 score in semi-supervised settings)\n",
+      "- Adaptable to both large and limited training data scenarios\n",
+      "\n",
+      "Scalability Benefits:\n",
+      "- Highly parallelizable architecture enables efficient scaling across multiple GPUs\n",
+      "- Constant computational complexity for relating any input/output positions\n",
+      "- Effective handling of long-range dependencies in sequences\n",
+      "- Maintains performance quality while scaling to larger datasets and model sizes\n",
+      "- Generalizes well across different tasks and domains without architectural changes\n",
+      "- Supports efficient inference and deployment in production environments\n",
+      "\n"
+     ]
+    }
+   ],
+   "source": [
+    "report_text = \"\\n\\n\".join([block.template for block in report.blocks])\n",
+    "print(report_text)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## 5. Edit the final report\n",
+    "\n",
+    "Now that we have a report, we can edit it.\n",
+    "\n",
+    "We can use the `asuggest_edits` method to get suggestions for edits, and then use the `aaccept_edit`/`areject_edit` methods to apply them.\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Justification for change: \n",
+      "I'd suggest changing \"TLDR\" to \"Executive Summary\" which is more appropriate for a professional or academic report. This term is widely used in formal documents and better reflects the nature of this concise overview section while maintaining the same function of providing a quick summary of the key points.\n",
+      "\n",
+      "Proposed changes:\n",
+      "## Executive Summary\n",
+      "\n",
+      "The Transformer introduced a revolutionary architecture that relies entirely on attention mechanisms, eliminating the need for recurrence or convolution in sequence processing. Its key innovations include multi-head self-attention for parallel processing of input sequences, scaled dot-product attention for efficient computation, and positional encodings for sequence order awareness. The model achieved breakthrough results in machine translation (28.4 BLEU on English-to-German, 41.8 BLEU on English-to-French) while requiring significantly less training time than previous approaches, training in 3.5 days on 8 GPUs. This architecture demonstrated that attention mechanisms alone are sufficient for state-of-the-art sequence modeling, setting a new direction for natural language processing.\n",
+      "==================\n"
+     ]
+    }
+   ],
+   "source": [
+    "suggestions = await report_client.asuggest_edits(\n",
+    "    \"Can you change the TLDR header to something more professional?\"\n",
+    ")\n",
+    "for suggestion in suggestions:\n",
+    "    print(\"Justification for change:\", suggestion.justification)\n",
+    "    print(\"Proposed changes:\")\n",
+    "    for block in suggestion.blocks:\n",
+    "        print(block.template)\n",
+    "        print(\"==================\")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Changing to \"Executive Summary\" sounds reasonable, lets accept that!\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "for suggestion in suggestions:\n",
+    "    await report_client.aaccept_edit(suggestion)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## 7. Print the final report\n",
+    "\n",
+    "Now that we have a report, we can print it."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "# Attention Is All You Need: A Pure Attention-Based Architecture for Neural Machine Translation\n",
+      "\n",
+      "## Executive Summary\n",
+      "\n",
+      "The Transformer introduced a revolutionary architecture that relies entirely on attention mechanisms, eliminating the need for recurrence or convolution in sequence processing. Its key innovations include multi-head self-attention for parallel processing of input sequences, scaled dot-product attention for efficient computation, and positional encodings for sequence order awareness. The model achieved breakthrough results in machine translation (28.4 BLEU on English-to-German, 41.8 BLEU on English-to-French) while requiring significantly less training time than previous approaches, training in 3.5 days on 8 GPUs. This architecture demonstrated that attention mechanisms alone are sufficient for state-of-the-art sequence modeling, setting a new direction for natural language processing.\n",
+      "\n",
+      "\n",
+      "## Architecture Details\n",
+      "\n",
+      "The Transformer architecture represents a groundbreaking approach to sequence processing, built entirely on attention mechanisms without recurrence or convolution. Here are its key technical details:\n",
+      "\n",
+      "Core Components:\n",
+      "- Encoder-decoder architecture with stacked self-attention and point-wise feed-forward layers\n",
+      "- Each layer contains two main sub-layers: multi-head self-attention mechanism and position-wise feed-forward network\n",
+      "- Layer normalization and residual connections between sub-layers\n",
+      "- No recurrent or convolutional elements, enabling parallel processing\n",
+      "\n",
+      "Self-Attention Mechanism:\n",
+      "- Processes relationships between all positions in a sequence simultaneously\n",
+      "- Computes attention weights using queries, keys, and values derived from input representations\n",
+      "- Implements scaled dot-product attention to prevent gradient issues with large input dimensions\n",
+      "- Allows direct modeling of dependencies regardless of positional distance\n",
+      "- Uses masking in decoder to prevent leftward information flow and maintain auto-regressive property\n",
+      "\n",
+      "Multi-Head Attention:\n",
+      "- Employs multiple attention heads operating in parallel\n",
+      "- Each head processes information in different representation subspaces\n",
+      "- Three types of attention applications:\n",
+      "  1. Encoder self-attention (all positions attend to each other)\n",
+      "  2. Decoder self-attention (each position attends to previous positions)\n",
+      "  3. Encoder-decoder attention (decoder queries attend to encoder outputs)\n",
+      "- Counteracts reduced resolution from attention averaging through parallel processing\n",
+      "\n",
+      "Position-wise Feed-Forward Network:\n",
+      "- Applied identically to each position separately\n",
+      "- Consists of two linear transformations with ReLU activation\n",
+      "- Structure: FFN(x) = max(0, xW1 + b1)W2 + b2\n",
+      "- Input and output dimensionality: dmodel = 512\n",
+      "- Inner-layer dimensionality: dff = 2048\n",
+      "- Parameters vary between layers but remain constant across positions\n",
+      "\n",
+      "Position Encoding:\n",
+      "- Adds positional information to input embeddings\n",
+      "- Enables the model to consider sequential order without recurrence\n",
+      "- Implements sinusoidal position encodings to allow model to attend to relative positions\n",
+      "- Maintains constant number of operations between any two positions, unlike convolutional approaches\n",
+      "- Allows effective modeling of both local and long-range dependencies\n",
+      "\n",
+      "\n",
+      "\n",
+      "## Performance and Applications\n",
+      "\n",
+      "The Transformer model demonstrates significant performance advantages and practical applications across multiple domains:\n",
+      "\n",
+      "Performance Advantages over RNN/CNN Models:\n",
+      "- Eliminates sequential computation constraints present in RNNs, enabling superior parallelization\n",
+      "- Reduces operations needed for relating distant positions to a constant number, compared to linear/logarithmic scaling in CNNs\n",
+      "- Processes all input and output positions simultaneously through self-attention mechanisms\n",
+      "- Achieves state-of-the-art results while requiring significantly less computational resources\n",
+      "\n",
+      "Machine Translation Benchmarks:\n",
+      "- WMT 2014 English-to-German: 28.4 BLEU score, exceeding previous best results by over 2 BLEU points\n",
+      "- WMT 2014 English-to-French: 41.8 BLEU score (single-model state-of-the-art)\n",
+      "- Surpasses performance of existing model ensembles in translation tasks\n",
+      "\n",
+      "Training Efficiency:\n",
+      "- Requires only 3.5 days of training on eight GPUs for state-of-the-art performance\n",
+      "- Achieves superior results at \"a small fraction of the training costs\" compared to previous models\n",
+      "- Enables significantly faster training through parallel processing of input/output sequences\n",
+      "- Can reach production-quality performance in as little as twelve hours on modern GPU hardware\n",
+      "\n",
+      "Real-world Applications:\n",
+      "- Machine translation systems\n",
+      "- Natural language understanding tasks\n",
+      "- Reading comprehension\n",
+      "- Abstractive summarization\n",
+      "- Text entailment analysis\n",
+      "- Constituency parsing (achieving 92.7 F1 score in semi-supervised settings)\n",
+      "- Adaptable to both large and limited training data scenarios\n",
+      "\n",
+      "Scalability Benefits:\n",
+      "- Highly parallelizable architecture enables efficient scaling across multiple GPUs\n",
+      "- Constant computational complexity for relating any input/output positions\n",
+      "- Effective handling of long-range dependencies in sequences\n",
+      "- Maintains performance quality while scaling to larger datasets and model sizes\n",
+      "- Generalizes well across different tasks and domains without architectural changes\n",
+      "- Supports efficient inference and deployment in production environments\n",
+      "\n"
+     ]
+    }
+   ],
+   "source": [
+    "report_response = await report_client.aget()\n",
+    "report_text = \"\\n\\n\".join([block.template for block in report_response.report.blocks])\n",
+    "print(report_text)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "We can also see the sources for each block!"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "0.99687636\n",
+      "# Abstract\n",
+      "\n",
+      "The dominant sequence transduction models are based on complex recurrent or convolutiona\n",
+      "==================\n",
+      "0.99591404\n",
+      "# 2 Background\n",
+      "\n",
+      "The goal of reducing sequential computation also forms the foundation of the Extende\n",
+      "==================\n",
+      "0.9951325\n",
+      "# 1 Introduction\n",
+      "\n",
+      "Recurrent neural networks, long short-term memory [13] and gated recurrent [7] neu\n",
+      "==================\n",
+      "0.99442345\n",
+      "# 7 Conclusion\n",
+      "\n",
+      "In this work, we presented the Transformer, the first sequence transduction model ba\n",
+      "==================\n",
+      "0.9967649\n",
+      "# 3.2.3 Applications of Attention in our Model\n",
+      "\n",
+      "The Transformer uses multi-head attention in three d\n",
+      "==================\n",
+      "0.99533635\n",
+      "# 2 Background\n",
+      "\n",
+      "The goal of reducing sequential computation also forms the foundation of the Extende\n",
+      "==================\n",
+      "0.9935868\n",
+      "# Abstract\n",
+      "\n",
+      "The dominant sequence transduction models are based on complex recurrent or convolutiona\n",
+      "==================\n",
+      "0.98780584\n",
+      "# Outputs\n",
+      "\n",
+      "(shifted right)\n",
+      "\n",
+      "Figure 1: The Transformer - model architecture.\n",
+      "\n",
+      "The Transformer follows\n",
+      "==================\n",
+      "0.9205043\n",
+      "# 3.3 Position-wise Feed-Forward Networks\n",
+      "\n",
+      "In addition to attention sub-layers, each of the layers i\n",
+      "==================\n",
+      "0.79581684\n",
+      "# 1 Introduction\n",
+      "\n",
+      "Recurrent neural networks, long short-term memory [13] and gated recurrent [7] neu\n",
+      "==================\n",
+      "0.9946774\n",
+      "# Abstract\n",
+      "\n",
+      "The dominant sequence transduction models are based on complex recurrent or convolutiona\n",
+      "==================\n",
+      "0.97079873\n",
+      "# 7 Conclusion\n",
+      "\n",
+      "In this work, we presented the Transformer, the first sequence transduction model ba\n",
+      "==================\n",
+      "0.9535353\n",
+      "# 6.3 English Constituency Parsing\n",
+      "\n",
+      "To evaluate if the Transformer can generalize to other tasks we \n",
+      "==================\n",
+      "0.9514138\n",
+      "# 2 Background\n",
+      "\n",
+      "The goal of reducing sequential computation also forms the foundation of the Extende\n",
+      "==================\n",
+      "0.9790758\n",
+      "# 1 Introduction\n",
+      "\n",
+      "Recurrent neural networks, long short-term memory [13] and gated recurrent [7] neu\n",
+      "==================\n",
+      "0.92262185\n",
+      "# Outputs\n",
+      "\n",
+      "(shifted right)\n",
+      "\n",
+      "Figure 1: The Transformer - model architecture.\n",
+      "\n",
+      "The Transformer follows\n",
+      "==================\n"
+     ]
+    }
+   ],
+   "source": [
+    "for block in report_response.report.blocks:\n",
+    "    # Each block has a list of sources, which are the nodes that were used to generate the block\n",
+    "    for source in block.sources:\n",
+    "        print(source.score)\n",
+    "        print(source.node.text[:100])\n",
+    "        print(\"==================\")"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "llama-parse-aNC435Vv-py3.10",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}

extract.md

@@ -0,0 +1,186 @@
+# LlamaExtract
+
+> **⚠️ EXPERIMENTAL**
+> This library is under active development with frequent breaking changes. APIs and functionality may change significantly between versions. If you're interested in being an early adopter, please contact us at [support@llamaindex.ai](mailto:support@llamaindex.ai) or join our [Discord](https://discord.com/invite/eN6D2HQ4aX).
+
+LlamaExtract provides a simple API for extracting structured data from unstructured documents like PDFs, text files and images (upcoming).
+
+## Quick Start
+
+```python
+from llama_cloud_services import LlamaExtract
+from pydantic import BaseModel, Field
+
+# Initialize client
+extractor = LlamaExtract()
+
+
+# Define schema using Pydantic
+class Resume(BaseModel):
+    name: str = Field(description="Full name of candidate")
+    email: str = Field(description="Email address")
+    skills: list[str] = Field(description="Technical skills and technologies")
+
+
+# Create extraction agent
+agent = extractor.create_agent(name="resume-parser", data_schema=Resume)
+
+# Extract data from document
+result = agent.extract("resume.pdf")
+print(result.data)
+```
+
+## Core Concepts
+
+- **Extraction Agents**: Reusable extractors configured with a specific schema and extraction settings.
+- **Data Schema**: Structure definition for the data you want to extract in the form of a JSON schema or a Pydantic model.
+- **Extraction Jobs**: Asynchronous extraction tasks that can be monitored.
+
+## Defining Schemas
+
+Schemas can be defined using either Pydantic models or JSON Schema:
+
+### Using Pydantic (Recommended)
+
+```python
+from pydantic import BaseModel, Field
+from typing import List, Optional
+
+
+class Experience(BaseModel):
+    company: str = Field(description="Company name")
+    title: str = Field(description="Job title")
+    start_date: Optional[str] = Field(description="Start date of employment")
+    end_date: Optional[str] = Field(description="End date of employment")
+
+
+class Resume(BaseModel):
+    name: str = Field(description="Candidate name")
+    experience: List[Experience] = Field(description="Work history")
+```
+
+### Using JSON Schema
+
+```python
+schema = {
+    "type": "object",
+    "properties": {
+        "name": {"type": "string", "description": "Candidate name"},
+        "experience": {
+            "type": "array",
+            "description": "Work history",
+            "items": {
+                "type": "object",
+                "properties": {
+                    "company": {
+                        "type": "string",
+                        "description": "Company name",
+                    },
+                    "title": {"type": "string", "description": "Job title"},
+                    "start_date": {
+                        "anyOf": [{"type": "string"}, {"type": "null"}],
+                        "description": "Start date of employment",
+                    },
+                    "end_date": {
+                        "anyOf": [{"type": "string"}, {"type": "null"}],
+                        "description": "End date of employment",
+                    },
+                },
+            },
+        },
+    },
+}
+
+agent = extractor.create_agent(name="resume-parser", data_schema=schema)
+```
+
+### Important restrictions on JSON/Pydantic Schema
+
+_LlamaExtract only supports a subset of the JSON Schema specification._ While limited, it should
+be sufficient for a wide variety of use-cases.
+
+- All fields are required by default. Nullable fields must be explicitly marked as such,
+  using `"anyOf"` with a `"null"` type. See `"start_date"` field above.
+- Root node must be of type `"object"`.
+- Schema nesting must be limited to within 5 levels.
+- The important fields are key names/titles, type and description. Fields for
+  formatting, default values, etc. are not supported.
+- There are other restrictions on number of keys, size of the schema, etc. that you may
+  hit for complex extraction use cases. In such cases, it is worth thinking how to restructure
+  your extraction workflow to fit within these constraints, e.g. by extracting subset of fields
+  and later merging them together.
+
+## Other Extraction APIs
+
+### Batch Processing
+
+Process multiple files asynchronously:
+
+```python
+# Queue multiple files for extraction
+jobs = await agent.queue_extraction(["resume1.pdf", "resume2.pdf"])
+
+# Check job status
+for job in jobs:
+    status = agent.get_extraction_job(job.id).status
+    print(f"Job {job.id}: {status}")
+
+# Get results when complete
+results = [agent.get_extraction_run_for_job(job.id) for job in jobs]
+```
+
+### Updating Schemas
+
+Schemas can be modified and updated after creation:
+
+```python
+# Update schema
+agent.data_schema = new_schema
+
+# Save changes
+agent.save()
+```
+
+### Managing Agents
+
+```python
+# List all agents
+agents = extractor.list_agents()
+
+# Get specific agent
+agent = extractor.get_agent(name="resume-parser")
+
+# Delete agent
+extractor.delete_agent(agent.id)
+```
+
+## Installation
+
+```bash
+pip install llama-extract==0.1.0
+```
+
+## Tips & Best Practices
+
+1. **Schema Design**:
+
+   - Try to limit schema nesting to 3-4 levels.
+   - Make fields optional when data might not always be present. Having required fields may force the model
+     to hallucinate when these fields are not present in the documents.
+   - When you want to extract a variable number of entities, use an `array` type. Note that you cannot use
+     an `array` type for the root node.
+   - Use descriptive field names and detailed descriptions. Use descriptions to pass formatting
+     instructions or few-shot examples.
+   - Start simple and iteratively build your schema to incorporate requirements.
+
+2. **Running Extractions**:
+   - Note that resetting `agent.schema` will not save the schema to the database,
+     until you call `agent.save`, but it will be used for running extractions.
+   - Check job status prior to accessing results. Any extraction error should be available as
+     part of `job.error` or `extraction_run.error` fields for debugging.
+   - Consider async operations (`queue_extraction`) for large-scale extraction once you have finalized your schema.
+
+## Additional Resources
+
+- [Example Notebook](examples/resume_screening.ipynb) - Detailed walkthrough of resume parsing
+- [Discord Community](https://discord.com/invite/eN6D2HQ4aX) - Get help and share feedback

llama_cloud_services/__init__.py

@@ -0,0 +1,11 @@
+from llama_cloud_services.parse import LlamaParse
+from llama_cloud_services.report import ReportClient, LlamaReport
+from llama_cloud_services.extract import LlamaExtract, ExtractionAgent
+
+__all__ = [
+    "LlamaParse",
+    "ReportClient",
+    "LlamaReport",
+    "LlamaExtract",
+    "ExtractionAgent",
+]

llama_cloud_services/extract/__init__.py

@@ -0,0 +1,3 @@
+from llama_cloud_services.extract.extract import LlamaExtract, ExtractionAgent
+
+__all__ = ["LlamaExtract", "ExtractionAgent"]

llama_cloud_services/extract/extract.py

@@ -0,0 +1,655 @@
+import asyncio
+import os
+import time
+from io import BufferedIOBase, BufferedReader, BytesIO
+from pathlib import Path
+from typing import List, Optional, Type, Union, Coroutine, Any, TypeVar
+import warnings
+import httpx
+from pydantic import BaseModel
+from llama_cloud import (
+    ExtractAgent as CloudExtractAgent,
+    ExtractConfig,
+    ExtractJob,
+    ExtractJobCreate,
+    ExtractRun,
+    File,
+    ExtractMode,
+    StatusEnum,
+    Project,
+    ExtractTarget,
+    LlamaExtractSettings,
+)
+from llama_cloud.client import AsyncLlamaCloud
+from llama_cloud_services.extract.utils import JSONObjectType, augment_async_errors
+from llama_index.core.schema import BaseComponent
+from llama_index.core.async_utils import run_jobs
+from llama_index.core.bridge.pydantic import Field, PrivateAttr
+from llama_index.core.constants import DEFAULT_BASE_URL
+from concurrent.futures import ThreadPoolExecutor
+
+T = TypeVar("T")
+
+FileInput = Union[str, Path, bytes, BufferedIOBase]
+SchemaInput = Union[JSONObjectType, Type[BaseModel]]
+
+DEFAULT_EXTRACT_CONFIG = ExtractConfig(
+    extraction_target=ExtractTarget.PER_DOC,
+    extraction_mode=ExtractMode.ACCURATE,
+)
+
+
+class ExtractionAgent:
+    """Class representing a single extraction agent with methods for extraction operations."""
+
+    def __init__(
+        self,
+        client: AsyncLlamaCloud,
+        agent: CloudExtractAgent,
+        project_id: Optional[str] = None,
+        organization_id: Optional[str] = None,
+        check_interval: int = 1,
+        max_timeout: int = 2000,
+        num_workers: int = 4,
+        show_progress: bool = True,
+        verbose: bool = False,
+    ):
+        self._client = client
+        self._agent = agent
+        self._project_id = project_id
+        self._organization_id = organization_id
+        self.check_interval = check_interval
+        self.max_timeout = max_timeout
+        self.num_workers = num_workers
+        self.show_progress = show_progress
+        self._verbose = verbose
+        self._data_schema: Union[JSONObjectType, None] = None
+        self._config: Union[ExtractConfig, None] = None
+        self._thread_pool = ThreadPoolExecutor(
+            max_workers=min(10, (os.cpu_count() or 1) + 4)
+        )
+
+    def _run_in_thread(self, coro: Coroutine[Any, Any, T]) -> T:
+        """Run coroutine in a separate thread to avoid event loop issues"""
+
+        def run_coro() -> T:
+            async def wrapped_coro() -> T:
+                async with httpx.AsyncClient(
+                    timeout=self._client._client_wrapper.httpx_client.timeout,
+                ) as client:
+                    original_client = self._client._client_wrapper.httpx_client
+                    self._client._client_wrapper.httpx_client = client
+                    try:
+                        return await coro
+                    finally:
+                        self._client._client_wrapper.httpx_client = original_client
+
+            return asyncio.run(wrapped_coro())
+
+        return self._thread_pool.submit(run_coro).result()
+
+    @property
+    def id(self) -> str:
+        return self._agent.id
+
+    @property
+    def name(self) -> str:
+        return self._agent.name
+
+    @property
+    def data_schema(self) -> dict:
+        return self._agent.data_schema if not self._data_schema else self._data_schema
+
+    @data_schema.setter
+    def data_schema(self, data_schema: SchemaInput) -> None:
+        processed_schema: JSONObjectType
+        if isinstance(data_schema, dict):
+            # TODO: if we expose a get_validated JSON schema method, we can use it here
+            processed_schema = data_schema  # type: ignore
+        elif isinstance(data_schema, type) and issubclass(data_schema, BaseModel):
+            processed_schema = data_schema.model_json_schema()
+        else:
+            raise ValueError(
+                "data_schema must be either a dictionary or a Pydantic model"
+            )
+        validated_schema = self._run_in_thread(
+            self._client.llama_extract.validate_extraction_schema(
+                data_schema=processed_schema
+            )
+        )
+        self._data_schema = validated_schema.data_schema
+
+    @property
+    def config(self) -> ExtractConfig:
+        return self._agent.config if not self._config else self._config
+
+    @config.setter
+    def config(self, config: ExtractConfig) -> None:
+        self._config = config
+
+    async def _upload_file(self, file_input: FileInput) -> File:
+        """Upload a file for extraction."""
+        if isinstance(file_input, BufferedIOBase):
+            upload_file = file_input
+        elif isinstance(file_input, bytes):
+            upload_file = BytesIO(file_input)
+        elif isinstance(file_input, (str, Path)):
+            upload_file = open(file_input, "rb")
+        else:
+            raise ValueError(
+                "file_input must be either a file path string, file bytes, or buffer object"
+            )
+
+        try:
+            return await self._client.files.upload_file(
+                project_id=self._project_id, upload_file=upload_file
+            )
+        finally:
+            if isinstance(upload_file, BufferedReader):
+                upload_file.close()
+
+    async def _wait_for_job_result(self, job_id: str) -> Optional[ExtractRun]:
+        """Wait for and return the results of an extraction job."""
+        start = time.perf_counter()
+        tries = 0
+        while True:
+            await asyncio.sleep(self.check_interval)
+            tries += 1
+            job = await self._client.llama_extract.get_job(
+                job_id=job_id,
+            )
+
+            if job.status == StatusEnum.SUCCESS:
+                return await self._client.llama_extract.get_run_by_job_id(
+                    job_id=job_id,
+                )
+            elif job.status == StatusEnum.PENDING:
+                end = time.perf_counter()
+                if end - start > self.max_timeout:
+                    raise Exception(f"Timeout while extracting the file: {job_id}")
+                if self._verbose and tries % 10 == 0:
+                    print(".", end="", flush=True)
+                continue
+            else:
+                warnings.warn(
+                    f"Failure in job: {job_id}, status: {job.status}, error: {job.error}"
+                )
+                return await self._client.llama_extract.get_run_by_job_id(
+                    job_id=job_id,
+                )
+
+    def save(self) -> None:
+        """Persist the extraction agent's schema and config to the database.
+
+        Returns:
+            ExtractionAgent: The updated extraction agent
+        """
+        self._agent = self._run_in_thread(
+            self._client.llama_extract.update_extraction_agent(
+                extraction_agent_id=self.id,
+                data_schema=self.data_schema,
+                config=self.config,
+            )
+        )
+
+    async def _queue_extraction_test(
+        self,
+        files: Union[FileInput, List[FileInput]],
+        extract_settings: LlamaExtractSettings,
+    ) -> Union[ExtractJob, List[ExtractJob]]:
+        if not isinstance(files, list):
+            files = [files]
+            single_file = True
+        else:
+            single_file = False
+
+        upload_tasks = [self._upload_file(file) for file in files]
+        with augment_async_errors():
+            uploaded_files = await run_jobs(
+                upload_tasks,
+                workers=self.num_workers,
+                desc="Uploading files",
+                show_progress=self.show_progress,
+            )
+
+        async def run_job(file: File) -> ExtractRun:
+            job_queued = await self._client.llama_extract.run_job_test_user(
+                job_create=ExtractJobCreate(
+                    extraction_agent_id=self.id,
+                    file_id=file.id,
+                    data_schema_override=self.data_schema,
+                    config_override=self.config,
+                ),
+                extract_settings=extract_settings,
+            )
+            return await self._wait_for_job_result(job_queued.id)
+
+        job_tasks = [run_job(file) for file in uploaded_files]
+        with augment_async_errors():
+            extract_jobs = await run_jobs(
+                job_tasks,
+                workers=self.num_workers,
+                desc="Running extraction jobs",
+                show_progress=self.show_progress,
+            )
+
+        if self._verbose:
+            for file, job in zip(files, extract_jobs):
+                file_repr = (
+                    str(file) if isinstance(file, (str, Path)) else "<bytes/buffer>"
+                )
+                print(
+                    f"Queued file extraction for file {file_repr} under job_id {job.id}"
+                )
+
+        return extract_jobs[0] if single_file else extract_jobs
+
+    async def queue_extraction(
+        self,
+        files: Union[FileInput, List[FileInput]],
+    ) -> Union[ExtractJob, List[ExtractJob]]:
+        """
+        Queue multiple files for extraction.
+
+        Args:
+            files (Union[FileInput, List[FileInput]]): The files to extract
+
+        Returns:
+            Union[ExtractJob, List[ExtractJob]]: The queued extraction jobs
+        """
+        """Queue one or more files for extraction concurrently."""
+        if not isinstance(files, list):
+            files = [files]
+            single_file = True
+        else:
+            single_file = False
+
+        upload_tasks = [self._upload_file(file) for file in files]
+        with augment_async_errors():
+            uploaded_files = await run_jobs(
+                upload_tasks,
+                workers=self.num_workers,
+                desc="Uploading files",
+                show_progress=self.show_progress,
+            )
+
+        job_tasks = [
+            self._client.llama_extract.run_job(
+                request=ExtractJobCreate(
+                    extraction_agent_id=self.id,
+                    file_id=file.id,
+                    data_schema_override=self.data_schema,
+                    config_override=self.config,
+                ),
+            )
+            for file in uploaded_files
+        ]
+        with augment_async_errors():
+            extract_jobs = await run_jobs(
+                job_tasks,
+                workers=self.num_workers,
+                desc="Creating extraction jobs",
+                show_progress=self.show_progress,
+            )
+
+        if self._verbose:
+            for file, job in zip(files, extract_jobs):
+                file_repr = (
+                    str(file) if isinstance(file, (str, Path)) else "<bytes/buffer>"
+                )
+                print(
+                    f"Queued file extraction for file {file_repr} under job_id {job.id}"
+                )
+
+        return extract_jobs[0] if single_file else extract_jobs
+
+    async def aextract(
+        self, files: Union[FileInput, List[FileInput]]
+    ) -> Union[ExtractRun, List[ExtractRun]]:
+        """Asynchronously extract data from one or more files using this agent.
+
+        Args:
+            files (Union[FileInput, List[FileInput]]): The files to extract
+
+        Returns:
+            Union[ExtractRun, List[ExtractRun]]: The extraction results
+        """
+        if not isinstance(files, list):
+            files = [files]
+            single_file = True
+        else:
+            single_file = False
+
+        # Queue all files for extraction
+        jobs = await self.queue_extraction(files)
+        # Wait for all results concurrently
+        result_tasks = [self._wait_for_job_result(job.id) for job in jobs]
+        with augment_async_errors():
+            results = await run_jobs(
+                result_tasks,
+                workers=self.num_workers,
+                desc="Extracting files",
+                show_progress=self.show_progress,
+            )
+
+        return results[0] if single_file else results
+
+    def extract(
+        self, files: Union[FileInput, List[FileInput]]
+    ) -> Union[ExtractRun, List[ExtractRun]]:
+        """Synchronously extract data from one or more files using this agent.
+
+        Args:
+            files (Union[FileInput, List[FileInput]]): The files to extract
+
+        Returns:
+            Union[ExtractRun, List[ExtractRun]]: The extraction results
+        """
+        return self._run_in_thread(self.aextract(files))
+
+    def get_extraction_job(self, job_id: str) -> ExtractJob:
+        """
+        Get the extraction job for a given job_id.
+
+        Args:
+            job_id (str): The job_id to get the extraction job for
+
+        Returns:
+            ExtractJob: The extraction job
+        """
+        return self._run_in_thread(self._client.llama_extract.get_job(job_id=job_id))
+
+    def get_extraction_run_for_job(self, job_id: str) -> ExtractRun:
+        """
+        Get the extraction run for a given job_id.
+
+        Args:
+            job_id (str): The job_id to get the extraction run for
+
+        Returns:
+            ExtractRun: The extraction run
+        """
+        return self._run_in_thread(
+            self._client.llama_extract.get_run_by_job_id(
+                job_id=job_id,
+            )
+        )
+
+    def list_extraction_runs(self) -> List[ExtractRun]:
+        """List extraction runs for the extraction agent.
+
+        Returns:
+            List[ExtractRun]: List of extraction runs
+        """
+        return self._run_in_thread(
+            self._client.llama_extract.list_extract_runs(
+                extraction_agent_id=self.id,
+            )
+        )
+
+    def __repr__(self) -> str:
+        return f"ExtractionAgent(id={self.id}, name={self.name})"
+
+
+class LlamaExtract(BaseComponent):
+    """Factory class for creating and managing extraction agents."""
+
+    api_key: str = Field(description="The API key for the LlamaExtract API.")
+    base_url: str = Field(description="The base URL of the LlamaExtract API.")
+    check_interval: int = Field(
+        default=1,
+        description="The interval in seconds to check if the extraction is done.",
+    )
+    max_timeout: int = Field(
+        default=2000,
+        description="The maximum timeout in seconds to wait for the extraction to finish.",
+    )
+    num_workers: int = Field(
+        default=4,
+        gt=0,
+        lt=10,
+        description="The number of workers to use sending API requests for extraction.",
+    )
+    show_progress: bool = Field(
+        default=True, description="Show progress when extracting multiple files."
+    )
+    verbose: bool = Field(
+        default=False, description="Show verbose output when extracting files."
+    )
+    _async_client: AsyncLlamaCloud = PrivateAttr()
+    _thread_pool: ThreadPoolExecutor = PrivateAttr()
+    _project_id: Optional[str] = PrivateAttr()
+    _organization_id: Optional[str] = PrivateAttr()
+
+    def __init__(
+        self,
+        api_key: Optional[str] = None,
+        base_url: Optional[str] = None,
+        check_interval: int = 1,
+        max_timeout: int = 2000,
+        num_workers: int = 4,
+        show_progress: bool = True,
+        project_id: Optional[str] = None,
+        organization_id: Optional[str] = None,
+        verbose: bool = False,
+    ):
+        if not api_key:
+            api_key = os.getenv("LLAMA_CLOUD_API_KEY", None)
+            if api_key is None:
+                raise ValueError("The API key is required.")
+
+        if not base_url:
+            base_url = os.getenv("LLAMA_CLOUD_BASE_URL", None) or DEFAULT_BASE_URL
+
+        super().__init__(
+            api_key=api_key,
+            base_url=base_url,
+            check_interval=check_interval,
+            max_timeout=max_timeout,
+            num_workers=num_workers,
+            show_progress=show_progress,
+            verbose=verbose,
+        )
+
+        self._async_client = AsyncLlamaCloud(
+            token=self.api_key, base_url=self.base_url, timeout=None
+        )
+        self._thread_pool = ThreadPoolExecutor(
+            max_workers=min(10, (os.cpu_count() or 1) + 4)
+        )
+        # Fetch default project id if not provided
+        if not project_id:
+            project_id = os.getenv("LLAMA_CLOUD_PROJECT_ID", None)
+            if not project_id:
+                print("No project_id provided, fetching default project.")
+                projects: List[Project] = self._run_in_thread(
+                    self._async_client.projects.list_projects()
+                )
+                default_project = [p for p in projects if p.is_default]
+                if not default_project:
+                    raise ValueError(
+                        "No default project found. Please provide a project_id."
+                    )
+                project_id = default_project[0].id
+
+        self._project_id = project_id
+        self._organization_id = organization_id
+
+    def _run_in_thread(self, coro: Coroutine[Any, Any, T]) -> T:
+        """Run coroutine in a separate thread to avoid event loop issues"""
+
+        def run_coro() -> T:
+            # Create a new client for this thread
+            async def wrapped_coro() -> T:
+                async with httpx.AsyncClient(
+                    timeout=self._async_client._client_wrapper.httpx_client.timeout,
+                ) as client:
+                    # Replace the client in the coro's context
+                    original_client = self._async_client._client_wrapper.httpx_client
+                    self._async_client._client_wrapper.httpx_client = client
+                    try:
+                        return await coro
+                    finally:
+                        self._async_client._client_wrapper.httpx_client = (
+                            original_client
+                        )
+
+            return asyncio.run(wrapped_coro())
+
+        return self._thread_pool.submit(run_coro).result()
+
+    def create_agent(
+        self,
+        name: str,
+        data_schema: SchemaInput,
+        config: Optional[ExtractConfig] = None,
+    ) -> ExtractionAgent:
+        """Create a new extraction agent.
+
+        Args:
+            name (str): The name of the extraction agent
+            data_schema (SchemaInput): The data schema for the extraction agent
+            config (Optional[ExtractConfig]): The extraction config for the agent
+
+        Returns:
+            ExtractionAgent: The created extraction agent
+        """
+
+        if isinstance(data_schema, dict):
+            data_schema = data_schema
+        elif issubclass(data_schema, BaseModel):
+            data_schema = data_schema.model_json_schema()
+        else:
+            raise ValueError(
+                "data_schema must be either a dictionary or a Pydantic model"
+            )
+
+        agent = self._run_in_thread(
+            self._async_client.llama_extract.create_extraction_agent(
+                name=name,
+                data_schema=data_schema,
+                config=config or DEFAULT_EXTRACT_CONFIG,
+                project_id=self._project_id,
+                organization_id=self._organization_id,
+            )
+        )
+
+        return ExtractionAgent(
+            client=self._async_client,
+            agent=agent,
+            project_id=self._project_id,
+            organization_id=self._organization_id,
+            check_interval=self.check_interval,
+            max_timeout=self.max_timeout,
+            num_workers=self.num_workers,
+            show_progress=self.show_progress,
+            verbose=self.verbose,
+        )
+
+    def get_agent(
+        self,
+        name: Optional[str] = None,
+        id: Optional[str] = None,
+    ) -> ExtractionAgent:
+        """Get extraction agents by name or extraction agent ID.
+
+        Args:
+            name (Optional[str]): Filter by name
+            extraction_agent_id (Optional[str]): Filter by extraction agent ID
+
+        Returns:
+            ExtractionAgent: The extraction agent
+        """
+        if id is not None and name is not None:
+            warnings.warn(
+                "Both name and extraction_agent_id are provided. Using extraction_agent_id."
+            )
+
+        if id:
+            agent = self._run_in_thread(
+                self._async_client.llama_extract.get_extraction_agent(
+                    extraction_agent_id=id,
+                )
+            )
+
+        elif name:
+            agent = self._run_in_thread(
+                self._async_client.llama_extract.get_extraction_agent_by_name(
+                    name=name,
+                    project_id=self._project_id,
+                )
+            )
+        else:
+            raise ValueError("Either name or extraction_agent_id must be provided.")
+
+        return ExtractionAgent(
+            client=self._async_client,
+            agent=agent,
+            project_id=self._project_id,
+            organization_id=self._organization_id,
+            check_interval=self.check_interval,
+            max_timeout=self.max_timeout,
+            num_workers=self.num_workers,
+            show_progress=self.show_progress,
+            verbose=self.verbose,
+        )
+
+    def list_agents(self) -> List[ExtractionAgent]:
+        """List all available extraction agents."""
+        agents = self._run_in_thread(
+            self._async_client.llama_extract.list_extraction_agents(
+                project_id=self._project_id,
+            )
+        )
+
+        return [
+            ExtractionAgent(
+                client=self._async_client,
+                agent=agent,
+                project_id=self._project_id,
+                organization_id=self._organization_id,
+                check_interval=self.check_interval,
+                max_timeout=self.max_timeout,
+                num_workers=self.num_workers,
+                show_progress=self.show_progress,
+                verbose=self.verbose,
+            )
+            for agent in agents
+        ]
+
+    def delete_agent(self, agent_id: str) -> None:
+        """Delete an extraction agent by ID.
+
+        Args:
+            agent_id (str): ID of the extraction agent to delete
+        """
+        self._run_in_thread(
+            self._async_client.llama_extract.delete_extraction_agent(
+                extraction_agent_id=agent_id
+            )
+        )
+
+
+if __name__ == "__main__":
+    from dotenv import load_dotenv
+
+    load_dotenv()
+
+    data_dir = Path(__file__).parent.parent / "tests" / "data"
+    extractor = LlamaExtract()
+    try:
+        agent = extractor.get_agent(name="test-agent")
+    except Exception:
+        agent = extractor.create_agent(
+            "test-agent",
+            {
+                "type": "object",
+                "properties": {
+                    "title": {"type": "string"},
+                    "summary": {"type": "string"},
+                },
+            },
+        )
+    results = agent.extract(data_dir / "slide" / "conocophilips.pdf")
+    extractor.delete_agent(agent.id)
+    print(results)

llama_cloud_services/extract/utils.py

@@ -0,0 +1,34 @@
+from typing import Any, Dict, List, Union, Generator
+from contextlib import contextmanager
+
+# Asyncio error messages
+nest_asyncio_err = "cannot be called from a running event loop"
+nest_asyncio_msg = (
+    "The event loop is already running. "
+    "Add `import nest_asyncio; nest_asyncio.apply()` to your code to fix this issue."
+)
+
+
+def is_jupyter() -> bool:
+    """Check if we're running in a Jupyter environment."""
+    try:
+        from IPython import get_ipython
+
+        return get_ipython().__class__.__name__ == "ZMQInteractiveShell"
+    except (ImportError, AttributeError):
+        return False
+
+
+@contextmanager
+def augment_async_errors() -> Generator[None, None, None]:
+    """Context manager to add helpful information for errors due to nested event loops."""
+    try:
+        yield
+    except RuntimeError as e:
+        if nest_asyncio_err in str(e):
+            raise RuntimeError(nest_asyncio_msg)
+        raise
+
+
+JSONType = Union[Dict[str, Any], List[Any], str, int, float, bool, None]
+JSONObjectType = Dict[str, JSONType]

llama_cloud_services/parse/__init__.py

@@ -0,0 +1,3 @@
+from llama_cloud_services.parse.base import LlamaParse, ResultType
+
+__all__ = ["LlamaParse", "ResultType"]

llama_cloud_services/parse/cli/main.py

@@ -0,0 +1,92 @@
+import click
+import json
+from enum import Enum
+from pathlib import Path
+from pydantic.fields import FieldInfo
+from typing import Any, Callable, List
+
+from llama_cloud_services.parse.base import LlamaParse
+
+
+def pydantic_field_to_click_option(name: str, field: FieldInfo) -> click.Option:
+    """Convert a Pydantic field to a Click option."""
+    kwargs = {
+        "default": field.default if field.default else None,
+        "help": field.description,
+    }
+
+    if isinstance(kwargs["default"], Enum):
+        kwargs["default"] = kwargs["default"].value
+
+    if field.annotation is bool:
+        kwargs["is_flag"] = True
+        if field.default and field.default is True:
+            name = f"no-{name}"
+    return click.option(f'--{name.replace("_", "-")}', **kwargs)
+
+
+def add_options(options: List[click.Option]) -> Callable:
+    def _add_options(func: Callable) -> Callable:
+        for option in reversed(options):
+            func = option(func)
+        return func
+
+    return _add_options
+
+
+@click.command()
+@click.argument("file_paths", nargs=-1, type=click.Path(exists=True, path_type=Path))
+@click.option(
+    "--output-file", type=click.Path(path_type=Path), help="Path to save the output"
+)
+@click.option("--output-raw-json", is_flag=True, help="Output the raw JSON result")
+@add_options(
+    [
+        pydantic_field_to_click_option(name, field)
+        for name, field in LlamaParse.model_fields.items()
+        if name not in ["custom_client"]
+    ]
+)
+def parse(**kwargs: Any) -> None:
+    """Parse files using LlamaParse and output the results."""
+    file_paths = kwargs.pop("file_paths")
+    output_file = kwargs.pop("output_file")
+    output_raw_json = kwargs.pop("output_raw_json")
+
+    # Remove None values to use LlamaParse defaults
+    kwargs = {k: v for k, v in kwargs.items() if v is not None}
+
+    # Remove no- prefix for boolean flags
+    kwargs = {k.replace("no_", ""): v for k, v in kwargs.items()}
+
+    parser = LlamaParse(**kwargs)
+    if output_raw_json:
+        results = parser.get_json_result(list(file_paths))
+
+        if output_file:
+            with output_file.open("w") as f:
+                json.dump(results, f)
+            click.echo(f"Results saved to {output_file}")
+        else:
+            click.echo(results)
+    else:
+        results = parser.load_data(list(file_paths))
+
+        if output_file:
+            with output_file.open("w") as f:
+                for i, doc in enumerate(results):
+                    f.write(f"File: {doc.metadata.get('file_path', 'Unknown')}\n")  # type: ignore
+                    f.write(doc.text)  # type: ignore
+                    if i < len(results) - 1:
+                        f.write("\n\n---\n\n")
+            click.echo(f"Results saved to {output_file}")
+        else:
+            for i, doc in enumerate(results):
+                click.echo(f"File: {doc.metadata.get('file_path', 'Unknown')}")  # type: ignore
+                click.echo(doc.text)  # type: ignore
+                if i < len(results) - 1:
+                    click.echo("\n---\n")
+
+
+if __name__ == "__main__":
+    parse()