File size: 336 Bytes
e5b884f
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
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