Research Knowledge Graph

CREATE KEYSPACE IF NOT EXISTS knowledge WITH replication = {
    'class': 'SimpleStrategy', 'replication_factor': 3
};

USE knowledge;

-- ============================================================
-- Vertex tables (entities)
-- ============================================================

-- Researcher: an academic with publications, metrics, and social links
CREATE TABLE IF NOT EXISTS researcher (
    researcher_id uuid PRIMARY KEY,
    name text,
    email text,
    title text,
    h_index int,
    citations_total int,
    orcid text,
    joined_at timestamp,
    active boolean,
    specializations set<text>,
    social_links map<text, text>
) WITH extensions = {
    'graph.type': 'vertex',
    'graph.label': 'Researcher'
};

-- Institution: a university or research lab
CREATE TABLE IF NOT EXISTS institution (
    institution_id uuid PRIMARY KEY,
    name text,
    country text,
    city text,
    institution_type text,
    ranking int,
    founded_year int,
    departments set<text>,
    metadata map<text, text>
) WITH extensions = {
    'graph.type': 'vertex',
    'graph.label': 'Institution'
};

-- Paper: a published research paper
CREATE TABLE IF NOT EXISTS paper (
    paper_id uuid PRIMARY KEY,
    title text,
    abstract_text text,
    doi text,
    venue text,
    year int,
    month int,
    citation_count int,
    paper_type text,
    keywords set<text>,
    urls map<text, text>
) WITH extensions = {
    'graph.type': 'vertex',
    'graph.label': 'Paper'
};

-- Topic: a research area or subfield
CREATE TABLE IF NOT EXISTS topic (
    topic_id uuid PRIMARY KEY,
    name text,
    parent_topic text,
    description text,
    paper_count int
) WITH extensions = {
    'graph.type': 'vertex',
    'graph.label': 'Topic'
};

-- Grant: a funding award from a government or private funder
CREATE TABLE IF NOT EXISTS grant_award (
    grant_id uuid PRIMARY KEY,
    title text,
    funder text,
    amount decimal,
    currency text,
    start_date date,
    end_date date,
    status text
) WITH extensions = {
    'graph.type': 'vertex',
    'graph.label': 'Grant'
};

-- ============================================================
-- Edge tables (relationships)
-- ============================================================

-- Researcher AUTHORED Paper
CREATE TABLE IF NOT EXISTS authored (
    source_id uuid,
    target_id uuid,
    author_position int,
    corresponding boolean,
    PRIMARY KEY (source_id, target_id)
) WITH extensions = {
    'graph.type': 'edge',
    'graph.label': 'AUTHORED',
    'graph.source': 'source_id',
    'graph.target': 'target_id',
    'graph.source_label': 'Researcher',
    'graph.target_label': 'Paper'
};

-- Paper CITES Paper
CREATE TABLE IF NOT EXISTS cites (
    source_id uuid,
    target_id uuid,
    context text,
    section text,
    PRIMARY KEY (source_id, target_id)
) WITH extensions = {
    'graph.type': 'edge',
    'graph.label': 'CITES',
    'graph.source': 'source_id',
    'graph.target': 'target_id',
    'graph.source_label': 'Paper',
    'graph.target_label': 'Paper'
};

-- Researcher AFFILIATED_WITH Institution
CREATE TABLE IF NOT EXISTS affiliated_with (
    source_id uuid,
    target_id uuid,
    department text,
    role text,
    start_year int,
    end_year int,
    PRIMARY KEY (source_id, target_id)
) WITH extensions = {
    'graph.type': 'edge',
    'graph.label': 'AFFILIATED_WITH',
    'graph.source': 'source_id',
    'graph.target': 'target_id',
    'graph.source_label': 'Researcher',
    'graph.target_label': 'Institution'
};

-- Paper COVERS Topic
CREATE TABLE IF NOT EXISTS covers_topic (
    source_id uuid,
    target_id uuid,
    relevance float,
    PRIMARY KEY (source_id, target_id)
) WITH extensions = {
    'graph.type': 'edge',
    'graph.label': 'COVERS',
    'graph.source': 'source_id',
    'graph.target': 'target_id',
    'graph.source_label': 'Paper',
    'graph.target_label': 'Topic'
};

-- Paper FUNDED_BY Grant
CREATE TABLE IF NOT EXISTS funded_by (
    source_id uuid,
    target_id uuid,
    role text,
    PRIMARY KEY (source_id, target_id)
) WITH extensions = {
    'graph.type': 'edge',
    'graph.label': 'FUNDED_BY',
    'graph.source': 'source_id',
    'graph.target': 'target_id',
    'graph.source_label': 'Paper',
    'graph.target_label': 'Grant'
};

-- Researcher COLLABORATES with Researcher
CREATE TABLE IF NOT EXISTS collaborates_with (
    source_id uuid,
    target_id uuid,
    paper_count int,
    first_collab_year int,
    last_collab_year int,
    PRIMARY KEY (source_id, target_id)
) WITH extensions = {
    'graph.type': 'edge',
    'graph.label': 'COLLABORATES',
    'graph.source': 'source_id',
    'graph.target': 'target_id',
    'graph.source_label': 'Researcher',
    'graph.target_label': 'Researcher'
};

-- ============================================================
-- Secondary indexes for CQL queries
-- ============================================================

