From 3b5e8bacfa6d5b7c223cb93f09c6c21d39542b43 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=AB=B9=E5=86=85=E8=AC=99=E5=A4=AA?=
 <49678973+kenta-takeuchi@users.noreply.github.com>
Date: Wed, 15 Nov 2023 13:31:12 +0900
Subject: [PATCH] FEAT Add some properties to NotionDBLoader (#13358)

<!-- Thank you for contributing to LangChain!

Replace this entire comment with:
  - **Description:** a description of the change,
  - **Issue:** the issue # it fixes (if applicable),
  - **Dependencies:** any dependencies required for this change,
- **Tag maintainer:** for a quicker response, tag the relevant
maintainer (see below),
- **Twitter handle:** we announce bigger features on Twitter. If your PR
gets announced, and you'd like a mention, we'll gladly shout you out!

Please make sure your PR is passing linting and testing before
submitting. Run `make format`, `make lint` and `make test` to check this
locally.

See contribution guidelines for more information on how to write/run
tests, lint, etc:

https://github.com/langchain-ai/langchain/blob/master/.github/CONTRIBUTING.md

If you're adding a new integration, please include:
1. a test for the integration, preferably unit tests that do not rely on
network access,
2. an example notebook showing its use. It lives in `docs/extras`
directory.

If no one reviews your PR within a few days, please @-mention one of
@baskaryan, @eyurtsev, @hwchase17.
 -->

fix #13356

Add supports following properties for metadata to NotionDBLoader.

- `checkbox`
- `email`
- `number`
- `select`

There are no relevant tests for this code to be updated.
---
 libs/langchain/langchain/document_loaders/notiondb.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/libs/langchain/langchain/document_loaders/notiondb.py b/libs/langchain/langchain/document_loaders/notiondb.py
index d7728efcc8..e24d6fe555 100644
--- a/libs/langchain/langchain/document_loaders/notiondb.py
+++ b/libs/langchain/langchain/document_loaders/notiondb.py
@@ -130,6 +130,14 @@ class NotionDBLoader(BaseLoader):
                 )
             elif prop_type == "created_time":
                 value = prop_data["created_time"] if prop_data["created_time"] else None
+            elif prop_type == "checkbox":
+                value = prop_data["checkbox"]
+            elif prop_type == "email":
+                value = prop_data["email"]
+            elif prop_type == "number":
+                value = prop_data["number"]
+            elif prop_type == "select":
+                value = prop_data["select"]["name"] if prop_data["select"] else None
             else:
                 value = None