Function get_openai_embed

llm/embedding.py:26–41

(model, input)

    interval=20
)
def get_openai_embed(model, input):
    openai.api_key = CONFIG['emb_api_key']
    openai.api_type = "azure"
    openai.api_base = "https://gpt-jp-az.openai.azure.com/"
    openai.api_version = "2023-05-15"
    openai.debug = True
    with requests.Session() as session:
        openai.requestssession = session
        try:
            response = openai.Embedding.create(
                engine=model, input=input)
            result = [e['embedding'] for e in response['data']]
        except Exception as e:
            result = 'error:{}'.format(e)
            print(result)
    return result


if __name__ == '__main__':