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

feat: add GraphRAG framework and integrate TuGraph #1506

Merged
merged 60 commits into from May 16, 2024

Conversation

fanzhidongyzby
Copy link
Contributor

Issue: #1505

@github-actions github-actions bot added the enhancement New feature or request label May 9, 2024
@Aries-ckt
Copy link
Collaborator

hi, @fanzhidongyzby ,Graph RAG is an amazing feature, could you give some use cases?

@fanzhidongyzby
Copy link
Contributor Author

fanzhidongyzby commented May 10, 2024

hi, @fanzhidongyzby ,Graph RAG is an amazing feature, could you give some use cases?

@Aries-ckt
Subsequent documentation sections illustrate the use and workflow of Graph RAG with examples.

@fanzhidongyzby fanzhidongyzby force-pushed the tugraph-rag branch 2 times, most recently from 0a5c432 to 12da5b3 Compare May 14, 2024 08:13
@fanzhidongyzby
Copy link
Contributor Author

fanzhidongyzby commented May 15, 2024

Here are tests for Graph RAG framework:

  • [Triplet/Keyword Extractor Test]: tests/intetration_tests/transformer/test_extactor.py
  • [MemoryGraph Test]: tests/unit_tests/graph/test_graph.py
  • [MemoryGraphStore Test]: tests/intetration_tests/graph_store/test_memgraph_store.py
  • [TuGraphStore Test]: tests/intetration_tests/graph_store/test_tugraph_store.py
  • [Graph RAG Example]: examples/rag/graph_rag_example.py

csunny
csunny previously approved these changes May 16, 2024
Copy link
Collaborator

@csunny csunny left a comment

Choose a reason for hiding this comment

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

Exciting advancement! Our imagination soars with expanded capabilities, encompassing databases, graphs, diverse data types, AI-native agent apps —unleashing the potential of structured, unstructured data, and beyond.

Very cool!
image

image

csunny
csunny previously approved these changes May 16, 2024
Copy link
Collaborator

@csunny csunny left a comment

Choose a reason for hiding this comment

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

r+

@KingSkyLi
Copy link
Contributor

KingSkyLi commented May 16, 2024

In this PR, we have implemented the ability to use the TuGraph database for retrieval-augmented generation (RAG). This feature is currently integrated into the Chat Knowledge module. TuGraph is supported from Version 4.3.0 and above.

When creating a knowledge base, users can choose between the following two storage types:

Knowledge Graph
Vector Store
Different storage types will have corresponding icons displayed in the frontend.

If the user chooses to create a knowledge base of the Knowledge Graph type and successfully creates it, they can click the "View Graph" button to view the graph of the current knowledge base.

Additionally, users can obtain relevant knowledge content through the Chat Knowledge conversation feature.

  1. Create Knowledge
    create_knowledge_graph

  2. Knowledge Graph and Open Graph Visual
    view_graph

  3. Graph Visual Page
    graph_data

  4. Chat Knowledge By Graph
    graph_rag_chat

Copy link
Collaborator

@Aries-ckt Aries-ckt left a comment

Choose a reason for hiding this comment

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

LGTM.

Copy link
Collaborator

@csunny csunny left a comment

Choose a reason for hiding this comment

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

LGTM~

@csunny csunny merged commit a9087c3 into eosphoros-ai:main May 16, 2024
4 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

5 participants