diff options
author | Nate Sesti <sestinj@gmail.com> | 2023-06-25 16:51:25 -0700 |
---|---|---|
committer | Nate Sesti <sestinj@gmail.com> | 2023-06-25 16:51:25 -0700 |
commit | 442f3d9aa5f3f1773b90733b12b369f34dd6be92 (patch) | |
tree | 3cf4bb2cd30c7f0528aa02cf55a8401457c0132d /continuedev/src | |
parent | d1e509b1c8a182f280348283fbc1c19d231add70 (diff) | |
download | sncontinue-442f3d9aa5f3f1773b90733b12b369f34dd6be92.tar.gz sncontinue-442f3d9aa5f3f1773b90733b12b369f34dd6be92.tar.bz2 sncontinue-442f3d9aa5f3f1773b90733b12b369f34dd6be92.zip |
remove python system message
Diffstat (limited to 'continuedev/src')
-rw-r--r-- | continuedev/src/continuedev/libs/llm/openai.py | 3 | ||||
-rw-r--r-- | continuedev/src/continuedev/libs/llm/proxy_server.py | 3 |
2 files changed, 2 insertions, 4 deletions
diff --git a/continuedev/src/continuedev/libs/llm/openai.py b/continuedev/src/continuedev/libs/llm/openai.py index 7621111f..f0b2e6d8 100644 --- a/continuedev/src/continuedev/libs/llm/openai.py +++ b/continuedev/src/continuedev/libs/llm/openai.py @@ -14,8 +14,7 @@ class OpenAI(LLM): def __init__(self, api_key: str, default_model: str, system_message: str = None): self.api_key = api_key self.default_model = default_model - self.system_message = (system_message or "") + \ - "\nDo not ever call the 'python' function." + self.system_message = system_message openai.api_key = api_key diff --git a/continuedev/src/continuedev/libs/llm/proxy_server.py b/continuedev/src/continuedev/libs/llm/proxy_server.py index 6b0336a7..d6eac3c9 100644 --- a/continuedev/src/continuedev/libs/llm/proxy_server.py +++ b/continuedev/src/continuedev/libs/llm/proxy_server.py @@ -18,8 +18,7 @@ class ProxyServer(LLM): def __init__(self, unique_id: str, default_model: Literal["gpt-3.5-turbo", "gpt-4"], system_message: str = None): self.unique_id = unique_id self.default_model = default_model - self.system_message = (system_message or "") + \ - "\nDo not ever call the 'python' function." + self.system_message = system_message self.name = default_model @property |