diff options
Diffstat (limited to 'continuedev/src')
| -rw-r--r-- | continuedev/src/continuedev/libs/llm/openai.py | 7 | ||||
| -rw-r--r-- | continuedev/src/continuedev/libs/llm/proxy_server.py | 3 |
2 files changed, 8 insertions, 2 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): diff --git a/continuedev/src/continuedev/libs/llm/proxy_server.py b/continuedev/src/continuedev/libs/llm/proxy_server.py index f75788d2..4ff57101 100644 --- a/continuedev/src/continuedev/libs/llm/proxy_server.py +++ b/continuedev/src/continuedev/libs/llm/proxy_server.py @@ -16,7 +16,8 @@ CHAT_MODELS = { "gpt-3.5-turbo", "gpt-4" } -SERVER_URL = "http://127.0.0.1:8002" +# SERVER_URL = "http://127.0.0.1:8002" +SERVER_URL = "https://proxy-server-l6vsfbzhba-uc.a.run.app" class ProxyServer(LLM): |