feature(#9): updated all the end points with firebase & pinecone configuration

Brain/src/common/brain_exception.py

@@ -7,10 +7,11 @@ from Brain.src.common.http_response_codes import responses
 class BrainException(Exception):
     JSON_PARSING_ISSUE_MSG = "Exception occurred in json paring."
 
-    def __init__(self, message: str = "Exception occurred in brain"):
+    def __init__(self, message: str = "Exception occurred in brain", code: int = 506):
         self.message = message
+        self.code = code
         super().__init__(self.message)
 
     def get_response_exp(self) -> Any:
-        responses[506] = ("Brain Exception", self.message)
-        return {"message": responses[506], "result": "", "status_code": 506}
+        responses[self.code] = ("Brain Exception", self.message)
+        return {"message": responses[self.code], "result": "", "status_code": self.code}

Brain/src/common/http_response_codes.py

@@ -49,4 +49,6 @@ responses = {
     504: ("Gateway Timeout", "The gateway server did not receive a timely response"),
     505: ("HTTP Version Not Supported", "Cannot fulfill request."),
     506: ("Brain Exception"),
+    507: ("Firebase Connection Error", "Your key is not correct."),
+    508: ("Pinecone Connection Error", "Your key or env is not correct."),
 }

Brain/src/common/utils.py

@@ -2,11 +2,8 @@ import json
 import os
 import re
 
-from firebase_admin import credentials
-from Brain.src.model.req_model import ReqModel
 
 # env variables
-DEFAULT_HOST_NAME = "test3-83ffc.appspot.com"
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 PINECONE_KEY = os.getenv("PINECONE_KEY")
 PINECONE_ENV = os.getenv("PINECONE_ENV")
@@ -19,7 +16,7 @@ API_URL = "http://localhost:5000/file/swagger.json"
 
 # firebase
 FIREBASE_STORAGE_ROOT = "images/"
-FIREBASE_STORAGE_BUCKET = DEFAULT_HOST_NAME
+FIREBASE_STORAGE_BUCKET = "test3-83ffc.appspot.com"
 
 # pinecone
 PINECONE_NAMESPACE = "risinglangchain-namespace"
@@ -42,17 +39,6 @@ AUTH_TOKEN = os.getenv("TWILIO_AUTH_TOKEN")
 HUGGINGFACEHUB_API_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
 
 
-def get_firebase_cred(setting: ReqModel):
-    if os.path.exists("Brain/firebase_cred.json"):
-        file = open("Brain/firebase_cred.json")
-        cred = json.load(file)
-        file.close()
-        return credentials.Certificate(cred)
-    else:
-        cred = json.loads(setting.firebase_key)
-        return credentials.Certificate(cred)
-
-
 class ProgramType:
     BROWSER = "browser"
     ALERT = "alert"

Brain/src/firebase/firebase.py

@@ -1,9 +1,14 @@
+import json
+import os
 from typing import Any
 
 import firebase_admin
+from firebase_admin import credentials
 
 from Brain.src.common.assembler import Assembler
-from Brain.src.common.utils import get_firebase_cred, FIREBASE_STORAGE_BUCKET
+from Brain.src.common.brain_exception import BrainException
+from Brain.src.common.http_response_codes import responses
+from Brain.src.common.utils import FIREBASE_STORAGE_BUCKET
 from Brain.src.logs import logger
 from Brain.src.model.req_model import ReqModel
 from Brain.src.model.requests.request_model import BasicReq
@@ -37,6 +42,19 @@ def firebase_admin_with_setting(data: BasicReq):
     # firebase admin init
     assembler = Assembler()
     setting = assembler.to_req_model(data)
-
-    firebase_app = initialize_app(setting)
+    try:
+        firebase_app = initialize_app(setting)
+    except Exception as ex:
+        raise BrainException(code=507, message=responses[507])
     return setting, firebase_app
+
+
+def get_firebase_cred(setting: ReqModel):
+    if os.path.exists("Brain/firebase_cred.json"):
+        file = open("Brain/firebase_cred.json")
+        cred = json.load(file)
+        file.close()
+        return credentials.Certificate(cred)
+    else:
+        cred = json.loads(setting.firebase_key)
+        return credentials.Certificate(cred)

Brain/src/model/req_model.py

@@ -1,6 +1,11 @@
 from typing import Any
 
-DEFAULT_HOST_NAME = "test3-83ffc.appspot.com"
+from Brain.src.common.utils import (
+    OPENAI_API_KEY,
+    PINECONE_KEY,
+    PINECONE_ENV,
+    FIREBASE_ENV,
+)
 
 
 class ReqModel:
@@ -8,7 +13,6 @@ class ReqModel:
         def __init__(self):
             self.temperature: float = 0.6
 
-    host_name: str
     openai_key: str
     pinecone_key: str
     pinecone_env: str
@@ -18,20 +22,17 @@ class ReqModel:
     settings: Settings
 
     def __init__(self, data: dict):
-        self.host_name = (
-            DEFAULT_HOST_NAME if data["host_name"] == "" else data["host_name"]
-        )
         self.openai_key = (
-            DEFAULT_HOST_NAME if data["openai_key"] == "" else data["openai_key"]
+            OPENAI_API_KEY if data["openai_key"] == "" else data["openai_key"]
         )
         self.pinecone_key = (
-            DEFAULT_HOST_NAME if data["pinecone_key"] == "" else data["pinecone_key"]
+            PINECONE_KEY if data["pinecone_key"] == "" else data["pinecone_key"]
         )
         self.pinecone_env = (
-            DEFAULT_HOST_NAME if data["pinecone_env"] == "" else data["pinecone_env"]
+            PINECONE_ENV if data["pinecone_env"] == "" else data["pinecone_env"]
         )
         self.firebase_key = (
-            DEFAULT_HOST_NAME if data["firebase_key"] == "" else data["firebase_key"]
+            FIREBASE_ENV if data["firebase_key"] == "" else data["firebase_key"]
         )
         self.token = data["token"]
         self.uuid = data["uuid"]
@@ -40,7 +41,6 @@ class ReqModel:
 
     def to_json(self):
         return {
-            "host_name": self.host_name,
             "openai_key": self.openai_key,
             "pinecone_key": self.pinecone_key,
             "pinecone_env": self.pinecone_env,

Brain/src/model/requests/request_model.py

@@ -70,7 +70,6 @@ class BasicReq(BaseModel):
     class Settings(BaseModel):
         temperature: float = 0.6
 
-    host_name: str
     openai_key: str
     pinecone_key: str
     pinecone_env: str
@@ -81,7 +80,6 @@ class BasicReq(BaseModel):
 
     def to_json(self):
         return {
-            "host_name": self.host_name,
             "openai_key": self.openai_key,
             "pinecone_key": self.pinecone_key,
             "pinecone_env": self.pinecone_env,
@@ -138,7 +136,6 @@ class ChatRising(BasicReq):
 
     history: list[Format]
     user_input: str
-    model: str
 
 
 """endpoint: /send_sms"""

Brain/src/rising_plugin/guardrails-config/actions/actions.py

@@ -104,7 +104,7 @@ async def general_question(query):
         if result["program"] == "image":
             if image_search:
                 result["content"] = {
-                    "image_name": query_image_text(result["content"], "", setting.uuid)
+                    "image_name": query_image_text(result["content"], "", setting)
                 }
         """ 2. check program is message to handle it with falcon llm """
         if result["program"] == "message":

Brain/src/rising_plugin/image_embedding.py

@@ -9,6 +9,7 @@ from .pinecone_engine import (
 )
 from ..model.basic_model import DataStatus
 from ..model.image_model import ImageModel
+from ..model.req_model import ReqModel
 
 
 def get_embeddings():
@@ -16,7 +17,7 @@ def get_embeddings():
 
 
 # def embed_image_text(image_text: str, image_name: str, uuid: str) -> str:
-def embed_image_text(image: ImageModel) -> str:
+def embed_image_text(image: ImageModel, setting: ReqModel) -> str:
     prompt_template = f"""
         This is the text about the image.
         ###
@@ -24,7 +25,7 @@ def embed_image_text(image: ImageModel) -> str:
         """
 
     embed_image = get_embeddings().embed_query(prompt_template)
-    index = init_pinecone(PINECONE_INDEX_NAME)
+    index = init_pinecone(index_name=PINECONE_INDEX_NAME, setting=setting)
 
     """create | update | delete in pinecone"""
     pinecone_namespace = get_pinecone_index_namespace(image.uuid)
@@ -46,16 +47,16 @@ def embed_image_text(image: ImageModel) -> str:
     return "success to embed image text"
 
 
-def query_image_text(image_content, message, uuid):
+def query_image_text(image_content, message, setting: ReqModel):
     embed_image = get_embeddings().embed_query(
         get_prompt_image_with_message(image_content, message)
     )
-    index = init_pinecone(PINECONE_INDEX_NAME)
+    index = init_pinecone(index_name=PINECONE_INDEX_NAME, setting=setting)
     relatedness_data = index.query(
         vector=embed_image,
         top_k=3,
         include_values=False,
-        namespace=get_pinecone_index_namespace(uuid),
+        namespace=get_pinecone_index_namespace(setting.uuid),
     )
     if len(relatedness_data["matches"]) > 0:
         return relatedness_data["matches"][0]["id"]

Brain/src/rising_plugin/pinecone_engine.py

@@ -1,6 +1,9 @@
 # initialize pinecone
 import pinecone
 from typing import Any
+
+from ..common.brain_exception import BrainException
+from ..common.http_response_codes import responses
 from ..common.utils import (
     PINECONE_INDEX_NAME,
     PINECONE_NAMESPACE,
@@ -14,14 +17,17 @@ POD_TYPE = "p1.x1"
 
 # get the existing index in pinecone or create a new one
 def init_pinecone(index_name, setting: ReqModel, flag=True):
-    pinecone.init(api_key=setting.pinecone_key, environment=setting.pinecone_env)
-    if flag:
-        return pinecone.Index(index_name)
-    else:
-        # create a new index in pinecone
-        return pinecone.create_index(
-            index_name, dimension=DIMENSION, metric=METRIC, pod_type=POD_TYPE
-        )
+    try:
+        pinecone.init(api_key=setting.pinecone_key, environment=setting.pinecone_env)
+        if flag:
+            return pinecone.Index(index_name)
+        else:
+            # create a new index in pinecone
+            return pinecone.create_index(
+                index_name, dimension=DIMENSION, metric=METRIC, pod_type=POD_TYPE
+            )
+    except Exception as ex:
+        raise BrainException(code=508, message=responses[508])
 
 
 """add item in pinecone"""

Brain/src/rising_plugin/risingplugin.py

@@ -21,6 +21,7 @@ from .csv_embed import get_embed
 from .llm.llms import get_llm, GPT_4, FALCON_7B
 from .pinecone_engine import init_pinecone
 from ..common.brain_exception import BrainException
+from ..common.http_response_codes import responses
 from ..common.utils import (
     OPENAI_API_KEY,
     FIREBASE_STORAGE_ROOT,
@@ -64,13 +65,15 @@ def llm_rails(
     """step 1: handle with gpt-4"""
 
     query_result = get_embed(query)
-    relatedness_data = index.query(
-        vector=query_result,
-        top_k=1,
-        include_values=False,
-        namespace=train_service.get_pinecone_index_train_namespace(),
-    )
-
+    try:
+        relatedness_data = index.query(
+            vector=query_result,
+            top_k=1,
+            include_values=False,
+            namespace=train_service.get_pinecone_index_train_namespace(),
+        )
+    except Exception as ex:
+        raise BrainException(code=508, message=responses[508])
     if len(relatedness_data["matches"]) == 0:
         return str({"program": "message", "content": ""})
     document_id = relatedness_data["matches"][0]["id"]
@@ -215,9 +218,9 @@ def query_image_ask(image_content, message, setting: ReqModel):
     return False
 
 
-def getTextFromImage(filename):
+def getTextFromImage(filename: str, firebase_app: firebase_admin.App) -> str:
     # Create a reference to the image file you want to download
-    bucket = storage.bucket()
+    bucket = storage.bucket(app=firebase_app)
     blob = bucket.blob(FIREBASE_STORAGE_ROOT.__add__(filename))
     download_url = ""
 

Brain/src/router/api.py

@@ -48,7 +48,6 @@ def construct_blueprint_api() -> APIRouter:
     # Service
 
     command_service = CommandService()
-    contacts_service = ContactsService()
 
     """@generator.response(
         status_code=200, schema={"message": "message", "result": "test_result"}
@@ -62,7 +61,10 @@ def construct_blueprint_api() -> APIRouter:
         data: Notification, client_info: ClientInfo = Depends(get_client_info)
     ):
         # firebase admin init
-        setting, firebase_app = firebase_admin_with_setting(data)
+        try:
+            setting, firebase_app = firebase_admin_with_setting(data)
+        except BrainException as ex:
+            return ex.get_response_exp()
 
         # cloud message
         cloud_message = CloudMessage(firebase_app=firebase_app)
@@ -81,6 +83,7 @@ def construct_blueprint_api() -> APIRouter:
 
         # check contact querying
         try:
+            contacts_service = ContactsService(setting=setting)
             if result["program"] == ProgramType.CONTACT:
                 # querying contacts to getting its expected results
                 contacts_results = contacts_service.query_contacts(
@@ -114,25 +117,33 @@ def construct_blueprint_api() -> APIRouter:
     @router.post("/uploadImage")
     def upload_image(data: UploadImage):
         # firebase admin init
-        setting, firebase_app = firebase_admin_with_setting(data)
+        try:
+            setting, firebase_app = firebase_admin_with_setting(data)
+        except BrainException as ex:
+            return ex.get_response_exp()
         # cloud message
-        cloud_message = CloudMessage(firebase_app=firebase_app)
+        try:
+            cloud_message = CloudMessage(firebase_app=firebase_app)
 
-        image_model = ImageModel()
-        token = data.token
+            image_model = ImageModel()
+            token = data.token
 
-        image_model.image_name = data.image_name
-        image_model.uuid = data.uuid
-        image_model.status = data.status
+            image_model.image_name = data.image_name
+            image_model.uuid = data.uuid
+            image_model.status = data.status
 
-        image_model.image_text = getTextFromImage(image_model.image_name)
+            image_model.image_text = getTextFromImage(
+                filename=image_model.image_name, firebase_app=firebase_app
+            )
 
-        embed_result = embed_image_text(image_model)
+            embed_result = embed_image_text(image=image_model, setting=setting)
 
-        notification = {"title": "alert", "content": embed_result}
+            notification = {"title": "alert", "content": embed_result}
 
-        state, value = cloud_message.send_message(notification, [token])
-        return assembler.to_response(200, value, image_model.to_json())
+            state, value = cloud_message.send_message(notification, [token])
+            return assembler.to_response(200, value, image_model.to_json())
+        except BrainException as ex:
+            return ex.get_response_exp()
 
     """@generator.response(
         status_code=200, schema={"message": "message", "result": "test_result"}
@@ -149,7 +160,10 @@ def construct_blueprint_api() -> APIRouter:
     @router.post("/image_relatedness")
     def image_relatedness(data: ImageRelatedness):
         # firebase admin init
-        setting, firebase_app = firebase_admin_with_setting(data)
+        try:
+            setting, firebase_app = firebase_admin_with_setting(data)
+        except BrainException as ex:
+            return ex.get_response_exp()
 
         # cloud message
         cloud_message = CloudMessage(firebase_app=firebase_app)
@@ -159,7 +173,7 @@ def construct_blueprint_api() -> APIRouter:
         token = data.token
         uuid = data.uuid
 
-        image_content = getTextFromImage(image_name)
+        image_content = getTextFromImage(filename=image_name, firebase_app=firebase_app)
         # check message type
         image_response = {}
         try:
@@ -169,11 +183,13 @@ def construct_blueprint_api() -> APIRouter:
             ):
                 image_response["image_desc"] = image_content
             else:
-                relatedness_data = query_image_text(image_content, message, uuid)
+                relatedness_data = query_image_text(image_content, message, setting)
 
                 image_response["image_name"] = relatedness_data
         except ValueError as e:
             print("image_relatedness parsing error for message chain data")
+            if isinstance(e, BrainException):
+                return e.get_response_exp()
 
         notification = {"title": "alert", "content": json.dumps(image_response)}
         state, value = cloud_message.send_message(notification, [token])
@@ -222,6 +238,8 @@ def construct_blueprint_api() -> APIRouter:
             # add the feedback
             feedback_service.add(feedback)
         except Exception as e:
+            if isinstance(e, BrainException):
+                return e.get_response_exp()
             return assembler.to_response(400, "failed to add", "")
         return assembler.to_response(200, "added successfully", "")
 
@@ -232,7 +250,10 @@ def construct_blueprint_api() -> APIRouter:
     @router.post("/feedback/{search}/{rating}")
     def get_feedback(search: str, rating: int, data: BasicReq):
         # firebase admin init
-        setting, firebase_app = firebase_admin_with_setting(data)
+        try:
+            setting, firebase_app = firebase_admin_with_setting(data)
+        except BrainException as ex:
+            return ex.get_response_exp()
 
         # cloud message
         cloud_message = CloudMessage(firebase_app=firebase_app)
@@ -274,7 +295,10 @@ def construct_blueprint_api() -> APIRouter:
     @router.post("/chat_rising")
     def message_agent(data: ChatRising):
         # firebase admin init
-        setting, firebase_app = firebase_admin_with_setting(data)
+        try:
+            setting, firebase_app = firebase_admin_with_setting(data)
+        except BrainException as ex:
+            return ex.get_response_exp()
 
         # cloud message
         cloud_message = CloudMessage(firebase_app=firebase_app)
@@ -323,7 +347,10 @@ def construct_blueprint_api() -> APIRouter:
     @router.post("/send_sms")
     def send_sms(data: SendSMS):
         # firebase admin init
-        setting, firebase_app = firebase_admin_with_setting(data)
+        try:
+            setting, firebase_app = firebase_admin_with_setting(data)
+        except BrainException as ex:
+            return assembler.to_response(ex.code, ex.message, "")
         try:
             token = data.token
             uuid = data.uuid
@@ -358,7 +385,10 @@ def construct_blueprint_api() -> APIRouter:
     @router.post("/train/contacts")
     def train_contacts(data: TrainContacts):
         # firebase admin init
-        setting, firebase_app = firebase_admin_with_setting(data)
+        try:
+            setting, firebase_app = firebase_admin_with_setting(data)
+        except BrainException as ex:
+            return assembler.to_response(ex.code, ex.message, "")
         try:
             token = data.token
             uuid = data.uuid
@@ -368,8 +398,11 @@ def construct_blueprint_api() -> APIRouter:
             for contact in data.contacts:
                 contacts.append(assembler.to_contact_model(contact))
             # train contact
+            contacts_service = ContactsService(setting=setting)
             contacts_service.train(uuid, contacts)
         except Exception as e:
+            if isinstance(e, BrainException):
+                return e.get_response_exp()
             return assembler.to_response(400, "Failed to train contacts", "")
         return assembler.to_response(200, "Trained successfully", "")
 
@@ -386,15 +419,21 @@ def construct_blueprint_api() -> APIRouter:
     @router.post("/train/contacts/delete")
     def delete_all_contacts(data: BasicReq):
         # firebase admin init
-        setting, firebase_app = firebase_admin_with_setting(data)
+        try:
+            setting, firebase_app = firebase_admin_with_setting(data)
+        except BrainException as ex:
+            return assembler.to_response(ex.code, ex.message, "")
         try:
             token = data.token
             uuid = data.uuid
 
             # parsing contacts
             # train contact
+            contacts_service = ContactsService(setting=setting)
             contacts_service.delete_all(uuid)
         except Exception as e:
+            if isinstance(e, BrainException):
+                return e.get_response_exp()
             return assembler.to_response(400, "Failed to delete contacts", "")
         return assembler.to_response(
             200, "Deleted all contacts from pinecone successfully", ""

Brain/src/router/browser_router.py

@@ -1,6 +1,7 @@
 from fastapi import APIRouter, Request, Depends
 
 from Brain.src.common.assembler import Assembler
+from Brain.src.common.brain_exception import BrainException
 from Brain.src.common.program_type import ProgramType
 from Brain.src.common.utils import parseUrlFromStr
 from Brain.src.firebase.firebase import firebase_admin_with_setting
@@ -30,7 +31,10 @@ def construct_blueprint_browser_api() -> APIRouter:
     @router.post("/item")
     def get_item(data: BrowserItem):
         # firebase admin init
-        setting, firebase_app = firebase_admin_with_setting(data)
+        try:
+            setting, firebase_app = firebase_admin_with_setting(data)
+        except BrainException as ex:
+            return ex.get_response_exp()
 
         item_link = ""
         try:
@@ -41,6 +45,8 @@ def construct_blueprint_browser_api() -> APIRouter:
             # train contact
             item_link = browser_service.query_item(items=data.items, query=data.prompt)
         except Exception as e:
+            if isinstance(e, BrainException):
+                return e.get_response_exp()
             return assembler.to_response(400, "Failed to get item in a browser", "")
         return assembler.to_response(
             200,

Brain/src/router/train_router.py

@@ -1,6 +1,7 @@
 from fastapi import APIRouter
 
 from Brain.src.common.assembler import Assembler
+from Brain.src.common.brain_exception import BrainException
 from Brain.src.firebase.firebase import firebase_admin_with_setting
 from Brain.src.model.requests.request_model import (
     Document,
@@ -22,9 +23,12 @@ def construct_blueprint_train_api() -> APIRouter:
     @router.get("")
     def read_all_documents(data: BasicReq):
         # parsing params
-        setting, firebase_app = firebase_admin_with_setting(data)
+        try:
+            setting, firebase_app = firebase_admin_with_setting(data)
+        except BrainException as ex:
+            return ex.get_response_exp()
         # Services
-        train_service = TrainService(firebase_app=firebase_app)
+        train_service = TrainService(firebase_app=firebase_app, setting=setting)
         try:
             result = train_service.read_all_documents()
         except Exception as e:
@@ -37,9 +41,12 @@ def construct_blueprint_train_api() -> APIRouter:
     @router.post("/all")
     def train_all_documents(data: BasicReq):
         # parsing params
-        setting, firebase_app = firebase_admin_with_setting(data)
+        try:
+            setting, firebase_app = firebase_admin_with_setting(data)
+        except BrainException as ex:
+            return ex.get_response_exp()
         # Services
-        train_service = TrainService(firebase_app=firebase_app)
+        train_service = TrainService(firebase_app=firebase_app, setting=setting)
         try:
             result = train_service.train_all_documents()
         except Exception as e:
@@ -52,9 +59,12 @@ def construct_blueprint_train_api() -> APIRouter:
     @router.post("/{document_id}")
     def read_one_document(document_id: str, data: BasicReq):
         # parsing params
-        setting, firebase_app = firebase_admin_with_setting(data)
+        try:
+            setting, firebase_app = firebase_admin_with_setting(data)
+        except BrainException as ex:
+            return ex.get_response_exp()
         # Services
-        train_service = TrainService(firebase_app=firebase_app)
+        train_service = TrainService(firebase_app=firebase_app, setting=setting)
         if document_id != "all":
             try:
                 result = train_service.read_one_document(document_id)
@@ -75,9 +85,12 @@ def construct_blueprint_train_api() -> APIRouter:
     @router.post("")
     def create_document_train(data: Document):
         # parsing params
-        setting, firebase_app = firebase_admin_with_setting(data)
+        try:
+            setting, firebase_app = firebase_admin_with_setting(data)
+        except BrainException as ex:
+            return ex.get_response_exp()
         # Services
-        train_service = TrainService(firebase_app=firebase_app)
+        train_service = TrainService(firebase_app=firebase_app, setting=setting)
         try:
             result = train_service.create_one_document(data.page_content)
         except Exception as e:
@@ -100,9 +113,12 @@ def construct_blueprint_train_api() -> APIRouter:
     @router.put("")
     def update_one_document(data: Document):
         # parsing params
-        setting, firebase_app = firebase_admin_with_setting(data)
+        try:
+            setting, firebase_app = firebase_admin_with_setting(data)
+        except BrainException as ex:
+            return ex.get_response_exp()
         # Services
-        train_service = TrainService(firebase_app=firebase_app)
+        train_service = TrainService(firebase_app=firebase_app, setting=setting)
         try:
             result = train_service.update_one_document(
                 data.document_id, data.page_content
@@ -125,12 +141,17 @@ def construct_blueprint_train_api() -> APIRouter:
     @router.post("/delete/{document_id}")
     def delete_one_document(document_id: str, data: BasicReq):
         # parsing params
-        setting, firebase_app = firebase_admin_with_setting(data)
+        try:
+            setting, firebase_app = firebase_admin_with_setting(data)
+        except BrainException as ex:
+            return ex.get_response_exp()
         # Services
-        train_service = TrainService(firebase_app=firebase_app)
+        train_service = TrainService(firebase_app=firebase_app, setting=setting)
         try:
             result = train_service.delete_one_document(document_id)
         except Exception as e:
+            if isinstance(e, BrainException):
+                return e.get_response_exp()
             return assembler.to_response(400, "fail to delete one train", "")
         return assembler.to_response(
             200, "deleted one document and train data successfully", result

Brain/src/service/contact_service.py

@@ -1,6 +1,7 @@
 """service to manage contacts"""
 from typing import List, Any
 
+from Brain.src.model.req_model import ReqModel
 from Brain.src.rising_plugin.csv_embed import get_embed
 from Brain.src.rising_plugin.pinecone_engine import (
     get_pinecone_index_namespace,
@@ -16,6 +17,9 @@ from Brain.src.model.contact_model import ContactModel, ContactStatus
 
 
 class ContactsService:
+    def __init__(self, setting: ReqModel):
+        self.setting = setting
+
     """train contacts (getting embedding) and update pinecone with embeddings by contact_id
     train datatype:
     key: contactId
@@ -33,13 +37,23 @@ class ContactsService:
             # create | update | delete pinecone
             if contact.status == ContactStatus.CREATED:
                 add_pinecone(
-                    namespace=pinecone_namespace, key=key, value=vectoring_values
+                    namespace=pinecone_namespace,
+                    key=key,
+                    value=vectoring_values,
+                    setting=self.setting,
                 )
             elif contact.status == ContactStatus.DELETED:
-                delete_pinecone(namespace=pinecone_namespace, key=key)
+                delete_pinecone(
+                    namespace=pinecone_namespace,
+                    key=key,
+                    setting=self.setting,
+                )
             elif contact.status == ContactStatus.UPDATED:
                 update_pinecone(
-                    namespace=pinecone_namespace, key=key, value=vectoring_values
+                    namespace=pinecone_namespace,
+                    key=key,
+                    value=vectoring_values,
+                    setting=self.setting,
                 )
 
     """"query contact with search text
@@ -47,7 +61,7 @@ class ContactsService:
 
     def query_contacts(self, uuid: str, search: str) -> List[str]:
         vector_data = get_embed(search)
-        index = init_pinecone(PINECONE_INDEX_NAME)
+        index = init_pinecone(index_name=PINECONE_INDEX_NAME, setting=self.setting)
         relatedness_data = index.query(
             vector=vector_data,
             top_k=5,
@@ -63,7 +77,9 @@ class ContactsService:
     """delete all items in the specific nanespace"""
 
     def delete_all(self, uuid: str) -> Any:
-        return delete_all_pinecone(self.get_pinecone_index_namespace(uuid))
+        return delete_all_pinecone(
+            namespace=self.get_pinecone_index_namespace(uuid), setting=self.setting
+        )
 
     """get pinecone namespace of pinecone"""