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

Multi-index querying for API layer #1644

Open
wants to merge 58 commits into
base: main
Choose a base branch
from

Conversation

KennyZhang1
Copy link
Contributor

Description

This PR adds support for local and global multi-index querying to the graphrag query API.

Related Issues

N/A

Proposed Changes

  • Implemented two new functions in the graphrag query API layer for multi index global and local search
  • Cleaned up graphrag API by moving helper functions to a separate util file
  • Added support for passing in lists of vector stores via the config file

Checklist

  • I have tested these changes locally.
  • I have reviewed the code changes.
  • I have updated the documentation (if necessary).
  • I have added appropriate unit tests (if applicable).

Additional Notes

  • This PR is a first step for integrating multi-index querying into the graphrag library. Once the config file changes and config loading methods are finalized, a separate PR will be opened for integrating multi-index querying at the CLI layer.

KennyZhang1 and others added 30 commits January 13, 2025 14:55
@KennyZhang1 KennyZhang1 requested a review from a team as a code owner January 21, 2025 16:29
graphrag/api/query.py Outdated Show resolved Hide resolved
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