'tools'

Tools.py

@@ -21,11 +21,18 @@ PADA_CSV_PATH = "Data/term_data_processed_v2.csv"
 
 class ScriptureDescriptionToolSpec(BaseToolSpec):
     '''
-    To obtain the description of the vedic scriptures, mandalas, kandahas, shukta, Anuvaka etc.
-    Sample Query:
-    1. Describe RigVeda?
-    2. What is the summary of the first mandala from RigVeda?
-    3. What is the brief description of the 4th Shukta of the 2nd Kandah from AtharvaVeda? 
+    Purpose: Obtains the description or summary about vedas, mandalas, kandas, shuktas, archakah, adhyaya, and other scriptural elements.
+    Returns: A dictionary containing the description or basic information about the specified scriptural element.
+    Structure of the levels in each Vedas:
+    RigVeda->Mandala->Shukta
+    ShuklaYajurVeda->Adhyaya
+    SamaVeda->Archika->Shukta
+    Atharvaveda->Kandah->Shukta
+    Sample query:
+    1. Describe the first kandah, second shukta from Atharvaveda? 
+       scripture_name: Atharvaveda, level_0: 1, level_1:2
+    2. Summarize ShuklaYajurVeda?
+    3. What is the difference between ShuklaYajurVeda and KrishnaYajurVeda?
     '''
     spec_functions = ["get_description"]
 
@@ -46,9 +53,9 @@ class ScriptureDescriptionToolSpec(BaseToolSpec):
         except IndexError as e:
             raise ValueError(f"Failed to get scripture description: {e}")
 
-    def get_description(self, level_0, level_1=None, level_2=None, level_3=None):
+    def get_description(self, scripture_name: str, level_1: int=None, level_2: int=None, level_3: int=None):
         try:
-            conditions = (self.df['scripture_name'].str.lower() == level_0.lower())
+            conditions = (self.df['scripture_name'].str.lower() == scripture_name.lower())
             if level_3 is not None:
                 conditions &= (self.df['level_1'] == str(level_1)) & (self.df['level_2'] == str(level_2)) & (self.df['level_3'] == str(level_3))
             elif level_2 is not None:
@@ -61,19 +68,19 @@ class ScriptureDescriptionToolSpec(BaseToolSpec):
 
 class MantraToolSpec(BaseToolSpec):
     '''
-    You can retrieve detailed information about Vedic mantras, including details such as vedamantra, padapatha, devata, chandah, 
-    and rishi, from all Vedas (RigVeda, AtharvaVeda, SamaVeda, KrishnaYajurVeda, and ShuklaYajurVeda) using the function 
-    `get_vedamantra_details`. Additionally, you can access a summary of the mantra, including anvaya, mantraVishaya, and 
-    adhibautic (or adhyatmic or adhidyvic) meaning (or bhavartha), purpose, usage, and tippani of the vedamantra with the function 
-    `get_vedamantra_summary`.
-
-    Here's a sample query format:
-    
-    1. Obtain the vedamantra of the mantra 1.1.1.1?
+    Use the function `get_vedamantra_details` to retrieve detailed information about Vedic mantras, including vedamantra, padapatha, devata, chandah, 
+    and rishi, from all Vedas (RigVeda, AtharvaVeda, SamaVeda, KrishnaYajurVeda, and ShuklaYajurVeda). 
+    Use the function `get_vedamantra_summary` to access the information such as anvaya of the mantra, mantraVishaya of the mantra, 
+    adhibautic (or adhyatmic or adhidyvic) meaning (or bhavarth) of the mantra, purpose of the mantra, usage of the mantra, and tippani of the mantra.
+    Sample Questions:
+    1. Obtain the vedamantra of the mantra whose id is 1.1.1.1?
     2. Retrieve the devata of the vedamantra from Rigveda, first mandala, first shukta, and first mantra.
     3. Provide the meaning of the vedamantra from Rigveda, first mandala, first shukta, and first mantra written by Tulsi Ram.
     4. Explain the adhibautic meaning of the first mantra from RigVeda, first mandala, and first shukta.
     5. Identify the mantraVishaya of the vedamantra from RigVeda, first mandala, first shukta, and first mantra.
+    6. What is the adibhautic meaning of the mantra 1.1.1.9?
+    7. What is the adhyatmic meaning of the mantra 1.1.1.7?
+    8. What is the adhidyic meaning of the 6th mantra from RigVeda, first mandala, and first shukta?
     '''
     spec_functions = ["get_vedamantra_details", "get_vedamantra_summary"]
 
