securing for production with env vars for creds
This commit is contained in:
@@ -16,12 +16,13 @@ if not os.path.exists(WORKING_DIR):
|
||||
rag = LightRAG(
|
||||
working_dir=WORKING_DIR,
|
||||
llm_model_func=gpt_4o_mini_complete, # Use gpt_4o_mini_complete LLM model
|
||||
kg="Neo4JStorage"
|
||||
kg="Neo4JStorage",
|
||||
log_level="INFO"
|
||||
# llm_model_func=gpt_4o_complete # Optionally, use a stronger model
|
||||
)
|
||||
|
||||
with open("./book.txt") as f:
|
||||
rag.insert(f.read())
|
||||
# with open("./book.txt") as f:
|
||||
# rag.insert(f.read())
|
||||
|
||||
# Perform naive search
|
||||
print(rag.query("What are the top themes in this story?", param=QueryParam(mode="naive")))
|
||||
|
Reference in New Issue
Block a user