CREATE INDEX IF NOT EXISTS ON researcher (name);
CREATE INDEX IF NOT EXISTS ON researcher (active);
CREATE INDEX IF NOT EXISTS ON paper (venue);
CREATE INDEX IF NOT EXISTS ON paper (year);
CREATE INDEX IF NOT EXISTS ON paper (doi);
CREATE INDEX IF NOT EXISTS ON institution (country);
CREATE INDEX IF NOT EXISTS ON institution (name);
CREATE INDEX IF NOT EXISTS ON topic (name);
CREATE INDEX IF NOT EXISTS ON grant_award (funder);
CREATE INDEX IF NOT EXISTS ON grant_award (status);

SELECT name, h_index, specializations FROM researcher
WHERE researcher_id = 11111111-0000-0000-0000-000000000001;

SELECT title, year, citation_count FROM paper WHERE venue = 'NeurIPS';
SELECT name, city, ranking FROM institution WHERE country = 'United States';

SELECT target_id, author_position, corresponding FROM authored
WHERE source_id = 11111111-0000-0000-0000-000000000001;

UPDATE researcher SET specializations = specializations + {'Reinforcement Learning'}
WHERE researcher_id = 11111111-0000-0000-0000-000000000001;

UPDATE researcher SET social_links['linkedin'] = 'https://linkedin.com/in/alicechen'
WHERE researcher_id = 11111111-0000-0000-0000-000000000001;

INSERT INTO researcher (researcher_id, name, email, active)
VALUES (11111111-0000-0000-0000-000000000099, 'New Researcher', 'new@test.com', true)
IF NOT EXISTS;

USE knowledge;

-- ============================================================
-- Basic lookups by primary key
-- ============================================================

-- Full row for a single researcher
SELECT * FROM researcher WHERE researcher_id = 11111111-0000-0000-0000-000000000001;

-- Project specific columns
SELECT name, h_index, specializations FROM researcher
WHERE researcher_id = 11111111-0000-0000-0000-000000000001;

-- Look up a paper by ID
SELECT title, venue, year, citation_count FROM paper
WHERE paper_id = 33333333-0000-0000-0000-000000000001;

-- Look up a grant
SELECT title, funder, amount, currency FROM grant_award
WHERE grant_id = 55555555-0000-0000-0000-000000000001;

-- ============================================================
-- Secondary index queries
-- ============================================================

-- Papers at a specific venue
SELECT title, year, citation_count FROM paper WHERE venue = 'NeurIPS';

-- Papers from a specific year
SELECT title, venue, citation_count FROM paper WHERE year = 2024;

-- Institutions in a country
SELECT name, city, ranking FROM institution WHERE country = 'United States';

-- Find a topic by name
SELECT * FROM topic WHERE name = 'Machine Learning';

-- Active grants from a specific funder
SELECT title, amount, currency, start_date, end_date FROM grant_award
WHERE funder = 'NSF';

-- Active grants by status
SELECT title, funder, amount FROM grant_award WHERE status = 'active';

-- Papers by DOI
SELECT title, venue, year FROM paper WHERE doi = '10.1234/neurips.2024.001';

-- ============================================================
-- Collection operations
-- ============================================================

-- Read a set column
SELECT specializations FROM researcher
WHERE researcher_id = 11111111-0000-0000-0000-000000000001;

-- Read a map column
SELECT social_links FROM researcher
WHERE researcher_id = 11111111-0000-0000-0000-000000000001;

-- Read keywords set from a paper
SELECT keywords FROM paper
WHERE paper_id = 33333333-0000-0000-0000-000000000001;

-- Read URLs map from a paper
SELECT urls FROM paper
WHERE paper_id = 33333333-0000-0000-0000-000000000001;

-- Read departments from institution
SELECT departments FROM institution
WHERE institution_id = 22222222-0000-0000-0000-000000000001;

-- ============================================================
-- Count and aggregation
-- ============================================================

SELECT COUNT(*) FROM paper;
SELECT COUNT(*) FROM researcher;
SELECT COUNT(*) FROM institution;
SELECT COUNT(*) FROM authored;
SELECT COUNT(*) FROM cites;

-- ============================================================
-- LIMIT
-- ============================================================

SELECT title, venue, year FROM paper LIMIT 5;
SELECT name, h_index FROM researcher LIMIT 3;

-- ============================================================
-- Edge table queries (clustering key traversals)
-- ============================================================

-- All papers authored by Alice
SELECT target_id, author_position, corresponding FROM authored
WHERE source_id = 11111111-0000-0000-0000-000000000001;

-- All citations from Paper 1 (Sparse Transformers)
SELECT target_id, context, section FROM cites
WHERE source_id = 33333333-0000-0000-0000-000000000001;

-- All citations from Paper 3 (Medical ViTs) -- multiple citations
SELECT target_id, context, section FROM cites
WHERE source_id = 33333333-0000-0000-0000-000000000003;

-- All topics covered by Paper 4 (Federated Privacy)
SELECT target_id, relevance FROM covers_topic
WHERE source_id = 33333333-0000-0000-0000-000000000004;

-- All papers funded by the DARPA grant
SELECT source_id, role FROM funded_by
WHERE target_id = 55555555-0000-0000-0000-000000000002 ALLOW FILTERING;