@@ -88,13 +95,30 @@ class MantraToolSpec(BaseToolSpec):
         except Exception as e:
             raise ValueError(f"Failed to get mantra details: {e}")
 
-    def _get_mantra_details_by_scripture(self, scripture_name=None, **kwargs):
+    def _get_mantra_details_by_scripture(self, scripture_name=None, KandahNumber=None,MandalaNumber=None, ArchikahNumber=None, 
+                                                           ShuktaNumber=None, PrapatakNumber=None, MantraNumber=None,
+                                                           AnuvakNumber=None, AdhyayaNumber=None):
         try:
+            condition = True
             if scripture_name:
                 condition = (self.df_vedamantra['scripture_name'].str.lower() == scripture_name.lower())
+            if KandahNumber:
+                condition &= (self.df_vedamantra['KandahNumber'] == KandahNumber)
+            if MandalaNumber:
+                condition &= (self.df_vedamantra['MandalaNumber'] == MandalaNumber)
+            if ArchikahNumber:
+                condition &= (self.df_vedamantra['ArchikahNumber'] == ArchikahNumber)
+            if ShuktaNumber:
+                condition &= (self.df_vedamantra['ShuktaNumber'] == ShuktaNumber)
+            if PrapatakNumber:
+                condition &= (self.df_vedamantra['PrapatakNumber'] == PrapatakNumber)
+            if MantraNumber:
+                condition &= (self.df_vedamantra['MantraNumber'] == MantraNumber)
+            if AnuvakNumber:
+                condition &= (self.df_vedamantra['AnuvakNumber'] == AnuvakNumber)
+            if AdhyayaNumber:
+                condition &= (self.df_vedamantra['AdhyayaNumber'] == AdhyayaNumber)
             
-            for key, value in kwargs.items():
-                condition &= (self.df_vedamantra[key] == value)
             filtered_df = self.df_vedamantra[condition]
             if not filtered_df.empty:
                 return filtered_df
@@ -104,30 +128,43 @@ class MantraToolSpec(BaseToolSpec):
             logging.error(f"Error in _get_pada_details_by_scripture: {e}")
 
 
-    def _get_mantra_id(self, scripture_name, **kwargs):
-        filtered_df = self._get_mantra_details_by_scripture(scripture_name, **kwargs)
-        mantraID = filtered_df['mantra_number']
-        return mantraID.values[0]
-
-    def get_vedamantra_details(self, mantraid=None, scripture_name=None,**kwargs):
+    def get_vedamantra_details(self, mantraid=None, scripture_name=None, KandahNumber=None,MandalaNumber=None, ArchikahNumber=None, 
+                                                           ShuktaNumber=None, PrapatakNumber=None, MantraNumber=None,
+                                                           AnuvakNumber=None, AdhyayaNumber=None):
         try:
             if mantraid:
                 query = f"SELECT mantra_json FROM veda_content WHERE mantra_number = '{mantraid}'"
             else:
-                mantra_id = self._get_mantra_id(scripture_name, **kwargs)
+                filter_df = self._get_mantra_details_by_scripture(scripture_name=scripture_name, KandahNumber=KandahNumber,MandalaNumber=MandalaNumber, ArchikahNumber=ArchikahNumber, 
+                                                           ShuktaNumber=ShuktaNumber, PrapatakNumber=PrapatakNumber, MantraNumber=MantraNumber,
+                                                           AnuvakNumber=AnuvakNumber, AdhyayaNumber=AdhyayaNumber)
+                mantra_id = filter_df['mantra_number'].values[0]
                 #print(mantra_id)
                 query = f"SELECT mantra_json FROM veda_content WHERE mantra_number = '{mantra_id}'"
             return self._get_mantra_details(query)
         except Exception as e:
             return {"error": str(e)}
 
-    def get_vedamantra_summary(self, mantraid=None, scripture_name=None, **kwargs):
+    def get_vedamantra_summary(self, mantraid=None, scripture_name=None, KandahNumber=None,MandalaNumber=None, ArchikahNumber=None, 
+                                                           ShuktaNumber=None, PrapatakNumber=None, MantraNumber=None,
+                                                           AnuvakNumber=None, AdhyayaNumber=None):
+        '''
+        Sample Query:
+        1. Obtain the anvaya of the mantra whose id (mantraid) is 1.1.1.1?
+        2. Retrieve tha adibhautic meaning of the mantra from RigVeda, first mandala, first shukta, and first mantra.
+        3. Provide the adhyatmic meaning of the mantra 1.1.1.9?
+        4. What is the tippani of the mantra 1.1.1.7?
+        5. What is the adhyatmic meaning of the mantra 1.1.1.7?
+        6. What is the mantravishaya of the 6th mantra from RigVeda, first mandala, and first shukta?
+        '''
         try:
             if mantraid:
                 query = f"SELECT mantra_json FROM veda_content WHERE mantra_number = '{mantraid}'"
             else:
-                mantra_id = self._get_mantra_id(scripture_name, **kwargs)
-                print(mantra_id)
+                filtered_df = self._get_mantra_details_by_scripture(scripture_name=scripture_name, KandahNumber=KandahNumber,MandalaNumber=MandalaNumber, ArchikahNumber=ArchikahNumber, 
+                                                           ShuktaNumber=ShuktaNumber, PrapatakNumber=PrapatakNumber, MantraNumber=MantraNumber,
+                                                           AnuvakNumber=AnuvakNumber, AdhyayaNumber=AdhyayaNumber)         
+                mantra_id = filtered_df['mantra_number'].values[0]
                 query = f"SELECT mantra_json FROM veda_content WHERE mantra_number = '{mantra_id}'"
             json_dict = get_details_mantra_json(query)
             mantra_summary = json_dict['mantraSummary']['language']
@@ -159,17 +196,33 @@ class PadaToolSpec(BaseToolSpec):
         self.df_terms = pd.read_csv(PADA_CSV_PATH, dtype={'AnuvakNumber': 'Int64', 'PrapatakNumber': 'Int64', 'KandahNumber': 'Int64', 'ShuktaNumber': 'Int64', 'ArchikahNumber': 'Int64', 'AdhyayaNumber': 'Int64', 'MandalaNumber': 'Int64', 'ParyayaNumber': 'Int64'}, encoding='utf-8')
         self.df_vedic_content = pd.read_csv(VEDAMANTRA_CSV_PATH,encoding = 'utf-8')
 
-    def _get_pada_details_by_scripture(self, pada, scripture_name=None, **kwargs):
+    def _get_pada_details_by_scripture(self, pada, scripture_name=None, KandahNumber=None,MandalaNumber=None, ArchikahNumber=None, 
+                                                           ShuktaNumber=None, PrapatakNumber=None, MantraNumber=None,
+                                                           AnuvakNumber=None, AdhyayaNumber=None):
         try:
             pada = iast_process(pada)
             condition = (self.df_terms['Pada'] == pada)
 
             if scripture_name:
                 condition &= (self.df_terms['scripture_name'].str.lower() == scripture_name.lower())
+
+            if KandahNumber:
+                condition &= (self.df_terms['KandahNumber'] == KandahNumber)
+            if MandalaNumber:
+                condition &= (self.df_terms['MandalaNumber'] == MandalaNumber)
+            if ArchikahNumber:
+                condition &= (self.df_terms['ArchikahNumber'] == ArchikahNumber)
+            if ShuktaNumber:
+                condition &= (self.df_terms['ShuktaNumber'] == ShuktaNumber)
+            if PrapatakNumber:
+                condition &= (self.df_terms['PrapatakNumber'] == PrapatakNumber)
+            if MantraNumber:
+                condition &= (self.df_terms['MantraNumber'] == MantraNumber)
+            if AnuvakNumber:
+                condition &= (self.df_terms['AnuvakNumber'] == AnuvakNumber)
+            if AdhyayaNumber:
+                condition &= (self.df_terms['AdhyayaNumber'] == AdhyayaNumber)
             
-            for key, value in kwargs.items():
-                condition &= (self.df_terms[key] == value)
-                
             filtered_df = self.df_terms[condition]
             
             if not filtered_df.empty:
@@ -205,7 +258,7 @@ class PadaToolSpec(BaseToolSpec):
             return best_meaning if best_meaning else {"error": "Required meaning associated with vedamantra is not available."}
         except Exception as e:
             logging.error(f"Error in _get_vedamantra_meaning: {e}")
