diff --git a/35-vanilla-agent-pdf/vanilla_agent_pdf/main.py b/35-vanilla-agent-pdf/vanilla_agent_pdf/main.py
index e6045c0..aabd2e0 100644
--- a/35-vanilla-agent-pdf/vanilla_agent_pdf/main.py
+++ b/35-vanilla-agent-pdf/vanilla_agent_pdf/main.py
@@ -9,7 +9,11 @@
 from sse_starlette.sse import EventSourceResponse
 
 from openbb_ai.models import (
+    Citation,
+    CitationHighlightBoundingBox,
+    CitationCollectionSSE,
     MessageChunkSSE,
+    FunctionCallSSE,
     QueryRequest,
     SingleFileReference,
     SingleDataContent,
@@ -18,7 +22,7 @@
     DataFileReferences,
     WidgetRequest,
 )
-from openbb_ai import message_chunk, get_widget_data
+from openbb_ai import message_chunk, get_widget_data, citations, cite
 
 from openai.types.chat import (
     ChatCompletionMessageParam,
@@ -87,12 +91,12 @@ async def query(request: QueryRequest) -> EventSourceResponse:
                 )
             )
 
-        async def retrieve_widget_data():
-            yield get_widget_data(widget_requests).model_dump()
+        async def retrieve_widget_data() -> AsyncGenerator[FunctionCallSSE, None]:
+            yield get_widget_data(widget_requests)
 
         # Early exit to retrieve widget data
         return EventSourceResponse(
-            content=retrieve_widget_data(),
+            content=(event.model_dump() async for event in retrieve_widget_data()),
             media_type="text/event-stream",
         )
 
@@ -105,6 +109,7 @@ async def retrieve_widget_data():
     ]
 
     context_str = ""
+    citations_list: list[Citation] = []
     for index, message in enumerate(request.messages):
         if message.role == "human":
             openai_messages.append(
@@ -124,11 +129,68 @@ async def retrieve_widget_data():
         elif message.role == "tool" and index == len(request.messages) - 1:
             context_str += await handle_widget_data(message.data)
 
+            # We also need to create citations for the widget data we retrieved.
+            for widget_data_request in message.input_arguments["data_sources"]:
+                filtered_widgets = list(
+                    filter(
+                        lambda w: str(w.uuid) == widget_data_request["widget_uuid"],
+                        request.widgets.primary,
+                    )
+                )
+                if filtered_widgets:
+                    quote_bounding_boxes = [
+                        [
+                            CitationHighlightBoundingBox(
+                                text="Some text chunk.",
+                                page=1,
+                                x0=72.0,
+                                top=117,
+                                x1=259,
+                                bottom=135,
+                            ),
+                            CitationHighlightBoundingBox(
+                                text="Some text chunk.",
+                                page=1,
+                                x0=110.0,
+                                top=140,
+                                x1=259,
+                                bottom=160,
+                            ),
+                        ],
+                        [
+                            CitationHighlightBoundingBox(
+                                text="Some text chunk.",
+                                page=1,
+                                x0=110,
+                                top=170,
+                                x1=275,
+                                bottom=185,
+                            ),
+                        ],
+                    ]
+                    citation = cite(
+                        widget=filtered_widgets[0],
+                        input_arguments=widget_data_request["input_args"],
+                        # You can add any extra details you want to the
+                        # citation using the `extra_details` argument.
+                        extra_details={
+                            "Widget Name": filtered_widgets[0].name,
+                            "Widget Input Arguments": widget_data_request["input_args"],
+                        },
+                    )
+                    # Add the bounding boxes to the citation.
+                    # This is just an example, you can modify the bounding boxes
+                    # as needed.
+                    citation.quote_bounding_boxes = quote_bounding_boxes
+                    citations_list.append(citation)
+
     if context_str:
         openai_messages[-1]["content"] += "\n\n" + context_str  # type: ignore
 
     # Define the execution loop.
-    async def execution_loop() -> AsyncGenerator[MessageChunkSSE, None]:
+    async def execution_loop() -> (
+        AsyncGenerator[MessageChunkSSE | CitationCollectionSSE, None]
+    ):
         client = openai.AsyncOpenAI()
         async for event in await client.chat.completions.create(
             model="gpt-4o",
@@ -136,11 +198,14 @@ async def execution_loop() -> AsyncGenerator[MessageChunkSSE, None]:
             stream=True,
         ):
             if chunk := event.choices[0].delta.content:
-                yield message_chunk(chunk).model_dump()
+                yield message_chunk(chunk)
+
+        if citations_list:
+            yield citations(citations_list)
 
     # Stream the SSEs back to the client.
     return EventSourceResponse(
-        content=execution_loop(),
+        content=(event.model_dump() async for event in execution_loop()),
         media_type="text/event-stream",
     )
 
diff --git a/poetry.lock b/poetry.lock
index 5113026..9b2d574 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1479,14 +1479,14 @@ voice-helpers = ["numpy (>=2.0.2)", "sounddevice (>=0.5.1)"]
 
 [[package]]
 name = "openbb-ai"
-version = "1.5.0"
+version = "1.6.2"
 description = "An SDK for building agents compatible with OpenBB Workspace"
 optional = false
 python-versions = ">=3.10"
 groups = ["main"]
 files = [
-    {file = "openbb_ai-1.5.0-py3-none-any.whl", hash = "sha256:4cb787724da55a097e48400c26a09549c4821c3997657ef6fb54bac14aec786a"},
-    {file = "openbb_ai-1.5.0.tar.gz", hash = "sha256:38e37774a9513afd771f379445125dfb82b90dba7eb06024da3a1218cf0069b2"},
+    {file = "openbb_ai-1.6.2-py3-none-any.whl", hash = "sha256:b82c7ada4a6e61effdcf346061fa86916d63658c91aba8484a69c6c48da5ef43"},
+    {file = "openbb_ai-1.6.2.tar.gz", hash = "sha256:718f18a945db5e84ba64909ab372507d83d101678f697d9ab90419961a86642b"},
 ]
 
 [package.dependencies]
@@ -3037,4 +3037,4 @@ type = ["pytest-mypy"]
 [metadata]
 lock-version = "2.1"
 python-versions = "^3.10"
-content-hash = "bf765f2019bcbea3b209beb6a85f5021bd12ec82bca5b5b5679ac4367b70fb9d"
+content-hash = "4d6e66517caeaa27a1cea0c718ad451b82a305fb38cae61b7ab6c398e8861deb"
diff --git a/pyproject.toml b/pyproject.toml
index 6254ce2..13a5928 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -16,7 +16,7 @@ pdfplumber = "^0.11.5"
 mypy = "^1.15.0"
 google-genai = "^1.11.0"
 asyncstdlib = "^3.13.1"
-openbb-ai = "^1.5.0"
+openbb-ai = "^1.6.0"
 
 
 [tool.poetry.group.development.dependencies]