openai_embed.py

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']