From fdd4ed6420a820622c530a19277ff811f5584a69 Mon Sep 17 00:00:00 2001
From: Javier Martinez <javiermartinezalvarez98@gmail.com>
Date: Wed, 24 Jul 2024 09:14:41 +0200
Subject: [PATCH] chore: update to use ollama in docker-compose

---
 docker-compose.yaml  | 1 +
 settings-docker.yaml | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/docker-compose.yaml b/docker-compose.yaml
index f12e1da..43dcc35 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -10,6 +10,7 @@ services:
       PORT: 8080
       PGPT_PROFILES: docker
       PGPT_MODE: ollama
+      PGPT_EMBED_MODE: ollama
   ollama:
     image: ollama/ollama:latest
     volumes:
diff --git a/settings-docker.yaml b/settings-docker.yaml
index 8e218f1..c0bcc93 100644
--- a/settings-docker.yaml
+++ b/settings-docker.yaml
@@ -6,7 +6,7 @@ llm:
   mode: ${PGPT_MODE:mock}
 
 embedding:
-  mode: ${PGPT_EMBEDDING_MODE:sagemaker}
+  mode: ${PGPT_EMBED_MODE:mock}
 
 llamacpp:
   llm_hf_repo_id: ${PGPT_HF_REPO_ID:TheBloke/Mistral-7B-Instruct-v0.1-GGUF}