ThaparGPT / get_embedding_function.py
0504ankitsharma's picture
Upload 9 files
c47212f verified
raw
history blame contribute delete
378 Bytes
from langchain_community.embeddings.ollama import OllamaEmbeddings
from langchain_community.embeddings.bedrock import BedrockEmbeddings
#
def get_embedding_function():
# embeddings = BedrockEmbeddings(
# credentials_profile_name="default", region_name="us-east-1"
# )
embeddings = OllamaEmbeddings(model="nomic-embed-text")
return embeddings