feature(#35): change get_pinecone_index_train_namespace function path.

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

@@ -48,19 +48,16 @@ from Brain.src.rising_plugin.pinecone_engine import (
 )
 
 
-def get_pinecone_index_train_namespace() -> str:
-    return get_pinecone_index_namespace(f"trains")
-
-
 @action()
 async def general_question(query):
     """step 0: convert string to json"""
     index = init_pinecone(PINECONE_INDEX_NAME)
+    train_service = TrainService()
     try:
         json_query = json.loads(query)
     except Exception as ex:
         raise BrainException(BrainException.JSON_PARSING_ISSUE_MSG)
-    """step 0-->: parsing parms from the json query"""
+    """step 0-->: parsing params from the json query"""
     query = json_query["query"]
     model = json_query["model"]
     uuid = json_query["uuid"]
@@ -75,14 +72,13 @@ async def general_question(query):
         vector=query_result,
         top_k=3,
         include_values=False,
-        namespace=get_pinecone_index_train_namespace(),
+        namespace=train_service.get_pinecone_index_train_namespace(),
     )
     documentId = ""
     if len(relatedness_data["matches"]) > 0:
         documentId = relatedness_data["matches"][0]["id"]
 
     docs = []
-    train_service = TrainService()
     documents = train_service.read_all_documents()
     for document in documents:
         if document["document_id"] == documentId:

Brain/src/service/train_service.py

@@ -23,10 +23,10 @@ def to_json(page_content: str):
 
 
 class TrainService:
-    """train (getting embedding) and update pinecone with embeddings by train_id
+    """train (getting embedding) and update pinecone with embeddings by document_id
     train datatype:
-    key: id
-    values: {id},{data}, {status}"""
+    key: document_id
+    values: {page_content}"""
 
     def __init__(self):
         self.db = firestore.client()
@@ -115,3 +115,6 @@ class TrainService:
 
     def get_pinecone_index_namespace(self) -> str:
         return get_pinecone_index_namespace(f"trains")
+
+    def get_pinecone_index_train_namespace(self) -> str:
+        return get_pinecone_index_namespace(f"trains")