Skip to content

Feature : Neo4j Search Tool Prototype#20

Closed
Charan6924 wants to merge 3 commits intoINCF:mainfrom
Charan6924:feature/graph-retreiver-demo
Closed

Feature : Neo4j Search Tool Prototype#20
Charan6924 wants to merge 3 commits intoINCF:mainfrom
Charan6924:feature/graph-retreiver-demo

Conversation

@Charan6924
Copy link
Copy Markdown

Overview

This PR adds a GraphRetriever class (backend/neo4j_search_tool.py) to enable the agent to query graph-structured data (NIFSTD ontology).

Why this is needed

As discussed in the roadmap, vector search often misses explicit relationships (e.g., Disease -> Affects -> Brain Region). This tool allows the agent to retrieve these direct connections to augment the context window.

Changes

  • Added neo4j dependency.
  • Implemented GraphRetriever with environment variable configuration.
  • Added connection verification and error handling.
  • Added Unit Tests: Included tests/test_neo4j_tool.py to ensure that the graph retrieval tool works.
Screenshot 2026-01-05 at 7 40 54 PM

@QuantumByte-01
Copy link
Copy Markdown
Collaborator

Clean prototype with good error handling and unit tests, and graph-based ontology search is a genuinely interesting direction for neuroscience dataset discovery. However there is no integration into the agent pipeline, no documentation on what Neo4j instance or dataset this targets (NIFSTD? Custom graph?), and neo4j is added as a core dependency when it should be optional. Please integrate it as an optional search tool in the agent, document the required graph schema/dataset, and make the dependency opt-in, then reopen.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants