langchain调用chatgpt对文本进行编码

科研小达人2024-11-02 23:31

1.导包

python 复制代码

from langchain_openai import OpenAIEmbeddings

2.加载编码器

python 复制代码

embeddings_model = OpenAIEmbeddings(model="text-embedding-3-large",base_url="https://api.chatanywhere.tech/v1")

3.编码

python 复制代码

embeded_result = embeddings_model.embed_documents(["Hello world!", "Hey bro"])

4.编码长度

python 复制代码

len(embeded_result)

输出：2

5.输出编码结果

python 复制代码

embeded_result

6.编码第一个位置长度

python 复制代码

len(embeded_result[0])

输出：3072

7.更改输出长度

python 复制代码

# 如果希望嵌入向量维度更小，可以通过dimensions参数进行指定
embeddings_model = OpenAIEmbeddings(model="text-embedding-3-large",base_url="https://api.chatanywhere.tech/v1",dimensions=1024)
embeded_result = embeddings_model.embed_documents(["Hello world!", "Hey bro"])
len(embeded_result[0])

输出：1024