langchain/templates/neo4j-semantic-layer
Tomaz Bratanic 3e0cd11f51
templates: Add neo4j semantic layer template (#15652)
Co-authored-by: Tomaz Bratanic <tomazbratanic@Tomazs-MacBook-Pro.local>
Co-authored-by: Erick Friis <erick@langchain.dev>
2024-01-09 15:33:44 -08:00
..
neo4j_semantic_layer templates: Add neo4j semantic layer template (#15652) 2024-01-09 15:33:44 -08:00
static templates: Add neo4j semantic layer template (#15652) 2024-01-09 15:33:44 -08:00
tests templates: Add neo4j semantic layer template (#15652) 2024-01-09 15:33:44 -08:00
ingest.py templates: Add neo4j semantic layer template (#15652) 2024-01-09 15:33:44 -08:00
main.py templates: Add neo4j semantic layer template (#15652) 2024-01-09 15:33:44 -08:00
poetry.lock templates: Add neo4j semantic layer template (#15652) 2024-01-09 15:33:44 -08:00
pyproject.toml templates: Add neo4j semantic layer template (#15652) 2024-01-09 15:33:44 -08:00
README.md templates: Add neo4j semantic layer template (#15652) 2024-01-09 15:33:44 -08:00

neo4j-semantic-layer

This template is designed to implement an agent capable of interacting with a graph database like Neo4j through a semantic layer using OpenAI function calling. The semantic layer equips the agent with a suite of robust tools, allowing it to interact with the graph databas based on the user's intent.

Workflow diagram

Tools

The agent utilizes several tools to interact with the Neo4j graph database effectively:

  1. Information tool:
    • Retrieves data about movies or individuals, ensuring the agent has access to the latest and most relevant information.
  2. Recommendation Tool:
    • Provides movie recommendations based upon user preferences and input.
  3. Memory Tool:
    • Stores information about user preferences in the knowledge graph, allowing for a personalized experience over multiple interactions.

Environment Setup

You need to define the following environment variables

OPENAI_API_KEY=<YOUR_OPENAI_API_KEY>
NEO4J_URI=<YOUR_NEO4J_URI>
NEO4J_USERNAME=<YOUR_NEO4J_USERNAME>
NEO4J_PASSWORD=<YOUR_NEO4J_PASSWORD>

Populating with data

If you want to populate the DB with an example movie dataset, you can run python ingest.py. The script import information about movies and their rating by users. Additionally, the script creates two fulltext indices, which are used to map information from user input to the database.

Usage

To use this package, you should first have the LangChain CLI installed:

pip install -U "langchain-cli[serve]"

To create a new LangChain project and install this as the only package, you can do:

langchain app new my-app --package neo4j-semantic-layer

If you want to add this to an existing project, you can just run:

langchain app add neo4j-semantic-layer

And add the following code to your server.py file:

from neo4j_semantic_layer import agent_executor as neo4j_semantic_agent

add_routes(app, neo4j_semantic_agent, path="/neo4j-semantic-layer")

(Optional) Let's now configure LangSmith. LangSmith will help us trace, monitor and debug LangChain applications. LangSmith is currently in private beta, you can sign up here. If you don't have access, you can skip this section

export LANGCHAIN_TRACING_V2=true
export LANGCHAIN_API_KEY=<your-api-key>
export LANGCHAIN_PROJECT=<your-project>  # if not specified, defaults to "default"

If you are inside this directory, then you can spin up a LangServe instance directly by:

langchain serve

This will start the FastAPI app with a server is running locally at http://localhost:8000

We can see all templates at http://127.0.0.1:8000/docs We can access the playground at http://127.0.0.1:8000/neo4j-semantic-layer/playground

We can access the template from code with:

from langserve.client import RemoteRunnable

runnable = RemoteRunnable("http://localhost:8000/neo4j-semantic-layer")