-            return {"error": f"An error occurred: {e}"}
+            return json.loads({"error": f"An error occurred: {e}"})
     
     def _get_pada_morphology(self, term_details, meanings):
         try:
@@ -240,27 +293,32 @@ class PadaToolSpec(BaseToolSpec):
             return meanings_list
         except Exception as e:
             logging.error(f"Error in get_pada_meaning: {e}")
-            return {"error": f"Required meaning associated with pada is not available. {e}"}    
+            return json.dumps({"error": f"Required meaning associated with pada is not available. {e}"})    
     
 
     def get_adibauatic_adidaivic_adhyatmic_meaning_of_pada(self, pada, mantraid=None, scripture_name=None, 
                                                            KandahNumber=None,MandalaNumber=None, ArchikahNumber=None, 
                                                            ShuktaNumber=None, PrapatakNumber=None, MantraNumber=None,
-                                                           AnuvakNumber=None, AdhyayaNumber=None, **kwargs):
-        pada = iast_process(pada)
+                                                           AnuvakNumber=None, AdhyayaNumber=None,MahatmaName=None):
+        '''
+        Sample query:
+        1. What is the meaning of pada 'agnim' from RigVeda, first mandala, first shukta and first mantra?
+        2. What is the adhyatmic meaning of the pada agnim in the context of the mantra whose id is '1.1.1.1?'
+        '''
         try:
+            pada = iast_process(pada)
             if mantraid:
                 details = self.df_terms[(self.df_terms['mantra_id'] == mantraid) & (self.df_terms['Pada'] == pada)]
-                
             else:
-                if scripture_name is not None:
-                    details = self._get_pada_details_by_scripture(pada, scripture_name, **kwargs)
+                details = self._get_pada_details_by_scripture(pada, scripture_name=scripture_name, KandahNumber=KandahNumber,MandalaNumber=MandalaNumber, ArchikahNumber=ArchikahNumber, 
+                                                           ShuktaNumber=ShuktaNumber, PrapatakNumber=PrapatakNumber, MantraNumber=MantraNumber,
+                                                           AnuvakNumber=AnuvakNumber, AdhyayaNumber=AdhyayaNumber)
         
             if not details.empty:
                 pada_details = details.iloc[0]
                 #print(pada_details)
                 mantraID = pada_details['mantra_id']
-                meanings = self._get_vedamantra_meaning(mantraID,MahatmaName=kwargs.get('MahatmaName'))
+                meanings = self._get_vedamantra_meaning(mantraID,MahatmaName=MahatmaName)
                 if 'error' in meanings:
                     return meanings
                 ab_term_morph_list = self._get_pada_morphology(pada_details, meanings['adibhautic'])
@@ -275,7 +333,7 @@ class PadaToolSpec(BaseToolSpec):
                     'vedamantra_adhyatmic_meaning': meanings['adhyatmic']
                 }
             else:
-                return {"error": f"No details found for pada '{pada}'"}
+                return json.dumps({"error": f"No details found for pada '{pada}'"})
         except Exception as e:
             logging.error(f"Error in get_adibauatic_adidaivic_adhyatmic_meaning_of_pada: {e}")
-            return {"error": f"Failed to get meaning of the word {pada}. {e}"}
+            return json.dumps({"error": f"Failed to get meaning of the word {pada}. {e}"})

app.py

