Files
AI/참고/neo4j-graphrag-python-main/examples/customize/embeddings/ollama_embeddings.py
2026-05-12 19:40:31 +09:00

13 lines
320 B
Python

"""This example demonstrate how to embed a text into a vector
using a local model served by Ollama.
"""
from neo4j_graphrag.embeddings import OllamaEmbeddings
embeder = OllamaEmbeddings(
model="<model_name>",
# host="...", # if using a remote server
)
res = embeder.embed_query("my question")
print(res[:10])