Skip to main content
Open In ColabOpen on GitHub

Metal

Metal is a managed service for ML Embeddings.

This notebook shows how to use Metal's retriever.

First, you will need to sign up for Metal and get an API key. You can do so here

%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)

Ingest Documents​

You only need to do this if you haven't already set up an index

metal.index({"text": "foo1"})
metal.index({"text": "foo"})
{'data': {'id': '642739aa7559b026b4430e42',
  'text': 'foo',
  'createdAt': '2023-03-31T19:51:06.748Z'}}

Query​

Now that our index is set up, we can set up a retriever and start querying it.

from langchain_community.retrievers import MetalRetriever
API Reference: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'})]