diff options
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 |