diviner/service/agent/init.py

# -*- coding: utf-8 -*-
from .__load__ import *
from langchain_huggingface import HuggingFaceEmbeddings
class Init(object):

    def __init__(self):
        model_name = "BAAI/bge-small-zh-v1.5"
        model_kwargs = {"device": "cpu"}
        encode_kwargs = {"normalize_embeddings": True}
        Demeter.embedding = HuggingFaceEmbeddings(
            model_name=model_name, model_kwargs=model_kwargs, encode_kwargs=encode_kwargs
        )