上代码
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