From 75cab46be2eba5956bfa4a3a9832fffcc80b150a Mon Sep 17 00:00:00 2001
From: Jordan <jordan.q.w@hotmail.com>
Date: Thu, 28 Sep 2023 22:13:45 +1000
Subject: [PATCH] Update run_localGPT.py add CHROMA_SETTINGS to Chroma

---
 run_localGPT.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/run_localGPT.py b/run_localGPT.py
index b6450e02..a553825d 100644
--- a/run_localGPT.py
+++ b/run_localGPT.py
@@ -26,6 +26,7 @@
 )
 
 from constants import (
+    CHROMA_SETTINGS,
     EMBEDDING_MODEL_NAME,
     PERSIST_DIRECTORY,
     MODEL_ID,
@@ -122,6 +123,7 @@ def retrieval_qa_pipline(device_type, use_history, promptTemplate_type="llama"):
     db = Chroma(
         persist_directory=PERSIST_DIRECTORY,
         embedding_function=embeddings,
+        client_settings=CHROMA_SETTINGS,
     )
     retriever = db.as_retriever()