diff --git a/src/OpenAIClient.py b/src/OpenAIClient.py index 44c72d9..014f0ff 100644 --- a/src/OpenAIClient.py +++ b/src/OpenAIClient.py @@ -1,5 +1,6 @@ from openai import OpenAI import SystemMsg +from dotenv import load_dotenv class OpenAIClient: @@ -7,6 +8,7 @@ def __init__(self): """ Initializes the OpenAIClient with the API key from the environment and creates the OpenAI client. """ + load_dotenv() self.client = OpenAI() self.messages = [] @@ -47,4 +49,4 @@ def clear_messages(self): """ Clears the list of messages. """ - self.messages = [] \ No newline at end of file + self.messages = []