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 | ec52ab29ff9826467dc3b06ab0cc3011fce931a9 (patch) | |
| tree | 9aa82d9ed3a414e794a29ed1d00b355a57b22de3 /continuedev/src | |
| parent | 86135303b04e85127b5b23128872ada3f3b609a8 (diff) | |
| download | sncontinue-ec52ab29ff9826467dc3b06ab0cc3011fce931a9.tar.gz sncontinue-ec52ab29ff9826467dc3b06ab0cc3011fce931a9.tar.bz2 sncontinue-ec52ab29ff9826467dc3b06ab0cc3011fce931a9.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 |