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

{agentgraph_server-0.5.3 → agentgraph_server-0.5.4}/.agents/skills/AgentGraph/SKILL.md

@@ -13,6 +13,10 @@ Prefer the `agentgraph` CLI when a shell is available. Use the equivalent MCP to
 when AgentGraph is connected directly to the agent. Do not query AgentGraph's SQLite
 database or connector internals directly.
 
+Use `agentgraph demo add` or `add_demo_tool` to add the fictional Atlas fixtures to
+the configured graph. Remove them afterward with `agentgraph demo remove` or
+`remove_demo_tool`.
+
 ## Commands that use localhost
 
 `agentgraph poll` and connector or authentication commands that queue a poll or

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentgraph-server
-Version: 0.5.3
+Version: 0.5.4
 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.3 → agentgraph_server-0.5.4}/agentgraph/backends/sqlite/backend.py

@@ -208,9 +208,7 @@ 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()
-        }
+        entity_columns = {str(row["name"]): bool(row["notnull"]) for row in await cursor.fetchall()}
         columns = set(entity_columns)
         if "cumulative_dwell_ms" in columns:
             await conn.execute(
@@ -249,9 +247,7 @@ class SQLiteBackend(StorageBackend):
             "UPDATE observations SET event_type = 'observation_threshold' "
             "WHERE event_type = 'dwell_threshold'"
         )
-        await conn.execute(
-            "UPDATE entities SET entity_type = 'Email' WHERE entity_type = 'Thread'"
-        )
+        await conn.execute("UPDATE entities SET entity_type = 'Email' WHERE entity_type = 'Thread'")
         await conn.execute(
             "CREATE INDEX IF NOT EXISTS idx_entities_bookmarked ON entities(bookmarked)"
         )
@@ -352,10 +348,13 @@ class SQLiteBackend(StorageBackend):
                 "AND cumulative_observation_duration_ms > 0 THEN observed_at ELSE NULL END"
             )
         observed = (
-            "CASE WHEN platform = 'rss' AND entity_type = 'Folder' THEN NULL "
-            f"ELSE {observed} END"
+            f"CASE WHEN platform = 'rss' AND entity_type = 'Folder' THEN NULL ELSE {observed} END"
+        )
+        duration = (
+            "cumulative_observation_duration_ms"
+            if "cumulative_observation_duration_ms" in columns
+            else "0"
         )
-        duration = "cumulative_observation_duration_ms" if "cumulative_observation_duration_ms" in columns else "0"
         bookmarked = "bookmarked" if "bookmarked" in columns else "0"
         await conn.execute("PRAGMA foreign_keys=OFF")
         try:
@@ -503,6 +502,7 @@ class SQLiteBackend(StorageBackend):
         for p in persons:
             canonical_key = p.canonical_email or f"{p.platform}:{p.platform_user_id}"
             meta: dict[str, str] = {}
+            meta.update({key: str(value) for key, value in p.metadata.items() if value is not None})
             if p.canonical_email:
                 meta["canonical_email"] = p.canonical_email
             meta[f"{p.platform}_user_id"] = p.platform_user_id
@@ -534,7 +534,9 @@ class SQLiteBackend(StorageBackend):
                 existing_metadata = json.loads(existing_row[3] or "{}") if existing_row else {}
                 merged_metadata = {**existing_metadata, **meta}
                 new_title = p.display_name if p.display_name is not None else existing_row[0]
