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