-- Alice's collaborators
SELECT target_id, paper_count, first_collab_year, last_collab_year FROM collaborates_with
WHERE source_id = 11111111-0000-0000-0000-000000000001;

-- ============================================================
-- Multiple WHERE conditions (partition + clustering)
-- ============================================================

-- Specific affiliation: Alice at MIT
SELECT department, role, start_year, end_year FROM affiliated_with
WHERE source_id = 11111111-0000-0000-0000-000000000001
  AND target_id = 22222222-0000-0000-0000-000000000001;

-- Specific authorship: Alice authored Paper 1
SELECT author_position, corresponding FROM authored
WHERE source_id = 11111111-0000-0000-0000-000000000001
  AND target_id = 33333333-0000-0000-0000-000000000001;

-- Specific citation: Paper 12 cites Paper 2
SELECT context, section FROM cites
WHERE source_id = 33333333-0000-0000-0000-000000000012
  AND target_id = 33333333-0000-0000-0000-000000000002;

-- ============================================================
-- UPDATE: collection manipulation
-- ============================================================

-- Add a specialization to a researcher's set
UPDATE researcher SET specializations = specializations + {'Reinforcement Learning'}
WHERE researcher_id = 11111111-0000-0000-0000-000000000001;

-- Add a social link to a researcher's map
UPDATE researcher SET social_links['linkedin'] = 'https://linkedin.com/in/alicechen'
WHERE researcher_id = 11111111-0000-0000-0000-000000000001;

-- Add a keyword to a paper
UPDATE paper SET keywords = keywords + {'efficiency'}
WHERE paper_id = 33333333-0000-0000-0000-000000000001;

-- Update a URL in a paper's map
UPDATE paper SET urls['slides'] = 'https://neurips.cc/slides/sparse-tf'
WHERE paper_id = 33333333-0000-0000-0000-000000000001;

-- Update a scalar column
UPDATE paper SET citation_count = 156
WHERE paper_id = 33333333-0000-0000-0000-000000000001;

-- Update h_index
UPDATE researcher SET h_index = 47
WHERE researcher_id = 11111111-0000-0000-0000-000000000001;

-- Add a department to an institution
UPDATE institution SET departments = departments + {'Linguistics'}
WHERE institution_id = 22222222-0000-0000-0000-000000000001;

-- Verify the updates
SELECT specializations, social_links FROM researcher
WHERE researcher_id = 11111111-0000-0000-0000-000000000001;
SELECT citation_count, keywords, urls FROM paper
WHERE paper_id = 33333333-0000-0000-0000-000000000001;
SELECT departments FROM institution
WHERE institution_id = 22222222-0000-0000-0000-000000000001;

-- ============================================================
-- DELETE: collection element removal
-- ============================================================

-- Remove a specialization from a set
UPDATE researcher SET specializations = specializations - {'Reinforcement Learning'}
WHERE researcher_id = 11111111-0000-0000-0000-000000000001;

-- Remove a social link from a map
DELETE social_links['linkedin'] FROM researcher
WHERE researcher_id = 11111111-0000-0000-0000-000000000001;

-- Remove a keyword from a paper
UPDATE paper SET keywords = keywords - {'efficiency'}
WHERE paper_id = 33333333-0000-0000-0000-000000000001;

-- Remove a URL from a paper's map
DELETE urls['slides'] FROM paper
WHERE paper_id = 33333333-0000-0000-0000-000000000001;

-- Remove a department from an institution
UPDATE institution SET departments = departments - {'Linguistics'}
WHERE institution_id = 22222222-0000-0000-0000-000000000001;

-- ============================================================
-- INSERT IF NOT EXISTS (lightweight transactions)
-- ============================================================

