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

Fix the lint issue

Browse Source
This commit is contained in:
Samuel Chan
2025-01-04 18:49:32 +08:00
parent 11f889a9df
commit 6c1b669f0f
4 changed files with 155 additions and 98 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
examples/lightrag_zhipu_postgres_demo.py
View File

@@ -53,7 +53,7 @@ async def main():
kv_storage="PGKVStorage",
doc_status_storage="PGDocStatusStorage",
graph_storage="PGGraphStorage",
vector_storage="PGVectorStorage"
vector_storage="PGVectorStorage",
)
# Set the KV/vector/graph storage's `db` property, so all operation will use same connection pool
rag.doc_status.db = postgres_db
@@ -77,27 +77,35 @@ async def main():
start_time = time.time()
# Perform naive search
print(
await rag.aquery("What are the top themes in this story?", param=QueryParam(mode="naive"))
await rag.aquery(
"What are the top themes in this story?", param=QueryParam(mode="naive")
)
)
print(f"Naive Query Time: {time.time() - start_time} seconds")
# Perform local search
print("**** Start Local Query ****")
start_time = time.time()
print(
await rag.aquery("What are the top themes in this story?", param=QueryParam(mode="local"))
await rag.aquery(
"What are the top themes in this story?", param=QueryParam(mode="local")
)
)
print(f"Local Query Time: {time.time() - start_time} seconds")
# Perform global search
print("**** Start Global Query ****")
start_time = time.time()
print(
await rag.aquery("What are the top themes in this story?", param=QueryParam(mode="global"))
await rag.aquery(
"What are the top themes in this story?", param=QueryParam(mode="global")
)
)
print(f"Global Query Time: {time.time() - start_time}")
# Perform hybrid search
print("**** Start Hybrid Query ****")
print(
await rag.aquery("What are the top themes in this story?", param=QueryParam(mode="hybrid"))
await rag.aquery(
"What are the top themes in this story?", param=QueryParam(mode="hybrid")
)
)
print(f"Hybrid Query Time: {time.time() - start_time} seconds")