First_agent / app.py
nmago's picture
Add a tool providing random ayah location
12596e3 verified
Raw
History Blame Contribute Delete
3.84 kB
from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
import datetime
import requests
import pytz
import yaml
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.
Args:
surah_number: The surah number.
ayah_number: The ayah number.
"""
url = f"https://api.alquran.cloud/v1/ayah/{surah_number}:{ayah_number}/editions/quran-simple,en.asad"
try:
response = requests.get(url)
response.raise_for_status()
data = response.json()
arabic_text = data["data"][0]["text"]
english_text = data["data"][1]["text"]
return f"The text for ayah {ayah_number} of surah {surah_number} is:\nArabic: {arabic_text}\nEnglish: {english_text}"
except requests.RequestException as e:
response_data = response.json()
if response_data.get("status") == "NOT FOUND":
return f"Error: NOT FOUND - Invalid surah/ayah reference ({surah_number}:{ayah_number})"
return f"Error fetching ayah {ayah_number} of surah {surah_number}: {e}"
@tool
def get_current_time_in_timezone(timezone: str) -> str:
"""A tool that fetches the current local time in a specified timezone.
Args:
timezone: A string representing a valid timezone (e.g., 'America/New_York').
"""
try:
# Create timezone object
tz = pytz.timezone(timezone)
# Get current time in that timezone
local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
return f"The current local time in {timezone} is: {local_time}"
except Exception as e:
return f"Error fetching time for timezone '{timezone}': {str(e)}"
final_answer = FinalAnswerTool()
# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
custom_role_conversions=None,
)
# Import tool from Hub
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
with open("prompts.yaml", 'r') as stream:
prompt_templates = yaml.safe_load(stream)
agent = CodeAgent(
model=model,
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,
planning_interval=None,
name=None,
description=None,
prompt_templates=prompt_templates
)
GradioUI(agent).launch()