Upload folder using huggingface_hub

graph_helper.py

@@ -11,6 +11,7 @@ from tools import (
     tool_push,
     tool_get_standardized_azhwar_names,
     tool_search_db_by_metadata,
+    tool_get_standardized_divya_desam_names,
 )
 from langgraph.prebuilt import ToolNode, tools_condition
 from langchain_core.messages import SystemMessage, ToolMessage, HumanMessage
@@ -32,6 +33,7 @@ def generate_graph() -> CompiledStateGraph:
         tool_search_db,
         tool_format_scripture_answer,
         tool_get_standardized_azhwar_names,
+        tool_get_standardized_divya_desam_names,
         tool_search_db_by_metadata,
     ]
     llm = ChatOpenAI(model="gpt-4o-mini").bind_tools(tools)

nalayiram_helper.py

@@ -1,11 +1,13 @@
 import json
 from dataclasses import dataclass
 
+
 @dataclass
 class Pasuram:
-    prabandham_code : str
-    azhwar_name : str
-    prabandham_name : str
+    prabandham_code: str
+    azhwar_name: str
+    prabandham_name: str
+
 
 def get_standardized_azhwar_names() -> list[Pasuram]:
     """
@@ -19,5 +21,28 @@ def get_standardized_azhwar_names() -> list[Pasuram]:
 
     return final_azhwars
 
+
+def get_standardized_divya_desam_names() -> list[dict]:
+    """
+    Get a list of divya desam names in divya_prabandham
+    """
+    with open("./data/divya_desams.json", "r", encoding="utf-8") as f:
+        divya_desams = json.load(f)  # FIXED
+
+    selected_fields = [
+        "title",
+        "other_names",
+        "name_ta",
+        "alwars",
+        "area",
+        "state",
+        "thirukolam",
+        "direction",
+        "sampradayam",
+        "divya_desam",
+    ]
+    return [{key : row[key] for key in selected_fields if key in row} for row in divya_desams["pageProps"]["hits"]]
+
+
 if __name__ == "__main__":
     print(get_standardized_azhwar_names())

tools.py

@@ -3,7 +3,7 @@ from langchain.agents import Tool
 from langchain_core.tools import StructuredTool
 
 from config import SanatanConfig
-from nalayiram_helper import get_standardized_azhwar_names
+from nalayiram_helper import get_standardized_azhwar_names, get_standardized_divya_desam_names
 from push_notifications_helper import push
 from serperdev_helper import search as search_web
 from sanatan_assistant import format_scripture_answer, query, query_by_metadata_field
@@ -31,6 +31,7 @@ tool_search_db_by_metadata = StructuredTool.from_function(
         " Use this to find relevant scripture verses or explanations."
         "if the user asks for a specific azhwar, use the `tool_get_standardized_azhwar_names` tool to get the standard name first and then pass to this tool to filter pasurams based on azhwar_name."
         "if the user asks for a specific prabandham name, use the `tool_get_standardized_azhwar_names` tool to get the standard prabandham name first and then pass to this tool to filter pasurams based on prabandham_name."
+        "if the user asks for a specific divya desam name, use the `tool_get_standardized_divya_desam_names` tool to get the standard divya desam name first and then pass to this tool to filter pasurams based on `divya_desams`."
         f"use this configuration for reference :\n{json.dumps(SanatanConfig.scriptures, indent=1)}\n"
         # "be aware that verse numbers are sometimes stored as strings and sometimes as mumbers, so if str search does not yield results, try passing in the metadata_value as a number instead"
         # "in the context of divya_prabandham, the verse/pasuram number is stored in metadata as the field `verse` and it is  stored as an int."
@@ -48,10 +49,21 @@ tool_format_scripture_answer = StructuredTool.from_function(format_scripture_ans
 
 tool_get_standardized_azhwar_names = StructuredTool.from_function(
     get_standardized_azhwar_names,
-    name="get_azhwar_names",
+    name="tool_get_standardized_azhwar_names",
     description=(
         "Get a list of standardized  azhwar names and the prabandhams they have written. "
         "Use this tool to standardize the names of the azhwars when the user asks for pasurams written by a specific azhwar."
         "Usually this is followed by passing that standardized azhwar_name for a metadata search using the `tool_search_db_by_metadata` tool."
     ),
 )
+
+
+tool_get_standardized_divya_desam_names = StructuredTool.from_function(
+    get_standardized_divya_desam_names,
+    name="tool_get_standardized_divya_desam_names",
+    description=(
+        "Get a list of standardized  divya desam names. "
+        "Use this tool to standardize the names of the divya desams when the user asks for pasurams written on a specific divya desam."
+        "Usually this is followed by passing that standardized divya desam name for a metadata search using the `tool_search_db_by_metadata` tool by using the fiels `divya_desams`."
+    ),
+)