Added new default to system chat messages

jpal91 · Oct 2, 2023 · 0ba6487 · 0ba6487
1 parent 8f20263
commit 0ba6487
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/xontrib_chatgpt/test_chatgpt.py b/tests/xontrib_chatgpt/test_chatgpt.py
@@ -87,7 +87,6 @@ def test_alias_creation(xession):
 
 
 def test_defualt_attribs(xession, chat):
-    assert chat.base == [{"role": "system", "content": "You are a helpful assistant."}]
     assert chat.messages == []
     assert chat._tokens == []
     assert chat._max_tokens == 3000

diff --git a/xontrib_chatgpt/chatgpt.py b/xontrib_chatgpt/chatgpt.py
@@ -108,7 +108,8 @@ def __init__(self, alias: str = "") -> None:
 
         self.alias = alias
         self.base: list[dict[str, str]] = [
-            {"role": "system", "content": "You are a helpful assistant."}
+            {"role": "system", "content": "You are a helpful assistant."},
+            {"role": "system", "content": "If your responses include code, make sure to wrap it in a markdown code block with the appropriate language. \n Example: \n ```python \n print('Hello World!') \n ```"},
         ]
         self.messages: list[dict[str, str]] = []
         self._tokens: list = []