13 lines
354 B
Python
13 lines
354 B
Python
|
"""This example demonstrate how to embed a text into a vector
|
||
|
|
using MistralAI models and API.
|
||
|
|
"""
|
||
|
|
|
||
|
|
from neo4j_graphrag.embeddings import MistralAIEmbeddings
|
||
|
|
|
||
|
|
# set api key here on in the MISTRAL_API_KEY env var
|
||
|
|
api_key = None
|
||
|
|
|
||
|
|
embeder = MistralAIEmbeddings(model="mistral-embed", api_key=api_key)
|
||
|
|
res = embeder.embed_query("my question")
|
||
|
|
print(res[:10])
|