跳至主要內容

PremAI

PremAI 是一個多合一平台,可簡化由生成式 AI 驅動的強大、可生產應用程式的建立。透過簡化開發流程,PremAI 讓您可以專注於增強使用者體驗並推動應用程式的整體成長。您可以在此快速開始使用我們的平台。

安裝與設定

我們先安裝 langchainpremai-sdk。您可以輸入以下指令進行安裝

pip install premai langchain

在繼續之前,請確保您已在 PremAI 上建立帳戶並已建立專案。如果沒有,請參閱快速入門指南,以開始使用 PremAI 平台。建立您的第一個專案並取得您的 API 金鑰。

PremEmbeddings

在本節中,我們將討論如何使用 LangChain 的 PremEmbeddings 來存取不同的嵌入模型。讓我們先匯入我們的模組並設定我們的 API 金鑰。

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

from langchain_community.embeddings import PremAIEmbeddings
API 參考:PremAIEmbeddings

匯入所需的模組後,讓我們設定我們的客戶端。現在讓我們假設我們的 project_id8。但請確保您使用您的 project-id,否則會拋出錯誤。

注意:與 ChatPremAI 不同,在 PremAIEmbeddings 中設定 model_name 參數是強制性的。

import getpass
import os

if os.environ.get("PREMAI_API_KEY") is None:
os.environ["PREMAI_API_KEY"] = getpass.getpass("PremAI API Key:")
model = "text-embedding-3-large"
embedder = PremAIEmbeddings(project_id=8, model=model)

我們支援許多最先進的嵌入模型。您可以在此處查看我們支援的 LLM 和嵌入模型列表。 現在,讓我們以 text-embedding-3-large 模型為例。

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]

最後,讓我們嵌入一個文檔

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]

此頁面是否對您有幫助?