graph_sage_predict

Generates node embeddings using a previously trained GraphSAGE model from the model catalog (see graph_sage_train).

Server Neo4j Gds neo4j-contrib/gds-agent
Category Execute
Risk class High
Parameters 00 required

What graph_sage_predict does on Neo4j Gds

AI agents invoke graph_sage_predict to trigger actions in Neo4j Gds. What it does depends on the arguments the agent supplies, and its effects often reach beyond the immediate call — builds kicked off, notifications sent, workflows started.

Why graph_sage_predict needs a policy

This tool runs a machine learning inference operation (GraphSAGE prediction) on a Neo4j graph database. It is not a simple read/query — it executes a trained model to generate embeddings, which constitutes an external computational operation. It doesn't delete data or move money, but it does execute a complex algorithm with side effects (embedding generation, potentially writing results to the graph).

From the tool's definition 'Generates node embeddings using a previously trained GraphSAGE model' — triggers execution of a trained ML model against graph data

Questions about graph_sage_predict

What does the graph_sage_predict tool do? +

Generates node embeddings using a previously trained GraphSAGE model from the model catalog (see graph_sage_train). It is categorised as a Execute tool in the Neo4j Gds MCP Server, which means it can trigger actions or run processes. Use rate limits and argument validation.

How do I enforce a policy on graph_sage_predict? +

Register the Neo4j Gds MCP server in PolicyLayer and add a rule for graph_sage_predict: allow, deny, rate-limit, or require approval. Point your MCP client at the PolicyLayer proxy URL and the rule is enforced on every call, before it reaches Neo4j Gds. Nothing to install.

What risk level is graph_sage_predict? +

graph_sage_predict is a Execute tool with high risk. Execute tools should be rate-limited and have argument validation enabled.

Can I rate-limit graph_sage_predict? +

Yes. Add a rate_limit block to the graph_sage_predict rule in your PolicyLayer policy. For example, setting max: 10 and window: 60 limits the tool to 10 calls per minute. Rate limits are tracked per agent session and reset automatically.

How do I block graph_sage_predict completely? +

Set action: deny in the PolicyLayer policy for graph_sage_predict. The AI agent will receive a policy violation error and cannot call the tool. You can also include a reason field to explain why the tool is blocked.

What MCP server provides graph_sage_predict? +

graph_sage_predict is provided by the Neo4j Gds MCP server (neo4j-contrib/gds-agent). PolicyLayer sits as a proxy in front of this server to enforce policies before tool calls reach the server.

// LOOK UP ANOTHER SERVER

Every MCP server has a record like this.

Type a name, get the same breakdown: verified identity, auth posture, risk grade, capabilities, recommended policy.

Teams ship this data inside their own products. See what a licence covers →

// GET IN TOUCH

Have a question or want to learn more? Send us a message.

Message sent.

We'll get back to you soon.