desde openai importar OpenAI
cliente = OpenAI (
base_url="https://yossapi.com/v1",
api_key=clave
)
respuesta = cliente.incrustaciones.create(
input="Tu cadena de texto va aquí",
modelo="incrustación-de-texto-3-pequeño"
)
imprimir(respuesta.datos[0].incrustación){
"objeto": "lista",
"datos": [
{
"objeto": "incrustación",
"índice": 0,
"incrustar": [
-0.006929283495992422,
-0.005336422007530928,
// ...más valores
],
}
],
"modelo": "incrustación de texto-3-pequeño",
"uso": {
"tokens_prompt": 5,
"total_tokens": 5
}
}| Modelo | Páginas por dólar | Evaluación de Desempeño MTEB | Entrada máxima |
|---|---|---|---|
| incrustación-de-texto-3-pequeño | 62.500 | 62,3% | 8191 |
| incrustación de texto-3-grande | 9.615 | 64,6% | 8191 |
| incrustación-de-texto-ada-002 | 12.500 | 61,0% | 8191 |
desde openai importar OpenAI
cliente = OpenAI (
base_url="https://yossapi.com/v1",
api_key=clave
)
def get_embedding(texto, modelo="text-embedding-3-small"):
texto = texto.reemplazar("\n", " ")
devolver client.embeddings.create(entrada = [texto], modelo=modelo).data[0].embedding
# Procesar el marco de datos
df['ada_embedding'] = df.combined.apply(lambda x: get_embedding(x, model='text-embedding-3-small'))
df.to_csv('salida/embedded_1k_reviews.csv', índice=Falso)
# Cargar incrustado guardado
importar pandas como pd
importar numpy como np
df = pd.read_csv('salida/embedded_1k_reviews.csv')
df['ada_embedding'] = df.ada_embedding.apply(eval).apply(np.array)