Metal
Metal 是一個用於 ML Embeddings 的託管服務。
這個筆記本展示了如何使用 Metal 的檢索器。
首先，您需要註冊 Metal 並取得 API 金鑰。您可以在這裡進行註冊
%pip install --upgrade --quiet 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
API 參考：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'})]