ClariDoc / app /embedding /embeder.py
Kshitijk20's picture
code push
e5b884f
raw
history blame contribute delete
336 Bytes
import os
from dotenv import load_dotenv
from typing import Union, List
class QueryEmbedding:
def __init__(self, query: str, embedding_model):
self.query = query
self.embedding_model = embedding_model
def get_embedding(self):
e_main = self.embedding_model.embed_query(self.query)
return e_main