Python
使用 LangChain Python 与 Metal 检索器集成。
Metal 是一个用于机器学习嵌入的托管服务。
pip install -qU metal_sdk
from metal_sdk.metal import Metal API_KEY = "" CLIENT_ID = "" INDEX_ID = "" metal = Metal(API_KEY, CLIENT_ID, INDEX_ID)
metal.index({"text": "foo1"}) metal.index({"text": "foo"})
{'data': {'id': '642739aa7559b026b4430e42', 'text': 'foo', 'createdAt': '2023-03-31T19:51:06.748Z'}}
from langchain_community.retrievers import MetalRetriever
retriever = MetalRetriever(metal, params={"limit": 2})
retriever.invoke("foo1")
[Document(page_content='foo1', metadata={'dist': '1.19209289551e-07', 'id': '642739a17559b026b4430e40', 'createdAt': '2023-03-31T19:50:57.853Z'}), Document(page_content='foo1', metadata={'dist': '4.05311584473e-06', 'id': '642738f67559b026b4430e3c', 'createdAt': '2023-03-31T19:48:06.769Z'})]