今天看啥  ›  专栏  ›  PaperAgent

一个更小、更快、更干净的GraphRAG!

PaperAgent  · 公众号  ·  · 2024-08-17 01:20

文章预览

nano-GraphRAG 一个简单,易于修改的GraphRAG实现 😭 GraphRAG很好很强大,但是官方实现很难/痛苦地阅读或修改。  😊 这个项目提供了一个更小、更快、更干净的GraphRAG,同时保留了核心功能(见基准测试和问题)。  🎁 除了测试和提示,nano-graphrag大约有800行代码。  👌 小而可扩展,异步且完全类型化。 pip  install nano-graphrag 下载查尔斯·狄更斯的《圣诞颂歌》副本 curl  https://raw.githubusercontent.com/gusye1234/nano-graphrag/main/tests/mock_data.txt > ./book.txt 使用下面的Python代码片段 from  nano_graphrag  import  GraphRAG, QueryParam graph_func = GraphRAG(working_dir= "./dickens" ) with open( "./book.txt" ) as f: graph_func.insert(f.read()) # Perform global graphrag search print(graph_func.query( "What are the top themes in this story?" )) # Perform local graphrag search (I think is better and more scalable one) print(graph_func.query( "What ………………………………

原文地址:访问原文地址
快照地址: 访问文章快照
总结与预览地址:访问总结与预览