|
from qdrant_client import QdrantClient |
|
from sentence_transformers import SentenceTransformer |
|
import os |
|
from dotenv import load_dotenv |
|
|
|
load_dotenv() |
|
|
|
|
|
url=os.environ.get("QDRANT_URL") |
|
api_key=os.environ.get("QDRANT_API_KEY") |
|
qdrant_client = QdrantClient( |
|
url=url, |
|
api_key=api_key, |
|
timeout=30.0 |
|
) |
|
|
|
|
|
model = SentenceTransformer("all-MiniLM-L6-v2") |
|
|
|
|
|
collection_name = "symptoms" |
|
|
|
def get_similar_symptoms(symptom): |
|
|
|
vector = model.encode(symptom) |
|
|
|
|
|
results = qdrant_client.search( |
|
collection_name=collection_name, |
|
query_vector=vector.tolist(), |
|
limit=5, |
|
with_payload=True |
|
) |
|
|
|
|
|
similar_symptoms = [result.payload["name"] for result in results] |
|
return similar_symptoms |
|
|