通过 OpenAI Embedding 接口计算相似度

上代码

from openai import OpenAI
import numpy as np

client = OpenAI(
    base_url="https://api.ikuaiai.com/v1",  # 这里是一个中转base_url 国内可用
    api_key='你的key',
    timeout=120
)

def get_embedding(text, model="text-embedding-3-small"):
   text = text.replace("\n", " ")
   return client.embeddings.create(input = [text], model=model).data[0].embedding

def get_similarity(text1, text2):
    embedding1 = get_embedding(text1)
    embedding2 = get_embedding(text2)
    cosine_sim = np.dot(embedding1, embedding2) / (np.linalg.norm(embedding1) * np.linalg.norm(embedding2))
    return cosine_sim

# 定义要比较的两段文本
text1 = "The cat jumped over the lazy dog."
text2 = "The quick brown fox jumps over the lazy dog."

embedding1 = get_embedding(text1)
embedding2 = get_embedding(text2)

# 计算两个向量的余弦相似度
cosine_sim = np.dot(embedding1, embedding2) / (np.linalg.norm(embedding1) * np.linalg.norm(embedding2))

# 输出相似度
print("相似度为：", cosine_sim)

OpenAI 国内中转API 推荐

快AI，有10万免费额度

官网：https://ikuaiai.com
使用文档：https://ikuaiai.com/about