gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

linting errors

Browse Source
This commit is contained in:
omdivyatej
2025-03-25 15:20:09 +05:30
parent f87c235a4c
commit f049f2f5c4
4 changed files with 36 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
examples/lightrag_multi_model_all_modes_demo.py
View File

@@ -9,6 +9,7 @@ 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,
@@ -21,6 +22,7 @@ async def initialize_rag():
return rag
def main():
# Initialize RAG instance
rag = asyncio.run(initialize_rag())
@@ -33,8 +35,7 @@ def main():
print("--- NAIVE mode ---")
print(
rag.query(
"What are the main themes in this story?",
param=QueryParam(mode="naive")
"What are the main themes in this story?", param=QueryParam(mode="naive")
)
)
@@ -42,8 +43,7 @@ def main():
print("\n--- LOCAL mode ---")
print(
rag.query(
"What are the main themes in this story?",
param=QueryParam(mode="local")
"What are the main themes in this story?", param=QueryParam(mode="local")
)
)
@@ -51,8 +51,7 @@ def main():
print("\n--- GLOBAL mode ---")
print(
rag.query(
"What are the main themes in this story?",
param=QueryParam(mode="global")
"What are the main themes in this story?", param=QueryParam(mode="global")
)
)
@@ -60,8 +59,7 @@ def main():
print("\n--- HYBRID mode ---")
print(
rag.query(
"What are the main themes in this story?",
param=QueryParam(mode="hybrid")
"What are the main themes in this story?", param=QueryParam(mode="hybrid")
)
)
@@ -69,8 +67,7 @@ def main():
print("\n--- MIX mode ---")
print(
rag.query(
"What are the main themes in this story?",
param=QueryParam(mode="mix")
"What are the main themes in this story?", param=QueryParam(mode="mix")
)
)
@@ -81,10 +78,11 @@ def main():
"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
)
model_func=gpt_4o_complete, # Override default model with more capable one
),
)
)
if __name__ == "__main__":
main()
main()