from openai import OpenAI


class LLMInterface:
    def __init__(self, personality):
        api_key = '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'
        self.client = OpenAI(api_key=api_key)
        self.personality = personality

    def get_reponse(self, query):
        completion = self.client.chat.completions.create(
            model="gpt-4o-mini",
            messages=[
                {"role": "system", "content": self.personality},
                {"role": "user", "content": query}
            ]
        )
        return completion.choices[0].message.content