Spaces:
Sleeping
Sleeping
| 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 | |
| def get_wikipedia_movie_summary(movie_title:str) -> str: | |
| """A tool that queries Wikipedia for a movie title in English and returns a page's introduction. | |
| Args: | |
| movie_title: the English name of the movie | |
| """ | |
| # Wikipedia API endpoint | |
| url = "https://en.wikipedia.org/w/api.php" | |
| params = { | |
| "action": "query", | |
| "format": "json", | |
| "titles": f"{movie_title} movie", | |
| "prop": "extracts", | |
| "explaintext": True, | |
| "exintro": True # Get only the introductory section | |
| } | |
| try: | |
| # Make the API request | |
| response = requests.get(url, params=params) | |
| data = response.json() | |
| # Extract the page content | |
| pages = data["query"]["pages"] | |
| page_id = next(iter(pages)) # Get the first page ID | |
| page_content = pages[page_id]["extract"] | |
| return page_content | |
| except Exception as e: | |
| return f"Error fetching Wikipedia data: {str(e)}" | |
| # 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, | |
| ) | |
| with open("prompts.yaml", 'r') as stream: | |
| prompt_templates = yaml.safe_load(stream) | |
| name = "Bootyful Bot" | |
| task_description = """ | |
| - Bootyful Bot will search wikipedia for a movie name | |
| - Bootyful Bot will read the page and tell whether the movie was inspired by factual events or not | |
| - In case it is, tell user about the events | |
| - In case it is not, tell about movie inspiration, if information is available. | |
| - Bootyful Bot should begin every sentence with 'bing bing' and end every sentence with 'bong bong'. | |
| - Bootyful Bot should begin answer with 'Hello, I am Bootyful Bot' and end it with 'Bye bye, I am Bootyful Bot'. | |
| """ | |
| final_answer = FinalAnswerTool(name=name, description=task_description) | |
| agent = CodeAgent( | |
| model=model, | |
| tools=[final_answer, get_wikipedia_movie_summary], ## add your tools here (don't remove final answer) | |
| max_steps=6, | |
| verbosity_level=1, | |
| grammar=None, | |
| planning_interval=None, | |
| name=None, | |
| description=task_description, | |
| prompt_templates=prompt_templates | |
| ) | |
| GradioUI(agent).launch() |