Skip to content

TXT2KG w/ txtkg_rag.py e2e workflow example #9992

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

Open
wants to merge 540 commits into
base: master
Choose a base branch
from
Open

Conversation

puririshi98
Copy link
Contributor

@puririshi98 puririshi98 commented Jan 29, 2025

TXT2kg docs before their merged into main:
https://pytorch-geometric--9992.org.readthedocs.build/en/9992/modules/nn.html?highlight=txt2kg#torch_geometric.nn.nlp.TXT2KG

example of how to use this in nvidia docker container:
git config --global credential.helper store; huggingface-cli login --token <insert_token>; cd /opt/pyg; pip uninstall -y torch-geometric; rm -rf pytorch_geometric; git clone -b latest-txt2kg https://github.com/pyg-team/pytorch_geometric.git; cd /opt/pyg/pytorch_geometric; pip install .; pip install openai

example to run:
python3 examples/llm/txt2kg_rag.py

thanks to @Kh4L @zaristei @rlratzel @rliu
for their contributions

using syntheticQA: 70%->80% acc over baselines
using TechQA:
Breaking into PRs:

  1. Cleanup G-retriever Example and APIs #10368
  2. Rag utils + testing for them
  3. TXT2KG & LLMJudge + example

puririshi98 and others added 30 commits January 27, 2025 10:23
pre-commit-ci bot and others added 12 commits July 23, 2025 17:27
253b ultra is not a good enough LLM Judge, looking for better.

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Ralph Liu <[email protected]>
(removes retrieved context from question when asking llm judge to eval)

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Ralph Liu <[email protected]>
@puririshi98 puririshi98 changed the title TXT2KG w/ hotpot_qa.pyGraph Retrieval Example and tech_qa.py HybridRAG e2e workflow example TXT2KG w/ txtkg_rag.py e2e workflow example Jul 28, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants