cortexflow-notion 0.1.0__tar.gz

cortexflow_notion-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Amit Chandra
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

cortexflow_notion-0.1.0/PKG-INFO

@@ -0,0 +1,47 @@
+Metadata-Version: 2.4
+Name: cortexflow-notion
+Version: 0.1.0
+Summary: CortexFlow plugin — search Notion pages and databases.
+Author-email: Amit Chandra <amit.vervebot@gmail.com>
+License: MIT
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: cortexflow-sdk>=0.1.0
+Requires-Dist: httpx>=0.27.0
+Provides-Extra: dev
+Requires-Dist: pytest>=8.2.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.23.0; extra == "dev"
+Dynamic: license-file
+
+# cortexflow-notion
+
+Example CortexFlow plugin: a `notion_search` tool that searches Notion pages
+and databases shared with your Notion integration.
+
+## Install
+
+```bash
+pip install -e ./cortexflow-sdk        # not yet on PyPI
+pip install -e examples/plugins/cortexflow-notion
+```
+
+## Setup
+
+1. Create a Notion integration at https://www.notion.so/my-integrations
+2. Share the pages/databases you want searchable with that integration
+3. Set `NOTION_TOKEN` in your environment to the integration's secret
+
+## Usage
+
+```python
+from cortexflow_notion import NotionSearchTool
+
+tool = NotionSearchTool(token="secret_...")
+result = await tool.execute(query="Q3 roadmap", limit=5)
+print(result.output)
+```
+
+Once installed alongside the CortexFlow gateway, `PluginRegistry.discover()`
+finds it via the `cortexflow.plugins` entry point declared in
+`pyproject.toml`.

cortexflow_notion-0.1.0/README.md

@@ -0,0 +1,31 @@
+# cortexflow-notion
+
+Example CortexFlow plugin: a `notion_search` tool that searches Notion pages
+and databases shared with your Notion integration.
+
+## Install
+
+```bash
+pip install -e ./cortexflow-sdk        # not yet on PyPI
+pip install -e examples/plugins/cortexflow-notion
+```
+
+## Setup
+
+1. Create a Notion integration at https://www.notion.so/my-integrations
+2. Share the pages/databases you want searchable with that integration
+3. Set `NOTION_TOKEN` in your environment to the integration's secret
+
+## Usage
+
+```python
+from cortexflow_notion import NotionSearchTool
+
+tool = NotionSearchTool(token="secret_...")
+result = await tool.execute(query="Q3 roadmap", limit=5)
+print(result.output)
+```
+
+Once installed alongside the CortexFlow gateway, `PluginRegistry.discover()`
+finds it via the `cortexflow.plugins` entry point declared in
+`pyproject.toml`.

cortexflow_notion-0.1.0/pyproject.toml

@@ -0,0 +1,31 @@
+[build-system]
+requires = ["setuptools>=68", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "cortexflow-notion"
+version = "0.1.0"
+description = "CortexFlow plugin — search Notion pages and databases."
+authors = [{ name = "Amit Chandra", email = "amit.vervebot@gmail.com" }]
+readme = "README.md"
+requires-python = ">=3.10"
+license = { text = "MIT" }
+dependencies = [
+    "cortexflow-sdk>=0.1.0",
+    "httpx>=0.27.0",
+]
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=8.2.0",
+    "pytest-asyncio>=0.23.0",
+]
+
+[project.entry-points."cortexflow.plugins"]
+cortexflow-notion = "cortexflow_notion.plugin:NotionPlugin"
+
+[tool.setuptools.packages.find]
+where = ["src"]
+
+[tool.pytest.ini_options]
+asyncio_mode = "auto"

cortexflow_notion-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

cortexflow_notion-0.1.0/src/cortexflow_notion/__init__.py

@@ -0,0 +1,7 @@
+"""cortexflow-notion — example CortexFlow plugin: Notion search."""
+
+from cortexflow_notion.plugin import NotionPlugin
+from cortexflow_notion.tool import NotionSearchTool
+
+__all__ = ["NotionPlugin", "NotionSearchTool"]
+__version__ = "0.1.0"

