from openai import OpenAI | |
with open('.openai-secret', 'r') as f: OPENAI_API_KEY = f.read() | |
class QueryEmbedder: | |
def __init__(self) -> None: | |
self.client = OpenAI(api_key=OPENAI_API_KEY) | |
def embed_query(self, query): | |
res = self.client.embeddings.create( | |
model="text-embedding-3-small", | |
input=query[0:8000], | |
encoding_format="float" | |
) | |
return res.dict()['data'][0]['embedding'] |