sentence-transformers How to use krishanusinha20/multi-agentic-sql-generator-model with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("krishanusinha20/multi-agentic-sql-generator-model")
sentences = [
"Find the most popular payment method used in 2024.",
"SELECT * FROM orders WHERE customer_id = 42;",
"SELECT customer_id, COUNT(order_id) AS order_count FROM orders WHERE order_date BETWEEN '2024-01-01' AND '2024-12-31' GROUP BY customer_id HAVING order_count >= 3;",
"SELECT payment_method, COUNT(*) AS usage_count FROM payments WHERE payment_date BETWEEN '2024-01-01' AND '2024-12-31' GROUP BY payment_method ORDER BY usage_count DESC LIMIT 1;"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