cortexflow_notion-0.1.0/src/cortexflow_notion/plugin.py

@@ -0,0 +1,28 @@
+"""NotionPlugin — registers NotionSearchTool with the CortexFlow gateway."""
+
+from __future__ import annotations
+
+import os
+
+from cortexflow_sdk import Plugin, PluginMetadata
+
+from cortexflow_notion.tool import NotionSearchTool
+
+
+class NotionPlugin(Plugin):
+    """Adds a notion_search tool. Reads NOTION_TOKEN from the environment."""
+
+    metadata = PluginMetadata(
+        name="cortexflow-notion",
+        version="0.1.0",
+        plugin_type="tool",
+        description="Search Notion pages and databases.",
+        permissions=["network"],
+        homepage="https://github.com/TheAmitChandra/CortexFlow",
+    )
+
+    def __init__(self) -> None:
+        self._token = os.getenv("NOTION_TOKEN")
+
+    def get_tools(self):
+        return [NotionSearchTool(token=self._token)]

cortexflow_notion-0.1.0/src/cortexflow_notion/tool.py

@@ -0,0 +1,77 @@
+"""NotionSearchTool — searches Notion pages and databases by query."""
+
+from __future__ import annotations
+
+from typing import Any
+
+from cortexflow_sdk import Tool, ToolResult
+
+_SEARCH_URL = "https://api.notion.com/v1/search"
+_NOTION_VERSION = "2022-06-28"
+
+
+def _extract_title(result: dict[str, Any]) -> str:
+    """Pull the plain-text title out of a Notion page/database object."""
+    for prop in result.get("properties", {}).values():
+        if prop.get("type") == "title":
+            texts = prop.get("title", [])
+            return "".join(t.get("plain_text", "") for t in texts) or "(untitled)"
+    return "(untitled)"
+
+
+class NotionSearchTool(Tool):
+    """Searches Notion pages and databases shared with the integration."""
+
+    name = "notion_search"
+    description = "Search Notion pages and databases by query text."
+    parameters = {
+        "query": {"type": "str", "description": "Search text", "required": True},
+        "limit": {"type": "int", "description": "Max results to return (default 10)", "required": False},
+    }
+    permissions = ["network"]
+
+    def __init__(self, token: str | None = None) -> None:
+        self._token = token
+
+    async def execute(self, query: str, limit: int = 10, **_) -> ToolResult:
+        if not self._token:
+            return ToolResult(tool=self.name, output=None, error="NOTION_TOKEN not set")
+
+        try:
+            import httpx
+        except ImportError:
+            return ToolResult(tool=self.name, output=None, error="pip install httpx")
+
+        headers = {
+            "Authorization": f"Bearer {self._token}",
+            "Notion-Version": _NOTION_VERSION,
+            "Content-Type": "application/json",
+        }
+
+        try:
+            async with httpx.AsyncClient() as client:
+                resp = await client.post(
+                    _SEARCH_URL,
+                    headers=headers,
+                    json={"query": query, "page_size": min(limit, 100)},
+                    timeout=10.0,
+                )
+                resp.raise_for_status()
+                data = resp.json()
+        except Exception as exc:
+            return ToolResult(tool=self.name, output=None, error=str(exc))
+
+        results = [
+            {
+                "id": r.get("id", ""),
+                "type": r.get("object", "unknown"),
+                "url": r.get("url", ""),
+                "title": _extract_title(r),
+            }
+            for r in data.get("results", [])[:limit]
+        ]
+        return ToolResult(
+            tool=self.name,
+            output=results,
+            metadata={"query": query, "count": len(results)},
+        )

cortexflow_notion-0.1.0/src/cortexflow_notion.egg-info/PKG-INFO

