from dotenv import load_dotenv
import os
import sys

from openai import OpenAI

import tiktoken

EMBEDDING_MODEL = os.getenv("EMBEDDING_MODEL")
GPT_MODEL = os.getenv("GPT_MODEL")

 
def get_openai_client():
    load_dotenv() 
    api_key = os.getenv("OPENAI_API_KEY")
    if not api_key:
        raise ValueError("API Key is not set in the environment variables.")
    return OpenAI(api_key=api_key)


def openai_chat(query):
    client = get_openai_client()
    response = client.chat.completions.create(
        messages=[
            {'role': 'user', 'content': query},
        ],
        model=GPT_MODEL,
        temperature=0,
    )

    return response.choices[0].message.content

 