diff --git a/examples/lightrag_multi_model_all_modes_demo.py b/examples/lightrag_multi_model_all_modes_demo.py deleted file mode 100644 index 16e18782..00000000 --- a/examples/lightrag_multi_model_all_modes_demo.py +++ /dev/null @@ -1,88 +0,0 @@ -import os -import asyncio -from lightrag import LightRAG, QueryParam -from lightrag.llm.openai import gpt_4o_mini_complete, gpt_4o_complete, openai_embed -from lightrag.kg.shared_storage import initialize_pipeline_status - -WORKING_DIR = "./lightrag_demo" - -if not os.path.exists(WORKING_DIR): - os.mkdir(WORKING_DIR) - - -async def initialize_rag(): - rag = LightRAG( - working_dir=WORKING_DIR, - embedding_func=openai_embed, - llm_model_func=gpt_4o_mini_complete, # Default model for queries - ) - - await rag.initialize_storages() - await initialize_pipeline_status() - - return rag - - -def main(): - # Initialize RAG instance - rag = asyncio.run(initialize_rag()) - - # Load the data - with open("./book.txt", "r", encoding="utf-8") as f: - rag.insert(f.read()) - - # Query with naive mode (default model) - print("--- NAIVE mode ---") - print( - rag.query( - "What are the main themes in this story?", param=QueryParam(mode="naive") - ) - ) - - # Query with local mode (default model) - print("\n--- LOCAL mode ---") - print( - rag.query( - "What are the main themes in this story?", param=QueryParam(mode="local") - ) - ) - - # Query with global mode (default model) - print("\n--- GLOBAL mode ---") - print( - rag.query( - "What are the main themes in this story?", param=QueryParam(mode="global") - ) - ) - - # Query with hybrid mode (default model) - print("\n--- HYBRID mode ---") - print( - rag.query( - "What are the main themes in this story?", param=QueryParam(mode="hybrid") - ) - ) - - # Query with mix mode (default model) - print("\n--- MIX mode ---") - print( - rag.query( - "What are the main themes in this story?", param=QueryParam(mode="mix") - ) - ) - - # Query with a custom model (gpt-4o) for a more complex question - print("\n--- Using custom model for complex analysis ---") - print( - rag.query( - "How does the character development reflect Victorian-era attitudes?", - param=QueryParam( - mode="global", - model_func=gpt_4o_complete, # Override default model with more capable one - ), - ) - ) - - -if __name__ == "__main__": - main() diff --git a/examples/copy_llm_cache_to_another_storage.py b/examples/unofficial-sample/copy_llm_cache_to_another_storage.py similarity index 100% rename from examples/copy_llm_cache_to_another_storage.py rename to examples/unofficial-sample/copy_llm_cache_to_another_storage.py diff --git a/examples/lightrag_bedrock_demo.py b/examples/unofficial-sample/lightrag_bedrock_demo.py similarity index 100% rename from examples/lightrag_bedrock_demo.py rename to examples/unofficial-sample/lightrag_bedrock_demo.py diff --git a/examples/lightrag_hf_demo.py b/examples/unofficial-sample/lightrag_hf_demo.py similarity index 100% rename from examples/lightrag_hf_demo.py rename to examples/unofficial-sample/lightrag_hf_demo.py diff --git a/examples/lightrag_llamaindex_direct_demo.py b/examples/unofficial-sample/lightrag_llamaindex_direct_demo.py similarity index 100% rename from examples/lightrag_llamaindex_direct_demo.py rename to examples/unofficial-sample/lightrag_llamaindex_direct_demo.py diff --git a/examples/lightrag_llamaindex_litellm_demo.py b/examples/unofficial-sample/lightrag_llamaindex_litellm_demo.py similarity index 100% rename from examples/lightrag_llamaindex_litellm_demo.py rename to examples/unofficial-sample/lightrag_llamaindex_litellm_demo.py diff --git a/examples/lightrag_lmdeploy_demo.py b/examples/unofficial-sample/lightrag_lmdeploy_demo.py similarity index 100% rename from examples/lightrag_lmdeploy_demo.py rename to examples/unofficial-sample/lightrag_lmdeploy_demo.py diff --git a/examples/lightrag_nvidia_demo.py b/examples/unofficial-sample/lightrag_nvidia_demo.py similarity index 100% rename from examples/lightrag_nvidia_demo.py rename to examples/unofficial-sample/lightrag_nvidia_demo.py diff --git a/examples/lightrag_openai_neo4j_milvus_redis_demo.py b/examples/unofficial-sample/lightrag_openai_neo4j_milvus_redis_demo.py similarity index 100% rename from examples/lightrag_openai_neo4j_milvus_redis_demo.py rename to examples/unofficial-sample/lightrag_openai_neo4j_milvus_redis_demo.py