/
spirzen
/
it-code-examples
Обзор
Документация
Войти
/
spirzen
/
it-code-examples
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
examples/python/lab-1149-006/main.py
26 строк
802 B
Spirzen
Code migration pack
09 июн 2026, 01:41
09 июн 2026, 01:41
cebc284
Код
Авторство
О чём код?
import numpy as np from openai import OpenAI client = OpenAI() def cosine(a, b): a, b = np.array(a), np.array(b) return float(np.dot(a, b) / (np.linalg.norm(a) * np.linalg.norm(b))) texts = [ "Как вызвать Chat Completions API", "REST-запрос к языковой модели", "Рецепт борща на кастрюлю", ] emb = client.embeddings.create(model="text-embedding-3-small", input=texts) vectors = [item.embedding for item in emb.data] query_vec = client.embeddings.create( model="text-embedding-3-small", input="Пример HTTP-вызова OpenAI", ).data[0].embedding scores = [(t, cosine(query_vec, v)) for t, v in zip(texts, vectors)] for title, score in sorted(scores, key=lambda x: -x[1]): print(f"{score:.3f} — {title}")