File size: 829 Bytes
e60c070 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import os
import sys
sys.path.append(os.path.dirname(os.path.dirname(__file__)))
from embedding.zhipuai_embedding import ZhipuAIEmbeddings
from langchain.embeddings.huggingface import HuggingFaceEmbeddings
from langchain.embeddings.openai import OpenAIEmbeddings
from llm.call_llm import parse_llm_api_key
def get_embedding(embedding: str, embedding_key: str=None, env_file: str=None):
if embedding == 'm3e':
return HuggingFaceEmbeddings(model_name="moka-ai/m3e-base")
if embedding_key is None:
embedding_key = parse_llm_api_key(embedding)
if embedding == "openai":
return OpenAIEmbeddings(openai_api_key=embedding_key)
elif embedding == "zhipuai":
return ZhipuAIEmbeddings(zhipuai_api_key=embedding_key)
else:
raise ValueError(f"embedding {embedding} not support ")
|