Spaces:
Sleeping
Sleeping
File size: 832 Bytes
3a2c9d3 73c8596 3a2c9d3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | import os
from dotenv import load_dotenv
from app.schemas.request_models import QuerySpec
from typing import Union, List
def get_query_embedding(query_spec: QuerySpec, embedding_model):
# load_dotenv()
# model_loader = ModelLoader(model_provider="openai")
# embedding_client = model_loader.load_llm()
q = query_spec.raw_query
e_main = embedding_model.embed_query(q)
expansions = []
# if "procedure" in query_spec.entities:
# procedure_value = query_spec.entities['procedure']
# # Handle both string and list values
# if isinstance(procedure_value, list):
# procedure_str = ", ".join(procedure_value)
# else:
# procedure_str = procedure_value
# expansions.append(f"{q} OR {procedure_str} procedures related")
return e_main,expansions
|