From f85a28edbdeac4ea07297ea4ca4f0683d485e8ae Mon Sep 17 00:00:00 2001 From: Luis Martinez Date: Sun, 21 May 2023 10:33:53 +0200 Subject: [PATCH] fix: use env var system message --- src/chatgpt.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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 }]