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