@@ -0,0 +1,47 @@
+Metadata-Version: 2.4
+Name: cortexflow-notion
+Version: 0.1.0
+Summary: CortexFlow plugin — search Notion pages and databases.
+Author-email: Amit Chandra <amit.vervebot@gmail.com>
+License: MIT
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: cortexflow-sdk>=0.1.0
+Requires-Dist: httpx>=0.27.0
+Provides-Extra: dev
+Requires-Dist: pytest>=8.2.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.23.0; extra == "dev"
+Dynamic: license-file
+
+# cortexflow-notion
+
+Example CortexFlow plugin: a `notion_search` tool that searches Notion pages
+and databases shared with your Notion integration.
+
+## Install
+
+```bash
+pip install -e ./cortexflow-sdk        # not yet on PyPI
+pip install -e examples/plugins/cortexflow-notion
+```
+
+## Setup
+
+1. Create a Notion integration at https://www.notion.so/my-integrations
+2. Share the pages/databases you want searchable with that integration
+3. Set `NOTION_TOKEN` in your environment to the integration's secret
+
+## Usage
+
+```python
+from cortexflow_notion import NotionSearchTool
+
+tool = NotionSearchTool(token="secret_...")
+result = await tool.execute(query="Q3 roadmap", limit=5)
+print(result.output)
+```
+
+Once installed alongside the CortexFlow gateway, `PluginRegistry.discover()`
+finds it via the `cortexflow.plugins` entry point declared in
+`pyproject.toml`.

cortexflow_notion-0.1.0/src/cortexflow_notion.egg-info/SOURCES.txt

@@ -0,0 +1,14 @@
+LICENSE
+README.md
+pyproject.toml
+src/cortexflow_notion/__init__.py
+src/cortexflow_notion/plugin.py
+src/cortexflow_notion/tool.py
+src/cortexflow_notion.egg-info/PKG-INFO
+src/cortexflow_notion.egg-info/SOURCES.txt
+src/cortexflow_notion.egg-info/dependency_links.txt
+src/cortexflow_notion.egg-info/entry_points.txt
+src/cortexflow_notion.egg-info/requires.txt
+src/cortexflow_notion.egg-info/top_level.txt
+tests/test_plugin.py
+tests/test_tool.py

cortexflow_notion-0.1.0/src/cortexflow_notion.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

cortexflow_notion-0.1.0/src/cortexflow_notion.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[cortexflow.plugins]
+cortexflow-notion = cortexflow_notion.plugin:NotionPlugin

cortexflow_notion-0.1.0/src/cortexflow_notion.egg-info/requires.txt

@@ -0,0 +1,6 @@
+cortexflow-sdk>=0.1.0
+httpx>=0.27.0
+
+[dev]
+pytest>=8.2.0
+pytest-asyncio>=0.23.0

cortexflow_notion-0.1.0/src/cortexflow_notion.egg-info/top_level.txt

@@ -0,0 +1 @@
+cortexflow_notion

cortexflow_notion-0.1.0/tests/test_plugin.py

@@ -0,0 +1,36 @@
+"""Unit tests for cortexflow_notion.plugin — NotionPlugin."""
+
+from __future__ import annotations
+
+from cortexflow_notion.plugin import NotionPlugin
+from cortexflow_notion.tool import NotionSearchTool
+
+
+def test_plugin_metadata():
+    p = NotionPlugin()
+    assert p.metadata.name == "cortexflow-notion"
+    assert p.metadata.plugin_type == "tool"
+
+
+def test_plugin_reads_token_from_env(monkeypatch):
+    monkeypatch.setenv("NOTION_TOKEN", "secret_fromenv")
+    p = NotionPlugin()
+    assert p._token == "secret_fromenv"
+
+
+def test_plugin_no_token_in_env(monkeypatch):
+    monkeypatch.delenv("NOTION_TOKEN", raising=False)
+    p = NotionPlugin()
+    assert p._token is None
+
+
+def test_get_tools_returns_notion_search_tool():
+    tools = NotionPlugin().get_tools()
+    assert len(tools) == 1
+    assert isinstance(tools[0], NotionSearchTool)
+
+
+def test_get_tools_tool_carries_plugin_token(monkeypatch):
+    monkeypatch.setenv("NOTION_TOKEN", "secret_carried")
+    tool = NotionPlugin().get_tools()[0]
+    assert tool._token == "secret_carried"

