core/resources/azure_openai.py

from langchain.embeddings import OpenAIEmbeddings
from core.environments import EMBEDDING_MODEL, COMPLETIONS_MODEL
import openai


def generate_response(messages: list):
    chat_response = openai.ChatCompletion.create(
       #deployment_id="chatbot", 
       model=COMPLETIONS_MODEL, messages=messages
    )
    print(chat_response)
    return chat_response["choices"][0]["message"]["content"].strip(" \n")

def get_embeddings():
    return OpenAIEmbeddings(model=EMBEDDING_MODEL)