Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Neo4j Graph Rag Integration #139

Merged
merged 15 commits into from
Dec 6, 2024
Merged

Conversation

Eric-Shang
Copy link
Collaborator

@Eric-Shang Eric-Shang commented Dec 3, 2024

Why are these changes needed?

This PR integrates Neo4j Graph Rag. It wraps Neo4j Property Graph Index as a GraphQueryEngine and could be added to an AG2 agent through GraphRagCapability.

I also include a notebook example to show how to use this feature.

Related issue number

part of #68

Checks

@Eric-Shang
Copy link
Collaborator Author

Need to add functionality to allow pre-defined schema and an unit test. The notebook functions as an integration test.

@AgentGenie
Copy link
Collaborator

Please add unit tests

setup.py Outdated Show resolved Hide resolved
@marklysze marklysze added the RAG label Dec 3, 2024
@Eric-Shang Eric-Shang requested a review from AgentGenie December 4, 2024 04:15
@AgentGenie AgentGenie marked this pull request as ready for review December 4, 2024 22:32
@qingyun-wu qingyun-wu requested a review from AgentGenie December 4, 2024 23:40
@AgentGenie
Copy link
Collaborator

Debugged a random pydantic exception
1 validation error for KGSchema triplets Field required [type=missing, input_value={}, input_type=dict] For further information visit https://errors.pydantic.dev/2.9/v/missing'

Resolved by downgrade the packages to
%pip install llama-index==0.11.15 llama-index-graph-stores-neo4j==0.3.0

Copy link
Collaborator

@AgentGenie AgentGenie left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

tentative approval for testing

.pre-commit-config.yaml Outdated Show resolved Hide resolved
@AgentGenie AgentGenie added this pull request to the merge queue Dec 6, 2024
Merged via the queue into ag2ai:main with commit 13989df Dec 6, 2024
206 of 214 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants