File size: 276 Bytes
79eca0e
 
 
 
 
 
 
9288345
1
2
3
4
5
6
7
8
9
import numpy as np
from models.model_loader import model

def generate_embeddings(messages: list):
    # Generate embeddings
    embeddings = model.encode(messages)
    embeddings = np.array(embeddings).tolist()  # Convert to list for JSON serialization
    return embeddings