使用 LlamaIndex 结合 MongoDB 和 Redis 实现索引存储

from llama_index import LlamaIndex
from pymongo import MongoClient
import redis
 
# 连接MongoDB和Redis
mongo_client = MongoClient("mongodb://localhost:27017")
redis_client = redis.StrictRedis(host='localhost', port=6379, db=0)
 
# 创建LlamaIndex实例，使用MongoDB和Redis作为数据存储
index = LlamaIndex(
    storage_mode="local",
    mongo_uri="mongodb://localhost:27017",
    mongo_db="llama_index_db",
    redis_host="localhost",
    redis_port=6379,
    redis_db=0,
    chunk_size=1024,
    n_data_workers=2,
    n_index_workers=2,
)
 
# 示例：向索引中添加文档
document_id = "example_document"
document_text = "这是一个示例文档用于演示如何使用LlamaIndex。"
index.add_document(document_id, document_text)
 
# 示例：搜索文档
query = "示例文档"
results = index.search(query)
print(results)

这段代码展示了如何使用LlamaIndex库结合MongoDB和Redis来创建和管理文本索引。首先，我们连接到MongoDB和Redis数据库，然后创建一个LlamaIndex实例，并使用它来添加和搜索文档。这个过程是在假设本地运行这些服务的基础上进行的，并且使用了默认的连接参数。在实际应用中，你可能需要根据你的环境配置修改连接参数。