agentgraph-server 0.5.4__tar.gz → 0.6.0__tar.gz

{agentgraph_server-0.5.4 → agentgraph_server-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentgraph-server
-Version: 0.5.4
+Version: 0.6.0
 Summary: A local-first CLI and MCP server that turns selected digital sources into a searchable graph for the AI agent you already use.
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown

{agentgraph_server-0.5.4 → agentgraph_server-0.6.0}/agentgraph/backends/sqlite/backend.py

@@ -64,7 +64,7 @@ _LIST_PAGE_ORDER_BY = {
 _COLUMN_FILTERS = {"platform", "platform_entity_id", "entity_type"}
 _FTS_DELETE_CHUNK_SIZE = 500
 _BUSY_TIMEOUT_MS = 5_000
-_SCHEMA_VERSION = 2
+_SCHEMA_VERSION = 3
 
 
 def _now() -> str:
@@ -208,7 +208,9 @@ class SQLiteBackend(StorageBackend):
     async def _run_migrations(self) -> None:
         conn = self._conn_or_raise()
         cursor = await conn.execute("PRAGMA table_info(entities)")
-        entity_columns = {str(row["name"]): bool(row["notnull"]) for row in await cursor.fetchall()}
+        table_info = await cursor.fetchall()
+        entity_columns = {str(row["name"]): bool(row["notnull"]) for row in table_info}
+        entity_defaults = {str(row["name"]): row["dflt_value"] for row in table_info}
         columns = set(entity_columns)
         if "cumulative_dwell_ms" in columns:
             await conn.execute(
@@ -234,6 +236,8 @@ class SQLiteBackend(StorageBackend):
             "last_accessed" in columns
             or not entity_columns.get("created_at", False)
             or not entity_columns.get("updated_at", False)
+            or entity_defaults.get("created_at") is None
+            or entity_defaults.get("updated_at") is None
             or entity_columns.get("observed_at", False)
             or "source_created_at" not in columns
             or "source_updated_at" not in columns
@@ -370,8 +374,8 @@ class SQLiteBackend(StorageBackend):
                     content TEXT,
                     content_embedding BLOB,
                     metadata TEXT NOT NULL DEFAULT '{}',
-                    created_at TEXT NOT NULL,
-                    updated_at TEXT NOT NULL,
+                    created_at TEXT NOT NULL DEFAULT (strftime('%Y-%m-%dT%H:%M:%SZ', 'now')),
+                    updated_at TEXT NOT NULL DEFAULT (strftime('%Y-%m-%dT%H:%M:%SZ', 'now')),
                     source_created_at TEXT,
                     source_updated_at TEXT,
                     synced_at TEXT,
@@ -660,8 +664,8 @@ class SQLiteBackend(StorageBackend):
                     """
                     INSERT INTO entities
                         (id, entity_type, platform, platform_entity_id, title, metadata,
-                         retention_policy, retention_parent_id)
-                    VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+                         created_at, updated_at, retention_policy, retention_parent_id)
+                    VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
                     ON CONFLICT (platform, platform_entity_id) DO UPDATE SET
                         entity_type = entities.entity_type
                     RETURNING id
@@ -673,6 +677,8 @@ class SQLiteBackend(StorageBackend):
                         e.platform_entity_id,
                         e.title,
                         json.dumps(dict(e.metadata)),
+                        now,
+                        now,
                         e.retention_policy,
                         parent_id,
                     ],
@@ -756,8 +762,8 @@ class SQLiteBackend(StorageBackend):
                     INSERT INTO entities
                         (id, entity_type, platform, platform_entity_id, title, content,
                          content_embedding, metadata, source_created_at, source_updated_at,
-                         synced_at, retention_policy, retention_parent_id)
-                    VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+                         synced_at, created_at, updated_at, retention_policy, retention_parent_id)
+                    VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
                     ON CONFLICT (platform, platform_entity_id) DO UPDATE SET
                         entity_type       = CASE WHEN entities.entity_type = 'Document' THEN EXCLUDED.entity_type ELSE entities.entity_type END,
                         title             = COALESCE(EXCLUDED.title, entities.title),
@@ -784,6 +790,8 @@ class SQLiteBackend(StorageBackend):
                         source_created,
                         source_updated,
                         now,
+                        now,
+                        local_updated,
                         e.retention_policy,
                         parent_id,
                         local_updated,

{agentgraph_server-0.5.4 → agentgraph_server-0.6.0}/agentgraph/cli.py

@@ -4,8 +4,12 @@ from __future__ import annotations
 
 import asyncio
 import json as _json
+import urllib.error
+import urllib.request
 from collections.abc import Iterator
 from contextlib import contextmanager
+from importlib.metadata import PackageNotFoundError
+from importlib.metadata import version as package_version
 from typing import TYPE_CHECKING, cast
 
 import typer
@@ -18,6 +22,33 @@ app = typer.Typer(
     help="Local knowledge graph for AI agents.",
     no_args_is_help=True,
 )
+
+
+def _version_callback(value: bool) -> None:
+    if not value:
+        return
+
+    try:
+        current = package_version("agentgraph-server")
+    except PackageNotFoundError:
+        current = "unknown"
+    typer.echo(f"agentgraph {current}")
+    raise typer.Exit()
+
+
+@app.callback()
+def main(
+    version: bool = typer.Option(
+        False,
+        "--version",
+        callback=_version_callback,
+        is_eager=True,
+        help="Show the installed version and exit.",
+    ),
+) -> None:
+    """Configure global CLI options."""
+
+
 demo_app = typer.Typer(
     help="Create reproducible local demo graphs.",
     no_args_is_help=True,
@@ -50,6 +81,20 @@ def _status_label(status: str) -> str:
     )
 
 
+def _server_is_running() -> bool:
+    """Return whether the configured AgentGraph server responds to health checks."""
+    from agentgraph.config import get_settings
+
+    settings = get_settings()
+    host = "127.0.0.1" if settings.server_host in ("", "0.0.0.0", "::") else settings.server_host
+    url = f"http://{host}:{settings.server_port}/health"
+    try:
+        with urllib.request.urlopen(url, timeout=1) as response:
+            return response.status == 200
+    except (OSError, urllib.error.URLError):
+        return False
+
+
 @demo_app.command("add")
 def add_demo_command(
     json: bool = typer.Option(False, "--json", help="Output as JSON"),
@@ -430,6 +475,7 @@ def serve(
     settings = get_settings()
     configure_logging(settings.log_level, settings.log_file)
     typer.echo(f"AgentGraph config directory: {get_config_paths()[0]}")
+    typer.echo(f"AgentGraph log file: {settings.log_file.expanduser()}")
     uvicorn.run(
         "agentgraph.server.app:app",
         host=settings.server_host,
@@ -595,8 +641,15 @@ def onboard() -> None:
         if label not in seen:
             seen[label] = connector
 
-    steps = list(seen.items())
-    total = len(steps)
+    # Only connectors with an explicit onboarding prompt own an interactive setup flow.
+    # Generic connectors such as Web are configured through connector commands instead.
+    steps = [
+        (label, connector)
+        for label, connector in seen.items()
+        if getattr(connector, "onboard_prompt", None)
+    ]
+    steps.sort(key=lambda item: getattr(item[1], "onboard_last", False))
+    total = len(steps) + 1
 
     typer.echo("=== AgentGraph Setup ===\n")
 
@@ -611,11 +664,12 @@ def onboard() -> None:
         if i < total:
             typer.echo()
 
+    typer.echo(f"\nStep {total}/{total}: Install the AgentGraph Chrome Extension")
     typer.echo(
-        "\nInstall the AgentGraph Chrome Extension: "
         "https://chromewebstore.google.com/detail/agentgraph-extension/iilkfclglabllelhjacijldknapbhidi"
     )
-    typer.echo("Run `agentgraph serve` to start the server.")
+    if not _server_is_running():
+        typer.echo("Run `agentgraph serve` to start the server.")
 
 
 @app.command()

{agentgraph_server-0.5.4 → agentgraph_server-0.6.0}/agentgraph/connectors/base.py

@@ -246,6 +246,7 @@ class BaseConnector(ABC):
     auth_label: ClassVar[str | None] = None
     auth_description: ClassVar[str | None] = None
     onboard_prompt: ClassVar[str | None] = None
+    onboard_last: ClassVar[bool] = False
     appears_in_auth_status: ClassVar[bool] = True
     """True for connectors backed by user/provider credentials.
 

{agentgraph_server-0.5.4 → agentgraph_server-0.6.0}/agentgraph_server.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentgraph-server
-Version: 0.5.4
+Version: 0.6.0
 Summary: A local-first CLI and MCP server that turns selected digital sources into a searchable graph for the AI agent you already use.
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown

{agentgraph_server-0.5.4 → agentgraph_server-0.6.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentgraph-server"
-version = "0.5.4"
+version = "0.6.0"
 description = "A local-first CLI and MCP server that turns selected digital sources into a searchable graph for the AI agent you already use."
 readme = "README.md"
 requires-python = ">=3.12"

{agentgraph_server-0.5.4 → agentgraph_server-0.6.0}/tests/test_cli.py

@@ -58,6 +58,38 @@ def test_help() -> None:
     assert "ingest" not in result.output
 
 
+def test_version() -> None:
+    result = runner.invoke(app, ["--version"])
+
+    assert result.exit_code == 0
+    assert result.output == "agentgraph 0.6.0\n"
+
+
+def test_serve_outputs_log_file_path() -> None:
+    log_file = Path("/tmp/agentgraph-test/agentgraph.log")
+    settings = SimpleNamespace(
+        log_level="INFO",
+        log_file=log_file,
+        server_host="127.0.0.1",
+        server_port=8765,
+    )
+    with (
+        patch("agentgraph.config.get_settings", return_value=settings),
+        patch("agentgraph.logging.configure_logging"),
+        patch("uvicorn.run") as run,
+    ):
+        result = runner.invoke(app, ["serve"])
+
+    assert result.exit_code == 0
+    assert f"AgentGraph log file: {log_file}" in result.output
+    run.assert_called_once_with(
+        "agentgraph.server.app:app",
+        host="127.0.0.1",
+        port=8765,
+        reload=False,
+    )
+
+
 def test_demo_add_creates_fixture_and_outputs_json(
     tmp_path: Path, monkeypatch: pytest.MonkeyPatch
 ) -> None:
@@ -781,10 +813,12 @@ def test_onboard_directs_users_to_install_chrome_extension() -> None:
     with (
         patch("agentgraph.connectors.registry.bootstrap"),
         patch("agentgraph.connectors.registry.get_all_connectors", return_value=[_FakeConnector()]),
+        patch("agentgraph.cli._server_is_running", return_value=False),
     ):
         result = runner.invoke(app, ["onboard"], input="n\n")
 
     assert result.exit_code == 0
+    assert "Step 2/2: Install the AgentGraph Chrome Extension" in result.output
     assert "Install the AgentGraph Chrome Extension" in result.output
     assert (
         "https://chromewebstore.google.com/detail/agentgraph-extension/"
@@ -793,6 +827,87 @@ def test_onboard_directs_users_to_install_chrome_extension() -> None:
     assert "Run `agentgraph serve` to start the server." in result.output
 
 
+def test_onboard_omits_server_hint_when_server_is_running() -> None:
+    with (
+        patch("agentgraph.connectors.registry.bootstrap"),
+        patch("agentgraph.connectors.registry.get_all_connectors", return_value=[_FakeConnector()]),
+        patch("agentgraph.cli._server_is_running", return_value=True),
+    ):
+        result = runner.invoke(app, ["onboard"], input="n\n")
+
+    assert result.exit_code == 0
+    assert "Step 2/2: Install the AgentGraph Chrome Extension" in result.output
+    assert "Run `agentgraph serve` to start the server." not in result.output
+
+
+def test_onboard_skips_connectors_without_an_onboarding_flow() -> None:
+    class _FakeWebConnector:
+        source = "web"
+        auth_label = None
+        auth_description = "Web"
+        onboard_prompt = None
+        onboard_last = False
+        poll_interval = None
+        poll_delegates: list[str] = []
+        url_patterns: list[str] = []
+
+    with (
+        patch("agentgraph.connectors.registry.bootstrap"),
+        patch(
+            "agentgraph.connectors.registry.get_all_connectors",
+            return_value=[_FakeWebConnector()],
+        ),
+    ):
+        result = runner.invoke(app, ["onboard"])
+
+    assert result.exit_code == 0
+    assert "Step 1/1: Install the AgentGraph Chrome Extension" in result.output
+    assert "Set up web?" not in result.output
+
+
+def test_onboard_places_last_step_connectors_at_the_end() -> None:
+    class _FakeRssOnboardConnector:
+        source = "rss"
+        auth_label = "rss"
+        auth_description = "RSS"
+        onboard_prompt = "Set up RSS feeds?"
+        onboard_last = True
+        poll_interval = None
+        poll_delegates: list[str] = []
+        url_patterns: list[str] = []
+        auth_calls = 0
+
+        @classmethod
+        def run_auth_flow(cls) -> None:
+            cls.auth_calls += 1
+
+    class _FakeSlackOnboardConnector:
+        source = "slack"
+        auth_label = "slack"
+        auth_description = "Slack"
+        onboard_prompt = "Set up Slack?"
+        onboard_last = False
+        poll_interval = None
+        poll_delegates: list[str] = []
+        url_patterns: list[str] = []
+
+        @classmethod
+        def run_auth_flow(cls) -> None:
+            pass
+
+    with (
+        patch("agentgraph.connectors.registry.bootstrap"),
+        patch(
+            "agentgraph.connectors.registry.get_all_connectors",
+            return_value=[_FakeRssOnboardConnector(), _FakeSlackOnboardConnector()],
+        ),
+    ):
+        result = runner.invoke(app, ["onboard"], input="n\nn\n")
+
+    assert result.exit_code == 0
+    assert result.output.index("Step 1/3: Slack") < result.output.index("Step 2/3: RSS")
+
+
 def test_auth_slack_accepts_noninteractive_options_after_provider() -> None:
     from agentgraph_connector_slack import SlackConnector
 

{agentgraph_server-0.5.4 → agentgraph_server-0.6.0}/tests/test_connectors.py

@@ -215,6 +215,12 @@ def gdocs_connector() -> GoogleDocsConnector:
     return GoogleDocsConnector()
 
 
+def test_gdocs_onboarding_description_covers_google_connectors() -> None:
+    assert GoogleDocsConnector.auth_description == (
+        "Google: Gmail, Google Drive, Google Docs, and Google Sheets."
+    )
+
+
 @pytest.mark.asyncio
 async def test_gdocs_fetch_fresh_returns_empty_batch(gdocs_connector: GoogleDocsConnector) -> None:
     """When data is fresh, connector returns an empty batch without mutation."""

{agentgraph_server-0.5.4 → agentgraph_server-0.6.0}/tests/test_schema.py

@@ -11,6 +11,7 @@ from unittest.mock import AsyncMock, patch
 import pytest
 
 from agentgraph.backends.sqlite.backend import SQLiteBackend
+from agentgraph.connectors.base import EntityBatch, EntityRecord
 from agentgraph.core.context import set_backend
 
 
@@ -378,6 +379,8 @@ async def test_existing_database_gets_columns_and_renames_threads_to_email(tmp_p
         by_name = {row["name"]: row for row in columns}
         assert by_name["created_at"]["notnull"] == 1
         assert by_name["updated_at"]["notnull"] == 1
+        assert by_name["created_at"]["dflt_value"] is not None
+        assert by_name["updated_at"]["dflt_value"] is not None
         assert by_name["observed_at"]["notnull"] == 0
 
         entity = await migrated.get_entity_by_platform("gmail", "test-thread-001")
@@ -394,9 +397,103 @@ async def test_existing_database_gets_columns_and_renames_threads_to_email(tmp_p
         assert updated is not None
         assert updated["cumulative_observation_duration_ms"] == 2221
         assert updated["observed_at"] is not None
+        await migrated.upsert_batch(
+            EntityBatch(
+                entities=[
+                    EntityRecord(
+                        entity_type="Message",
+                        platform="slack",
+                        platform_entity_id="T/C/123.456",
+                        content="A migrated database must accept new entities",
+                    )
+                ]
+            ),
+            {},
+            {},
+        )
+        inserted = await migrated.get_entity_by_platform("slack", "T/C/123.456")
+        assert inserted is not None
+        assert inserted["created_at"] is not None
+        assert inserted["updated_at"] is not None
         event_type = await migrated._fetchval(
             "SELECT event_type FROM observations WHERE id = ?", ["legacy-observation"]
         )
         assert event_type == "observation_threshold"
     finally:
         await migrated.close()
+
+
+async def test_version_two_database_gets_timestamp_default_repair(tmp_path: Path) -> None:
+    db_path = tmp_path / "missing-timestamp-defaults.db"
+    conn = sqlite3.connect(db_path)
+    conn.executescript(
+        """
+        CREATE TABLE entities (
+            id TEXT PRIMARY KEY,
+            entity_type TEXT NOT NULL,
+            platform TEXT NOT NULL,
+            platform_entity_id TEXT NOT NULL,
+            title TEXT,
+            content TEXT,
+            content_embedding BLOB,
+            metadata TEXT NOT NULL DEFAULT '{}',
+            created_at TEXT NOT NULL,
+            updated_at TEXT NOT NULL,
+            source_created_at TEXT,
+            source_updated_at TEXT,
+            synced_at TEXT,
+            observed_at TEXT,
+            retention_policy TEXT NOT NULL DEFAULT 'observed'
+                CHECK (retention_policy IN ('observed', 'owned', 'connected', 'persistent')),
+            retention_parent_id TEXT REFERENCES entities(id) ON DELETE CASCADE,
+            cumulative_observation_duration_ms INTEGER NOT NULL DEFAULT 0,
+            bookmarked INTEGER NOT NULL DEFAULT 0,
+            UNIQUE (platform, platform_entity_id)
+        );
+        CREATE VIRTUAL TABLE entities_fts USING fts5(id UNINDEXED, title, content);
+        CREATE TABLE observations (
+            id TEXT PRIMARY KEY,
+            event_type TEXT NOT NULL,
+            url TEXT NOT NULL,
+            title TEXT,
+            tab_id INTEGER,
+            timestamp TEXT NOT NULL,
+            evaluated INTEGER NOT NULL DEFAULT 0,
+            meta TEXT,
+            created_at TEXT NOT NULL DEFAULT (strftime('%Y-%m-%dT%H:%M:%SZ', 'now'))
+        );
+        INSERT INTO entities (
+            id, entity_type, platform, platform_entity_id, created_at, updated_at
+        ) VALUES ('existing', 'Document', 'web', 'https://example.com', '2026-01-01T00:00:00Z', '2026-01-01T00:00:00Z');
+        PRAGMA user_version=2;
+        """
+    )
+    conn.commit()
+    conn.close()
+
+    migrated = SQLiteBackend(str(db_path))
+    await migrated.initialize()
+    try:
+        columns = {
+            row["name"]: row for row in await migrated._fetchall("PRAGMA table_info(entities)")
+        }
+        assert columns["created_at"]["dflt_value"] is not None
+        assert columns["updated_at"]["dflt_value"] is not None
+        await migrated.upsert_batch(
+            EntityBatch(
+                entities=[
+                    EntityRecord(
+                        entity_type="Message",
+                        platform="slack",
+                        platform_entity_id="T/C/123.456",
+                    )
+                ]
+            ),
+            {},
+            {},
+        )
+        inserted = await migrated.get_entity_by_platform("slack", "T/C/123.456")
+        assert inserted is not None
+        assert inserted["created_at"] is not None
+    finally:
+        await migrated.close()