from langchain_openai import OpenAIEmbeddings
from openai import OpenAI

from env_util import OPENAI_BASE_URL, OPENAI_API_KEY



# 这里采用Langchain的API访问,模型还是OpenAI的模型
# 1、 拿到OpenAI的Client
llm = OpenAIEmbeddings(
    base_url=OPENAI_BASE_URL,
    api_key=OPENAI_API_KEY,
    model='text-embedding-3-large',
    dimensions=512,  # 向量的维度~~
);

# 2、 调用Embedding模型，获取text的向量
text = "i like LLM.";
resp = llm.embed_query(text);

# 3、输出结果
print(resp)
# print(len(resp))