PremAI is an all-in-one platform that simplifies the creation of robust, production-ready applications powered by Generative AI. By streamlining the development process, PremAI allows you to concentrate on enhancing user experience and driving overall growth for your application. You can quickly start using our platformhere.

Installation and setup​

We start by installinglangchain andpremai-sdk. You can type the following command to install:

pip install premai langchain

Before proceeding further, please make sure that you have made an account on PremAI and already created a project. If not, please refer to thequick start guide to get started with the PremAI platform. Create your first project and grab your API key.

PremEmbeddings​

In this section we are going to dicuss how we can get access to different embedding model usingPremEmbeddings with LangChain. Let's start by importing our modules and setting our API Key.

# Let's start by doing some imports and define our embedding object

from langchain_community.embeddingsimport PremAIEmbeddings

Once we imported our required modules, let's setup our client. For now let's assume that ourproject_id is8. But make sure you use your project-id, otherwise it will throw error.

Note: Settingmodel_name argument in mandatory for PremAIEmbeddings unlikeChatPremAI.

import getpass
import os

if os.environ.get("PREMAI_API_KEY")isNone:
    os.environ["PREMAI_API_KEY"]= getpass.getpass("PremAI API Key:")

model="text-embedding-3-large"
embedder= PremAIEmbeddings(project_id=8, model=model)

We support lots of state of the art embedding models. You can view our list of supported LLMs and embedding modelshere. For now let's go fortext-embedding-3-large model for this example.

query="Hello, this is a test query"
query_result= embedder.embed_query(query)

# Let's print the first five elements of the query embedding vector

print(query_result[:5])

[-0.02129288576543331, 0.0008162345038726926, -0.004556538071483374, 0.02918623760342598, -0.02547479420900345]

Finally let's embed a document

documents=["This is document1","This is document2","This is document3"]

doc_result= embedder.embed_documents(documents)

# Similar to previous result, let's print the first five element
# of the first document vector

print(doc_result[0][:5])

[-0.0030691148713231087, -0.045334383845329285, -0.0161729846149683, 0.04348714277148247, -0.0036920777056366205]

Related​

• Embedding modelconceptual guide
• Embedding modelhow-to guides