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.

Sign up for GitHub

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

Jump to bottom

PGVector and GraphRAG support (Retrieval for RAG and GraphRAG) #167

Open
k2ai opened this issue Sep 12, 2024 · 2 comments
Open

PGVector and GraphRAG support (Retrieval for RAG and GraphRAG) #167

k2ai opened this issue Sep 12, 2024 · 2 comments

Comments

@k2ai
Copy link

k2ai commented Sep 12, 2024

Hi,
I have my own RAG and GraphRAG systems and I wanted to integrate STROM Retrieval for PGVector and GraphRAG.
Can you please integrate support for such Retrievals?
@shaoyijia
Copy link
Collaborator

Hi @k2ai , are you proposing integrating PGVector and GraphRAG to this project?

I have my own RAG and GraphRAG systems

Currently, we haven't used PGVector and GraphRAG. If you are willing to start a draft PR to delineate the integration, we are happy to help.

vishwamartur added a commit to vishwamartur/storm that referenced this issue Jan 3, 2025
@vishwamartur
Add PGVector retriever support
c31981e 
Related to stanford-oval#167

Add support for PGVector and GraphRAG retrievals.

* **knowledge_storm/rm.py**:
  - Add `PGVectorRetriever` class to support PGVector retrieval.
  - Add `GraphRAGRM` class to support GraphRAG retrieval.
  - Update import statements to include necessary libraries for PGVector and GraphRAG.

* **examples/storm_examples/run_storm_wiki_claude.py** and **examples/storm_examples/run_storm_wiki_gemini.py**:
  - Update environment variable descriptions to include PGVector and GraphRAG.
  - Add cases for `pgvector` and `graphrag` retrievers in the main function.

* **knowledge_storm/collaborative_storm/engine.py**:
  - Add `pgvector_retriever` attribute to `CollaborativeStormLMConfigs` class.
  - Add `set_pgvector_retriever` method to `CollaborativeStormLMConfigs` class.
  - Update import statements to include `PGVectorRetriever`.

* **tests/test_pgvector_retriever.py**:
  - Add unit tests for the new `PGVectorRetriever` class.
@vishwamartur vishwamartur mentioned this issue Jan 3, 2025
Open
@vishwamartur
Copy link

@shaoyijia PR is open please take the task from here

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@k2ai @vishwamartur @shaoyijia