-                new_content = p.canonical_email if p.canonical_email is not None else existing_row[1]
+                new_content = (
+                    p.canonical_email if p.canonical_email is not None else existing_row[1]
+                )
                 new_embedding = emb_blob if emb_blob is not None else existing_row[2]
                 changed = (
                     new_title != existing_row[0]
@@ -647,9 +649,7 @@ class SQLiteBackend(StorageBackend):
                 if parent_id is None:
                     parent_id = await self._resolve_existing_entity_id(conn, e.platform, parent_ref)
                 if parent_id is None:
-                    raise ValueError(
-                        f"Retention parent {e.platform}:{parent_ref} is not available"
-                    )
+                    raise ValueError(f"Retention parent {e.platform}:{parent_ref} is not available")
             elif e.retention_parent_platform_entity_id is not None:
                 raise ValueError(
                     f"Entity {e.platform}:{e.platform_entity_id} has a parent but is not owned"
@@ -689,8 +689,12 @@ class SQLiteBackend(StorageBackend):
                     [e.platform, e.platform_entity_id],
                 )
                 existing_row = await existing_cursor.fetchone()
-                existing_title = str(existing_row["title"]) if existing_row and existing_row["title"] else ""
-                existing_content = str(existing_row["content"]) if existing_row and existing_row["content"] else ""
+                existing_title = (
+                    str(existing_row["title"]) if existing_row and existing_row["title"] else ""
+                )
+                existing_content = (
+                    str(existing_row["content"]) if existing_row and existing_row["content"] else ""
+                )
                 fts_title = e.title if e.title is not None else existing_title
                 fts_content = e.content if e.content is not None else existing_content
                 rewrite_fts = (
@@ -1734,7 +1738,14 @@ class SQLiteBackend(StorageBackend):
             WHERE platform = ? AND platform_entity_id = ?
               AND retention_policy = 'observed'
             """,
-            [observation_duration_ms, now, observation_duration_ms, now, platform, platform_entity_id],
+            [
+                observation_duration_ms,
+                now,
+                observation_duration_ms,
+                now,
+                platform,
+                platform_entity_id,
+            ],
         )
 
     async def record_observation_once(
@@ -1772,7 +1783,14 @@ class SQLiteBackend(StorageBackend):
                         WHERE platform = ? AND platform_entity_id = ?
                           AND retention_policy = 'observed'
                         """,
-                        [observation_duration_ms, now, observation_duration_ms, now, platform, platform_entity_id],
+                        [
+                            observation_duration_ms,
+                            now,
+                            observation_duration_ms,
+                            now,
+                            platform,
+                            platform_entity_id,
+                        ],
                     )
                 await conn.execute("COMMIT")
                 return inserted is not None
@@ -1871,8 +1889,12 @@ def _row_to_entity(row: Any) -> EntityResult:
         "synced_at": row["synced_at"] if "synced_at" in keys else None,
         "observed_at": row["observed_at"] if "observed_at" in keys else None,
         "retention_policy": row["retention_policy"] if "retention_policy" in keys else "observed",
-        "retention_parent_id": row["retention_parent_id"] if "retention_parent_id" in keys else None,
-        "cumulative_observation_duration_ms": row["cumulative_observation_duration_ms"] if "cumulative_observation_duration_ms" in keys else 0,
+        "retention_parent_id": row["retention_parent_id"]
+        if "retention_parent_id" in keys
+        else None,
+        "cumulative_observation_duration_ms": row["cumulative_observation_duration_ms"]
+        if "cumulative_observation_duration_ms" in keys
+        else 0,
         "bookmarked": bool(row["bookmarked"]) if "bookmarked" in keys else False,
         "score": row["score"] if "score" in keys else None,
     }

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

@@ -50,30 +50,44 @@ def _status_label(status: str) -> str:
     )
 
 
-@demo_app.command("seed")
-def seed_demo_command(
-    force: bool = typer.Option(False, "--force", help="Replace an existing demo database"),
+@demo_app.command("add")
+def add_demo_command(
     json: bool = typer.Option(False, "--json", help="Output as JSON"),
 ) -> None:
-    """Seed the fictional Atlas graph used by the launch demo."""
+    """Add the fictional Atlas graph to the configured database."""
     from agentgraph.config import get_config_paths
-    from agentgraph.demo import seed_demo
+    from agentgraph.demo import add_demo
 
     try:
-        result = asyncio.run(seed_demo(get_config_paths()[0], force=force))
-    except (FileExistsError, ValueError) as exc:
+        result = asyncio.run(add_demo(get_config_paths()[0]))
+    except ValueError as exc:
         typer.echo(str(exc), err=True)
         raise typer.Exit(code=1) from exc
 
     if json:
         typer.echo(_json.dumps(result, indent=2))
         return
-    typer.echo(f"Seeded Atlas demo graph at {result['database']}")
+    typer.echo(f"Added Atlas demo fixtures to {result['database']}")
     typer.echo(
         f"{result['entities']} entities, {result['persons']} people, {result['edges']} edges"
     )
 
 
+@demo_app.command("remove")
+def remove_demo_command(
+    json: bool = typer.Option(False, "--json", help="Output as JSON"),
+) -> None:
+    """Remove Atlas demo fixtures from the configured database."""
+    from agentgraph.config import get_config_paths
+    from agentgraph.demo import remove_demo
+
+    result = asyncio.run(remove_demo(get_config_paths()[0]))
+    if json:
+        typer.echo(_json.dumps(result, indent=2))
+        return
+    typer.echo(f"Removed {result['removed']} Atlas demo entities from {result['database']}")
+
+
 def _remove_auth_credentials(provider: str, account_id: str | None) -> dict[str, object]:
     from agentgraph.auth.credentials import remove_platform, remove_platform_account
 

{agentgraph_server-0.5.3 → agentgraph_server-0.5.4}/agentgraph/config.py

@@ -125,7 +125,7 @@ class Settings(BaseSettings):
     )
     # Logging
     log_level: str = Field(default="INFO")
