diff options
| author | Nate Sesti <sestinj@gmail.com> | 2023-06-15 13:07:47 -0700 |
|---|---|---|
| committer | Nate Sesti <sestinj@gmail.com> | 2023-06-15 13:07:47 -0700 |
| commit | 096628f5486815fff7a0ee2c03651f5991c6c2b1 (patch) | |
| tree | 046e6f64111c651307210aafb9cdde339a6acfa9 /continuedev | |
| parent | 00ffcc779571ee876c478c680ea1590a455d0d70 (diff) | |
| download | sncontinue-096628f5486815fff7a0ee2c03651f5991c6c2b1.tar.gz sncontinue-096628f5486815fff7a0ee2c03651f5991c6c2b1.tar.bz2 sncontinue-096628f5486815fff7a0ee2c03651f5991c6c2b1.zip | |
openai init
Diffstat (limited to 'continuedev')
| -rw-r--r-- | continuedev/src/continuedev/libs/llm/openai.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/continuedev/src/continuedev/libs/llm/openai.py b/continuedev/src/continuedev/libs/llm/openai.py index 8fec2046..2b4c3fad 100644 --- a/continuedev/src/continuedev/libs/llm/openai.py +++ b/continuedev/src/continuedev/libs/llm/openai.py @@ -22,7 +22,12 @@ CHAT_MODELS = { class OpenAI(LLM): api_key: str completion_count: int = 0 - default_model: str = "gpt-3.5-turbo" + default_model: str + + def __init__(self, api_key: str, default_model: str = "gpt-3.5-turbo", system_message: str = None): + self.api_key = api_key + self.default_model = default_model + self.system_message = system_message @validator("api_key", pre=True, always=True) def validate_api_key(cls, v): |