DSPy Neo4j 集成

DSPy 是一个用于算法优化 LM 提示和权重的框架,尤其是在 LM 在管道中使用一次或多次时。

Neo4j 集成允许进行向量搜索。

以下是 DSPy 集成 的概述。

安装

pip install dspy neo4j

功能包括

  • Neo4jRM - 是一个典型的检索器组件,可用于查询向量存储索引并查找相关文档。

from dspy.retrieve.neo4j_rm import Neo4jRM
import os

os.environ["NEO4J_URI"] = 'bolt://localhost:7687'
os.environ["NEO4J_USERNAME"] = 'neo4j'
os.environ["NEO4J_PASSWORD"] = 'password'
os.environ["OPENAI_API_KEY"] = 'sk-'

retriever_model = Neo4jRM(
    index_name="vector",
    text_node_property="text"
)

results = retriever_model("Explore the significance of quantum computing", k=3)

for passage in results:
    print("Document:", passage, "\n")