15 lines
552 B
Python
15 lines
552 B
Python
from typing import List
|
|
from openai import OpenAI
|
|
import os
|
|
def get_embeddings(embedding_texts: List[str]):
|
|
response = OpenAI(
|
|
api_key=os.getenv("OPENAI_API_KEY","EMPTY"),
|
|
base_url=os.getenv("OPENAI_API_EMBEDDING_BASE","http://172.18.30.122:40060/v1"),
|
|
).embeddings.create(
|
|
model=os.getenv("OPENAI_EMBEDDING_MODEL","bge-m3"),
|
|
input=embedding_texts,
|
|
)
|
|
emebdding_list = response.data[0].embedding
|
|
return emebdding_list
|
|
|
|
print(get_embeddings(["你好","世界"])) |