Add a tool providing random ayah location

app.py

@@ -7,6 +7,21 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
+@tool
+def get_random_ayah_number(surah_number: int = None) -> str:
+    """Returns random Quran ayah location info.
+    Args:
+        surah_number: The surah number. If None, a random surah is chosen.
+    """
+    import random
+    ayahs_meta = [7, 286, 200, 176, 120, 165, 206, 75, 129, 109, 123, 111, 43, 52, 99, 128,111, 110, 98, 135, 112, 78, 118, 64, 77, 227, 93, 88, 69, 60, 34, 30, 73, 54, 45, 83, 182, 88, 75, 85, 54, 53, 89, 59, 37, 35, 38, 29, 18, 45, 60, 49, 62, 55, 78, 96, 29, 22, 24, 13, 14, 11, 11, 18, 12, 12, 30, 52, 52, 44, 28, 28, 20, 56, 40, 31, 50, 40, 46, 42, 29, 19, 36, 25, 22, 17, 19, 26, 30, 20, 15, 21, 11, 8, 8, 19, 5, 8, 8, 11, 11, 8, 3, 9, 5, 4, 7, 3, 6, 3, 5, 4, 5, 6]
+    if surah_number is not None:
+        ayah = random.randint(1, ayahs_meta[surah_number-1])
+    else:
+        surah = random.randint(1, 114)
+        ayah = random.randint(1, ayahs_meta[surah-1])
+    return "Surah " + str(surah) + " ayah " + str(ayah)
+
 @tool
 def get_quran_ayah(surah_number: int, ayah_number: int) -> str:
     """Fetches a specific ayah arabic and translation textes from the Quran API for the specified surah and ayah numbers.
@@ -66,7 +81,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[get_quran_ayah, final_answer], ## add your tools here (don't remove final answer)
+    tools=[get_random_ayah_number, get_quran_ayah, final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,