-- Conditionally insert a new researcher (only if UUID doesn't exist)
INSERT INTO researcher (researcher_id, name, email, active)
VALUES (11111111-0000-0000-0000-000000000099, 'New Researcher', 'new@test.com', true)
IF NOT EXISTS;

-- Conditionally insert a duplicate (should return applied=false)
INSERT INTO researcher (researcher_id, name, email, active)
VALUES (11111111-0000-0000-0000-000000000001, 'Duplicate Alice', 'dup@test.com', true)
IF NOT EXISTS;

-- Conditionally insert a new topic
INSERT INTO topic (topic_id, name, description, paper_count)
VALUES (44444444-0000-0000-0000-000000000099, 'Quantum Computing',
        'Computation using quantum mechanical phenomena', 3200)
IF NOT EXISTS;

-- ============================================================
-- USING TTL (time-to-live for expiring data)
-- ============================================================

-- Insert a preprint that expires after 24 hours
INSERT INTO paper (paper_id, title, year, paper_type)
VALUES (33333333-0000-0000-0000-000000000099, 'Preprint Draft: Preliminary Results', 2024, 'Preprint')
USING TTL 86400;

-- Insert a temporary collaboration record (30-day TTL)
INSERT INTO collaborates_with (source_id, target_id, paper_count, first_collab_year, last_collab_year)
VALUES (11111111-0000-0000-0000-000000000099, 11111111-0000-0000-0000-000000000001, 0, 2024, 2024)
USING TTL 2592000;

-- ============================================================
-- USING TIMESTAMP (explicit write timestamps)
-- ============================================================

-- Insert a topic with an explicit microsecond timestamp
INSERT INTO topic (topic_id, name, description, paper_count)
VALUES (44444444-0000-0000-0000-000000000098, 'Quantum Computing',
        'Computation using quantum mechanical phenomena', 3200)
USING TIMESTAMP 1704067200000000;

-- Update with an explicit timestamp
UPDATE researcher USING TIMESTAMP 1704067200000000
SET citations_total = 13000
WHERE researcher_id = 11111111-0000-0000-0000-000000000001;

-- ============================================================
-- DELETE operations
-- ============================================================

-- Delete a specific row
DELETE FROM collaborates_with
WHERE source_id = 11111111-0000-0000-0000-000000000099
  AND target_id = 11111111-0000-0000-0000-000000000001;

-- Delete a specific column value
DELETE email FROM researcher
WHERE researcher_id = 11111111-0000-0000-0000-000000000099;

-- Clean up test data
DELETE FROM researcher WHERE researcher_id = 11111111-0000-0000-0000-000000000099;
DELETE FROM paper WHERE paper_id = 33333333-0000-0000-0000-000000000099;
DELETE FROM topic WHERE topic_id = 44444444-0000-0000-0000-000000000099;
DELETE FROM topic WHERE topic_id = 44444444-0000-0000-0000-000000000098;

-- ============================================================
-- ALLOW FILTERING (full-table scans with non-indexed columns)
-- ============================================================

-- Find researchers by title (no index on title)
SELECT name, title, h_index FROM researcher WHERE title = 'Professor' ALLOW FILTERING;

-- Find papers of a specific type
SELECT title, venue, year FROM paper WHERE paper_type = 'Journal' ALLOW FILTERING;

-- Find corresponding authors
SELECT source_id, target_id FROM authored WHERE corresponding = true ALLOW FILTERING;

-- Find high-relevance topic coverage
SELECT source_id, target_id, relevance FROM covers_topic WHERE relevance > 0.9 ALLOW FILTERING;

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH (r:Researcher {name: \"Alice Chen\"}) RETURN r.name, r.h_index, r.specializations", "keyspace": "knowledge"}'

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH (i:Institution) RETURN i.name, i.country, i.ranking ORDER BY i.ranking", "keyspace": "knowledge"}'

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH (r:Researcher {name: \"Alice Chen\"})-[:AUTHORED]->(p:Paper) RETURN p.title, p.year, p.venue ORDER BY p.year DESC", "keyspace": "knowledge"}'

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH (r:Researcher)-[:AFFILIATED_WITH]->(i:Institution {name: \"MIT\"}) RETURN r.name, r.h_index, r.title ORDER BY r.h_index DESC", "keyspace": "knowledge"}'

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH (p1:Paper)-[:CITES]->(p2:Paper)-[:CITES]->(p3:Paper) RETURN p1.title AS citing, p2.title AS via, p3.title AS cited LIMIT 10", "keyspace": "knowledge"}'

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH (r:Researcher {name: \"Alice Chen\"})-[:AUTHORED]->(p:Paper)-[:COVERS]->(t:Topic) RETURN DISTINCT t.name", "keyspace": "knowledge"}'

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH (r1:Researcher)-[:AFFILIATED_WITH]->(i1:Institution), (r2:Researcher)-[:AFFILIATED_WITH]->(i2:Institution), (r1)-[:COLLABORATES]->(r2) WHERE i1.name <> i2.name RETURN r1.name, i1.name AS inst1, r2.name, i2.name AS inst2", "keyspace": "knowledge"}'

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH (p:Paper)-[:FUNDED_BY]->(g:Grant), (p)-[:COVERS]->(t:Topic) RETURN g.funder, g.title, p.title, t.name ORDER BY g.funder", "keyspace": "knowledge"}'

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH path = shortestPath((a:Researcher {name: \"Alice Chen\"})-[*]-(b:Researcher {name: \"Eva Schmidt\"})) RETURN path", "keyspace": "knowledge"}'

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH path = shortestPath((a:Researcher {name: \"Alice Chen\"})-[:COLLABORATES*]-(b:Researcher {name: \"Jun Tanaka\"})) RETURN [n IN nodes(path) | n.name] AS chain", "keyspace": "knowledge"}'

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH path = (p:Paper)-[:CITES*1..3]->(end:Paper) WHERE p.title = \"Vision Transformers for Medical Image Segmentation\" RETURN [n IN nodes(path) | n.title] AS chain LIMIT 5", "keyspace": "knowledge"}'

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH (p:Paper)-[:COVERS]->(t:Topic) RETURN t.name, COUNT(p) AS paper_count ORDER BY paper_count DESC", "keyspace": "knowledge"}'

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH (r:Researcher)-[:AFFILIATED_WITH]->(i:Institution), (r)-[:AUTHORED]->(p:Paper) RETURN i.name, COUNT(DISTINCT p) AS papers ORDER BY papers DESC", "keyspace": "knowledge"}'

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH (p:Paper) RETURN p.venue, AVG(p.citation_count) AS avg_citations, COUNT(p) AS papers ORDER BY avg_citations DESC", "keyspace": "knowledge"}'

#!/usr/bin/env bash
set -euo pipefail

# Cypher graph queries for the research knowledge graph tutorial.
# Requires a running Ferrosa cluster with FERROSA_GRAPH_ENABLED=true.

FERROSA_GRAPH_HOST="${FERROSA_GRAPH_HOST:-localhost}"
FERROSA_GRAPH_PORT="${FERROSA_GRAPH_PORT:-7474}"
BASE_URL="http://${FERROSA_GRAPH_HOST}:${FERROSA_GRAPH_PORT}"

PASS=0
FAIL=0

run_query() {
    local label="$1"
    local endpoint="$2"
    local method="${3:-GET}"
    local body="${4:-}"

    printf "%-65s " "${label}..."

    if [ "${method}" = "GET" ]; then
        HTTP_CODE=$(curl -s -o /dev/null -w "%{http_code}" "${BASE_URL}${endpoint}")
    else
        HTTP_CODE=$(curl -s -o /dev/null -w "%{http_code}" \
            -X "${method}" "${BASE_URL}${endpoint}" \
            -H "Content-Type: application/json" \
            -d "${body}")
    fi

    if [ "${HTTP_CODE}" -eq 200 ]; then
        echo "PASS (${HTTP_CODE})"
        PASS=$((PASS + 1))
    else
        echo "FAIL (${HTTP_CODE})"
        FAIL=$((FAIL + 1))
    fi
}

echo "========================================"
echo "  Research Knowledge Graph Cypher Tests"
echo "  Host: ${FERROSA_GRAPH_HOST}:${FERROSA_GRAPH_PORT}"
echo "========================================"
echo ""

# ------------------------------------------------------------------
# Health and schema
# ------------------------------------------------------------------

run_query "Graph health check" "/graph/health"
run_query "Graph schema introspection" "/graph/schema"

echo ""
echo "--- Basic Node Queries ---"

# ------------------------------------------------------------------
# Basic node queries
# ------------------------------------------------------------------

run_query "Find researcher by name" \
    "/graph/query" "POST" \
    '{"query": "MATCH (r:Researcher {name: \"Alice Chen\"}) RETURN r.name, r.h_index, r.specializations", "keyspace": "knowledge"}'

run_query "All institutions ordered by ranking" \
    "/graph/query" "POST" \
    '{"query": "MATCH (i:Institution) RETURN i.name, i.country, i.ranking ORDER BY i.ranking", "keyspace": "knowledge"}'

run_query "Papers published in 2024" \
    "/graph/query" "POST" \
    '{"query": "MATCH (p:Paper) WHERE p.year = 2024 RETURN p.title, p.venue ORDER BY p.citation_count DESC", "keyspace": "knowledge"}'

run_query "Topics ordered by paper count" \
    "/graph/query" "POST" \
    '{"query": "MATCH (t:Topic) RETURN t.name, t.paper_count ORDER BY t.paper_count DESC", "keyspace": "knowledge"}'

run_query "Active grants with amounts" \
    "/graph/query" "POST" \
    '{"query": "MATCH (g:Grant) WHERE g.status = \"active\" RETURN g.title, g.funder, g.amount ORDER BY g.amount DESC", "keyspace": "knowledge"}'

echo ""
echo "--- Single-Hop Relationship Traversals ---"

# ------------------------------------------------------------------
# Single-hop relationship traversals
# ------------------------------------------------------------------

run_query "Papers authored by Alice Chen" \
    "/graph/query" "POST" \
    '{"query": "MATCH (r:Researcher {name: \"Alice Chen\"})-[:AUTHORED]->(p:Paper) RETURN p.title, p.year, p.venue ORDER BY p.year DESC", "keyspace": "knowledge"}'

run_query "Coauthors of Sparse Transformers paper" \
    "/graph/query" "POST" \
    '{"query": "MATCH (r:Researcher)-[:AUTHORED]->(p:Paper {title: \"Attention Is All You Need 2.0: Sparse Transformers at Scale\"}) RETURN r.name, r.title ORDER BY r.name", "keyspace": "knowledge"}'

run_query "Researchers at MIT" \
    "/graph/query" "POST" \
    '{"query": "MATCH (r:Researcher)-[:AFFILIATED_WITH]->(i:Institution {name: \"MIT\"}) RETURN r.name, r.h_index, r.title ORDER BY r.h_index DESC", "keyspace": "knowledge"}'

run_query "Papers citing Consensus Verification" \
    "/graph/query" "POST" \
    '{"query": "MATCH (citing:Paper)-[:CITES]->(p:Paper {title: \"Formal Verification of Distributed Consensus Protocols\"}) RETURN citing.title, citing.year", "keyspace": "knowledge"}'

run_query "Papers funded by DARPA" \
    "/graph/query" "POST" \
    '{"query": "MATCH (p:Paper)-[:FUNDED_BY]->(g:Grant {funder: \"DARPA\"}) RETURN p.title, g.title AS grant_title", "keyspace": "knowledge"}'

run_query "Alice'\''s direct collaborators" \
    "/graph/query" "POST" \
    '{"query": "MATCH (r:Researcher {name: \"Alice Chen\"})-[:COLLABORATES]->(other:Researcher) RETURN other.name, other.title", "keyspace": "knowledge"}'

echo ""
echo "--- Multi-Hop Traversals ---"

# ------------------------------------------------------------------
# Multi-hop traversals
# ------------------------------------------------------------------

run_query "Citation chain (2 hops)" \
    "/graph/query" "POST" \
    '{"query": "MATCH (p1:Paper)-[:CITES]->(p2:Paper)-[:CITES]->(p3:Paper) RETURN p1.title AS citing, p2.title AS via, p3.title AS cited LIMIT 10", "keyspace": "knowledge"}'

run_query "Researcher topics via papers" \
    "/graph/query" "POST" \
    '{"query": "MATCH (r:Researcher {name: \"Alice Chen\"})-[:AUTHORED]->(p:Paper)-[:COVERS]->(t:Topic) RETURN DISTINCT t.name", "keyspace": "knowledge"}'

run_query "Cross-institution collaborations" \
    "/graph/query" "POST" \
    '{"query": "MATCH (r1:Researcher)-[:AFFILIATED_WITH]->(i1:Institution), (r2:Researcher)-[:AFFILIATED_WITH]->(i2:Institution), (r1)-[:COLLABORATES]->(r2) WHERE i1.name <> i2.name RETURN r1.name, i1.name AS inst1, r2.name, i2.name AS inst2", "keyspace": "knowledge"}'

run_query "Funding chain: funder -> grant -> paper -> topic" \
    "/graph/query" "POST" \
    '{"query": "MATCH (p:Paper)-[:FUNDED_BY]->(g:Grant), (p)-[:COVERS]->(t:Topic) RETURN g.funder, g.title, p.title, t.name ORDER BY g.funder", "keyspace": "knowledge"}'

run_query "Researcher -> paper -> citation -> coauthor" \
    "/graph/query" "POST" \
    '{"query": "MATCH (r:Researcher {name: \"Bob Martinez\"})-[:AUTHORED]->(p:Paper)-[:CITES]->(cited:Paper)<-[:AUTHORED]-(coauthor:Researcher) WHERE coauthor.name <> \"Bob Martinez\" RETURN DISTINCT coauthor.name, cited.title", "keyspace": "knowledge"}'

echo ""
echo "--- Aggregation Queries ---"

# ------------------------------------------------------------------
# Aggregation queries
# ------------------------------------------------------------------

run_query "Most cited papers (top 5)" \
    "/graph/query" "POST" \
    '{"query": "MATCH (p:Paper) RETURN p.title, p.citation_count ORDER BY p.citation_count DESC LIMIT 5", "keyspace": "knowledge"}'

run_query "Papers per topic" \
    "/graph/query" "POST" \
    '{"query": "MATCH (p:Paper)-[:COVERS]->(t:Topic) RETURN t.name, COUNT(p) AS paper_count ORDER BY paper_count DESC", "keyspace": "knowledge"}'

run_query "Researcher publication count" \
    "/graph/query" "POST" \
    '{"query": "MATCH (r:Researcher)-[:AUTHORED]->(p:Paper) RETURN r.name, COUNT(p) AS publications ORDER BY publications DESC", "keyspace": "knowledge"}'

run_query "Collaboration count per researcher" \
    "/graph/query" "POST" \
    '{"query": "MATCH (r:Researcher)-[:COLLABORATES]->(other:Researcher) RETURN r.name, COUNT(other) AS collabs ORDER BY collabs DESC", "keyspace": "knowledge"}'

run_query "Average citations per venue" \
    "/graph/query" "POST" \
    '{"query": "MATCH (p:Paper) RETURN p.venue, AVG(p.citation_count) AS avg_citations, COUNT(p) AS papers ORDER BY avg_citations DESC", "keyspace": "knowledge"}'

run_query "Institution publication count" \
    "/graph/query" "POST" \
    '{"query": "MATCH (r:Researcher)-[:AFFILIATED_WITH]->(i:Institution), (r)-[:AUTHORED]->(p:Paper) RETURN i.name, COUNT(DISTINCT p) AS papers ORDER BY papers DESC", "keyspace": "knowledge"}'

run_query "Grant impact (papers per grant)" \
    "/graph/query" "POST" \
    '{"query": "MATCH (g:Grant)<-[:FUNDED_BY]-(p:Paper) RETURN g.title, g.funder, COUNT(p) AS papers ORDER BY papers DESC", "keyspace": "knowledge"}'

echo ""
echo "--- Path Queries ---"

# ------------------------------------------------------------------
# Path queries
# ------------------------------------------------------------------

run_query "Shortest path between Alice and Eva" \
    "/graph/query" "POST" \
    '{"query": "MATCH path = shortestPath((a:Researcher {name: \"Alice Chen\"})-[*]-(b:Researcher {name: \"Eva Schmidt\"})) RETURN path", "keyspace": "knowledge"}'

run_query "All paths to a paper (max 3 hops)" \
    "/graph/query" "POST" \
    '{"query": "MATCH path = (r:Researcher)-[*1..3]->(p:Paper {title: \"Attention Is All You Need 2.0: Sparse Transformers at Scale\"}) RETURN path LIMIT 10", "keyspace": "knowledge"}'

run_query "Shortest collaboration path (Alice to Jun)" \
    "/graph/query" "POST" \
    '{"query": "MATCH path = shortestPath((a:Researcher {name: \"Alice Chen\"})-[:COLLABORATES*]-(b:Researcher {name: \"Jun Tanaka\"})) RETURN [n IN nodes(path) | n.name] AS chain", "keyspace": "knowledge"}'

run_query "Citation path depth 3" \
    "/graph/query" "POST" \
    '{"query": "MATCH path = (p:Paper)-[:CITES*1..3]->(end:Paper) WHERE p.title = \"Vision Transformers for Medical Image Segmentation\" RETURN [n IN nodes(path) | n.title] AS chain LIMIT 5", "keyspace": "knowledge"}'

echo ""
echo "--- Filtering and Predicates ---"

# ------------------------------------------------------------------
# Filtering and predicates
# ------------------------------------------------------------------

run_query "High h-index researchers (>40)" \
    "/graph/query" "POST" \
    '{"query": "MATCH (r:Researcher) WHERE r.h_index > 40 RETURN r.name, r.h_index, r.title ORDER BY r.h_index DESC", "keyspace": "knowledge"}'

run_query "Highly cited papers (>100)" \
    "/graph/query" "POST" \
    '{"query": "MATCH (p:Paper) WHERE p.citation_count > 100 RETURN p.title, p.citation_count, p.venue ORDER BY p.citation_count DESC", "keyspace": "knowledge"}'

run_query "Funded research with amounts" \
    "/graph/query" "POST" \
    '{"query": "MATCH (p:Paper)-[:FUNDED_BY]->(g:Grant) RETURN p.title, g.funder, g.amount, g.currency ORDER BY g.amount DESC", "keyspace": "knowledge"}'

run_query "Papers with keyword filter" \
    "/graph/query" "POST" \
    '{"query": "MATCH (p:Paper) WHERE \"consensus\" IN p.keywords RETURN p.title, p.venue", "keyspace": "knowledge"}'

run_query "Researchers with specific specialization" \
    "/graph/query" "POST" \
    '{"query": "MATCH (r:Researcher) WHERE \"Distributed Systems\" IN r.specializations RETURN r.name, r.title, r.h_index", "keyspace": "knowledge"}'

run_query "Corresponding authors only" \
    "/graph/query" "POST" \
    '{"query": "MATCH (r:Researcher)-[a:AUTHORED]->(p:Paper) WHERE a.corresponding = true RETURN r.name, p.title ORDER BY r.name", "keyspace": "knowledge"}'

echo ""
echo "--- Complex Graph Analytics ---"

# ------------------------------------------------------------------
# Complex graph analytics
# ------------------------------------------------------------------

run_query "Research influence (papers citing my papers)" \
    "/graph/query" "POST" \
    '{"query": "MATCH (r:Researcher {name: \"Alice Chen\"})-[:AUTHORED]->(p:Paper)<-[:CITES]-(citing:Paper) RETURN p.title AS original, citing.title AS cited_by", "keyspace": "knowledge"}'

run_query "Collaboration network (collect)" \
    "/graph/query" "POST" \
    '{"query": "MATCH (r:Researcher)-[:COLLABORATES]->(other:Researcher) RETURN r.name, COLLECT(other.name) AS collaborators ORDER BY SIZE(COLLECT(other.name)) DESC", "keyspace": "knowledge"}'

run_query "Topic overlap between MIT and Stanford" \
    "/graph/query" "POST" \
    '{"query": "MATCH (i1:Institution {name: \"MIT\"})<-[:AFFILIATED_WITH]-(r1:Researcher)-[:AUTHORED]->(p1:Paper)-[:COVERS]->(t:Topic)<-[:COVERS]-(p2:Paper)<-[:AUTHORED]-(r2:Researcher)-[:AFFILIATED_WITH]->(i2:Institution {name: \"Stanford\"}) RETURN DISTINCT t.name AS shared_topic", "keyspace": "knowledge"}'

run_query "Mutual citation detection" \
    "/graph/query" "POST" \
    '{"query": "MATCH (a:Paper)-[:CITES]->(b:Paper)-[:CITES]->(a) RETURN a.title, b.title", "keyspace": "knowledge"}'

run_query "Researcher reach (papers at distance 2)" \
    "/graph/query" "POST" \
    '{"query": "MATCH (r:Researcher {name: \"Grace Okonkwo\"})-[:AUTHORED]->(p1:Paper)-[:CITES]->(p2:Paper) RETURN r.name, p1.title AS authored, p2.title AS influences", "keyspace": "knowledge"}'

run_query "Co-citation analysis" \
    "/graph/query" "POST" \
    '{"query": "MATCH (p1:Paper)-[:CITES]->(common:Paper)<-[:CITES]-(p2:Paper) WHERE id(p1) < id(p2) RETURN p1.title, p2.title, common.title AS co_cited LIMIT 10", "keyspace": "knowledge"}'

run_query "Topic-spanning researchers" \
    "/graph/query" "POST" \
    '{"query": "MATCH (r:Researcher)-[:AUTHORED]->(p:Paper)-[:COVERS]->(t:Topic) WITH r, COLLECT(DISTINCT t.name) AS topics WHERE SIZE(topics) >= 2 RETURN r.name, topics ORDER BY SIZE(topics) DESC", "keyspace": "knowledge"}'

run_query "Unfunded highly-cited papers" \
    "/graph/query" "POST" \
    '{"query": "MATCH (p:Paper) WHERE p.citation_count > 50 AND NOT (p)-[:FUNDED_BY]->(:Grant) RETURN p.title, p.citation_count ORDER BY p.citation_count DESC", "keyspace": "knowledge"}'

run_query "Institution collaboration graph" \
    "/graph/query" "POST" \
    '{"query": "MATCH (r1:Researcher)-[:AFFILIATED_WITH]->(i1:Institution), (r2:Researcher)-[:AFFILIATED_WITH]->(i2:Institution), (r1)-[:COLLABORATES]->(r2) WHERE i1.name < i2.name RETURN i1.name, i2.name, COUNT(*) AS collaboration_count", "keyspace": "knowledge"}'

run_query "Topic expertise by institution" \
    "/graph/query" "POST" \
    '{"query": "MATCH (i:Institution)<-[:AFFILIATED_WITH]-(r:Researcher)-[:AUTHORED]->(p:Paper)-[:COVERS]->(t:Topic) RETURN i.name, t.name, COUNT(DISTINCT p) AS papers ORDER BY i.name, papers DESC", "keyspace": "knowledge"}'

echo ""
echo "--- WITH Chaining and Subqueries ---"

# ------------------------------------------------------------------
# WITH chaining and subqueries
# ------------------------------------------------------------------

run_query "Top researcher per institution" \
    "/graph/query" "POST" \
    '{"query": "MATCH (r:Researcher)-[:AFFILIATED_WITH]->(i:Institution) WITH i, r ORDER BY r.h_index DESC WITH i, COLLECT(r)[0] AS top RETURN i.name, top.name, top.h_index", "keyspace": "knowledge"}'

run_query "Research pipeline: grant -> paper -> topic" \
    "/graph/query" "POST" \
    '{"query": "MATCH (g:Grant)<-[:FUNDED_BY]-(p:Paper)-[:COVERS]->(t:Topic) WITH g, COLLECT(DISTINCT t.name) AS topics, COUNT(DISTINCT p) AS papers RETURN g.title, g.funder, papers, topics", "keyspace": "knowledge"}'

run_query "Prolific collaborators (>= 2 collabs)" \
    "/graph/query" "POST" \
    '{"query": "MATCH (r:Researcher)-[c:COLLABORATES]->(other:Researcher) WITH r, COUNT(c) AS total_collabs WHERE total_collabs >= 2 RETURN r.name, total_collabs ORDER BY total_collabs DESC", "keyspace": "knowledge"}'

echo ""
echo "========================================"
echo "Results: ${PASS} passed, ${FAIL} failed"
echo "========================================"

if [ "${FAIL}" -gt 0 ]; then
    exit 1
fi

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH (r:Researcher {name: \"Alice Chen\"})-[:AUTHORED]->(p:Paper)<-[:CITES]-(citing:Paper) RETURN p.title AS original, citing.title AS cited_by", "keyspace": "knowledge"}'

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH (a:Paper)-[:CITES]->(b:Paper)-[:CITES]->(a) RETURN a.title, b.title", "keyspace": "knowledge"}'

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH (p1:Paper)-[:CITES]->(common:Paper)<-[:CITES]-(p2:Paper) WHERE id(p1) < id(p2) RETURN p1.title, p2.title, common.title AS co_cited LIMIT 10", "keyspace": "knowledge"}'

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH (r:Researcher)-[:COLLABORATES]->(other:Researcher) RETURN r.name, COLLECT(other.name) AS collaborators ORDER BY SIZE(COLLECT(other.name)) DESC", "keyspace": "knowledge"}'

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH (r1:Researcher)-[:AFFILIATED_WITH]->(i1:Institution), (r2:Researcher)-[:AFFILIATED_WITH]->(i2:Institution), (r1)-[:COLLABORATES]->(r2) WHERE i1.name < i2.name RETURN i1.name, i2.name, COUNT(*) AS collaboration_count", "keyspace": "knowledge"}'

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH (r:Researcher)-[:AUTHORED]->(p:Paper)-[:COVERS]->(t:Topic) WITH r, COLLECT(DISTINCT t.name) AS topics WHERE SIZE(topics) >= 2 RETURN r.name, topics ORDER BY SIZE(topics) DESC", "keyspace": "knowledge"}'

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH (i1:Institution {name: \"MIT\"})<-[:AFFILIATED_WITH]-(r1:Researcher)-[:AUTHORED]->(p1:Paper)-[:COVERS]->(t:Topic)<-[:COVERS]-(p2:Paper)<-[:AUTHORED]-(r2:Researcher)-[:AFFILIATED_WITH]->(i2:Institution {name: \"Stanford\"}) RETURN DISTINCT t.name AS shared_topic", "keyspace": "knowledge"}'

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH (i:Institution)<-[:AFFILIATED_WITH]-(r:Researcher)-[:AUTHORED]->(p:Paper)-[:COVERS]->(t:Topic) RETURN i.name, t.name, COUNT(DISTINCT p) AS papers ORDER BY i.name, papers DESC", "keyspace": "knowledge"}'

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH (p:Paper) WHERE p.citation_count > 50 AND NOT (p)-[:FUNDED_BY]->(:Grant) RETURN p.title, p.citation_count ORDER BY p.citation_count DESC", "keyspace": "knowledge"}'

curl -X POST http://localhost:7474/graph/query \
  -H "Content-Type: application/json" \
  -d '{"query": "MATCH (g:Grant)<-[:FUNDED_BY]-(p:Paper)-[:COVERS]->(t:Topic) WITH g, COLLECT(DISTINCT t.name) AS topics, COUNT(DISTINCT p) AS papers RETURN g.title, g.funder, papers, topics", "keyspace": "knowledge"}'