From 5193860ee94cae6329d1518a4088f77b19721466 Mon Sep 17 00:00:00 2001
From: fanpt <320622572@qq.com>
Date: Wed, 6 Mar 2024 11:27:28 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=E4=B8=BB=E6=B5=81=E7=A8=8B?=
 =?UTF-8?q?=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 qa_Ask.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/qa_Ask.py b/qa_Ask.py
index 42ddb7c..9cf3479 100644
--- a/qa_Ask.py
+++ b/qa_Ask.py
@@ -1,17 +1,27 @@
 # qa_operations.py
 import json
-
 from sentence_transformers import CrossEncoder
 from faiss_kb_service import FaissKBService
 from langchain.docstore.document import Document
 from base_kb import KnowledgeFile
 from pydantic import BaseModel
 
+
+import yaml
+
+# Read the configuration file
+with open('config/config.yaml', 'r') as config_file:
+    config = yaml.safe_load(config_file)
+
+# Access the 'bge-large-zh-v1.5' configuration
+bge_large_zh_v1_5_config = config.get('bge_large_zh_v1_5', {})
+embed_model_path = bge_large_zh_v1_5_config.get('embed_model_path', 'default_path_if_not_provided')
+
 class QAService():
     def __init__(self, kb_name, device=None) -> None:
         self.kb_name = kb_name
         self.device = device
-        self.fkbs = FaissKBService(kb_name, embed_model_path='bge-large-zh-v1.5', device=device)
+        self.fkbs = FaissKBService(kb_name, embed_model_path=embed_model_path, device=device)
         self.fkbs.do_create_kb()
 
     def update_qa_doc(self, qa_file_id, doc_list, id_list):