MistralAI 임베딩

MistralAI 임베딩 (MistralAI Embeddings)

MistralAI의 mistral-embed 모델을 LlamaIndex에서 사용하는 방법을 보여주는 예시예요. API 키로 임베딩 모델을 만들고 텍스트를 벡터화하는 흐름이에요.

출처: 문서

본문

colab에서 이 노트북을 열고 있다면 LlamaIndex 🦙를 설치해야 할 거예요.

%pip install llama-index-embeddings-mistralai

!pip install llama-index

# imports
from llama_index.embeddings.mistralai import MistralAIEmbedding

# get API key and create embeddings
api_key = "YOUR API KEY"
model_name = "mistral-embed"
embed_model = MistralAIEmbedding(model_name=model_name, api_key=api_key)


embeddings = embed_model.get_text_embedding("La Plateforme - The Platform")

print(f"Dimension of embeddings: {len(embeddings)}")

Dimension of embeddings: 1024

embeddings[:5]

[-0.0299224853515625,
 -0.0028362274169921875,
 0.0282745361328125,
 -0.034759521484375,
 -0.0017366409301757812]