DSPy Neo4j 集成

DSPy 是一个框架，用于在算法层面优化语言模型（LM）的提示和权重，尤其是在流水线中一次或多次使用语言模型时。

Neo4j 集成实现了向量搜索功能。

安装

pip install dspy neo4j

功能包括

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

from dspy.retrieve.neo4j_rm import Neo4jRM
import os

os.environ["NEO4J_URI"] = 'bolt://: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")