diff options
| author | Tuowen Zhao <ztuowen@gmail.com> | 2023-10-19 00:04:44 -0700 |
|---|---|---|
| committer | Tuowen Zhao <ztuowen@gmail.com> | 2023-10-19 00:04:44 -0700 |
| commit | 2128f5fe9386dcf2f0597c8035f951c5b60d7562 (patch) | |
| tree | ac3ab65a87bd4971275ae91d7b61176eced13774 /server/tests/util/config.py | |
| parent | 08f38574fa2633bbf709d24e1c79417d4285ba61 (diff) | |
| download | sncontinue-2128f5fe9386dcf2f0597c8035f951c5b60d7562.tar.gz sncontinue-2128f5fe9386dcf2f0597c8035f951c5b60d7562.tar.bz2 sncontinue-2128f5fe9386dcf2f0597c8035f951c5b60d7562.zip | |
cleanup server
Diffstat (limited to 'server/tests/util/config.py')
| -rw-r--r-- | server/tests/util/config.py | 15 |
1 files changed, 9 insertions, 6 deletions
diff --git a/server/tests/util/config.py b/server/tests/util/config.py index 370933a0..82811bc1 100644 --- a/server/tests/util/config.py +++ b/server/tests/util/config.py @@ -1,15 +1,18 @@ from continuedev.core.config import ContinueConfig from continuedev.core.models import Models -from continuedev.libs.llm.openai_free_trial import OpenAIFreeTrial +from continuedev.libs.llm import Ollama config = ContinueConfig( allow_anonymous_telemetry=False, models=Models( - default=OpenAIFreeTrial(api_key="", model="gpt-4"), - summarize=OpenAIFreeTrial( - api_key="", - model="gpt-3.5-turbo", - ), + default=Ollama( + title="CodeLlama-7b-Instruct", + model="codellama:7b-instruct" + ), + summarize=Ollama( + title="CodeLlama-7b-Instruct", + model="codellama:7b-instruct" + ) ), system_message=None, temperature=0.5, |