Spaces:
Sleeping
Sleeping
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
|