From 3e67e21d38927e929ebf43a35237075354e3dfe1 Mon Sep 17 00:00:00 2001
From: Pablo Orgaz <pabloogc@gmail.com>
Date: Thu, 25 Jan 2024 10:55:32 +0100
Subject: [PATCH] Add embedding mode config (#1541)

---
 settings-docker.yaml | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/settings-docker.yaml b/settings-docker.yaml
index 49b3961..6b915cc 100644
--- a/settings-docker.yaml
+++ b/settings-docker.yaml
@@ -5,6 +5,9 @@ server:
 llm:
   mode: ${PGPT_MODE:mock}
 
+embedding:
+  mode: ${PGPT_MODE:sagemaker}
+
 local:
   llm_hf_repo_id: ${PGPT_HF_REPO_ID:TheBloke/Mistral-7B-Instruct-v0.1-GGUF}
   llm_hf_model_file: ${PGPT_HF_MODEL_FILE:mistral-7b-instruct-v0.1.Q4_K_M.gguf}
@@ -16,4 +19,4 @@ sagemaker:
 
 ui:
   enabled: true
-  path: /
\ No newline at end of file
+  path: /