diff --git a/src/chatgpt.py b/src/chatgpt.py index 4b532ab..26b582e 100644 --- a/src/chatgpt.py +++ b/src/chatgpt.py @@ -1,3 +1,4 @@ +import os from src.models import ModelInterface @@ -7,7 +8,7 @@ def __init__(self, model: ModelInterface): def get_response(self, text: str) -> str: messages = [{ - 'role': 'system', 'content': 'You are a helpful assistant.' + 'role': 'system', 'content': os.getenv('SYSTEM_MESSAGE') }, { 'role': 'user', 'content': text }]