@@ -94,10 +94,10 @@ query_engine_tools = [
                 Helpful to get semantic information from the documents. These documents containing comprehensive information about the Vedas.\
                 They also covers various aspects, including general details about the Vedas, fundamental terminology associated with Vedic literature, \
                 and detailed information about Vedamantras for each Veda. The Vedamantra details encompass essential elements such as padapatha, rishi, chandah,\
-                devata, and swarah.This tool is very useful to answer general questions related to vedas.\
-                Sample Query:\
-                1. What is the meaning of devata ?\
-                2. What are the different Brahmanas associated with SamaVeda?\
+                devata, and swarah.This tool is very useful to answer general questions related to vedas.
+                Sample Query:
+                1. What is the meaning of devata ?
+                2. What are the different Brahmanas associated with SamaVeda?
                 3. What is the difference between Shruti and Smriti.
                '''
             ),
@@ -111,12 +111,12 @@ query_engine_tools = [
                 '''A powerful tool designed to handle queries related to counting information about vedic content document. This document is a .csv file with different columns as follows:\
                 'mantra_id', 'scripture_name', 'KandahNumber', 'PrapatakNumber','AnuvakNumber', 'MantraNumber', 'DevataName', 'RishiName', 'SwarahName', 'ChandaName',\
                 'padapatha', 'vedamantra', 'AdhyayaNumber', 'ArchikahNumber', 'ArchikahName', 'ShuktaNumber', 'keyShukta', 'ParyayaNumber', 'MandalaNumber'.\
-                Always provide the final answer after excuting pandas query which is equivalent to user query.\
-                Sample Query:\
-                1. How many mantras are there in RigVeda whose swarah is gāndhāraḥ?\
-                2. How many different devata present in rigveda?\
-                3. Which Kandah has the maximum number of in KrishnaYajurVeda?\
-                4. Find the number of mantras from AtharvaVeda whose devata is vācaspatiḥ and chandah is anuṣṭup?\
+                Always provide the final answer after excuting pandas query which is equivalent to user query.
+                Sample Query:
+                1. How many mantras are there in RigVeda whose swarah is gāndhāraḥ?
+                2. How many different devata present in RigVeda?
+                3. Which Kandah has the maximum number of in KrishnaYajurVeda?
+                4. Find the number of mantras from AtharvaVeda whose devata is vācaspatiḥ and chandah is anuṣṭup?
                 5. count the mantras in RigVeda whose swarah is gāndhāraḥ?
                '''
             ),
@@ -127,13 +127,13 @@ query_engine_tools = [
         metadata=ToolMetadata(
             name="pandas_engine_padas",
             description=(
-                '''A powerful tool designed to handle queries related to counting information about pada or words from vedic documents. This document is a .csv file with different columns as follows:\
-                    'Pada', 'scripture_name', 'mantra_id', 'MantraNumber', 'AnuvakNumber', 'PrapatakNumber', 'KandahNumber', 'Pada_position', 'term_index', 'Segmentation', 'Morphology', 'ShuktaNumber',\
-                        'ArchikahNumber', 'AdhyayaNumber', 'MandalaNumber', 'ParyayaNumber' \
-                Always provide the final answer after excuting pandas query which is equivalent to user query.\
-                Sample Query:\
-                1. How many padas are there in RigVeda?\
-                2. How many padas present in both rigveda and samaveda?
+                '''A powerful tool designed to handle queries related to counting information about pada or words from vedic documents. This document is a .csv file with different columns as follows:
+                    'Pada', 'scripture_name', 'mantra_id', 'MantraNumber', 'AnuvakNumber', 'PrapatakNumber', 'KandahNumber', 'Pada_position', 'term_index', 'Segmentation', 'Morphology', 'ShuktaNumber',
+                    'ArchikahNumber', 'AdhyayaNumber', 'MandalaNumber', 'ParyayaNumber'.
+                    Always provide the final answer after excuting pandas query which is equivalent to user query.
+                    Sample Query:
+                    1. How many padas are there in RigVeda?
+                    2. How many padas present in both rigveda and samaveda?
                '''
             ),
         ),
@@ -149,15 +149,16 @@ tools = [*mantra_tools,*pada_tools,*description_tools,*query_engine_tools]
 context = """
   You are an expert on Vedas and related scriptures.
   Your role is to respond to questions about vedic scriptures (RigVeda, AtharvaVeda, SamaVeda, ShuklaYajurVeda, and KrishnaYajurVeda) and associated information based on available sources.\
-  For every query, you must use tool first. If the input args, kwargs and tools for the given query is same as in the history or retrieved context is sufficient, then use the history as context.
+  For every query, you first use an appropriate tool. If the input args, kwargs and tools for the given query is same as in the history or retrieved context is sufficient, then use the history as context.
   Please provide well-informed answers. If the context or history is not sufficient, you can say that you don't have sufficient information. Don't use prior knowledge. 
-  If your response is based on "Implicit" thought then you can say that you don't have sufficient information. Don't use prior knowledge.
+  If your response is based on "Implicit" thought then you can caution the user that the answer is not from the 'Svarupa Knowledge base'. 
   Also, provide three follow-up questions based on the input query and the context in the following format.
+
   *****
   You may also try the following questions: 
-  1. <i>Question1</i>
-  2. <i>Question2</i>
-  3. <i>Question3</i>
+  1. Question1
+  2. Question2
+  3. Question3
 """
 
 # Function to create ReActAgent instance (change it based on your initialization logic)