-    log_file: Path = Field(default=Path("/tmp/agentgraph.log"))
+    log_file: Path = Field(default_factory=lambda: get_config_paths()[0] / "agentgraph.log")
 
     def __init__(self, **values: Any) -> None:
         # Resolve the config-directory .env at instantiation time as well as

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

@@ -34,13 +34,13 @@ _URL_RE = re.compile(r"https?://\S+")
 
 # Maps ResourceType values to entity_type strings stored in the DB
 RESOURCE_TYPE_TO_ENTITY_TYPE: dict[str, str] = {
-    "channel":     "Channel",
-    "dm":          "Channel",
-    "document":    "Document",
-    "folder":      "Folder",
-    "message":     "Message",
+    "channel": "Channel",
+    "dm": "Channel",
+    "document": "Document",
+    "folder": "Folder",
+    "message": "Message",
     "spreadsheet": "Spreadsheet",
-    "thread":      "Email",
+    "thread": "Email",
 }
 
 
@@ -76,10 +76,11 @@ class PersonRecord(BaseModel):
     platform_username: str | None = None
     canonical_email: str | None = None
     display_name: str | None = None
+    metadata: dict[str, str | int | float | bool | None] = {}
 
 
 class EntityRecord(BaseModel):
-    entity_type: str          # 'Message' | 'Document' | 'Channel' | 'Task'
+    entity_type: str  # 'Message' | 'Document' | 'Channel' | 'Task'
     platform: str
     platform_entity_id: str
     title: str | None = None
@@ -87,13 +88,13 @@ class EntityRecord(BaseModel):
     source_created_at: datetime | None = None
     source_updated_at: datetime | None = None
     metadata: dict[str, str | int | float | bool | None] = {}
-    is_stub: bool = False     # True → placeholder pending a full fetch; preserves synced_at=NULL
+    is_stub: bool = False  # True → placeholder pending a full fetch; preserves synced_at=NULL
     retention_policy: RetentionPolicy = "observed"
     retention_parent_platform_entity_id: str | None = None
 
 
 class EdgeRecord(BaseModel):
-    edge_type: str            # 'authored' | 'posted_in' | 'replied_to' | 'mentions'
+    edge_type: str  # 'authored' | 'posted_in' | 'replied_to' | 'mentions'
     source_platform_entity_id: str | None = None
     source_platform_user_id: str | None = None
     target_platform_entity_id: str | None = None
@@ -132,18 +133,22 @@ class EntityBatch(BaseModel):
             if ref.source == entity.platform and ref.resource_id == entity.platform_entity_id:
                 continue
             seen.add(key)
-            self.entities.append(EntityRecord(
-                entity_type=RESOURCE_TYPE_TO_ENTITY_TYPE[ref.resource_type],
-                platform=ref.source,
-                platform_entity_id=ref.resource_id,
-                is_stub=True,
-            ))
-            self.edges.append(EdgeRecord(
-                edge_type="references",
-                source_platform_entity_id=entity.platform_entity_id,
-                target_platform_entity_id=ref.resource_id,
-                platform="cross",
-            ))
+            self.entities.append(
+                EntityRecord(
+                    entity_type=RESOURCE_TYPE_TO_ENTITY_TYPE[ref.resource_type],
+                    platform=ref.source,
+                    platform_entity_id=ref.resource_id,
+                    is_stub=True,
+                )
+            )
+            self.edges.append(
+                EdgeRecord(
+                    edge_type="references",
+                    source_platform_entity_id=entity.platform_entity_id,
+                    target_platform_entity_id=ref.resource_id,
+                    platform="cross",
+                )
+            )
 
 
 async def get_known_channel_syncs(
@@ -195,7 +200,6 @@ class ConnectorAccount(BaseModel):
     metadata: dict[str, str] = Field(default_factory=dict)
 
 
-
 class FetchPolicy:
     """Encapsulates refresh policy decisions for a resource."""
 
@@ -222,7 +226,7 @@ class FetchPolicy:
 
 
 class BaseConnector(ABC):
-    source: ClassVar[str]           # platform name, e.g. "slack" — must be set by subclass
+    source: ClassVar[str]  # platform name, e.g. "slack" — must be set by subclass
     fetch_policy: ClassVar[FetchPolicy]  # staleness policy — must be set by subclass
     is_generic_url_fallback: ClassVar[bool] = False
     """True for broad fallback connectors that should not claim URLs during discovery."""
@@ -280,14 +284,16 @@ class BaseConnector(ABC):
         user = cls.get_authenticated_user()
         if user is None:
             return []
-        return [ConnectorAccount(
-            account_id=cls.source,
-            label=user,
-            auth_group=cls.auth_label or cls.source,
-            source=cls.source,
-            user_id=user,
-            email=user if "@" in user else None,
-        )]
+        return [
+            ConnectorAccount(
+                account_id=cls.source,
+                label=user,
+                auth_group=cls.auth_label or cls.source,
+                source=cls.source,
+                user_id=user,
+                email=user if "@" in user else None,
+            )
+        ]
 
     @classmethod
     async def verify_auth(cls, account_id: str | None = None) -> tuple[str, str | None]:
@@ -339,9 +345,7 @@ class BaseConnector(ABC):
         _ = (args, result)
         return ConnectorCommandEffects()
 
-    def normalise_fetch_id(
-        self, resource_id: str, entity_type: str
-    ) -> tuple[str, ResourceType]:
+    def normalise_fetch_id(self, resource_id: str, entity_type: str) -> tuple[str, ResourceType]:
         """Map a stored resource_id + entity_type to the (id, resource_type) that fetch() expects.
 
         The default maps entity_type to ResourceType using the standard table and

{agentgraph_server-0.5.3 → agentgraph_server-0.5.4}/agentgraph/demo.py

@@ -11,6 +11,8 @@ from agentgraph.backends.sqlite.backend import SQLiteBackend
 from agentgraph.connectors.base import EdgeRecord, EntityBatch, EntityRecord, PersonRecord
 
 DEMO_DATABASE_NAME = "agentgraph.db"
+DEMO_FIXTURE_METADATA_KEY = "_agentgraph_fixture"
+DEMO_FIXTURE_METADATA_VALUE = "atlas-demo"
 DEMO_CREATED_AT = "2026-08-14T08:00:00Z"
 WEBHOOK_ARTICLE_URL = (
     "https://github.com/simonexmachina/agent-graph/blob/main/"
@@ -28,10 +30,7 @@ def utc(year: int, month: int, day: int, hour: int, minute: int = 0) -> datetime
 
 def load_research_fixture(filename: str) -> str:
     return (
-        files("agentgraph")
-        .joinpath("demo_fixtures", filename)
-        .read_text(encoding="utf-8")
-        .strip()
+        files("agentgraph").joinpath("demo_fixtures", filename).read_text(encoding="utf-8").strip()
     )
 
 
@@ -44,6 +43,7 @@ def build_demo_batch() -> EntityBatch:
     drive_folder_id = "atlas-program-folder"
     drive_plan_id = "atlas-integration-plan"
 
+    fixture_metadata = {DEMO_FIXTURE_METADATA_KEY: DEMO_FIXTURE_METADATA_VALUE}
     entities = [
         EntityRecord(
             entity_type="Channel",
@@ -53,6 +53,7 @@ def build_demo_batch() -> EntityBatch:
             source_created_at=utc(2026, 7, 1, 9),
             source_updated_at=utc(2026, 8, 10, 14),
             metadata={
+                **fixture_metadata,
                 "web_url": "https://app.slack.com/client/TDEMO/CATLAS",
                 "workspace_name": "AgentGraph Demo",
             },
@@ -69,7 +70,10 @@ def build_demo_batch() -> EntityBatch:
             ),
             source_created_at=utc(2026, 8, 10, 10),
             source_updated_at=utc(2026, 8, 10, 10),
-            metadata={"web_url": "https://app.slack.com/client/TDEMO/CATLAS/thread-1"},
+            metadata={
+                **fixture_metadata,
+                "web_url": "https://app.slack.com/client/TDEMO/CATLAS/thread-1",
+            },
             retention_policy="owned",
             retention_parent_platform_entity_id=channel_id,
         ),
@@ -83,7 +87,10 @@ def build_demo_batch() -> EntityBatch:
             ),
             source_created_at=utc(2026, 8, 10, 10, 30),
             source_updated_at=utc(2026, 8, 10, 10, 30),
-            metadata={"web_url": "https://app.slack.com/client/TDEMO/CATLAS/thread-1-reply-1"},
+            metadata={
+                **fixture_metadata,
+                "web_url": "https://app.slack.com/client/TDEMO/CATLAS/thread-1-reply-1",
+            },
             retention_policy="owned",
             retention_parent_platform_entity_id=channel_id,
         ),
@@ -97,7 +104,10 @@ def build_demo_batch() -> EntityBatch:
             ),
             source_created_at=utc(2026, 8, 10, 11),
             source_updated_at=utc(2026, 8, 10, 11),
-            metadata={"web_url": "https://app.slack.com/client/TDEMO/CATLAS/thread-1-reply-2"},
+            metadata={
+                **fixture_metadata,
+                "web_url": "https://app.slack.com/client/TDEMO/CATLAS/thread-1-reply-2",
+            },
             retention_policy="owned",
             retention_parent_platform_entity_id=channel_id,
         ),
@@ -116,6 +126,7 @@ def build_demo_batch() -> EntityBatch:
             source_created_at=utc(2026, 8, 9, 15),
             source_updated_at=utc(2026, 8, 9, 15),
             metadata={
+                **fixture_metadata,
                 "web_url": "https://mail.google.com/mail/u/0/#all/atlas-maya-thread",
                 "snippet": "For the September 30 Atlas launch...",
             },
@@ -127,7 +138,10 @@ def build_demo_batch() -> EntityBatch:
             title="Atlas program",
             source_created_at=utc(2026, 7, 1, 8),
             source_updated_at=utc(2026, 8, 6, 9),
-            metadata={"web_url": "https://drive.google.com/drive/folders/atlas-program-folder"},
+            metadata={
+                **fixture_metadata,
+                "web_url": "https://drive.google.com/drive/folders/atlas-program-folder",
+            },
         ),
         EntityRecord(
             entity_type="Document",
@@ -143,6 +157,7 @@ def build_demo_batch() -> EntityBatch:
             source_created_at=utc(2026, 8, 2, 9),
             source_updated_at=utc(2026, 8, 6, 16),
             metadata={
+                **fixture_metadata,
                 "web_url": "https://drive.google.com/file/d/atlas-integration-plan/view",
                 "mime_type": "text/markdown",
             },
@@ -155,7 +170,7 @@ def build_demo_batch() -> EntityBatch:
             content=load_research_fixture("reliable-webhooks.md"),
             source_created_at=utc(2026, 8, 4, 9),
             source_updated_at=utc(2026, 8, 4, 9),
-            metadata={"web_url": WEBHOOK_ARTICLE_URL, "author": "Nadia Okafor"},
+            metadata={**fixture_metadata, "web_url": WEBHOOK_ARTICLE_URL, "author": "Nadia Okafor"},
         ),
         EntityRecord(
             entity_type="Document",
@@ -165,7 +180,7 @@ def build_demo_batch() -> EntityBatch:
             content=load_research_fixture("retry-guidance.md"),
             source_created_at=utc(2026, 8, 8, 9),
             source_updated_at=utc(2026, 8, 8, 9),
-            metadata={"web_url": RETRY_GUIDE_URL, "revision": "3"},
+            metadata={**fixture_metadata, "web_url": RETRY_GUIDE_URL, "revision": "3"},
         ),
     ]
 
@@ -176,12 +191,14 @@ def build_demo_batch() -> EntityBatch:
             platform_username="alex",
             canonical_email="alex@agentgraph.demo",
             display_name="Alex Chen",
+            metadata=fixture_metadata,
         ),
         PersonRecord(
             platform="gdrive",
             platform_user_id="alex@agentgraph.demo",
             canonical_email="alex@agentgraph.demo",
             display_name="Alex Chen",
+            metadata=fixture_metadata,
         ),
         PersonRecord(
             platform="slack",
@@ -189,12 +206,14 @@ def build_demo_batch() -> EntityBatch:
             platform_username="priya",
             canonical_email="priya@agentgraph.demo",
             display_name="Priya Raman",
+            metadata=fixture_metadata,
         ),
         PersonRecord(
             platform="gmail",
             platform_user_id="maya@customer.example",
             canonical_email="maya@customer.example",
             display_name="Maya Patel",
+            metadata=fixture_metadata,
         ),
     ]
 
@@ -284,21 +303,6 @@ def build_demo_batch() -> EntityBatch:
     return EntityBatch(entities=entities, persons=persons, edges=edges)
 
 
-def validate_demo_config_dir(config_dir: Path) -> Path:
-    resolved = config_dir.expanduser().resolve()
-    default_dir = (Path.home() / ".agentgraph").resolve()
-    if resolved == default_dir:
-        raise ValueError("Refusing to seed the default ~/.agentgraph directory")
-    return resolved
-
-
-def remove_demo_database(database_path: Path) -> None:
-    for suffix in ("", "-wal", "-shm"):
-        candidate = Path(f"{database_path}{suffix}")
-        if candidate.exists():
-            candidate.unlink()
-
-
 def apply_demo_timestamps(database_path: Path) -> None:
     observations = {
         ("slack", "TDEMO/CATLAS"): "2026-08-10T11:05:00Z",
@@ -308,29 +312,41 @@ def apply_demo_timestamps(database_path: Path) -> None:
     }
     with sqlite3.connect(database_path) as conn:
         conn.execute(
-            "UPDATE entities SET created_at = ?, updated_at = ?, synced_at = ?",
-            [DEMO_CREATED_AT, DEMO_CREATED_AT, DEMO_CREATED_AT],
+            """
+            UPDATE entities
+            SET created_at = ?, updated_at = ?, synced_at = ?
+            WHERE json_extract(metadata, ?) = ?
+            """,
+            [
+                DEMO_CREATED_AT,
+                DEMO_CREATED_AT,
+                DEMO_CREATED_AT,
+                f"$.{DEMO_FIXTURE_METADATA_KEY}",
+                DEMO_FIXTURE_METADATA_VALUE,
+            ],
         )
         for (platform, platform_entity_id), observed_at in observations.items():
             conn.execute(
                 """
                 UPDATE entities
                 SET observed_at = ?, cumulative_observation_duration_ms = 3000
-                WHERE platform = ? AND platform_entity_id = ?
+                WHERE platform = ?
+                  AND platform_entity_id = ?
+                  AND json_extract(metadata, ?) = ?
                 """,
-                [observed_at, platform, platform_entity_id],
+                [
+                    observed_at,
+                    platform,
+                    platform_entity_id,
+                    f"$.{DEMO_FIXTURE_METADATA_KEY}",
+                    DEMO_FIXTURE_METADATA_VALUE,
+                ],
             )
 
 
-async def seed_demo(config_dir: Path, *, force: bool = False) -> dict[str, object]:
-    safe_dir = validate_demo_config_dir(config_dir)
-    database_path = safe_dir / DEMO_DATABASE_NAME
-    if database_path.exists() and not force:
-        raise FileExistsError(f"Demo database already exists: {database_path}. Use --force to replace it.")
-    safe_dir.mkdir(parents=True, exist_ok=True)
-    if force:
-        remove_demo_database(database_path)
-
+async def add_demo(config_dir: Path) -> dict[str, object]:
+    database_path = config_dir / DEMO_DATABASE_NAME
+    config_dir.mkdir(parents=True, exist_ok=True)
     batch = build_demo_batch()
     backend = SQLiteBackend(str(database_path), vector_mode="bm25-only")
     await backend.initialize()
@@ -339,13 +355,11 @@ async def seed_demo(config_dir: Path, *, force: bool = False) -> dict[str, objec
     finally:
         await backend.close()
     apply_demo_timestamps(database_path)
-
-    with sqlite3.connect(database_path) as conn:
-        entity_count = int(conn.execute("SELECT count(*) FROM entities").fetchone()[0])
-        person_count = int(
-            conn.execute("SELECT count(*) FROM entities WHERE entity_type = 'Person'").fetchone()[0]
-        )
-        edge_count = int(conn.execute("SELECT count(*) FROM edges").fetchone()[0])
+    entity_count = len(batch.entities)
+    person_count = len(
+        {p.canonical_email or f"{p.platform}:{p.platform_user_id}" for p in batch.persons}
+    )
+    edge_count = len(batch.edges)
     return {
         "database": str(database_path),
         "entities": entity_count,
@@ -354,3 +368,29 @@ async def seed_demo(config_dir: Path, *, force: bool = False) -> dict[str, objec
         "webhook_article": WEBHOOK_ARTICLE_URL,
         "retry_guide": RETRY_GUIDE_URL,
     }
+
+
+async def remove_demo(config_dir: Path) -> dict[str, object]:
+    database_path = config_dir / DEMO_DATABASE_NAME
+    if not database_path.exists():
+        return {"database": str(database_path), "removed": 0}
+
+    with sqlite3.connect(database_path) as conn:
+        rows = conn.execute(
+            """
+            SELECT id FROM entities
+            WHERE json_extract(metadata, ?) = ?
+            """,
+            [f"$.{DEMO_FIXTURE_METADATA_KEY}", DEMO_FIXTURE_METADATA_VALUE],
+        ).fetchall()
+    entity_ids = [str(row[0]) for row in rows]
+
+    backend = SQLiteBackend(str(database_path), vector_mode="bm25-only")
+    await backend.initialize()
+    try:
+        for entity_id in entity_ids:
+            if await backend.get_entity_by_id(entity_id) is not None:
+                await backend.delete_entity(entity_id)
+    finally:
+        await backend.close()
+    return {"database": str(database_path), "removed": len(entity_ids)}

{agentgraph_server-0.5.3 → agentgraph_server-0.5.4}/agentgraph/mcp/server.py

@@ -326,6 +326,40 @@ async def run_connector_command_tool(source: str, args: list[str]) -> str:
         return json.dumps({"error": str(exc)})
 
 
+@mcp.tool(
+    annotations=_tool_annotations(
+        "Add AgentGraph demo fixtures",
+        read_only=False,
+        destructive=False,
+        idempotent=True,
+        open_world=False,
+    )
+)
+async def add_demo_tool() -> str:
+    """Add the fictional Atlas demo fixtures to the configured database."""
+    from agentgraph.config import get_config_paths
+    from agentgraph.demo import add_demo
+
+    return json.dumps(await add_demo(get_config_paths()[0]))
+
+
+@mcp.tool(
+    annotations=_tool_annotations(
+        "Remove AgentGraph demo fixtures",
+        read_only=False,
+        destructive=True,
+        idempotent=True,
+        open_world=False,
+    )
+)
+async def remove_demo_tool() -> str:
+    """Remove marked Atlas demo fixtures from the configured database."""
+    from agentgraph.config import get_config_paths
+    from agentgraph.demo import remove_demo
+
+    return json.dumps(await remove_demo(get_config_paths()[0]))
+
+
 async def _enrich_results(results: list[dict[str, Any]]) -> None:
     """Let each owning connector apply result presentation fixes in-place."""
     from agentgraph.connectors.registry import bootstrap, get_connector

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentgraph-server
-Version: 0.5.3
+Version: 0.5.4
 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