MCP Toolbox Neo4j 集成

sources:
  my-neo4j-source:
    kind: neo4j
    uri: bolt://:7687
    user: neo4j
    password: my-password # Use environment variables in production

tools:
  search-movies-by-actor:
    kind: neo4j-cypher
    source: my-neo4j-source
    description: "Searches for movies an actor has appeared in based on their name. Useful for questions like 'What movies has Tom Hanks been in?'"
    parameters:
      - name: actor_name
        type: string
        description: The full name of the actor to search for.
    statement: |
      MATCH (p:Person {name: $actor_name}) -[:ACTED_IN]-> (m:Movie)
      RETURN m.title AS title, m.year AS year, m.genre AS genre

  get-actor-for-movie:
    kind: neo4j-cypher
    source: my-neo4j-source
    description: "Finds the actors who starred in a specific movie. Useful for questions like 'Who acted in Inception?'"
    parameters:
      - name: movie_title
        type: string
        description: The exact title of the movie.
    statement: |
      MATCH (p:Person) -[:ACTED_IN]-> (m:Movie {title: $movie_title})
      RETURN p.name AS actor

  find-nearest-cinema:
    kind: neo4j-cypher
    source: my-neo4j-source
    description: "Find the nearest cinema to a given city. The city must be an exact match."
    parameters:
      - name: city_name
        type: string
        description: The name of the city to find cinemas near.
    statement: |
      MATCH (city:City {name: $city_name})
      MATCH (cinema:Cinema)
      WITH
          city.latitude AS fromLat,
          city.longitude AS fromLon,
          cinema.latitude AS toLat,
          cinema.longitude AS toLon,
          cinema.name AS cinemaName
      RETURN
          cinemaName,
          point.distance(
              point({latitude: fromLat, longitude: fromLon}),
              point({latitude: toLat, longitude: toLon})
          ) AS distance
      ORDER BY distance
      LIMIT 1

  get-movie-list:
    kind: neo4j-cypher
    source: my-neo4j-source
    description: "Get a list of movies, optionally filtering by year. This is a very useful general tool for getting movies."
    parameters:
      - name: year
        type: integer
        optional: true
        description: The year the movie was released.
    statement: |
      MATCH (movie:Movie)
      WHERE $year IS NULL OR movie.released = $year
      RETURN movie.title, movie.released
      ORDER BY movie.released DESC
      LIMIT 10

  get-movie:
    kind: neo4j-cypher
    source: my-neo4j-source
    description: "Get all information about a specific movie by its title. If a user asks a question about a movie and provides the title, this is the tool to use."
    parameters:
      - name: title
        type: string
        description: The title of the movie.
    statement: |
      MATCH (movie:Movie {title: $title})
      OPTIONAL MATCH (movie)<-[:ACTED_IN]-(actor)
      OPTIONAL MATCH (movie)<-[:DIRECTED]-(director)
      RETURN movie, collect(actor.name) AS actors, collect(director.name) AS directors

toolbox --tools-file "tools.yaml"

sources:
  companies-demo:
    kind: neo4j
    uri: neo4j+s://demo.neo4jlabs.com:7687
    user: companies
    password: companies
    database: companies

tools:
  get-schema:
    kind: neo4j-schema
    source: companies-demo
    description: "Extracts the database schema."

  get-organizations-alphabetical:
    kind: neo4j-cypher
    source: companies-demo
    description: "Retrieves a list of organizations sorted alphabetically."
    parameters:
      - name: limit
        type: integer
        description: "The number of results to return."
    statement: |
      MATCH (o:Organization)
      RETURN o.name as OrganizationName
      ORDER BY o.name ASC
      LIMIT $limit

import asyncio
import os
from mcp import StdioServerParameters
from mcp.client.stdio import stdio_client
from mcp import ClientSession

# Configuration: Launch toolbox with the companies yaml
server_params = StdioServerParameters(
    command="toolbox",
    args=["--tools-file", "tools_companies.yaml", "--stdio"],
    env=os.environ.copy()
)

async def main():
    print("Connecting to MCP Toolbox (Companies Demo)...")

    async with stdio_client(server_params) as (read, write):
        async with ClientSession(read, write) as session:
            await session.initialize()
            print("Connected!")

            print("\n--- Question: What are first 5 organizations alphabetically? ---")

            result = await session.call_tool(
                "get-organizations-alphabetical",
                arguments={
                    "limit": 5
                }
            )

            # Display the result
            for content in result.content:
                print(content.text)

if __name__ == "__main__":
    asyncio.run(main())