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

[GraphRAG] FalkorDB graph rag integration #111

Merged
merged 10 commits into from
Nov 29, 2024
Merged

Conversation

AgentGenie
Copy link
Collaborator

Why are these changes needed?

This PR integrates FalkorDB. It wraps FalkorDB 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

@marklysze
Copy link
Collaborator

Thanks @AgentGenie, updates:

  • Text in notebook
  • Updated version support of GraphRAG SDK to 0.3.3 - this version has a chat session inbuilt that can keep track of its own query history, so utilising that.
  • Default OpenAI model now GPT-4o (as recommended by GraphRAG SDK)
  • Tests have been updated to support v0.3.3

In the future I'll add another example. I'm thinking it would be good if we could use the auto-detect ontology capability. Do you think this could be incorporated? Would make it easy for users to load data and query it without having to create a schema/ontology.

@AgentGenie
Copy link
Collaborator Author

@marklysze Two main changes:

  1. Use GenerativeModel as arg type and use OpenAI model as default. This would not limit the user to use other LLM models.
  2. Update the code that auto generate ontology if not defined by user.

Copy link
Collaborator

@qingyun-wu qingyun-wu left a comment

Choose a reason for hiding this comment

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

Approving right now but may need more eyes on this if you can get more reviewers. @AgentGenie please feel free to create an issue to remind us.

@qingyun-wu qingyun-wu added this pull request to the merge queue Nov 29, 2024
Merged via the queue into main with commit 5ab5522 Nov 29, 2024
204 of 212 checks passed
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.

4 participants