create-context-graph 0.1.0__py3-none-any.whl

create_context_graph/__init__.py

@@ -0,0 +1,3 @@
+"""Create Context Graph - Interactive CLI scaffolding tool for domain-specific context graph applications."""
+
+__version__ = "0.1.0"

create_context_graph/__main__.py

@@ -0,0 +1,5 @@
+"""Allow running as `python -m create_context_graph`."""
+
+from create_context_graph.cli import main
+
+main()

create_context_graph/cli.py

@@ -0,0 +1,157 @@
+"""CLI entry point for create-context-graph."""
+
+from __future__ import annotations
+
+from pathlib import Path
+
+import click
+from rich.console import Console
+from rich.progress import Progress, SpinnerColumn, TextColumn
+
+from create_context_graph.config import SUPPORTED_FRAMEWORKS, ProjectConfig
+from create_context_graph.ontology import list_available_domains, load_domain
+from create_context_graph.renderer import ProjectRenderer
+
+console = Console()
+
+
+@click.command()
+@click.argument("project_name", required=False)
+@click.option(
+    "--domain",
+    type=str,
+    help="Domain ID (e.g., financial-services, healthcare, software-engineering)",
+)
+@click.option(
+    "--framework",
+    type=click.Choice(SUPPORTED_FRAMEWORKS, case_sensitive=False),
+    help="Agent framework to use",
+)
+@click.option("--demo-data", is_flag=True, help="Generate synthetic demo data")
+@click.option("--ingest", is_flag=True, help="Ingest generated data into Neo4j")
+@click.option("--neo4j-uri", envvar="NEO4J_URI", help="Neo4j connection URI")
+@click.option("--neo4j-username", envvar="NEO4J_USERNAME", default="neo4j")
+@click.option("--neo4j-password", envvar="NEO4J_PASSWORD", default="password")
+@click.option("--anthropic-api-key", envvar="ANTHROPIC_API_KEY", help="Anthropic API key for LLM generation")
+@click.option("--output-dir", type=click.Path(), help="Output directory (default: ./<project-name>)")
+@click.option("--list-domains", is_flag=True, help="List available domains and exit")
+@click.version_option(package_name="create-context-graph")
+def main(
+    project_name: str | None,
+    domain: str | None,
+    framework: str | None,
+    demo_data: bool,
+    ingest: bool,
+    neo4j_uri: str | None,
+    neo4j_username: str,
+    neo4j_password: str,
+    anthropic_api_key: str | None,
+    output_dir: str | None,
+    list_domains: bool,
+) -> None:
+    """Create a domain-specific context graph application.
+
+    Generates a full-stack application with a FastAPI backend,
+    Next.js frontend, Neo4j knowledge graph, and AI agent—
+    all customized for your industry domain.
+    """
+    # List domains mode
+    if list_domains:
+        domains = list_available_domains()
+        console.print("\n[bold]Available domains:[/bold]\n")
+        for d in domains:
+            console.print(f"  {d['id']:30s} {d['name']}")
+        console.print()
+        return
+
+    # If all required args are provided, skip wizard
+    if project_name and domain and framework:
+        config = ProjectConfig(
+            project_name=project_name,
+            domain=domain,
+            framework=framework,
+            data_source="demo" if demo_data else "demo",
+            neo4j_uri=neo4j_uri or "neo4j://localhost:7687",
+            neo4j_username=neo4j_username,
+            neo4j_password=neo4j_password,
+            neo4j_type="docker",
+            anthropic_api_key=anthropic_api_key,
+            generate_data=demo_data,
+        )
+    else:
+        # Launch interactive wizard
+        from create_context_graph.wizard import run_wizard
+
+        config = run_wizard()
+
+    # Resolve output directory
+    out = Path(output_dir) if output_dir else Path.cwd() / config.project_slug
+    if out.exists() and any(out.iterdir()):
+        console.print(f"[red]Error:[/red] Directory {out} already exists and is not empty.")
+        raise SystemExit(1)
+
+    # Load domain ontology
+    try:
+        ontology = load_domain(config.domain)
+    except FileNotFoundError:
+        console.print(f"[red]Error:[/red] Domain '{config.domain}' not found.")
+        available = list_available_domains()
+        console.print("Available domains: " + ", ".join(d["id"] for d in available))
+        raise SystemExit(1)
+
+    # Generate project
+    with Progress(
+        SpinnerColumn(),
+        TextColumn("[progress.description]{task.description}"),
+        console=console,
+    ) as progress:
+        task = progress.add_task("Creating project scaffold...", total=None)
+
+        renderer = ProjectRenderer(config, ontology)
+        renderer.render(out)
+
+        progress.update(task, description="Project generated!")
+
+    # Generate demo data if requested
+    fixture_path = out / "data" / "fixtures.json"
+    if config.generate_data or demo_data:
+        console.print("\n[bold]Generating demo data...[/bold]")
+        from create_context_graph.generator import generate_fixture_data
+
+        generate_fixture_data(
+            ontology,
+            fixture_path,
+            api_key=config.anthropic_api_key or anthropic_api_key,
+        )
+
+    # Ingest into Neo4j if requested
+    if ingest and fixture_path.exists():
+        console.print("\n[bold]Ingesting data into Neo4j...[/bold]")
+        from create_context_graph.ingest import ingest_data
+
+        ingest_data(
+            fixture_path,
+            ontology,
+            config.neo4j_uri,
+            config.neo4j_username,
+            config.neo4j_password,
+        )
+
+    # Success message
+    console.print()
+    console.print(f"[bold green]Done![/bold green] Your {ontology.domain.name} context graph app is ready.")
+    console.print()
+    try:
+        display_path = out.relative_to(Path.cwd())
+    except ValueError:
+        display_path = out
+    console.print(f"  [bold]cd {display_path}[/bold]")
+    console.print(f"  [bold]make install[/bold]       # Install dependencies")
+    if config.neo4j_type == "docker":
+        console.print(f"  [bold]make docker-up[/bold]    # Start Neo4j")
+    console.print(f"  [bold]make seed[/bold]          # Seed sample data")
+    console.print(f"  [bold]make start[/bold]         # Start backend + frontend")
+    console.print()
+    console.print(f"  Backend:  http://localhost:8000")
+    console.print(f"  Frontend: http://localhost:3000")
+    console.print()

create_context_graph/config.py

@@ -0,0 +1,76 @@
+"""Project configuration model."""
+
+from __future__ import annotations
+
+import re
+from typing import Literal
+
+from pydantic import BaseModel, Field, computed_field
+
+
+SUPPORTED_FRAMEWORKS = [
+    "pydanticai",
+    "claude-agent-sdk",
+    "strands",
+    "google-adk",
+    "openai-agents",
+    "langgraph",
+    "crewai",
+    "maf",
+]
+
+FRAMEWORK_DISPLAY_NAMES = {
+    "pydanticai": "PydanticAI",
+    "claude-agent-sdk": "Claude Agent SDK",
+    "strands": "Strands (AWS)",
+    "google-adk": "Google ADK",
+    "openai-agents": "OpenAI Agents SDK",
+    "langgraph": "LangGraph",
+    "crewai": "CrewAI",
+    "maf": "MAF (Modular Agent Framework)",
+}
+
+FRAMEWORK_DEPENDENCIES = {
+    "pydanticai": ["pydantic-ai>=0.1"],
+    "claude-agent-sdk": ["claude-agent-sdk>=0.1"],
+    "strands": ["strands-agents>=0.1"],
+    "google-adk": ["google-adk>=0.1"],
+    "openai-agents": ["openai-agents>=0.1"],
+    "langgraph": ["langgraph>=0.1"],
+    "crewai": ["crewai>=0.1"],
+    "maf": ["maf>=0.1"],
+}
+
+
+class ProjectConfig(BaseModel):
+    """All configuration collected from the wizard or CLI flags."""
+
+    project_name: str = Field(description="Human-readable project name")
+    domain: str = Field(description="Domain ID from ontology YAML")
+    framework: str = Field(description="Agent framework key")
+    data_source: Literal["demo", "saas"] = Field(default="demo")
+    neo4j_uri: str = Field(default="neo4j://localhost:7687")
+    neo4j_username: str = Field(default="neo4j")
+    neo4j_password: str = Field(default="password")
+    neo4j_type: Literal["aurads", "aura-free", "docker", "existing"] = Field(
+        default="docker"
+    )
+    anthropic_api_key: str | None = Field(default=None)
+    openai_api_key: str | None = Field(default=None)
+    generate_data: bool = Field(default=False)
+
+    @computed_field
+    @property
+    def project_slug(self) -> str:
+        """Kebab-case slug derived from project name."""
+        slug = self.project_name.lower().strip()
+        slug = re.sub(r"[^a-z0-9]+", "-", slug)
+        return slug.strip("-")
+
+    @property
+    def framework_display_name(self) -> str:
+        return FRAMEWORK_DISPLAY_NAMES.get(self.framework, self.framework)
+
+    @property
+    def framework_deps(self) -> list[str]:
+        return FRAMEWORK_DEPENDENCIES.get(self.framework, [])

create_context_graph/domains/_base.yaml

@@ -0,0 +1,82 @@
+# Base POLE+O ontology - shared foundation for all domains
+# Person, Object, Location, Event + Organization
+version: "1.0"
+
+base_entity_types:
+  Person:
+    pole_type: PERSON
+    color: "#22c55e"
+    icon: user
+    base_properties:
+      - name: name
+        type: string
+        required: true
+      - name: email
+        type: string
+      - name: role
+        type: string
+      - name: description
+        type: string
+
+  Organization:
+    pole_type: ORGANIZATION
+    color: "#3b82f6"
+    icon: building
+    base_properties:
+      - name: name
+        type: string
+        required: true
+      - name: description
+        type: string
+      - name: industry
+        type: string
+
+  Location:
+    pole_type: LOCATION
+    color: "#a855f7"
+    icon: map-pin
+    base_properties:
+      - name: name
+        type: string
+        required: true
+      - name: address
+        type: string
+      - name: latitude
+        type: float
+      - name: longitude
+        type: float
+
+  Event:
+    pole_type: EVENT
+    color: "#f97316"
+    icon: calendar
+    base_properties:
+      - name: name
+        type: string
+        required: true
+      - name: date
+        type: datetime
+      - name: description
+        type: string
+
+  Object:
+    pole_type: OBJECT
+    color: "#eab308"
+    icon: box
+    base_properties:
+      - name: name
+        type: string
+        required: true
+      - name: description
+        type: string
+
+base_relationships:
+  WORKS_FOR:
+    source: Person
+    target: Organization
+  LOCATED_AT:
+    source: Organization
+    target: Location
+  PARTICIPATED_IN:
+    source: Person
+    target: Event