cortexflow_notion-0.1.0/tests/test_tool.py

@@ -0,0 +1,103 @@
+"""Unit tests for cortexflow_notion.tool — NotionSearchTool."""
+
+from __future__ import annotations
+
+from unittest.mock import AsyncMock, MagicMock, patch
+
+import pytest
+from cortexflow_notion.tool import NotionSearchTool, _extract_title
+
+_SAMPLE_RESULT = {
+    "id": "page-1",
+    "object": "page",
+    "url": "https://notion.so/page-1",
+    "properties": {
+        "Name": {"type": "title", "title": [{"plain_text": "Q3 "}, {"plain_text": "Roadmap"}]},
+    },
+}
+
+
+def _make_mock_client(json_body, raise_on_status: Exception | None = None):
+    mock_resp = MagicMock()
+    mock_resp.raise_for_status = MagicMock(side_effect=raise_on_status)
+    mock_resp.json = MagicMock(return_value=json_body)
+
+    mock_client = MagicMock()
+    mock_client.__aenter__ = AsyncMock(return_value=mock_client)
+    mock_client.__aexit__ = AsyncMock(return_value=False)
+    mock_client.post = AsyncMock(return_value=mock_resp)
+    return mock_client
+
+
+def test_extract_title_joins_rich_text_segments():
+    assert _extract_title(_SAMPLE_RESULT) == "Q3 Roadmap"
+
+
+def test_extract_title_untitled_when_no_title_property():
+    assert _extract_title({"properties": {}}) == "(untitled)"
+
+
+@pytest.mark.asyncio
+async def test_execute_no_token_returns_error():
+    tool = NotionSearchTool(token=None)
+    result = await tool.execute(query="x")
+    assert not result.success
+    assert "NOTION_TOKEN" in result.error
+
+
+@pytest.mark.asyncio
+async def test_execute_returns_formatted_results():
+    tool = NotionSearchTool(token="secret_abc")
+    mock_client = _make_mock_client({"results": [_SAMPLE_RESULT]})
+
+    with patch("httpx.AsyncClient", return_value=mock_client):
+        result = await tool.execute(query="roadmap")
+
+    assert result.success
+    assert result.output[0]["title"] == "Q3 Roadmap"
+    assert result.output[0]["type"] == "page"
+    assert result.metadata["query"] == "roadmap"
+
+
+@pytest.mark.asyncio
+async def test_execute_sends_notion_version_header():
+    tool = NotionSearchTool(token="secret_abc")
+    mock_client = _make_mock_client({"results": []})
+
+    with patch("httpx.AsyncClient", return_value=mock_client):
+        await tool.execute(query="x")
+
+    headers = mock_client.post.call_args[1]["headers"]
+    assert headers["Authorization"] == "Bearer secret_abc"
+    assert headers["Notion-Version"] == "2022-06-28"
+
+
+@pytest.mark.asyncio
+async def test_execute_http_error_returns_error_result():
+    tool = NotionSearchTool(token="secret_abc")
+    mock_client = _make_mock_client({}, raise_on_status=RuntimeError("401 Unauthorized"))
+
+    with patch("httpx.AsyncClient", return_value=mock_client):
+        result = await tool.execute(query="x")
+
+    assert not result.success
+    assert "401" in result.error
+
+
+@pytest.mark.asyncio
+async def test_execute_missing_httpx_returns_error():
+    import builtins
+
+    real_import = builtins.__import__
+
+    def fake_import(name, *args, **kwargs):
+        if name == "httpx":
+            raise ImportError("No module named 'httpx'")
+        return real_import(name, *args, **kwargs)
+
+    tool = NotionSearchTool(token="secret_abc")
+    with patch("builtins.__import__", side_effect=fake_import):
+        result = await tool.execute(query="x")
+
+    assert not result.success
+    assert "pip install httpx" in result.error